]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
bpo-42740: Fix get_args for PEP 585 collections.abc.Callable (GH-23963)
authorKen Jin <28750310+Fidget-Spinner@users.noreply.github.com>
Mon, 28 Dec 2020 20:06:19 +0000 (04:06 +0800)
committerGitHub <noreply@github.com>
Mon, 28 Dec 2020 20:06:19 +0000 (12:06 -0800)
PR 1/2. Needs backport to 3.9.

Lib/test/test_typing.py
Lib/typing.py

index c340c8a898289d09f07b2fd9a6c61d09530dd803..3dcc31fe5be56c32324444248c378ddcf674b06f 100644 (file)
@@ -3048,6 +3048,11 @@ class GetUtilitiesTestCase(TestCase):
         self.assertEqual(get_args(Callable), ())
         self.assertEqual(get_args(list[int]), (int,))
         self.assertEqual(get_args(list), ())
+        self.assertEqual(get_args(collections.abc.Callable[[int], str]), ([int], str))
+        self.assertEqual(get_args(collections.abc.Callable[..., str]), (..., str))
+        self.assertEqual(get_args(collections.abc.Callable[[], str]), ([], str))
+        self.assertEqual(get_args(collections.abc.Callable[[int], str]),
+                         get_args(Callable[[int], str]))
 
 
 class CollectionsAbcTests(BaseTestCase):
index 7b79876d4ebc700b6b6655066fce4ebd8909143a..67e95dd78f34d48233612945e405863490f14d8c 100644 (file)
@@ -1684,13 +1684,11 @@ def get_args(tp):
     """
     if isinstance(tp, _AnnotatedAlias):
         return (tp.__origin__,) + tp.__metadata__
-    if isinstance(tp, _GenericAlias):
+    if isinstance(tp, (_GenericAlias, GenericAlias)):
         res = tp.__args__
         if tp.__origin__ is collections.abc.Callable and res[0] is not Ellipsis:
             res = (list(res[:-1]), res[-1])
         return res
-    if isinstance(tp, GenericAlias):
-        return tp.__args__
     return ()