写一个简单的pytest用例
时间: 2024-12-09 16:17:54 浏览: 10
Pytest是一个流行的Python测试框架,它允许开发者编写简洁、易于理解的测试代码。下面是一个简单的pytest用例示例,这个例子测试了一个函数`add`,该函数接受两个整数并返回它们的和:
```python
# 定义要测试的函数
def add(x, y):
return x + y
# 使用pytest的测试模块
import pytest
# 测试用例 - 验证加法是否按预期工作
@pytest.mark.parametrize("x, y, expected", [(1, 2, 3), (0, 5, 5), (-3, 7, 4)])
def test_addition(x, y, expected):
assert add(x, y) == expected, f"Addition failed for {x} and {y}. Expected {expected}, got {add(x, y)}"
# 运行测试
if __name__ == "__main__":
pytest.run()
```
在这个测试用例中:
- `@pytest.mark.parametrize`装饰器用于定义一组参数化的输入值,每个参数对都会独立运行一次测试。
- `assert`语句用于断言`add`函数的结果是否等于期望值。
- `if __name__ == "__main__": pytest.run()`确保只有当脚本直接运行时才会执行测试,而不是作为模块导入。
运行这个文件时,pytest会自动查找以`test_`开头的函数,并报告每个测试的结果。
阅读全文