returnes

导航

装饰器测试函数执行时间 

 1 # -*- coding: utf-8 -*-
 2 import time
 3 import datetime
 4 
 5 def calculate_function_run_time(func):
 6     """
 7     Calculate the running time of the function
 8     :param func: the function need to been calculated
 9     :return:
10     """
11 
12     def call_fun(*args, **kwargs):
13         start_time = time.time()
14         f = func(*args, **kwargs)
15         end_time = time.time()
16         print('%s() run time：%s s' % (func.__name__, int(end_time - start_time)))
17         return f
18 
19     return call_fun
20 
21 
22 def calculate_function_run_time_ms(func):
23     def call_fun(*args, **kwargs):
24         start_time = time.time()
25         f = func(*args, **kwargs)
26         end_time = time.time()
27         print('%s() run time：%s ms' % (func.__name__, int(1000 * (end_time - start_time))))
28         return f
29 
30     return call_fun

 

posted on 2020-11-27 19:32  returnes  阅读(76)  评论(0)    收藏  举报