mirror of https://github.com/python/cpython
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:
parent
f5afb7f233
commit
0e15c31c7e
|
@ -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):
|
||||||
|
|
|
@ -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])
|
||||||
|
|
|
@ -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.
|
|
@ -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);
|
||||||
|
|
Loading…
Reference in New Issue