【pytest】paramtrize 参数化、数据驱动

最新推荐文章于 2025-09-25 15:20:28 发布

原创最新推荐文章于 2025-09-25 15:20:28 发布 · 2.4k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#paramtrize #参数化 #数据驱动

pytest 专栏收录该内容

19 篇文章

订阅专栏

本文深入讲解pytest中的参数化功能，包括单参数、多参数及多次使用参数化的具体应用，通过实例展示如何有效利用参数化提高测试效率。

paramtrize 参数化：

pytest 中可以使用 @pytest.mak.paramtrize(argsnames, argsvalues, ids=None) 来实现参数化；

主要参数说明：

argsnames ：参数名，是个字符串，如中间用逗号分隔则表示为多个参数名；
argsvalues ：参数值，参数组成的列表，列表中有几个元素，就会生成几条测试用例；

使用方法：

使用 @pytest.mak.paramtrize() 装饰测试方法；
parametrize('data', param) 中的 ‘data’ 是自定义的参数名，param 是引入的参数列表；
将自定义的参数名 data 作为参数传给测试用例 test_func；
然后就可以在测试用例内部使用 data 的参数了；

整个过程中，pytest 将参数列表 [1, 2, 3] 中的 3 个元素取出来，用每个元素生成 1 条测试用例，并且将元素赋值给 data 作为测试用例的参数，由测试用例自由使用；

# test_moduleName.py

import pytest

class TestClass_1:
    """测试类1"""

    param = [1, 2, 3]
    
    # 'data' ：自定义的参数名
    # param  ：参数列表
    @pytest.mark.parametrize('data', param)
    def test_func(self, data):  # 将参数名作为测试方法的参数
        """测试方法"""

        print('\n参数：', data)  # 实际得到的参数

# 执行结果：
============================= test session starts =============================
test_moduleName.py::TestClass_1::test_func[1] 
参数：data =  1
PASSED
test_moduleName.py::TestClass_1::test_func[2] 
参数：data =  2
PASSED
test_moduleName.py::TestClass_1::test_func[3] 
参数：data =  3
PASSED

============================== 3 passed in 0.03s ==============================

Process finished with exit code 0

多个参数名的用法：

前面讲了参数名可以就用逗号分隔，表示为多个参数名，当使用多个参数名时要注意，参数列表中的元素必须是可拆包的数组，并且数组内的元素数必须和参数名的数量相同；

# test_moduleName.py

import pytest


class TestClass_1:
    """测试类1"""

    param = [('admin1', 'pwd001'), ('admin2', 'pwd002'), ('admin3', 'pwd003')]
    # 因为参数名是 2 个，所以列表的每个元素都必须是长度为 2 的数组；
    @pytest.mark.parametrize('uname, pwd', param)
    def test_func(self, uname, pwd):
        """测试方法"""

        print('\n参数：uname = ', uname)
        print('参数：pwd = ', pwd)

# 执行结果
============================= test session starts =============================
test_moduleName.py::TestClass_1::test_func[admin1-pwd001] 
参数：uname =  admin1
参数：pwd =  pwd001
PASSED
test_moduleName.py::TestClass_1::test_func[admin2-pwd002] 
参数：uname =  admin2
参数：pwd =  pwd002
PASSED
test_moduleName.py::TestClass_1::test_func[admin3-pwd003] 
参数：uname =  admin3
参数：pwd =  pwd003
PASSED

============================== 3 passed in 0.02s ==============================

多次使用 paramtrize ：

同一个测试用例还可以同时适用多个 @pytest.mak.paramtrize()，多个 paramtrize 的所有元素互相组合（类似笛卡儿乘积）生成大量测试用例；

# test_moduleName.py

import pytest


class TestClass_1:
    """测试类1"""

    @pytest.mark.parametrize('date3', ['value3-1', 'value3-2', 'value3-3'])
    @pytest.mark.parametrize('date2', ['value2-1', 'value2-2', 'value2-3'])
    @pytest.mark.parametrize('date1', ['value1-1', 'value1-2', 'value1-3'])
    def test_func(self, date1, date2, date3):
        """测试方法"""

        print('\n参数：date1 = ', date1)
        print('参数：date2 = ', date2)
        print('参数：date3 = ', date3)

如下所示，3 个 paramtrize() ，共有 3 * 3 * 3 = 27 种组合，pytest 便会生成 27 条测试用例！

test_moduleName.py::TestClass_1::test_func[value1.1-value2.1-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.1-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.1-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.2-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.2-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.2-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.3-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.3-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.1-value2.3-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.1-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.1-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.1-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.2-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.2-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.2-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.3-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.3-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.2-value2.3-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.1-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.1-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.1-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.2-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.2-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.2-value3.3] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.3-value3.1] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.3-value3.2] 
test_moduleName.py::TestClass_1::test_func[value1.3-value2.3-value3.3]