Update test

Author: aisk

Lib/test/test_capi/test_opt.py

@@ -2734,6 +2734,28 @@ class C():
         self.assertLessEqual(count_ops(ex, "_POP_TOP"), 2)
         self.assertIn("_POP_TOP_NOP", uops)
 
+    def test_load_attr_module(self):
+        def testfunc(n):
+            import math
+            x = 0
+            for _ in range(n):
+                y = math.pi
+                if y:
+                    x += 1
+            return x
+
+        res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
+        self.assertEqual(res, TIER2_THRESHOLD)
+        self.assertIsNotNone(ex)
+        uops = get_opnames(ex)
+        for uop in uops:
+            print(uop)
+        return
+
+        self.assertIn("_LOAD_ATTR_MODULE", uops)
+        self.assertLessEqual(count_ops(ex, "_POP_TOP"), 3)
+        self.assertIn("_POP_TOP_NOP", uops)
+
     def test_load_attr_with_hint(self):
         def testfunc(n):
             class C:
@@ -2774,24 +2796,6 @@ class C:
         self.assertLessEqual(count_ops(ex, "_POP_TOP"), 2)
         self.assertIn("_POP_TOP_NOP", uops)
 
-    def test_load_attr_module(self):
-        import platform
-        # Use default argument to capture module as local variable
-        def testfunc(n, mod=platform):
-            x = 0
-            for _ in range(n):
-                # Access module attribute multiple times
-                x = mod.python_version
-            return x
-        res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
-        self.assertIsNotNone(ex)
-        uops = get_opnames(ex)
-
-        self.assertIn("_LOAD_ATTR_MODULE", uops)
-        # _POP_TOP_MODULE should be optimized to _POP_TOP_NOP for borrowed refs
-        self.assertIn("_POP_TOP_NOP", uops)
-        self.assertLessEqual(count_ops(ex, "_POP_TOP"), 2)
-
     def test_int_add_op_refcount_elimination(self):
         def testfunc(n):
             c = 1