Update test_generated_cases.py

Fidget-Spinner · Jan 21, 2025 · c2c68de · c2c68de
1 parent 47f973b
commit c2c68de
Showing 1 changed file with 36 additions and 32 deletions.
diff --git a/Lib/test/test_generated_cases.py b/Lib/test/test_generated_cases.py
@@ -412,10 +412,12 @@ def test_predictions(self):
             frame->instr_ptr = next_instr;
             next_instr += 1;
             INSTRUCTION_STATS(OP1);
-            PREDICTED(OP1);
-            _PyStackRef res;
-            res = Py_None;
-            stack_pointer[-1] = res;
+            PRED_OP1:
+            {
+                _PyStackRef res;
+                res = Py_None;
+                stack_pointer[-1] = res;
+            }
             DISPATCH();
         }
 
@@ -646,36 +648,38 @@ def test_macro_instruction(self):
             frame->instr_ptr = next_instr;
             next_instr += 6;
             INSTRUCTION_STATS(OP);
-            PREDICTED(OP);
-            _Py_CODEUNIT* const this_instr = next_instr - 6;
-            (void)this_instr;
-            _PyStackRef left;
-            _PyStackRef right;
-            _PyStackRef arg2;
-            _PyStackRef res;
-            // _OP1
+            PRED_OP:
             {
-                right = stack_pointer[-1];
-                left = stack_pointer[-2];
-                uint16_t counter = read_u16(&this_instr[1].cache);
-                (void)counter;
-                _PyFrame_SetStackPointer(frame, stack_pointer);
-                op1(left, right);
-                stack_pointer = _PyFrame_GetStackPointer(frame);
-            }
-            /* Skip 2 cache entries */
-            // OP2
-            {
-                arg2 = stack_pointer[-3];
-                uint32_t extra = read_u32(&this_instr[4].cache);
-                (void)extra;
-                _PyFrame_SetStackPointer(frame, stack_pointer);
-                res = op2(arg2, left, right);
-                stack_pointer = _PyFrame_GetStackPointer(frame);
+                _Py_CODEUNIT* const this_instr = next_instr - 6;
+                (void)this_instr;
+                _PyStackRef left;
+                _PyStackRef right;
+                _PyStackRef arg2;
+                _PyStackRef res;
+                // _OP1
+                {
+                    right = stack_pointer[-1];
+                    left = stack_pointer[-2];
+                    uint16_t counter = read_u16(&this_instr[1].cache);
+                    (void)counter;
+                    _PyFrame_SetStackPointer(frame, stack_pointer);
+                    op1(left, right);
+                    stack_pointer = _PyFrame_GetStackPointer(frame);
+                }
+                /* Skip 2 cache entries */
+                // OP2
+                {
+                    arg2 = stack_pointer[-3];
+                    uint32_t extra = read_u32(&this_instr[4].cache);
+                    (void)extra;
+                    _PyFrame_SetStackPointer(frame, stack_pointer);
+                    res = op2(arg2, left, right);
+                    stack_pointer = _PyFrame_GetStackPointer(frame);
+                }
+                stack_pointer[-3] = res;
+                stack_pointer += -2;
+                assert(WITHIN_STACK_BOUNDS());
             }
-            stack_pointer[-3] = res;
-            stack_pointer += -2;
-            assert(WITHIN_STACK_BOUNDS());
             DISPATCH();
         }