summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/tests/documentation.py18
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()