@pytest.mark.parametrize 装饰器用于参数化测试,使单个测试函数可以接受多个参数集,从而进行多次测试。

import pytest

@pytest.mark.parametrize("a, b, expected", [
    (3, 4, 7),
    (-1, 1, 0),
    (-1, -1, -2),
])
def test_add(a, b, expected):
    assert a + b == expected