improve tests

test/fragments/test_all.rb

@@ -2,6 +2,7 @@
 require_relative "test_foo"
 require_relative "test_if"
 require_relative "test_hello"
+require_relative "test_class"
 require_relative "test_putint"
 require_relative "test_functions"
 require_relative "test_recursive_fibo"

test/fragments/test_class.rb

@@ -0,0 +1,21 @@
+require_relative 'helper'
+
+class TestBasicClass < MiniTest::Test
+  include Fragments
+
+  def test_class_basic
+    @string_input = <<HERE
+module Foo
+  class Bar
+    int buh()
+      return 1
+    end
+  end
+end
+HERE
+    @expect =  [ Virtual::Return ]
+    check
+  end
+
+
+end