前言
当我们要使用pytest
输入多个数据对一个功能进行测试时,如果写多个测试用例的话,那就完全体现不出通过代码来执行测试的优势了,这个时候,就轮到pytest
的参数化功能闪亮登场了。所谓参数化,就是就是把测试过程中的数据提取出来,通过参数传递不同的数据来驱动用例运行。其实也就是数据驱动的概念。本文就给大家介绍一下pytest
的参数化和数据驱动。
参数化方式
在unittest
中,我们使用ddt库配合unittest
实现数据驱动,在pytest中并不需要额外的库,通过pytest.mark.parametrize()
即可实现参数化。parametrize()
的第一个参数是用逗号分割的字符串列表,第二个参数是一个值列表。
pytest
有三种传参方式:
@pytest.mark.parametrize()
通过装饰器方式进行参数化(最常使用)pytest.fixture()
方式进行参数化,fixture
装饰的函数可以作为参数传入其他函数conftest.py
文件中存放参数化函数,可作用于模块内的所有测试用例
单个传参
示例代码如下:
代码语言:python代码运行次数:0复制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
执行结果如下:
代码语言:python代码运行次数:0复制============================= 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()
装饰器接收两个参数,第一个参数是以字符串的形式标识用例函数的参数,第二个参数以列表或元组的形式传递测试数据。
多个传参
示例代码如下:
代码语言:python代码运行次数:0复制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
运行结果如下:
代码语言:python代码运行次数:0复制============================= 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
testcase2test_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等文件来实现更高层次的参数化和数据驱动测试,希望本文能够帮到大家!