090. 编写一个函数,实现简单的性能测试工具
090. 编写一个函数,实现简单的性能测试工具
在 Python 中,可以使用 time
模块来实现一个简单的性能测试工具。这个工具可以帮助你测量代码片段的执行时间,从而评估其性能。以下是一个简单的性能测试工具的实现,支持测量函数的执行时间。
示例代码
以下代码实现了一个简单的性能测试工具,支持测量函数的执行时间。
import time
def measure_performance(func, *args, **kwargs):
"""
测量函数的执行时间
:param func: 要测量的函数
:param args: 函数的参数
:param kwargs: 函数的关键字参数
:return: 函数的执行时间(秒)
"""
start_time = time.time() # 记录开始时间
result = func(*args, **kwargs) # 调用函数
end_time = time.time() # 记录结束时间
execution_time = end_time - start_time # 计算执行时间
return execution_time, result
# 示例用法
def example_function(n):
"""示例函数,计算从 1 到 n 的和"""
return sum(range(1, n + 1))
if __name__ == "__main__":
n = 1000000
execution_time, result = measure_performance(example_function, n)
print(f"函数执行时间:{
execution_time:.6f} 秒")
print(