Merge pull request #300 from staftermath/use_wraps

Use functools.wraps in profile decorator

Author: fabianp

memory_profiler.py

@@ -1130,13 +1130,15 @@ def profile(func=None, stream=None, precision=1, backend='psutil'):
             show_results, stream=stream, precision=precision
         )
         if iscoroutinefunction(func):
+            @wraps(wrapped=func)
             @coroutine
             def wrapper(*args, **kwargs):
                 prof = get_prof()
                 val = yield from prof(func)(*args, **kwargs)
                 show_results_bound(prof)
                 return val
         else:
+            @wraps(wrapped=func)
             def wrapper(*args, **kwargs):
                 prof = get_prof()
                 val = prof(func)(*args, **kwargs)

test/test_attributes.py

@@ -0,0 +1,12 @@
+from memory_profiler import profile
+
+
+@profile
+def test_with_profile(arg1):
+    """dummy doc"""
+    return None
+
+
+if __name__ == '__main__':
+    assert test_with_profile.__doc__ == "dummy doc"
+    assert test_with_profile.__name__ == "test_with_profile"