使用parametrize()函数生成不同组合的测试数据

parametrize()函数是pytest框架中的一个内置函数，它用于为测试函数提供不同的参数组合。通过使用parametrize()函数，我们可以简单且优雅地生成多个测试数据，并将这些数据传递给测试函数进行测试。下面是一个1000字的具体示例，展示了parametrize()函数的不同用法和使用示例。

首先，我们需要导入pytest库和parametrize()函数：

import pytest
from pytest import parametrize

接下来，我们可以创建一个测试函数来接收参数并执行测试。在这个例子中，我们创建了一个名为"test_addition"的函数，用于测试两个数相加的结果是否正确：

def test_addition(a, b, expected):
    result = a + b
    assert result == expected

在上面的测试函数中，我们接收了三个参数a、b和expected。这些参数将在parametrize()函数中进行定义。

然后，我们可以使用parametrize()函数为测试函数提供不同的参数组合。parametrize()函数接受两个参数：参数名称和参数值。参数值可以是一个列表，其中包含多个元组或多个字典，每个元组或字典表示一个参数组合。下面是parametrize()函数的几种常见用法示例：

1. 使用元组列表：

@parametrize('a, b, expected', [(1, 2, 3), (3, 4, 7), (5, 6, 11)])
def test_addition(a, b, expected):
    result = a + b
    assert result == expected

在这个示例中，我们使用一个包含多个元组的列表作为参数值。每个元组表示一个参数组合，其中元组的每个元素分别对应测试函数中定义的参数a、b和expected。这样，测试函数将运行三次，分别使用不同的参数组合进行测试。

2. 使用字典列表：

@parametrize('data', [{'a': 1, 'b': 2, 'expected': 3}, {'a': 3, 'b': 4, 'expected': 7}, {'a': 5, 'b': 6, 'expected': 11}])
def test_addition(data):
    result = data['a'] + data['b']
    assert result == data['expected']

在这个示例中，我们使用一个包含多个字典的列表作为参数值。每个字典表示一个参数组合，其中字典的键对应测试函数中定义的参数名称，而字典的值对应参数的值。这样，测试函数将运行三次，分别使用不同的参数组合进行测试。

3. 使用元组和标签：

@parametrize('a, b, expected', [(1, 2, 3), (3, 4, 7), pytest.param(5, 6, 11, marks=pytest.mark.skip)])
def test_addition(a, b, expected):
    result = a + b
    assert result == expected

在这个示例中，我们使用元组列表作为参数值。不同之处在于，我们使用了pytest.param()函数为某些参数组合添加了标签(mark)。在这里，我们为最后一个参数组合添加了一个名为"skip"的标签，表示这个参数组合将被跳过。这样，测试函数将根据参数组合的标签进行相应的处理。

通过使用parametrize()函数，我们可以方便地生成不同的测试数据，并能根据需要进行灵活的标记和过滤。这种参数化的测试方法可以帮助我们更全面地测试代码，并提高测试覆盖率。