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 # C version of GenericAlias
self.assertEqual(list[A()].__parameters__, (T,)) 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): class ClassVarTests(BaseTestCase):
def test_basics(self): def test_basics(self):

View File

@ -1437,6 +1437,10 @@ class _GenericAlias(_BaseGenericAlias, _root=True):
new_args = [] new_args = []
for old_arg in self.__args__: for old_arg in self.__args__:
if isinstance(old_arg, type):
new_args.append(old_arg)
continue
substfunc = getattr(old_arg, '__typing_subst__', None) substfunc = getattr(old_arg, '__typing_subst__', None)
if substfunc: if substfunc:
new_arg = substfunc(new_arg_by_param[old_arg]) new_arg = substfunc(new_arg_by_param[old_arg])

View File

@ -0,0 +1,4 @@
Fix subscription of type aliases containing bare generic types or types like
:class:`~typing.TypeVar`: for example ``tuple[A, T][int]`` and
``tuple[TypeVar, T][int]``, where ``A`` is a generic type, and ``T`` is a
type variable.

View File

@ -458,6 +458,13 @@ _Py_subs_parameters(PyObject *self, PyObject *args, PyObject *parameters, PyObje
} }
for (Py_ssize_t iarg = 0, jarg = 0; iarg < nargs; iarg++) { for (Py_ssize_t iarg = 0, jarg = 0; iarg < nargs; iarg++) {
PyObject *arg = PyTuple_GET_ITEM(args, iarg); PyObject *arg = PyTuple_GET_ITEM(args, iarg);
if (PyType_Check(arg)) {
Py_INCREF(arg);
PyTuple_SET_ITEM(newargs, jarg, arg);
jarg++;
continue;
}
int unpack = _is_unpacked_typevartuple(arg); int unpack = _is_unpacked_typevartuple(arg);
if (unpack < 0) { if (unpack < 0) {
Py_DECREF(newargs); Py_DECREF(newargs);