pytest的参数化操作

在使用pytest进行测试时,我们经常需要针对不同的输入值执行相同的测试逻辑。为了提高测试效率,避免重复编写相似的测试用例,pytest提供了参数化测试的功能。这一功能允许我们轻松执行同一测试用例的多个变体,从而覆盖更多的测试场景。

一、基本用法

pytest的参数化测试主要通过@pytest.mark.parametrize装饰器实现。该装饰器接收两个参数:第一个是字符串,表示参数的名称;第二个是参数的值列表,每个值都会作为一次独立的测试执行。

python 复制代码
import pytest  
  
@pytest.mark.parametrize("input,expected", [  
    (1, 2),  
    (2, 3),  
    (3, 4)  
])  
def test_increment(input, expected):  
    assert input + 1 == expected

|---|
| |

在这个例子中,test_increment函数会被执行三次,每次都会使用parametrize装饰器中定义的一组参数。

二、高级用法

除了基本用法外,pytest的参数化测试还支持一些高级功能,如使用ids为测试用例设置别名、使用indirect参数进行间接参数化等。

使用ids设置别名

ids参数允许我们为每个测试用例设置一个别名,这样在运行测试时,pytest会显示这些别名而不是参数的默认值。

python 复制代码
import pytest  
  
@pytest.mark.parametrize("input,expected", [  
    pytest.param(1, 2, marks=pytest.mark.basic),  
    pytest.param(2, 3, marks=pytest.mark.basic, id="basic_2"),  
    pytest.param(3, 4, id="basic_3"),  
])  
def test_increment(input, expected):  
    assert input + 1 == expected

|---|
| |

在这个例子中,第二个和第三个测试用例分别使用了id参数来设置别名。

使用indirect进行间接参数化

indirect参数允许我们将参数传递给fixture函数,而不是直接传递给测试用例函数。这可以用于更复杂的测试场景,其中fixture函数需要根据不同的参数来设置测试环境。

python 复制代码
import pytest  
  
@pytest.fixture  
def my_fixture(request):  
    param = request.param  
    return param  
  
@pytest.mark.parametrize("my_fixture", [1, 2, 3], indirect=True)  
def test_with_fixture(my_fixture):  
    assert my_fixture > 0

|---|
| |

在这个例子中,my_fixture是一个fixture函数,它接收一个参数并返回该参数。test_with_fixture函数使用@pytest.mark.parametrize装饰器进行参数化,并通过indirect=True告诉pytest将参数传递给fixture函数。

相关推荐
留不住丨晚霞33 分钟前
说说SpringBoot常用的注解?
java·开发语言
大模型真好玩36 分钟前
准确率飙升!GraphRAG如何利用知识图谱提升RAG答案质量(额外篇)——大规模文本数据下GraphRAG实战
人工智能·python·mcp
198937 分钟前
【零基础学AI】第30讲:生成对抗网络(GAN)实战 - 手写数字生成
人工智能·python·深度学习·神经网络·机器学习·生成对抗网络·近邻算法
hardStudy_h43 分钟前
C++——内联函数与Lambda表达式
开发语言·jvm·c++
applebomb1 小时前
没合适的组合wheel包,就自行编译flash_attn吧
python·ubuntu·attention·flash
爱学习的茄子1 小时前
React Hooks驱动的Todo应用:现代函数式组件开发实践与组件化架构深度解析
前端·react.js·面试
艾莉丝努力练剑1 小时前
【C语言】学习过程教训与经验杂谈:思想准备、知识回顾(三)
c语言·开发语言·数据结构·学习·算法
Chasing__Dreams1 小时前
python--杂识--18.1--pandas数据插入sqlite并进行查询
python·sqlite·pandas
彭泽布衣2 小时前
python2.7/lib-dynload/_ssl.so: undefined symbol: sk_pop_free
python·sk_pop_free
witton3 小时前
Go语言网络游戏服务器模块化编程
服务器·开发语言·游戏·golang·origin·模块化·耦合