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

This commit is contained in:
Nikita Sobolev 2024-05-04 18:08:38 +03:00 committed by GitHub
parent f34e965e52
commit 5f547585fa
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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]: