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函数。

相关推荐
liu****16 分钟前
4.基础开发工具(一)
linux·开发语言·1024程序员节
文火冰糖的硅基工坊16 分钟前
[人工智能-大模型-72]:模型层技术 - 模型训练六大步:①数据预处理 - 基本功能与对应的基本组成函数
开发语言·人工智能·python
小龙报21 分钟前
《C语言疑难点 --- 字符函数和字符串函数专题(上)》
c语言·开发语言·c++·算法·学习方法·业界资讯·visual studio
凭君语未可26 分钟前
深度解析Java的多态特性
java·开发语言
csbysj202037 分钟前
DTD 元素:XML 与 SGML 文档结构解析指南
开发语言
傻童:CPU1 小时前
C语言练习题
c语言·开发语言
极地星光1 小时前
协程:实战与系统集成(高级篇)
开发语言
0和1的舞者1 小时前
《Git:从入门到精通(八)——企业级git开发相关内容》
大数据·开发语言·git·搜索引擎·全文检索·软件工程·初学者
liulilittle2 小时前
LwIP协议栈MPA多进程架构
服务器·开发语言·网络·c++·架构·lwip·通信
水淹萌龙2 小时前
玩转 Go 表达式引擎:expr 实战指南
开发语言·后端·golang