def test_print_hello_world():
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code('print("hello")')
vm.reset_std_stream()
assert string_output.getvalue() == "hello\n"
def test_recursive_function_call():
package_path = os.path.join(test_packages, "recursive_function_call")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
assert string_output.getvalue() == "89\n573147844013817084101\n"
def test_class_usage():
package_path = os.path.join(test_packages, "class_usage")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
assert string_output.getvalue() == "name is haha, id is hehe\n"
def test_multi_layer_closure():
package_path = os.path.join(test_packages, "multi_layer_closure")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = "[]\n[10]\n[10, 20]\n"
assert string_output.getvalue() == target
def test_even_odd_loop():
package_path = os.path.join(test_packages, "even_odd_loop")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
target = ["0", "2", "4", "6"]
assert string_output.getvalue().split() == target
def test_classic_closure_example():
package_path = os.path.join(test_packages, "classic_closure_example")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
target = list(map(str, [21] * 10))
assert string_output.getvalue().split() == target
def test_dict_iter_item():
package_path = os.path.join(test_packages, 'dict_iter_item')
test_file = os.path.join(package_path, 'example.py')
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = '1 2\n3 4\n'
assert string_output.getvalue() == target
def test_complicated_arguments_passing():
package_path = os.path.join(test_packages, "complicated_arguments_passing")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
target = ["301", "20", "30", "name", "haha", "pwd", "hehe"]
assert string_output.getvalue().split() == target
def test_complicated_iterator_scenario():
package_path = os.path.join(test_packages, "complicated_iterator_scenario")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
assert string_output.getvalue().split() == list(
map(str, [5, 8, 11, 14, 17, 20, 23, 26, 29, 32]))
def test_so_many_const_values_scenario():
package_path = os.path.join(test_packages, 'so_many_const_values_scenario')
test_file = os.path.join(package_path, 'example.py')
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = '1000\n324\n735\n'
assert string_output.getvalue() == target
def test_global_variable_manipulation():
package_path = os.path.join(test_packages, "global_variable_manipulation")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = list(map(str, [10, 100]))
assert string_output.getvalue().split() == target
def test_class_inheritance():
package_path = os.path.join(test_packages, "class_inheritance")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = "-14\n"
assert string_output.getvalue() == target
def test_while_loop():
package_path = os.path.join(test_packages, "while_loop")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = list(map(str, [0, 1, 3, 7, 15, 31, 63]))
assert string_output.getvalue().split() == target
def test_slice_usage_in_object():
package_path = os.path.join(test_packages, "slice_usage_in_object")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = "[1, 4, 7]\n"
assert string_output.getvalue() == target
def test_import_self_defined_module():
package_path = os.path.join(test_packages, "import_self_defined_module")
test_file = os.path.join(package_path, "a.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_python_file(test_file)
vm.reset_std_stream()
target = "77\n"
assert string_output.getvalue() == target
def test_import_system_library():
package_path = os.path.join(test_packages, "import_system_library")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
target = "a/b\nc/d\n"
assert string_output.getvalue() == target
def test_mix_of_class_and_generator_function():
package_path = os.path.join(test_packages,
"mix_of_class_and_generator_function")
test_file = os.path.join(package_path, "example.py")
string_output = io.StringIO()
vm = VirtualMachine()
vm.set_std_stream(vm_out=string_output)
vm.run_compiled_code(open(test_file).read())
vm.reset_std_stream()
target = list(map(str, list(range(1, 10, 2))))
assert string_output.getvalue().split() == target
