summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorGregory Beauregard <greg@greg.red>2022-03-03 01:14:52 (GMT)
committerGitHub <noreply@github.com>2022-03-03 01:14:52 (GMT)
commit75d2d945b4e28ca34506b2d4902367b61a8dff82 (patch)
treefa0598788ccb8c94268c1f9e1c1f8e21d30da1de /Lib
parent50ec3453c50c490eca7e990103c79671bd08ab2e (diff)
downloadcpython-75d2d945b4e28ca34506b2d4902367b61a8dff82.zip
cpython-75d2d945b4e28ca34506b2d4902367b61a8dff82.tar.gz
cpython-75d2d945b4e28ca34506b2d4902367b61a8dff82.tar.bz2
bpo-46643: Fix stringized P.args/P.kwargs with get_type_hints (GH-31238)
Diffstat (limited to 'Lib')
-rw-r--r--Lib/test/test_typing.py12
-rw-r--r--Lib/typing.py3
2 files changed, 14 insertions, 1 deletions
diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py
index 8fcc24c..bd99204 100644
--- a/Lib/test/test_typing.py
+++ b/Lib/test/test_typing.py
@@ -5187,6 +5187,18 @@ class ParamSpecTests(BaseTestCase):
self.assertEqual(repr(P.args), "P.args")
self.assertEqual(repr(P.kwargs), "P.kwargs")
+ def test_stringized(self):
+ P = ParamSpec('P')
+ class C(Generic[P]):
+ func: Callable["P", int]
+ def foo(self, *args: "P.args", **kwargs: "P.kwargs"):
+ pass
+
+ self.assertEqual(gth(C, globals(), locals()), {"func": Callable[P, int]})
+ self.assertEqual(
+ gth(C.foo, globals(), locals()), {"args": P.args, "kwargs": P.kwargs}
+ )
+
def test_user_generics(self):
T = TypeVar("T")
P = ParamSpec("P")
diff --git a/Lib/typing.py b/Lib/typing.py
index 9d668b3..6e0c68c 100644
--- a/Lib/typing.py
+++ b/Lib/typing.py
@@ -181,7 +181,8 @@ def _type_check(arg, msg, is_argument=True, module=None, *, allow_special_forms=
return arg
if isinstance(arg, _SpecialForm) or arg in (Generic, Protocol):
raise TypeError(f"Plain {arg} is not valid as type argument")
- if isinstance(arg, (type, TypeVar, ForwardRef, types.UnionType, ParamSpec)):
+ if isinstance(arg, (type, TypeVar, ForwardRef, types.UnionType, ParamSpec,
+ ParamSpecArgs, ParamSpecKwargs)):
return arg
if not callable(arg):
raise TypeError(f"{msg} Got {arg!r:.100}.")