Python测试利器——pytest框架

在Python开发中，单元测试是确保代码质量的重要手段。一个好的测试框架可以帮助我们快速编写和执行测试用例，发现潜在的错误，并确保代码的稳定性。pytest是Python社区中广泛使用的一个测试框架，具有简单易用、灵活多变的特点。
一、安装pytest
要使用pytest，首先需要安装pytest库。可以使用pip命令进行安装：

pip install pytest

二、编写测试用例
在Python中，测试用例通常以函数的形式编写，并使用def关键字定义。每个测试用例函数都应该以test_开头，以便pytest能够识别并执行它。下面是一个简单的例子：

def test_addition():
assert 2 + 2 == 4

在这个例子中，我们定义了一个名为test_addition的函数，用于测试加法运算。我们使用assert关键字来断言2 + 2的结果应该等于4。如果断言失败，pytest将报告一个错误。
三、运行测试用例
安装好pytest后，可以使用pytest命令来运行测试用例。只需在终端中进入包含测试用例的目录，并运行以下命令：

pytest

pytest将自动查找并执行所有以test_开头的函数。运行结束后，它将输出通过的测试用例数量和失败的测试用例数量。如果有失败的测试用例，它将显示详细的错误信息。
四、参数化测试用例
有时候我们需要对同一组数据执行相同的测试逻辑，但每次使用不同的输入和预期输出。这时可以使用pytest的参数化功能。在函数定义中，我们可以使用*args和**kwargs来接收参数，并在断言中使用这些参数。下面是一个示例：

def test_addition(arg1, expected):
assert arg1 + arg1 == expected

在这个例子中，我们将arg1和expected作为参数传递给测试函数。然后，在断言中使用这些参数来检查加法运算的结果是否符合预期。要运行参数化的测试用例，可以使用以下命令：

pytest -k test_addition --arg1=2 --expected=4

这将只运行名为test_addition的测试用例，并将arg1设置为2，将expected设置为4。
五、使用fixture注入依赖项
在测试用例中，有时需要共享一些依赖项或初始化一些状态。这时可以使用pytest的fixture功能。Fixture可以在测试函数执行之前和之后执行一些操作，并可以将依赖项注入到测试函数中。下面是一个示例：

import pytest
@pytest.fixture()
def setup_data():
data = {'key': 'value'}
yield data
print('Teardown')