diff options
| -rw-r--r-- | src/tests/documentation.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/tests/documentation.py b/src/tests/documentation.py index 35224cb..13d1415 100644 --- a/src/tests/documentation.py +++ b/src/tests/documentation.py @@ -1818,6 +1818,24 @@ def singledispatch_example2(): return g, V +@decorator +def warn_slow(func, duration=0, *args, **kwargs): + t0 = time.time() + res = func(*args, **kwargs) + dt = time.time() - t0 + if dt > duration: + print('%s is slow' % func.__name__) + return res + + +@warn_slow() +def operation(): + """ + >>> operation() + operation is slow + """ + + if __name__ == '__main__': import doctest doctest.testmod() |
