软件测试|详解 Pytest 参数化：简化测试用例的编写

news2025/1/11 11:39:17

简介

Pytest 是一个广泛使用的 Python 测试框架，它提供了丰富的功能来编写和执行测试用例。其中一个强大的特性是参数化，它允许我们通过一种简洁的方式运行多个输入参数的相似测试用例，从而减少冗余的代码。本文将详细介绍 Pytest 的参数化功能以及如何使用它来简化测试用例的编写。

参数化测试的优势

参数化测试允许我们为一个测试函数提供多组输入数据，以验证函数在不同输入条件下的行为。这有助于更全面地覆盖测试用例，减少重复代码，以及在代码变更时更容易维护测试。

参数化测试的基本用法

以下是使用 Pytest 参数化的基本步骤：

导入pytest

import pytest

创建测试函数，并在函数上使用 @pytest.mark.parametrize 装饰器，指定参数名称和参数值列表：

@pytest.mark.parametrize("input_value, expected_output", [
    (1, 2),
    (2, 4),
    (3, 6)
])
def test_multiply_by_two(input_value, expected_output):
    result = input_value * 2
    assert result == expected_output

在这个例子中，我们创建了一个名为 test_multiply_by_two 的测试函数。通过 @pytest.mark.parametrize 装饰器，我们传递了两个参数列表：input_value 和 expected_output。每个元组中的值会分别传递给测试函数，使我们能够在多组输入条件下运行测试。

执行测试

我们可以在命令行中执行这个用例，也可以在界面中，点击绿色小三角形运行用例，如下图：

参数化测试案例

假设我们要测试一个函数，该函数接收两个数字并返回它们的和。我们可以通过参数化测试来验证函数的正确性：

def add(a, b):
    return a + b

@pytest.mark.parametrize("a, b, expected_result", [
    (1, 2, 3),
    (0, 0, 0),
    (-1, 1, 0),
])
def test_add_function(a, b, expected_result):
    result = add(a, b)
    assert result == expected_result

在这个例子中，我们测试了不同的输入情况，以确保 add 函数在各种情况下都能正确运行。

使用参数化测试进行边界测试

参数化测试特别适用于边界测试，因为可以轻松地为边界值和边界值附近的值设置多组输入。例如，假设我们有一个函数来计算年龄组别：

def age_group(age):
    if age < 18:
        return "Underage"
    elif age >= 18 and age < 65:
        return "Adult"
    else:
        return "Senior"

我们可以通过参数化测试来测试各种年龄值的输出：

@pytest.mark.parametrize("age, expected_group", [
    (10, "Underage"),
    (20, "Adult"),
    (70, "Senior"),
    (18, "Adult"),
    (65, "Senior"),
])
def test_age_group_function(age, expected_group):
    result = age_group(age)
    assert result == expected_group

总结

Pytest 的参数化功能是一个强大的工具，可以显著简化测试用例的编写，同时还能更全面地覆盖不同的测试情况。本文介绍了 Pytest 参数化的基本用法，以及如何在测试函数中使用参数化装饰器来指定多组输入参数。无论是边界测试还是一般性测试，参数化测试都能帮助我们更好地编写测试用例，从而提高代码质量和可维护性。

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走：