在Python中进行基于性能的(基准)单元测试,可以使用timeit
模块。timeit
模块可以帮助您测量代码的执行时间,并提供统计数据,以便更好地了解代码的性能。
以下是一个使用timeit
模块的示例:
import timeit
def test_function():
return sum(range(1000))
def test_function2():
return sum(range(1, 1001))
# 测量test_function的执行时间
time_taken = timeit.timeit(test_function, number=1000)
print(f"test_function执行时间: {time_taken:.6f}秒")
# 测量test_function2的执行时间
time_taken2 = timeit.timeit(test_function2, number=1000)
print(f"test_function2执行时间: {time_taken2:.6f}秒")
在这个示例中,我们定义了两个函数test_function
和test_function2
,并使用timeit.timeit()
函数分别测量它们的执行时间。number
参数表示要运行的测试次数,这里我们设置为1000次。
输出结果将显示每个函数的执行时间,您可以根据这些信息来比较不同函数的性能。
请注意,timeit
模块会在多次运行之间进行随机化,以减少其他进程对测试结果的影响。如果您需要更精确的测试结果,可以使用timeit.repeat()
函数,该函数将运行多次测试并返回一个结果列表,您可以从中计算平均值和标准差。
领取专属 10元无门槛券
手把手带您无忧上云