test assignment to a boolean operation

As title
numba · Jan 30, 2025 · ba4e132 · ba4e132
1 parent b5bb5a9
commit ba4e132
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/numba_rvsdg/tests/test_ast_transforms.py b/numba_rvsdg/tests/test_ast_transforms.py
@@ -1592,6 +1592,45 @@ def function(x: int, y: int) -> int:
             ],
         )
 
+    def test_bool_op_assignment(self):
+        def function(x: int, y: int) -> bool:
+            result = x and y
+            return result
+
+        expected = {
+            "0": {
+                "instructions": [
+                    "__scfg_bool_op_1__ = x",
+                    "__scfg_bool_op_1__",
+                ],
+                "jump_targets": ["1", "2"],
+                "name": "0",
+            },
+            "1": {
+                "instructions": ["__scfg_bool_op_1__ = y"],
+                "jump_targets": ["2"],
+                "name": "1",
+            },
+            "2": {
+                "instructions": [
+                    "result = __scfg_bool_op_1__",
+                    "return result",
+                ],
+                "jump_targets": [],
+                "name": "2",
+            },
+        }
+        self.compare(
+            function,
+            expected,
+            arguments=[
+                (0, 0),  # Both false
+                (0, 1),  # x false, y true
+                (1, 0),  # x true, y false
+                (1, 1),  # Both true
+            ],
+        )
+
 
 class TestEntryPoints(TestCase):