### 使用 Python 和 pytest 编写单元测试用例 pytest 是一种强大的 Python 测试框架，能够帮助开发者快速编写高质量的测试用例。以下是关于如何使用 pytest 编写单元测试的具体说明。 #### 1. 安装 pytest 在开始之前，需要安装 pytest 库。可以通过 pip 工具完成安装： ```bash pip install pytest ``` #### 2. 创建简单的测试函数 pytest 的核心理念之一是简化测试过程。通常情况下，不需要额外导入模块即可定义测试函数[^4]。下面是一个基本的例子： 假设有一个名为 `calculator.py` 的文件，其中包含如下代码： ```python def add(a, b): return a + b ``` 可以为其创建一个对应的测试文件 `test_calculator.py`，内容如下： ```python def test_add(): from calculator import add result = add(1, 2) assert result == 3 ``` 上述代码中，`assert` 关键字用于验证预期结果是否与实际结果一致[^1]。 #### 3. 运行测试 保存并运行以下命令来执行测试： ```bash pytest test_calculator.py ``` 如果一切正常，pytest 将报告成功的结果；如果有错误，则会显示具体的失败原因。 #### 4. 参数化测试 为了减少重复代码量，pytest 提供了参数化的功能。例如，对于加法运算的不同输入组合，可以用以下方式实现批量测试： ```python import pytest from calculator import add @pytest.mark.parametrize("a,b,expected", [ (1, 2, 3), (-1, -1, -2), (0, 0, 0), ]) def test_add_parametrized(a, b, expected): result = add(a, b) assert result == expected ``` 此方法允许一次性定义多个测试场景，从而提高效率[^3]。 #### 5. 封装测试用例 当项目规模增大时，合理地封装测试逻辑变得尤为重要。通过将公共设置或清理操作提取到 fixture 中，可以使测试更加清晰和维护友好[^2]。例如： ```python import pytest from calculator import subtract @pytest.fixture def setup_data(): return {"num1": 10, "num2": 5} def test_subtract(setup_data): num1 = setup_data["num1"] num2 = setup_data["num2"] result = subtract(num1, num2) assert result == 5 ``` --- ### 总结 以上介绍了如何利用 pytest 框架为 Python 代码构建高效的单元测试环境。从基础概念到高级特性如参数化和 fixtures 都进行了探讨[^2]。