pytest中的pytest-benchmark插件,可以来执行基准测试也可以用来进行比较,它能测试代码的性能、内存使用率、CPU使用率等,并生成性能报告,本文就来详细介绍下pytest-benchmark如何做性能测试。
官网地址:
https://pytest-benchmark.readthedocs.io/en/latest/
1. 安装 pytest-benchmark
安装命令:
pip install pytest-benchmark
2. 编写性能测试用例
编写一个测试函数,并使用pytest.mark.benchmark标记将其标记为性能测试用例。
代码实例:
import pytest
@pytest.mark.benchmark
def test_something(benchmark):
result = benchmark(some_function, some_arguments)
assert result == expected_result
在这个性能测试用例中使用pytest-benchmark插件提供的benchmark装饰器将测试函数标记为性能测试用例,并将测试结果存储在一个result对象中,使用assert语句检查结果是否符合预期。
3. 运行性能测试
在命令行中运行pytest命令,使用`--benchmark-autosave`选项将测试结果保存到指定文件中。
命令