Add more AST tests

Author: mdbrnowski

test_main.py

@@ -1,3 +1,4 @@
+import pytest
 from typer.testing import CliRunner
 from main import app
 
@@ -27,9 +28,10 @@ def test_parser_error():
     assert result.stdout.count("line") == 5
 
 
-def test_ast():
-    result = runner.invoke(app, ["ast", "tests/ast/input_1.txt"])
+@pytest.mark.parametrize("n", [1, 2, 3])
+def test_ast(n):
+    result = runner.invoke(app, ["ast", f"tests/ast/input_{n}.txt"])
     assert result.exit_code == 0
-    with open("tests/ast/output_1.txt", encoding="utf-8") as f:
+    with open(f"tests/ast/output_{n}.txt", encoding="utf-8") as f:
         output = f.read()
     assert result.stdout == output

tests/ast/input_2.txt

@@ -0,0 +1,8 @@
+n = 10;
+for i = 0:n {
+    j = i;
+    while (i > 0)
+        i -= 1;
+    if (i == 0)
+        print "OK";
+}

tests/ast/input_3.txt

@@ -0,0 +1,3 @@
+a = 2 * (2 + 2) + 2 * 2;
+b = 3 * -3;
+C = 2 .+ 2 .* ones(3)';

tests/ast/output_2.txt

@@ -0,0 +1,25 @@
+=
+|  n
+|  10
+for
+|  i
+|  range
+|  |  0
+|  |  n
+|  =
+|  |  j
+|  |  i
+|  while
+|  |  >
+|  |  |  i
+|  |  |  0
+|  |  -=
+|  |  |  i
+|  |  |  1
+|  if
+|  |  ==
+|  |  |  i
+|  |  |  0
+|  then
+|  |  print
+|  |  |  "OK"

tests/ast/output_3.txt

@@ -0,0 +1,26 @@
+=
+|  a
+|  +
+|  |  *
+|  |  |  2
+|  |  |  +
+|  |  |  |  2
+|  |  |  |  2
+|  |  *
+|  |  |  2
+|  |  |  2
+=
+|  b
+|  *
+|  |  3
+|  |  -
+|  |  |  3
+=
+|  C
+|  .+
+|  |  2
+|  |  .*
+|  |  |  2
+|  |  |  transpose
+|  |  |  |  ones
+|  |  |  |  |  3