Add test cases

JelleZijlstra · JelleZijlstra · commit 394d1288f111 · 2023-04-26T18:48:48.000-06:00
diff --git a/Lib/test/test_type_params.py b/Lib/test/test_type_params.py
@@ -305,6 +305,42 @@ class Foo[T: Undefined, U: (Undefined,)]:
         self.assertEqual(type_params[1].__constraints__, ("defined",))
 
 
+class TypeParamsClassScopeTest(unittest.TestCase):
+    def test_alias(self):
+        class X:
+            T = int
+            type U = T
+        self.assertIs(X.U.__value__, int)
+
+        code = textwrap.dedent("""\
+            glb = "global"
+            class X:
+                cls = "class"
+                type U = (glb, cls)
+
+            assert X.U.__value__ == ("global", "class"), X.U.__value__
+        """)
+        exec(code, {})
+
+    def test_bound(self):
+        class X:
+            T = int
+            def foo[U: T](self): ...
+        self.assertIs(X.foo.__type_params__[0].__bound__, int)
+
+        code = textwrap.dedent("""\
+            glb = "global"
+            class X:
+                cls = "class"
+                def foo[T: glb, U: cls](self): ...
+
+            T, U = X.foo.__type_params__
+            assert T.__bound__ == "global"
+            assert U.__bound__ == "class"
+        """)
+        exec(code, {})
+
+
 class TypeParamsTraditionalTypeVars(unittest.TestCase):
     def test_traditional_01(self):
         code = textwrap.dedent("""\
diff --git a/Python/symtable.c b/Python/symtable.c
@@ -1958,8 +1958,8 @@ symtable_visit_typeparam(struct symtable *st, typeparam_ty tp)
             VISIT(st, expr, tp->v.TypeVar.bound);
             if (!symtable_exit_block(st))
                 VISIT_QUIT(st, 0);
+        }
         break;
-    }
     case TypeVarTuple_kind:
         if (!symtable_add_def(st, tp->v.TypeVarTuple.name, DEF_TYPE_PARAM | DEF_LOCAL, LOCATION(tp)))
             VISIT_QUIT(st, 0);
