python pytest框架

# 示例代码：使用 pytest 框架进行单元测试

# 导入 pytest 模块
import pytest

# 定义一个简单的函数，用于测试
def add(a, b):
    return a + b

# 编写测试用例
def test_add():
    # 测试两个正数相加
    assert add(1, 2) == 3
    # 测试一个正数和一个负数相加
    assert add(-1, 1) == 0
    # 测试两个负数相加
    assert add(-1, -1) == -2

# 使用 pytest.mark.parametrize 进行参数化测试
@pytest.mark.parametrize("a, b, expected", [
    (1, 2, 3),
    (-1, 1, 0),
    (-1, -1, -2),
    (0, 0, 0)
])
def test_add_parametrize(a, b, expected):
    assert add(a, b) == expected

# 运行此文件时，pytest 会自动发现并运行所有以 test_ 开头的函数
if __name__ == "__main__":
    pytest.main()

解释说明：

1. 导入模块：我们首先导入了 pytest 模块。
2. 定义函数：编写了一个简单的 add 函数，用于加法运算。
3. 编写测试用例：使用 def test_add(): 定义了一个测试用例，使用 assert 关键字来验证函数的输出是否符合预期。
4. 参数化测试：使用 @pytest.mark.parametrize 装饰器来进行参数化测试，可以一次性测试多个输入组合。
5. 运行测试：当直接运行此文件时，pytest.main() 会自动发现并执行所有的测试用例。

通过这种方式，你可以轻松地对 Python 代码进行单元测试。