X-Git-Url: http://git.subgeniuskitty.com/vvhitespace/.git/blobdiff_plain/6e942bd8524e9393f61a0358a0282b2641f9d3d7..0c56152e6ad67ef7e311f1e394250279864edd1a:/vv_test.py

diff --git a/vv_test.py b/vv_test.py
index 46f5bc9..a2779ab 100755
--- a/vv_test.py
+++ b/vv_test.py
@@ -14,17 +14,39 @@ path_to_tests = './tests/'
 src_extension = '.pvvs'
 
 tests = [
-        ['0001_push_printchar_exit', 'A'],
-        ['1001_stack_operations', 'BB']
+        # Format: ['filename_without_extension', 'string for stdin', 'string for expected stdout']
+        ['0001_push_printchar_exit', '', 'A'],
+        ['1001_stack_push', '', 'A'],
+        ['1002_stack_dup', '', 'AA'],
+        ['1003_stack_swap', '', 'AB'],
+        ['1004_stack_drop', '', 'A'],
+        ['2001_arithmetic_addition', '', 'B'],
+        ['2002_arithmetic_subtraction', '', 'A'],
+        ['2003_arithmetic_multiplication', '', 'B'],
+        ['2004_arithmetic_division', '', 'A'],
+        ['2005_arithmetic_remainder', '', 'A'],
+        ['3001_heap', '', 'BCA'],
+        ['4001_flowcontrol_exit', '', ''],
+        ['4002_flowcontrol_unconditional_jump_to_label', '', 'A'],
+        ['4003_flowcontrol_jump_if_tos_is_zero', '', 'A'],
+        ['4004_flowcontrol_jump_if_tos_is_negative', '', 'A'],
+        ['4005_flowcontrol_jump_to_subroutine', '', 'A'],
+        ['4006_flowcontrol_return_from_subroutine', '', 'A'],
+        ['4100_flowcontrol_branches_not_taken', '', 'A'],
+        ['5001_io_output_character', '', 'A'],
+        ['5002_io_output_digit', '', '2'],
+        ['5003_io_input_character', 'A', 'A'],
+        ['5004_io_input_digit', '1', '1'],
         ] 
 
 for test in tests:
+    # TODO: Catch stderr
     subprocess.run([compiler_path, '-i', path_to_tests + test[0] + src_extension, '-o', temp_file])
-    result = subprocess.run([interpreter_path, '-i', temp_file], stdout=subprocess.PIPE)
-    if result.stdout.decode('utf-8') != test[1]:
-        print(test[0])
+    result = subprocess.run([interpreter_path, '-i', temp_file], stdout=subprocess.PIPE, input=test[1].encode('utf-8'))
+    if result.stdout.decode('utf-8') != test[2]:
+        print('\n' + test[0])
     else:
-        print('.', end='')
+        print('.', end='', flush=True)
     os.remove(temp_file)
 
 print("")