Fix signatures if a decorator has no signatures, fixes #1705

2025-12-06 22:14:27 +08:00 · 2020-12-28 00:47:10 +01:00
parent 04c1c0f871
commit 5d2aed34f4
2 changed files with 10 additions and 0 deletions
--- a/test/test_inference/test_signature.py
+++ b/test/test_inference/test_signature.py
@@ -102,6 +102,8 @@ class X:
        (partialmethod_code + 'X().d(', None),
        (partialmethod_code + 'X.c(', 'func(a, b)'),
        (partialmethod_code + 'X.d(', None),
+
+        ('import contextlib\n@contextlib.contextmanager\ndef f(x): pass\nf(', 'f(x)'),
    ]
 )
 def test_tree_signature(Script, environment, code, expected):