python的@pytest.mark.parametrize

使用 pytest.mark.parametrize 实现参数化测试

简介

pytest 是一个功能强大的 Python 测试框架，它提供了许多丰富的功能来帮助我们编写高质量的测试代码。其中之一就是 @pytest.mark.parametrize 装饰器，它可以用来实现参数化测试。

参数化测试是一种测试方法，它允许我们使用多组不同的输入参数来运行同一段测试代码。这样可以更全面地覆盖不同的测试场景，提高测试的可靠性和可维护性。

在本文中，我将向你介绍如何使用 @pytest.mark.parametrize 来实现参数化测试。

流程

下面是使用 @pytest.mark.parametrize 实现参数化测试的基本步骤：

接下来，我将详细介绍每个步骤需要做什么，并提供相应的代码示例。

1. 定义测试函数

首先，我们需要定义一个测试函数，用来编写我们的测试代码。这个函数将会被 @pytest.mark.parametrize 装饰器装饰，以支持参数化测试。

def test_example():
    # 测试代码
    pass

这是一个示例函数，你可以根据你的实际需求来编写测试代码。

2. 使用 @pytest.mark.parametrize 装饰器

接下来，我们需要使用 @pytest.mark.parametrize 装饰器来装饰我们的测试函数。这个装饰器将接收参数和参数值，并在测试函数中使用这些参数值来运行测试。

import pytest

@pytest.mark.parametrize("param", [value1, value2, ...])
def test_example(param):
    # 测试代码
    pass

在上面的代码中，param 是参数的名称，[value1, value2, ...] 是参数的值列表。你可以根据你的实际需求来定义参数的名称和值列表。

3. 指定参数和参数值

在上一步中，我们定义了参数和参数值的名称和列表。现在，我们需要为这些参数值指定具体的值。

import pytest

@pytest.mark.parametrize("param", [value1, value2, ...])
def test_example(param):
    # 测试代码
    pass

在上面的代码中，value1、value2 等是具体的参数值。

4. 在测试函数中使用参数

最后，我们需要在测试函数中使用参数。测试函数将会根据参数值的不同运行多次，每次都使用不同的参数值来运行测试。

import pytest

@pytest.mark.parametrize("param", [value1, value2, ...])
def test_example(param):
    # 测试代码
    pass

在上面的代码中，param 是测试函数的参数名称，我们可以在测试函数内部使用它来引用参数值。

示例

下面是一个完整的示例，展示了如何使用 @pytest.mark.parametrize 实现参数化测试：

import pytest

def fibonacci(n):
    if n <= 0:
        return []
    elif n == 1:
        return [0]
    elif n == 2:
        return [0, 1]
    else:
        fib = [0, 1]
        for i in range(2, n):
            fib.append(fib[i-1] + fib[i-2])
        return fib

@pytest.mark.parametrize("n, expected", [
    (0, []),
    (1, [0]),
    (2, [0, 1]),
    (5, [0, 1, 1, 2, 3]),
    (10, [0, 1, 1, 2, 3, 5, 8, 13, 21, 34])
])
def test_fibonacci(n, expected):
    assert fibonacci(n) == expected

在上面的示例中，我们定义了一个斐波那契数列函数 fibonacci，然后使用 @pytest.mark.parametrize 装饰器来测试这个函数