summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorYury Selivanov <yselivanov@sprymix.com>2015-05-28 01:56:53 (GMT)
committerYury Selivanov <yselivanov@sprymix.com>2015-05-28 01:56:53 (GMT)
commit46c759d76d02cb873bae55fac37dfe71750e3bef (patch)
tree42418ab91cb39107010584ef758513f2f03179a5 /Lib
parent68fe9aa58c139e3a6f8c9188760bb237added7d6 (diff)
downloadcpython-46c759d76d02cb873bae55fac37dfe71750e3bef.zip
cpython-46c759d76d02cb873bae55fac37dfe71750e3bef.tar.gz
cpython-46c759d76d02cb873bae55fac37dfe71750e3bef.tar.bz2
Issue 24298: Fix signature() to properly unwrap wrappers around bound methods
Diffstat (limited to 'Lib')
-rw-r--r--Lib/inspect.py8
-rw-r--r--Lib/test/test_inspect.py13
2 files changed, 21 insertions, 0 deletions
diff --git a/Lib/inspect.py b/Lib/inspect.py
index a91f749..49e66ce 100644
--- a/Lib/inspect.py
+++ b/Lib/inspect.py
@@ -1911,6 +1911,14 @@ def _signature_internal(obj, follow_wrapper_chains=True, skip_bound_arg=True):
# Was this function wrapped by a decorator?
if follow_wrapper_chains:
obj = unwrap(obj, stop=(lambda f: hasattr(f, "__signature__")))
+ if isinstance(obj, types.MethodType):
+ # If the unwrapped object is a *method*, we might want to
+ # skip its first parameter (self).
+ # See test_signature_wrapped_bound_method for details.
+ return _signature_internal(
+ obj,
+ follow_wrapper_chains=follow_wrapper_chains,
+ skip_bound_arg=skip_bound_arg)
try:
sig = obj.__signature__
diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py
index 8d92f82..58a6cd2 100644
--- a/Lib/test/test_inspect.py
+++ b/Lib/test/test_inspect.py
@@ -1939,6 +1939,19 @@ class TestSignatureObject(unittest.TestCase):
with self.assertRaisesRegex(ValueError, 'invalid method signature'):
self.signature(Test())
+ def test_signature_wrapped_bound_method(self):
+ # Issue 24298
+ class Test:
+ def m1(self, arg1, arg2=1) -> int:
+ pass
+ @functools.wraps(Test().m1)
+ def m1d(*args, **kwargs):
+ pass
+ self.assertEqual(self.signature(m1d),
+ ((('arg1', ..., ..., "positional_or_keyword"),
+ ('arg2', 1, ..., "positional_or_keyword")),
+ int))
+
def test_signature_on_classmethod(self):
class Test:
@classmethod