什么是参数化测试？

参数化测试是一种测试方法，允许开发者使用多组不同的输入数据运行同一个测试逻辑，从而避免为每组数据重复编写相似的测试代码。

主要优势：提高代码复用性、增加测试覆盖率、简化测试维护、减少重复代码

使用unittest实现参数化测试

Python标准库中的unittest可以通过子测试或第三方库(如ddt)实现参数化测试。

方法1：使用子测试(subTest)

import unittest

class TestMathOperations(unittest.TestCase):
    # 参数化测试数据
    test_data = [
        (1, 2, 3),
        (5, -3, 2),
        (0, 0, 0),
        (100, 200, 300)
    ]
    
    def test_addition(self):
        """测试加法操作"""
        for a, b, expected in self.test_data:
            with self.subTest(a=a, b=b, expected=expected):
                result = a + b
                self.assertEqual(result, expected)

if __name__ == '__main__':
    unittest.main()

方法2：使用ddt库

ddt库为unittest提供了更简洁的参数化语法：

import unittest
from ddt import ddt, data, unpack

@ddt
class TestStringMethods(unittest.TestCase):
    @data(
        ("hello", "HELLO"),
        ("WORLD", "world"),
        ("PyThOn", "pYtHoN")
    )
    @unpack
    def test_swap_case(self, input_str, expected):
        """测试字符串大小写转换"""
        result = input_str.swapcase()
        self.assertEqual(result, expected)

使用pytest实现参数化测试

pytest框架内置了强大的参数化功能，使用@pytest.mark.parametrize装饰器。

import pytest

# 参数化测试数据
login_data = [
    ("admin", "secret", True),
    ("user", "password123", True),
    ("guest", "wrongpass", False),
    ("", "", False)
]

@pytest.mark.parametrize("username, password, expected", login_data)
def test_login(username, password, expected):
    """测试登录功能"""
    # 实际项目中这里会调用登录函数
    result = login(username, password)  # 假设的登录函数
    assert result == expected

import pytest
from datetime import date

# 使用ID自定义测试名称
user_data = [
    pytest.param("Alice", date(1990, 5, 20), "user", id="valid_user_alice"),
    pytest.param("Bob", date(2005, 3, 15), "child", id="underage_user"),
    pytest.param("", date(1985, 12, 1), "invalid", id="empty_username"),
    pytest.param("Charlie", date(2025, 1, 1), "invalid", id="future_birthdate")
]

@pytest.mark.parametrize("name, dob, expected", user_data)
def test_user_validation(name, dob, expected):
    """测试用户验证逻辑"""
    user_type = validate_user(name, dob)  # 假设的验证函数
    assert user_type == expected

参数化测试的最佳实践

常见参数化应用场景