软件测试/测试开发|Pytest参数化神器，pytest.mark.parametrize()使用

发布时间：2024年01月04日

在这里插入图片描述

前言

当我们要使用pytest输入多个数据对一个功能进行测试时，如果写多个测试用例的话，那就完全体现不出通过代码来执行测试的优势了，这个时候，就轮到pytest的参数化功能闪亮登场了。所谓参数化，就是就是把测试过程中的数据提取出来，通过参数传递不同的数据来驱动用例运行。其实也就是数据驱动的概念。本文就给大家介绍一下pytest的参数化和数据驱动。

参数化方式

在unittest中，我们使用ddt库配合unittest实现数据驱动，在pytest中并不需要额外的库，通过pytest.mark.parametrize()即可实现参数化。parametrize()的第一个参数是用逗号分割的字符串列表，第二个参数是一个值列表。

pytest有三种传参方式：

@pytest.mark.parametrize() 通过装饰器方式进行参数化（最常使用）
pytest.fixture()方式进行参数化，fixture装饰的函数可以作为参数传入其他函数
conftest.py文件中存放参数化函数，可作用于模块内的所有测试用例

单个传参

示例代码如下：

import pytest

def add(a,b):
    return a+b

class TestParams:

    @pytest.mark.parametrize('a',[1,2,3,4,5])
    def test_par1(self,a):
        assert add(a,1)==a+1

执行结果如下：

============================= test session starts =============================
collecting ... collected 5 items

test_c.py::TestParams::test_par1[1] PASSED
test_c.py::TestParams::test_par1[2] PASSED
test_c.py::TestParams::test_par1[3] PASSED
test_c.py::TestParams::test_par1[4] PASSED
test_c.py::TestParams::test_par1[5] PASSED

============================== 5 passed in 0.04s ==============================

注：@pytest.mark.parametrize() 装饰器接收两个参数，第一个参数是以字符串的形式标识用例函数的参数，第二个参数以列表或元组的形式传递测试数据。

在这里插入图片描述

多个传参

示例代码如下：

import pytest

def add(a,b):
    return a+b

class TestParams:

    @pytest.mark.parametrize('a,b,c',[[1,2,3],[3,4,5],[1,3,4],[2,5,7]])
    def test_par1(self,a,b,c):
        assert add(a,b)==c

运行结果如下：

============================= test session starts =============================
collecting ... collected 4 items

test_c.py::TestParams::test_par1[1-2-3] PASSED
test_c.py::TestParams::test_par1[3-4-5] FAILED
testcase2\test_c.py:19 (TestParams.test_par1[3-4-5])
7 != 5

Expected :5
Actual   :7
<Click to see difference>

self = <testcase2.test_c.TestParams object at 0x000001789DB13108>, a = 3, b = 4
c = 5

    @pytest.mark.parametrize('a,b,c',[[1,2,3],[3,4,5],[1,3,4],[2,5,7]])
    def test_par1(self,a,b,c):
>       assert add(a,b)==c
E       assert 7 == 5
E        +  where 7 = add(3, 4)

test_c.py:22: AssertionError





test_c.py::TestParams::test_par1[1-3-4] PASSED
test_c.py::TestParams::test_par1[2-5-7] PASSED

========================= 1 failed, 3 passed in 0.06s =========================

注：有多个参数时，参数之间要用逗号分隔，参数名称和个数要一一对应

总结

参数化测试是pytest中一个强大的功能，可以帮助开发者更高效地编写测试用例，提高代码覆盖率，并简化测试代码。后续我们还将介绍使用yaml等文件来实现更高层次的参数化和数据驱动测试，希望本文能够帮到大家！

获取更多技术资料，请点击！