testpython.git

@@ -8,6 +8,7 @@ License: MIT

 import cProfile

 import tempfile

 import pstats

+import functools

 from typing import Union, Optional, Tuple, List, Callable

@@ -20,6 +21,7 @@ def profile(sort: Union[Tuple, List, str]='cumulative',

     then it's the percentage of total lines. (Eg., .2 = 20%)

     Output goes to stdout if filename is None, otherwise to filename."""

     def outer(fun: Callable) -> Callable:

+        @functools.wraps(fun)

         def inner(*args, **kwargs):

             prof = cProfile.Profile()

             ret = prof.runcall(fun, *args, **kwargs)