安装pytest-benchmark

pip install pytest-benchmark

代码编写

导入pytest

import pytest

我们以“在120以内的正整数中找出和120互质的正整数，并统计个数和求和”为例，比较穷举和使用数学技巧完成该任务的性能。

def coprime_120_1():count = 0sum = 0for i in range(1, 121):if i % 2 != 0 and i % 3 != 0 and i % 5 != 0:count += 1sum += ireturn count, sum   

结果返回(32, 1920)。

def coprime_120_2():num_1 = (1 + 120) * 120 / 2num_2 = (2 + 120) * 60 / 2num_3 = (3 + 120) * 40 / 2num_5 = (5 + 120) * 24 / 2num_2_3 = (6 + 120) * 20 / 2num_2_5 = (10 + 120) * 12 / 2num_3_5 = (15 + 120) * 8 / 2num_2_3_5 = (30 + 120) * 4 / 2count = 120 - 60 - 40 - 24 + 20 + 12 + 8 - 4sum = num_1 - num_2 - num_3 - num_5 + num_2_3 + num_2_5 + num_3_5 - num_2_3_5return count, sum

def test_coprime_120_1(benchmark):result = benchmark(coprime_120_1)assert (32, 1920) == resultdef test_coprime_120_2(benchmark):result = benchmark(coprime_120_2)assert (32, 1920) == result   

Benchmark测试

执行ptest命令，可以运行基准测试。假定包含benchmark的文件名称为coprime120.py，那么命令如下：

ptest coprime120.py

运行效果如下图：

可见，数学在程序开发中起到很大的作用。