Add more tests

Author: Fidget-Spinner

Lib/test/test_generated_cases.py

@@ -2288,5 +2288,72 @@ def test_pure_uop_body_copied_in(self):
         """
         self.run_cases_test(input, input2, output)
 
+    def test_pure_uop_body_copied_in_complex(self):
+        input = """
+        pure op(OP, (foo -- res)) {
+            if (foo) {
+                res = body(foo);
+            }
+            else {
+                res = 1;
+            }
+        }
+        """
+        input2 = """
+        op(OP, (foo -- res)) {
+            res = sym_new_unknown(ctx);
+        }
+        """
+        output = """
+        case OP: {
+            JitOptSymbol *res;
+            if (
+                sym_is_const(ctx, foo)
+            ) {
+                JitOptSymbol *foo_sym = foo;
+                _PyStackRef foo = sym_get_const_as_stackref(ctx, foo_sym);
+                _PyStackRef res_stackref;
+                /* Start of pure uop copied from bytecodes for constant evaluation */
+                if (foo) {
+                    res_stackref = body(foo);
+                }
+                else {
+                    res_stackref = 1;
+                }
+                /* End of pure uop copied from bytecodes for constant evaluation */
+                res = sym_new_const_steal(ctx, PyStackRef_AsPyObjectBorrow(res_stackref));
+                stack_pointer[-1] = res;
+            }
+            else {
+                res = sym_new_unknown(ctx);
+                stack_pointer[-1] = res;
+            }
+            break;
+        }
+        """
+        self.run_cases_test(input, input2, output)
+
+    def test_pure_uop_reject_array_effects(self):
+        input = """
+        pure op(OP, (foo[2] -- res)) {
+            if (foo) {
+                res = body(foo);
+            }
+            else {
+                res = 1;
+            }
+        }
+        """
+        input2 = """
+        op(OP, (foo[2] -- res)) {
+            res = sym_new_unknown(ctx);
+        }
+        """
+        output = """
+        """
+        with self.assertRaisesRegex(AssertionError,
+                                    "Unsafe to convert a symbol to an array-like StackRef."):
+            self.run_cases_test(input, input2, output)
+
 if __name__ == "__main__":
     unittest.main()