Add test for FOR_ITER_RANGE

gvanrossum · gvanrossum · commit e6c6603302e8 · 2023-07-08T08:23:10.000-07:00
diff --git a/Lib/test/test_capi/test_misc.py b/Lib/test/test_capi/test_misc.py
@@ -2517,6 +2517,21 @@ def testfunc(x):
         uops = {opname for opname, _ in ex}
         self.assertIn("UNPACK_SEQUENCE", uops)
 
+    def test_for_iter(self):
+        def testfunc(x):
+            for i in range(x):
+                i += 1
+
+        opt = _testinternalcapi.get_uop_optimizer()
+
+        with temporary_optimizer(opt):
+            testfunc(10)
+
+        ex = get_first_executor(testfunc.__code__)
+        self.assertIsNotNone(ex)
+        uops = {opname for opname, _ in ex}
+        self.assertIn("FOR_ITER_RANGE", uops)
+
 
 if __name__ == "__main__":
     unittest.main()
