简介
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 参数化的基本用法，以及如何在测试函数中使用参数化装饰器来指定多组输入参数。无论是边界测试还是一般性测试，参数化测试都能帮助我们更好地编写测试用例，从而提高代码质量和可维护性。