pytest测试函数返回值无法被同文件后续测试使用的最优解决方法
原写法的问题
- 执行逻辑错误:
test_load_sample中return语句位于assert之前,断言永远不会执行,第一个函数的测试逻辑完全失效 - pytest用例默认隔离:pytest框架设计上默认每个测试函数独立运行,不会存储测试函数的返回值,因此你无法直接通过返回值在不同测试用例之间传递数据;如果强行使用全局变量存储,会导致测试用例高度耦合,前序用例失败会直接引发后续所有关联用例报错,增加问题排查难度
最佳实现方案:使用pytest fixture共享测试数据
pytest内置的fixture工具专门用于实现测试数据、前置依赖的共享,支持自定义作用域,灵活控制数据生成的频次,是这类场景的官方推荐方案。
基础使用示例(适配你的vcf文件加载场景)
首先在测试文件中定义fixture,用来生成共享的测试数据:
import pytest # 导入你自己的业务函数和配置 from file import load_sample, NAME_FILE_1 # scope设为module表示整个测试模块运行期间仅生成一次数据,适合大文件加载场景 @pytest.fixture(scope="module") def loaded_sample_data(): data = load_sample(NAME_FILE_1) return data
之后所有需要用到该数据的测试用例,直接将fixture名称作为参数传入即可:
def test_load_sample(loaded_sample_data): '''Verify all rows of the body of the vcf file are taken''' assert len(loaded_sample_data) == 1400 def test_your_second_function(loaded_sample_data): # 直接使用fixture返回的共享数据作为入参 result = your_second_function(loaded_sample_data) # 编写你的断言逻辑即可 assert result == "你的预期结果"
适配你的简化业务场景示例
针对你给出的function_1、function_2的测试需求,对应的测试代码如下:
import pytest from file import function_1, function_2 # 基础输入数据fixture @pytest.fixture(scope="module") def base_input(): return 1 # function_1生成的结果fixture,共享给后续测试 @pytest.fixture(scope="module") def func1_output(base_input): res = function_1(base_input) assert res == 2 return res def test_function_1(func1_output): assert func1_output == 2 def test_function_2(func1_output): # 直接使用前序生成的共享数据 res = function_2(func1_output) assert res == 3
可选补充说明
如果你的场景确实要求测试用例严格按顺序执行,且后序用例必须依赖前序用例的执行状态,可以安装pytest-dependency插件实现用例依赖,但fixture方案耦合度更低、可维护性更高,是绝大多数场景的首选。
内容的提问来源于stack exchange,提问作者Manolo Dominguez Becerra
相关产品推荐
相关产品推荐

