测试Python函数用时

132 阅读1分钟

开始定义这块代码,要测函数用时直接在函数上面加 @fn_timer 例如:

from functools import wraps

def fn_timer(function):
     @wraps(function)
     def function_timer(*args, **kwargs):
         t0 = time.time()
         result = function(*args, **kwargs)
         t1 = time.time()
         print("Total time running %s: %s seconds" %
             (function.__name__, str(t1-t0))
             )
         return result
     return function_timer
@ fn_timer
def test():
    for i in range(1,100000):
        a = i*2
test()