gh-98852: Fix subscription of type aliases (GH-98920)

Fix subscription of type aliases containing bare generic types or types
like TypeVar: for example tuple[A, T][int] and tuple[TypeVar, T][int],
where A is a generic type, and T is a type variable.
This commit is contained in:
Serhiy Storchaka 2022-11-01 09:33:09 +02:00 committed by GitHub
parent f5afb7f233
commit 0e15c31c7e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 43 additions and 0 deletions

View file

@ -3916,6 +3916,34 @@ class GenericTests(BaseTestCase):
# C version of GenericAlias
self.assertEqual(list[A()].__parameters__, (T,))
def test_non_generic_subscript(self):
T = TypeVar('T')
class G(Generic[T]):
pass
class A:
__parameters__ = (T,)
for s in (int, G, A, List, list,
TypeVar, TypeVarTuple, ParamSpec,
types.GenericAlias, types.UnionType):
for t in Tuple, tuple:
with self.subTest(tuple=t, sub=s):
self.assertEqual(t[s, T][int], t[s, int])
self.assertEqual(t[T, s][int], t[int, s])
a = t[s]
with self.assertRaises(TypeError):
a[int]
for c in Callable, collections.abc.Callable:
with self.subTest(callable=c, sub=s):
self.assertEqual(c[[s], T][int], c[[s], int])
self.assertEqual(c[[T], s][int], c[[int], s])
a = c[[s], s]
with self.assertRaises(TypeError):
a[int]
class ClassVarTests(BaseTestCase):
def test_basics(self):