[英]Decorator for timeit.timeit method?
I'm trying to write a simple time decorator to measure time taken by functions. 我正在尝试编写一个简单的时间装饰器来测量函数所用的时间。 However the code below is giving our recursion error.
但是下面的代码给出了我们的递归错误。 What's wrong with it?
它出什么问题了?
import timeit
def measure(func):
def wrapper():
func_name = func.__name__
setup="from __main__ import {}".format(func_name)
op_time = timeit.timeit('{}()'.format(func_name), number = 2, setup=setup)
print(ot)
return wrapper
@measure
def sample():
return 10
sample()
Output 产量
RecursionError Traceback (most recent call last)
<ipython-input-61-e079e1bd7fba> in <module>()
15 return 10
16
---> 17 sample()
<ipython-input-61-e079e1bd7fba> in wrapper()
7 func_name = func.__name__
8 setup="from __main__ import {}".format(func_name)
----> 9 op_time = timeit.timeit('{}()'.format(func_name), number = 2, setup=setup)
10 print(ot)
11 return wrapper
~/anaconda3/lib/python3.6/timeit.py in timeit(stmt, setup, timer, number, globals)
231 number=default_number, globals=None):
232 """Convenience function to create Timer object and call timeit method."""
--> 233 return Timer(stmt, setup, timer, globals).timeit(number)
234
235 def repeat(stmt="pass", setup="pass", timer=default_timer,
~/anaconda3/lib/python3.6/timeit.py in timeit(self, number)
176 gc.disable()
177 try:
--> 178 timing = self.inner(it, self.timer)
179 finally:
180 if gcold:
~/anaconda3/lib/python3.6/timeit.py in inner(_it, _timer)
... last 4 frames repeated, from the frame below ...
<ipython-input-61-e079e1bd7fba> in wrapper()
7 func_name = func.__name__
8 setup="from __main__ import {}".format(func_name)
----> 9 op_time = timeit.timeit('{}()'.format(func_name), number = 2, setup=setup)
10 print(ot)
11 return wrapper
RecursionError: maximum recursion depth exceeded while calling a Python object
I'm interested in learning what's wrong with my existing code, please don't post alternative solutions. 我有兴趣了解我现有代码的错误,请不要发布替代解决方案。
from functools import wraps
from time import time
def measure(func):
@wraps(func)
def _time_it(*args, **kwargs):
start = int(round(time() * 1000))
try:
return func(*args, **kwargs)
finally:
end_ = int(round(time() * 1000)) - start
print(f"Total execution time: {end_ if end_ > 0 else 0} ms")
return _time_it
@measure
def hello():
print('hello world')
hello()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.