Fixes memoize stripping __doc__ on decorated class

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:

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',

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)