diff options
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/reprlib.py | 1 | ||||
| -rw-r--r-- | Lib/test/test_reprlib.py | 9 |
2 files changed, 10 insertions, 0 deletions
diff --git a/Lib/reprlib.py b/Lib/reprlib.py index a92b3e3..840dd0e 100644 --- a/Lib/reprlib.py +++ b/Lib/reprlib.py @@ -29,6 +29,7 @@ def recursive_repr(fillvalue='...'): wrapper.__name__ = getattr(user_function, '__name__') wrapper.__qualname__ = getattr(user_function, '__qualname__') wrapper.__annotations__ = getattr(user_function, '__annotations__', {}) + wrapper.__wrapped__ = user_function return wrapper return decorating_function diff --git a/Lib/test/test_reprlib.py b/Lib/test/test_reprlib.py index e7216d4..502287b 100644 --- a/Lib/test/test_reprlib.py +++ b/Lib/test/test_reprlib.py @@ -765,5 +765,14 @@ class TestRecursiveRepr(unittest.TestCase): for name in assigned: self.assertIs(getattr(wrapper, name), getattr(wrapped, name)) + def test__wrapped__(self): + class X: + def __repr__(self): + return 'X()' + f = __repr__ # save reference to check it later + __repr__ = recursive_repr()(__repr__) + + self.assertIs(X.f, X.__repr__.__wrapped__) + if __name__ == "__main__": unittest.main() |