test exercising transformer from string input

As title
numba · Apr 29, 2024 · 4b7d46b · 4b7d46b
1 parent d3137c5
commit 4b7d46b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/numba_rvsdg/tests/test_ast_transforms.py b/numba_rvsdg/tests/test_ast_transforms.py
@@ -1,4 +1,5 @@
 # mypy: ignore-errors
+import textwrap
 from typing import Callable, Any
 from unittest import main, TestCase
 
@@ -36,6 +37,21 @@ def function() -> int:
         }
         self.compare(function, expected)
 
+    def test_solo_return_from_string(self):
+        function = textwrap.dedent("""
+            def function() -> int:
+                return 1
+        """)
+
+        expected = {
+            "0": {
+                "instructions": ["return 1"],
+                "jump_targets": [],
+                "name": "0",
+            }
+        }
+        self.compare(function, expected)
+
     def test_solo_assign(self):
         def function() -> None:
             x = 1  # noqa: F841