pytest是Python中最轻量、灵活且易上手的测试框架,支持函数式写法、自动发现测试、友好断言及丰富插件;三步可搭好环境:规范目录结构、安装pytest与pytest-cov、配置pyproject.toml并运行首个测试。

为什么从pytest开始写单元测试
pytest不是唯一选择,但它是目前Python生态中最轻量、最灵活、最易上手的测试框架。它天然支持函数式测试写法,无需继承TestCase类,用普通函数就能写测试;自动发现test_开头的函数和test_*.py文件;断言失败时能清晰展示变量值,比unittest更友好;插件丰富(如pytest-cov测覆盖率、pytest-xdist跑并发),适合从小项目快速铺开到中大型工程。
三步搭建可运行的测试环境
先确保项目结构合理:把源码放在src/或直接放根目录(不推荐混放),测试代码统一放在tests/目录下,与源码同级。接着安装核心工具:
- pip install pytest pytest-cov —— 基础框架 + 覆盖率统计
- 在项目根目录加pyproject.toml,配置pytest默认行为(比如自动加载conftest.py、设置测试路径)
- 写一个极简测试函数,例如tests/test_math.py里放
def test_add(): assert 1 + 1 == 2,终端运行pytest,看到绿色的.就说明环境通了
写好单个测试函数的关键细节
一个测试函数应该只验证一件事,名字要见名知意(如test_calculate_total_with_discount而非test_func1)。避免在测试里做I/O、网络请求或依赖全局状态。常用技巧包括:
- 用
@pytest.mark.parametrize批量验证多组输入输出,减少重复代码 - 用
pytest.raises(ValidationError)断言某段代码必须抛出指定异常 - 对浮点数比较用
assert abs(a - b) 或pytest.approx(),避免精度误差误报失败 - 测试前准备数据用
setup函数或fixture,而不是在每个测试里手动构造对象
用fixture管理测试依赖与生命周期
fixture是pytest最强大的机制之一,用来复用测试资源(如数据库连接、临时文件、mock对象)。它比setup/teardown更灵活,支持函数、类、模块、会话多个作用域。例如:
立即学习“Python免费学习笔记(深入)”;
- 定义一个
@pytest.fixture函数返回一个初始化好的API客户端,所有需要它的测试函数只需把函数名当参数传入,pytest自动注入 - 加
scope="session"让数据库连接在整个测试会话只创建一次,提升速度 - 用
autouse=True让日志清理、临时目录创建等通用动作自动生效,不用显式声明 - 在conftest.py里声明fixture,它会自动被同级及子目录下所有测试文件识别
让测试真正跑进CI并持续有效
本地能过不等于上线安全。把测试纳入CI流程,关键是稳定、快速、可读。建议:
- 在CI脚本中运行
pytest --cov=src --cov-report=term-missing,强制检查未覆盖的关键分支 - 用
--maxfail=3避免单次运行失败太多而卡住CI队列 - 给耗时测试打
@pytest.mark.slow标签,CI默认跳过,开发时可选执行 - 定期用
pytest --cache-clear和--tb=short排查缓存干扰与错误堆栈噪音 - 把
pytest.ini或pyproject.toml中的配置同步进团队文档,避免“在我机器上是好的”问题










