From: Nikita Sobolev <mail@sobolevn.me>
Date: Sat, 4 May 2024 15:08:38 +0000 (+0300)
Subject: gh-118569: Add a test for dynamic PEP695 classes (#118570)
X-Git-Tag: v3.13.0b1~110
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=5f547585fa56c94c5d836b5313a7200f4937ebc4;p=thirdparty%2FPython%2Fcpython.git

gh-118569: Add a test for dynamic PEP695 classes (#118570)
---

diff --git a/Lib/test/test_type_params.py b/Lib/test/test_type_params.py
index 4c5bf6bfd33c..82f1007f9ac9 100644
--- a/Lib/test/test_type_params.py
+++ b/Lib/test/test_type_params.py
@@ -776,6 +776,31 @@ class TypeParamsClassScopeTest(unittest.TestCase):
         self.assertIn(int, C.D.__bases__)
         self.assertIs(C.D.x, str)
 
+
+class DynamicClassTest(unittest.TestCase):
+    def _set_type_params(self, ns, params):
+        ns['__type_params__'] = params
+
+    def test_types_new_class_with_callback(self):
+        T = TypeVar('T', infer_variance=True)
+        Klass = types.new_class('Klass', (Generic[T],), {},
+                                lambda ns: self._set_type_params(ns, (T,)))
+
+        self.assertEqual(Klass.__bases__, (Generic,))
+        self.assertEqual(Klass.__orig_bases__, (Generic[T],))
+        self.assertEqual(Klass.__type_params__, (T,))
+        self.assertEqual(Klass.__parameters__, (T,))
+
+    def test_types_new_class_no_callback(self):
+        T = TypeVar('T', infer_variance=True)
+        Klass = types.new_class('Klass', (Generic[T],), {})
+
+        self.assertEqual(Klass.__bases__, (Generic,))
+        self.assertEqual(Klass.__orig_bases__, (Generic[T],))
+        self.assertEqual(Klass.__type_params__, ())  # must be explicitly set
+        self.assertEqual(Klass.__parameters__, (T,))
+
+
 class TypeParamsManglingTest(unittest.TestCase):
     def test_mangling(self):
         class Foo[__T]: