pytest单元测试实例

pytest是一个流行的Python测试框架，它使得编写和组织单元测试变得简单而直观。下面是一个简单的pytest单元测试的例子：

# 首先，我们有一个函数，这个函数是我们想要测试的对象
def add(a, b):
    return a + b

# 使用pytest，我们需要创建一个测试模块。这里我们导入pytest模块，并对add函数进行测试
import pytest

# 测试添加两个正数的情况
def test_add_positive_numbers():
    assert add(2, 3) == 5

# 测试当其中一个参数为零时，结果是否正确
def test_add_one_to_zero():
    assert add(1, 0) == 1

# 还可以测试异常处理情况
def test_add_with_invalid_input():
    with pytest.raises(TypeError):
        add("one", 2)

# 运行pytest命令，它会自动找到并运行所有以test_开头的函数
if __name__ == "__main__":
    pytest.main()

在这个例子中，`test_add_positive_numbers()`、`test_add_one_to_zero()` 和 `test_add_with_invalid_input()` 分别是单元测试用例，它们通过断言验证了`add()`函数的行为。