基于pytest的JUnit风格数据驱动测试setup与teardown实现问询

我拥有大量执行相同操作但使用不同数据的测试用例，希望通过pytest实现数据驱动测试，目前已采用典型的JUnit风格完成了如下代码实现，现咨询该实现的相关问题（代码如下：）

import pytest
import random
d = {1: 'Hi',2: 'How',3: 'Are',4:'You ?'}
def setup_function(function):
    print("setUp",flush=True)
def teardown_function(functions):
    print("tearDown",flush=True)
@pytest.mark.parametrize("test_input", [1,2,3,4])
def test_one(test_input):
    print("Test with data " + str(test_input))
    print(d[test_input])  # 补全你未写完的代码部分

看起来你已经用pytest的核心参数化特性搭好了数据驱动测试的基础框架，我来帮你梳理下这个实现的关键点，以及你可能关心的常见问题和优化方向：

1. 当前实现的核心逻辑说明 #

• 数据驱动的核心：@pytest.mark.parametrize
  这个装饰器会把你传入的[1,2,3,4]列表拆分成独立的测试用例，相当于自动帮你写了4个逻辑相同但参数不同的test_one函数，每个参数对应一次测试执行。
• JUnit风格的固件：setup_function/teardown_function
  这两个函数是pytest的函数级固件，会在每个测试用例执行前运行setup_function，执行后运行teardown_function，和JUnit里的setUp/tearDown行为完全一致。
• 数据存储：字典d
  你用字典把测试输入和对应的预期输出/关联内容绑定在一起，这种方式对于小规模测试数据来说简洁直观。

2. 常见问题与优化建议 #

如何同时传递输入和预期结果？ #

如果你的测试需要验证输出是否符合预期，直接把输入和预期结果配对传入参数化装饰器即可：

@pytest.mark.parametrize("test_input, expected", [(1, 'Hi'), (2, 'How'), (3, 'Are'), (4, 'You ?')])
def test_one(test_input, expected):
    print(f"Test with data {test_input}")
    # 增加断言逻辑，让测试更有意义
    assert d[test_input] == expected

如何让固件只运行一次（而非每个用例都跑）？ #

当前的setup_function是函数级的，每个测试用例都会触发一次。如果想在所有测试用例前后只执行一次初始化/清理操作，可以改用模块级固件：

def setup_module(module):
    print("全局初始化：在所有测试用例前只运行一次", flush=True)
def teardown_module(module):
    print("全局清理：在所有测试用例后只运行一次", flush=True)

测试数据量很大时，如何管理？ #

如果测试数据有成百上千条，把数据写在代码里会很臃肿，建议从外部文件加载，比如JSON：

import json

# 从外部JSON文件加载测试数据
with open('test_data.json', 'r', encoding='utf-8') as f:
    test_data = json.load(f)

@pytest.mark.parametrize("test_input, expected", test_data.items())
def test_one(test_input, expected):
    assert d[test_input] == expected

test_data.json的内容可以是：

{"1": "Hi", "2": "How", "3": "Are", "4": "You ?"}

如何跳过特定的测试用例？ #

如果某条数据暂时不需要测试，可以用pytest.param给对应的参数标记跳过：

@pytest.mark.parametrize("test_input", [
    1,
    pytest.param(2, marks=pytest.mark.skip(reason="暂时跳过该用例，待修复数据")),
    3,
    4
])
def test_one(test_input):
    print(f"Test with data {test_input}")
    print(d[test_input])

如何给不同的测试用例添加自定义标记？ #

如果需要对部分用例分组（比如标记为“冒烟测试”），同样可以用pytest.param添加标记：

@pytest.mark.parametrize("test_input", [
    pytest.param(1, marks=pytest.mark.smoke),
    2,
    pytest.param(3, marks=pytest.mark.smoke),
    4
])
def test_one(test_input):
    print(f"Test with data {test_input}")

运行时可以通过pytest -m smoke只执行标记为冒烟测试的用例。

内容的提问来源于stack exchange，提问作者David Fliguer