Add tests for augmenting assigns recursively (#376)

Author: JSAbrahams

tests/resource/valid/definition/assign_with_if.mamba

@@ -0,0 +1 @@
+def a:= if True then 20 else 10

tests/resource/valid/definition/assign_with_if_check.py

@@ -0,0 +1 @@
+a: int = 20 if True else 10

tests/resource/valid/definition/assign_with_match.mamba

@@ -0,0 +1,4 @@
+def a:= match 40
+    2 => 3
+    4 => 30
+    _ => 300

tests/resource/valid/definition/assign_with_match_check.py

@@ -0,0 +1,7 @@
+match 40:
+    case 2:
+        a = 3
+    case 4:
+        a = 30
+    case _:
+        a = 300

tests/resource/valid/definition/assign_with_nested_if.mamba

@@ -0,0 +1,8 @@
+def x := 20
+def a: Int := if x > 10 then
+    if x > 30 then
+        x + 1
+    else
+        x - 1
+else
+    x

tests/resource/valid/definition/assign_with_nested_if_check.py

@@ -0,0 +1,8 @@
+x: int = 20
+if x > 10:
+    if x > 30:
+        a = x + 1
+    else:
+        a = x - 1
+else:
+    a = x

tests/resource/valid/definition/assign_with_try_except.mamba

@@ -0,0 +1,5 @@
+def g() -> Int raise [Exception] =>
+    raise Exception("A")
+
+def a := g() handle
+    err: Exception => x + 10

tests/resource/valid/definition/assign_with_try_except_check.py

@@ -0,0 +1,8 @@
+def g() -> int:
+    raise Exception("A")
+
+a = None
+try:
+    a: int = g()
+except Exception as err:
+    a = x + 10

tests/resource/valid/definition/function_with_try_except_check.py

@@ -4,5 +4,5 @@ def g() -> int =>
 def f(x: int) -> int =>
     try:
         return g()
-    except:
+    except Exception as err:
         return x + 10

tests/system/valid/definition.rs

@@ -10,6 +10,26 @@ fn assign_tuples() -> OutTestRet {
     test_directory(true, &["definition"], &["definition", "target"], "assign_tuples")
 }
 
+#[test]
+fn assign_with_if() -> OutTestRet {
+    test_directory(true, &["definition"], &["definition", "target"], "assign_with_if")
+}
+
+#[test]
+fn assign_with_match() -> OutTestRet {
+    test_directory(true, &["definition"], &["definition", "target"], "assign_with_match")
+}
+
+#[test]
+fn assign_with_nested_if() -> OutTestRet {
+    test_directory(true, &["definition"], &["definition", "target"], "assign_with_nested_if")
+}
+
+#[test]
+fn assign_with_try_except() -> OutTestRet {
+    test_directory(true, &["definition"], &["definition", "target"], "assign_with_try_except")
+}
+
 #[test]
 fn function_ret_super_in_class() -> OutTestRet {
     test_directory(true, &["definition"], &["definition", "target"], "function_ret_super_in_class")