pytest是Python领域常用的测试框架，相比Python自带的unittest框架，它的语法更简洁，不需要编写复杂的测试类结构，同时支持丰富的插件生态，能够满足不同场景的测试需求。无论是小型脚本测试还是大型项目的集成测试，都可以通过pytest快速实现。
pytest环境安装
使用pip命令即可快速安装pytest，安装完成后可以通过命令验证是否安装成功。
# 安装pytest pip install pytest # 验证安装版本 pytest --version
编写基础测试用例
pytest对测试用例的命名和文件命名有默认规范，遵循规范可以让框架自动识别测试用例，不需要额外配置。
命名规范
- 测试文件命名：以
test_开头或者以_test结尾，例如test_demo.py、user_test.py - 测试函数命名：以
test_开头，例如test_add() - 测试类命名：以
Test开头，且类中不能包含__init__方法，测试类中的测试方法同样以test_开头
简单测试示例
创建一个test_math.py文件，编写两个简单的测试函数：
# 待测试的函数
def add(a, b):
return a + b
def subtract(a, b):
return a - b
# 测试用例1：测试加法函数
def test_add():
result = add(1, 2)
assert result == 3
# 测试用例2：测试减法函数
def test_subtract():
result = subtract(5, 3)
assert result == 2
常用断言方式
pytest直接使用Python内置的assert关键字进行断言，不需要像unittest那样使用自带的断言方法，使用起来更直观。
| 断言场景 | 断言写法 | 说明 |
|---|---|---|
| 相等判断 | assert a == b | 判断a和b的值相等 |
| 不等判断 | assert a != b | 判断a和b的值不相等 |
| 包含判断 | assert item in container | 判断item在container中 |
| 异常判断 | with pytest.raises(异常类型): | 判断代码块会抛出指定异常 |
异常断言的示例代码如下：
import pytest
def divide(a, b):
if b == 0:
raise ValueError("除数不能为0")
return a / b
def test_divide_zero():
# 断言调用divide(1,0)时会抛出ValueError异常
with pytest.raises(ValueError):
divide(1, 0)
执行测试用例
在命令行中进入测试文件所在目录，直接执行pytest命令即可自动发现并执行所有符合规范的测试用例。
常用执行参数
pytest -v：显示详细的测试执行信息，包括每个测试用例的执行结果pytest test_math.py：只执行指定的测试文件pytest -k "add"：只执行名称中包含add的测试用例pytest --maxfail=2：当失败用例达到2个时停止执行后续测试
测试夹具fixture的使用
fixture是pytest的核心功能之一，可以用来做测试前的准备工作、测试后的清理工作，或者提供测试依赖的数据。
定义一个fixture只需要在函数上添加@pytest.fixture装饰器，然后在测试函数中把fixture函数名作为参数传入即可使用。
import pytest
# 定义fixture，模拟测试前的数据库连接准备
@pytest.fixture
def db_connection():
print("建立数据库连接")
connection = {"status": "connected"}
yield connection # 返回连接对象，测试执行完后执行后续清理
print("关闭数据库连接")
# 使用fixture的测试函数
def test_db_query(db_connection):
# db_connection就是fixture返回的连接对象
assert db_connection["status"] == "connected"
执行这个测试函数时，会先执行fixture中yield之前的逻辑，再执行测试函数，最后执行yield之后的清理逻辑。
参数化测试
如果需要对同一个测试函数传入多组不同的参数进行测试，可以使用@pytest.mark.parametrize装饰器实现参数化，避免重复编写测试函数。
import pytest
def add(a, b):
return a + b
# 参数化测试，传入三组参数
@pytest.mark.parametrize("a,b,expected", [
(1, 2, 3),
(0, 0, 0),
(-1, 1, 0)
])
def test_add_param(a, b, expected):
result = add(a, b)
assert result == expected
执行这个测试函数时，pytest会自动用三组参数分别运行三次测试，每次传入不同的a、b和expected值。