Add astcompile tests for compound assignment (AugAssign, AugSetItem, AugSetAttr)

Author: antocuni

spy/tests/test_astcompile.py

@@ -177,3 +177,41 @@ def foo() -> i32:
             return x
         """
         self.assert_dump_decls(expected)
+
+    def test_augassign(self):
+        self.compile_src("""
+        def foo(x: i32) -> i32:
+            x += 1
+            return x
+        """)
+        expected = """
+        def foo(x: i32) -> i32:
+            x = x + 1
+            return x
+        """
+        self.assert_dump(expected)
+
+    def test_augsetitem(self):
+        self.compile_src("""
+        def foo(items: dynamic, idx: i32, val: i32) -> None:
+            items[idx] += val
+        """)
+        expected = """
+        def foo(items: dynamic, idx: i32, val: i32) -> None:
+            _$aug_target0 = items
+            _$aug_arg0_0 = idx
+            _$aug_target0[_$aug_arg0_0] = _$aug_target0[_$aug_arg0_0] + val
+        """
+        self.assert_dump(expected)
+
+    def test_augsetattr(self):
+        self.compile_src("""
+        def foo(obj: dynamic, val: i32) -> None:
+            obj.x += val
+        """)
+        expected = """
+        def foo(obj: dynamic, val: i32) -> None:
+            _$aug_target0 = obj
+            _$aug_target0.x = _$aug_target0.x + val
+        """
+        self.assert_dump(expected)

spy/tests/test_cli.py

@@ -145,18 +145,6 @@ def main() -> None:
         assert "while " in stdout
         assert stdout.startswith("def main() -> None:")
 
-    def test_astcompile_spy_output_augsetitem(self):
-        src = """
-        def main() -> None:
-            items = make_items()
-            items[index()] += value()
-        """
-        f = self.write("test.spy", src)
-        _, stdout = self.run("astcompile", "--format", "spy", f)
-        assert "+=" not in stdout
-        assert "_$aug_target0" in stdout
-        assert "_$aug_arg0_0" in stdout
-
     def test_execute(self):
         argsets = [["execute"], []]  # No subcommand is equivalent to execute command
         for argset in argsets: