简单的测试工具代码,它可以输出不同程序执行的时间。我们可以使用Python的time
模块来实现这个功能。
import time
def test_function(func, *args, **kwargs):
"""
测试函数执行时间的工具函数
:param func: 待测试的函数
:param *args: 函数的位置参数
:param **kwargs: 函数的关键字参数
"""
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
execution_time = end_time - start_time
print(f"函数 '{func.__name__}' 执行时间: {execution_time:.6f} 秒")
return result
# 示例测试函数
def example_function(n):
"""
这是一个示例函数,模拟一个耗时的操作
"""
total = 0
for i in range(n):
total += i
time.sleep(0.1) # 模拟耗时操作
return total
if __name__ == "__main__":
# 在这里调用测试函数并传入待测试的函数和参数
test_function(example_function, 5)
定义了一个test_function
来测试其他函数的执行时间。你可以调用test_function
,并将要测试的函数和参数传递给它。在这个示例中,我们测试了example_function
函数的执行时间,并传递了参数 5
。运行代码后,将输出该函数的执行时间。你可以根据需要测试其他函数,只需在if __name__ == "__main__":
这个条件下调用test_function
即可。