Python使用pytest-benchmark做基准测试

安装pytest-benchmark

pip install pytest-benchmark

代码编写

导入pytest

import pytest

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

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

结果返回(32, 1920)。

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

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

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

Benchmark测试

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

ptest coprime120.py

运行效果如下图：

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