Add regression tests

Author: brandtbucher

Lib/test/test_capi/test_opt.py

@@ -1620,6 +1620,80 @@ def f(n):
         self.assertIn("_TO_BOOL_STR", uops)
         self.assertNotIn("_GUARD_TOS_UNICODE", uops)
 
+    def test_remove_guard_for_known_type_dict(self):
+        def f(n):
+            x = 0
+            for _ in range(n):
+                d = {}
+                d["Spam"] = 1  # Guarded...
+                x += d["Spam"]  # ...unguarded!
+            return x
+
+        res, ex = self._run_with_optimizer(f, TIER2_THRESHOLD)
+        self.assertEqual(res, TIER2_THRESHOLD)
+        self.assertIsNotNone(ex)
+        uops = get_opnames(ex)
+        self.assertEqual(uops.count("_GUARD_NOS_DICT"), 1)
+        self.assertEqual(uops.count("_STORE_SUBSCR_DICT"), 1)
+        self.assertEqual(uops.count("_BINARY_OP_SUBSCR_DICT"), 1)
+
+    def test_remove_guard_for_known_type_list(self):
+        def f(n):
+            x = 0
+            for _ in range(n):
+                l = [0]
+                l[0] = 1  # Guarded...
+                [a] = l  # ...unguarded!
+                b = l[0]  # ...unguarded!
+                if l:  # ...unguarded!
+                    x += a + b
+            return x
+
+        res, ex = self._run_with_optimizer(f, TIER2_THRESHOLD)
+        self.assertEqual(res, 2 * TIER2_THRESHOLD)
+        self.assertIsNotNone(ex)
+        uops = get_opnames(ex)
+        self.assertEqual(uops.count("_GUARD_NOS_LIST"), 1)
+        self.assertEqual(uops.count("_STORE_SUBSCR_LIST_INT"), 1)
+        self.assertEqual(uops.count("_GUARD_TOS_LIST"), 0)
+        self.assertEqual(uops.count("_UNPACK_SEQUENCE_LIST"), 1)
+        self.assertEqual(uops.count("_BINARY_OP_SUBSCR_LIST_INT"), 1)
+        self.assertEqual(uops.count("_TO_BOOL_LIST"), 1)
+
+    def test_remove_guard_for_known_type_set(self):
+        def f(n):
+            x = 0
+            for _ in range(n):
+                x += "Spam" in {"Spam"}  # Unguarded!
+            return x
+
+        res, ex = self._run_with_optimizer(f, TIER2_THRESHOLD)
+        self.assertEqual(res, TIER2_THRESHOLD)
+        self.assertIsNotNone(ex)
+        uops = get_opnames(ex)
+        self.assertNotIn("_GUARD_TOS_ANY_SET", uops)
+        self.assertIn("_CONTAINS_OP_SET", uops)
+
+    def test_remove_guard_for_known_type_tuple(self):
+        def f(n):
+            x = 0
+            for _ in range(n):
+                t = (1, 2, (3, (4,)))
+                t_0, t_1, (t_2_0, t_2_1) = t  # Unguarded!
+                t_2_1_0 = t_2_1[0]  # Unguarded!
+                x += t_0 + t_1 + t_2_0 + t_2_1_0
+            return x
+
+        res, ex = self._run_with_optimizer(f, TIER2_THRESHOLD)
+        self.assertEqual(res, 10 * TIER2_THRESHOLD)
+        self.assertIsNotNone(ex)
+        uops = get_opnames(ex)
+        self.assertNotIn("_GUARD_TOS_TUPLE", uops)
+        self.assertIn("_UNPACK_SEQUENCE_TUPLE", uops)
+        self.assertIn("_UNPACK_SEQUENCE_TWO_TUPLE", uops)
+        self.assertNotIn("_GUARD_NOS_TUPLE", uops)
+        self.assertIn("_BINARY_OP_SUBSCR_TUPLE_INT", uops)
+
 
 def global_identity(x):
     return x