Fixes memoize stripping __doc__ on decorated class

cevans87 · Jul 13, 2019 · 6ae230f · 6ae230f
1 parent aa3b50f
commit 6ae230f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/atools/memoize_decorator.py b/atools/memoize_decorator.py
@@ -323,7 +323,7 @@ def __call__(cls, *args, **kwargs):
             class Wrapped(decoratee, metaclass=WrappedMeta):
                 pass
 
-            return type(decoratee.__name__, (Wrapped,), {})
+            return type(decoratee.__name__, (Wrapped,), {'__doc__': decoratee.__doc__})
 
     @staticmethod
     def reset_all() -> None:

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='atools',
-    version='0.6.1',
+    version='0.6.2',
     packages=['', 'atools'],
     python_requires='>=3.6',
     url='https://github.com/cevans87/atools',

diff --git a/test/test_memoize_decorator.py b/test/test_memoize_decorator.py
@@ -482,6 +482,14 @@ def foo(_: Foo) -> None:
         del f
         assert r() is None
 
+    def test_memoize_class_preserves_doc(self) -> None:
+
+        @memoize
+        class Foo:
+            """Foo doc"""
+
+        assert Foo.__doc__ == "Foo doc"
+
 
 if __name__ == '__main__':
     unittest.main(verbosity=2)