Pytest学习笔记3——参数化
前言
在讲pytest与unittest的区别文章中，我们知道其中一个区别就是参数化，unittest框架使用的第三方库ddt来参数化的，而pytest框架就直接使用装饰器@pytest.mark.parametrize来对测试用例进行传参。这个是针对测试方法来参数化，还有一种是针对前置处理函数来传参。但往往这两种是可以结合使用。
Params参数化(单个)
前面讲fixture函数时，它有个参数params用来传递参数，并且与request结合使用，先看单个：
import pytest
seq = [1, 2, 3]
@pytest.fixture(params=seq)
def test_data(request):
print("参数")
return request.param
class TestData:
def test_1(self, test_data):
print("用例", test_data)
if __name__ == '__main__':
pytest.main()运行结果：
fixture_test03.py::TestData::test_1[1] 参数
PASSED [ 33%]用例 1
fixture_test03.py::TestData::test_1[2] 参数
PASSED [ 66%]用例 2
fixture_test03.py::TestData::test_1[3] 参数
PASSED [100%]用例 3
============================== 3 passed in 0.05s ==============================Params参数化(多个)
列表里有多个dict数据，request会自动化循环读取每个索引下的值，从0开始，看下面例子：
import pytest
seq = [{'a':1,'b':2},{'c':3,'d':4}]
@pytest.fixture(params=seq)
def test_data(request):
print("参数：%s"%request.param)
return request.param
class TestData:
def test_1(self, test_data):
print("用例", test_data)
if __name__ == '__main__':
pytest.main()运行结果：
============================== 2 passed in 0.04s ==============================
Process finished with exit code 0
参数：{'a': 1, 'b': 2}
PASSED [ 50%]用例 {'a': 1, 'b': 2}
参数：{'c': 3, 'd': 4}
PASSED [100%]用例 {'c': 3, 'd': 4}注意：params需要与request一起使用。
装饰器@pytest.mark.parametrize参数化(单个)
我们做接口测试时，编写测试用例需要输入一组数据，然后执行程序，得到输出数据，经过断言将实际结果与预期结果进行比较，从而得到这条用例执行的结果。但是输入的数据肯定不是一两个，这个时候需要参数化，比如登录的接口，我们可以用测试账号1登录，也可用账号2，账号3...进行登录，如果每一个参数写一条测试用例肯定是不行的，所以引入@pytest.mark.parametrize()装饰器进行参数化，这就是背景。
格式：@pytest.mark.parametrize("参数名称"，"参数列表")，第一个是参数的名称，第二个是参数列表(list)，也就是参数的值。
如：@pytest.mark.parametrize("x",1, 2, 3)
如：@pytest.mark.parametrize('请求方式,接口地址,传参,预期结果',('get','www.baidu.com','{"page":1}','{"code":0,"msg":"成功"})',('post','www.baidu.com','{"page":2}','{"code":0,"msg":"成功"}'))
1、第一个参数是字符串，多个参数中间用逗号隔开
2、第二个参数是list,多组数据用元祖类型;传三个或更多参数也是这样传。list的每个元素都是一个元组，元组里的每个元素和按参数顺序一一对应
3、传一个参数 @pytest.mark.parametrize('参数名'，list) 进行参数化
4、传两个参数@pytest.mark.parametrize('参数名1，参数名2'，[(参数1_data0, 参数2_data0),(参数1_data1, 参数2_data1)]) 进行参数化
seq = [1, 2, 3]
def get_data(x):
print(x)
pass
@pytest.mark.parametrize("x",seq)
def test_data(x):
a = get_data(x)
print(a)
if __name__ == '__main__':
pytest.main()运行结果：
============================== 3 passed in 0.04s ==============================
Process finished with exit code 0
PASSED [ 33%]1
None
PASSED [ 66%]2
None
PASSED [100%]3
None备注：第一个函数输出的是参数值，类中的函数打印的是外面那个函数的返回值，因为没有return，默认返回是None。
与request结合使用
如果装饰器@pytest.mark.parametrize与request结合使用，如果测试方法写在类中，则@pytest.mark.parametrize的参数名称要与@pytest.fixture函数名称保持一致。并且要添加indirect=True参数，目的是把ss_data当做函数去执行，而不是参数。
如：
import pytest
seq = [1,2,3]
@pytest.fixture()
def ss_data(request):
print("参数：%s"%request.param)
return request.param
class TestData:
@pytest.mark.parametrize("ss_data",seq,indirect=True)
def test_1(self,ss_data):
print("用例", ss_data)
if __name__ == '__main__':
pytest.main()运行结果：
collecting ... collected 3 items
fixture_test03.py::TestData::test_1[1]
fixture_test03.py::TestData::test_1[2]
fixture_test03.py::TestData::test_1[3]
============================== 3 passed in 0.04s ==============================
Process finished with exit code 0
参数：1
PASSED [ 33%]用例 1
参数：2
PASSED [ 66%]用例 2
参数：3
PASSED [100%]用例 3注意1：如果将@pytest.mark.parametrize参数名称改成x的执行，就会报错如：
注意2：如果不使用fixture装饰器函数，也不使用外部函数，也是可以的，例子如下：
import pytest
seq = [1,2,3]
class TestData:
@pytest.mark.parametrize("x",seq,indirect=False)
def test_1(self,x):
print("用例", x)
if __name__ == '__main__':
pytest.main()运行结果：
============================== 3 passed in 0.04s ==============================
Process finished with exit code 0
PASSED [ 33%]用例 1
PASSED [ 66%]用例 2
PASSED [100%]用例 3可以看出，装饰器@pytest.mark.parametrize参数传递的是参数名称了，而不是函数了。根据开关indirect自由控制，默认是False.
装饰器@pytest.mark.parametrize参数化(多个)
多个参数和单个参数写法差不多，只不过多个参数中间用逗号隔开，列表中用元组来分组，举个例子(不带类的测试方法和没有使用request参数）：
import pytest
seq = [(1,2),(3,4)]
def get_data(x,y):
print(x)
print(y)
pass
@pytest.mark.parametrize("x,y",seq)
def test_data(x,y):
get_data(x,y)
# print(a)
if __name__ == '__main__':
pytest.main()运行结果：
============================== 2 passed in 0.04s ==============================
Process finished with exit code 0
PASSED [ 50%]1
2
PASSED [100%]3
4这就是把参数列表list里面的元组中两个值，看作一对参数(x,y)。
与request结合使用
前面介绍过单个使用的场景，现在是多个参数的时候，如何使用呢？
其实差不多，但是需要注意一些问题，先看个例子：
import pytest
seq = [(1,2),(3,4)]
@pytest.fixture()
def get_data(request):
x = request.param[0]
y = request.param[1]
print(x,111)
print(y,222)
# print(request.param)
pass
# indirect=True声明x是个函数
@pytest.mark.parametrize("get_data",seq)
def test_data(get_data):
print(get_data)
if __name__ == '__main__':
pytest.main()运行结果：
plugins: allure-pytest-2.8.6, rerunfailures-5.0
collecting ... collected 2 items
fixture_005.py::test_data[get_data0] PASSED [ 50%](1, 2)
fixture_005.py::test_data[get_data1] PASSED [100%](3, 4)
============================== 2 passed in 0.02s ==============================
Process finished with exit code 0我们可以看出来fixture装饰器函数根本没有执行，这是为什么呢？
因为使用@pytest.mark.parametrize装饰器参数化时，默认是以参数而不是函数，这里的@pytest.mark.parametrize("get_data",seq)中get_data是参数名称，不是函数，如果要声明为函数，需要添加：indirect=True。
我们将其改之，再运行：
import pytest
seq = [(1,2),(3,4)]
@pytest.fixture()
def get_data(request):
x = request.param[0]
y = request.param[1]
print(x,111)
print(y,222)
# print(request.param)
pass
# indirect=True声明x是个函数
@pytest.mark.parametrize("get_data",seq,indirect=True)
def test_data(get_data):
print(get_data)
if __name__ == '__main__':
pytest.main()运行结果：
fixture_005.py::test_data[get_data0] 1 111
2 222
PASSED [ 50%]None
fixture_005.py::test_data[get_data1] 3 111
4 222
PASSED [100%]None
============================== 2 passed in 0.02s ==============================
Process finished with exit code 0多个fixture与@pytest.mark.parametrize组合
用例前面可以有多个条件，参数化装饰器也可以叠加，使用parametrize装饰器叠加时，用例组合是2个参数个数相乘
import pytest
seq1 = [1,2,3]
seq2 = [4,5,6]
@pytest.fixture()
def get_seq1(request):
seq1 = request.param
print("seq1:",seq1)
return seq1
@pytest.fixture()
def get_seq2(request):
seq2 = request.param
print("seq2:", seq2)
return seq2
@pytest.mark.parametrize("get_seq1",seq1,indirect=True)
@pytest.mark.parametrize("get_seq2",seq2,indirect=True)
def test_1(get_seq1,get_seq2):
print(get_seq1,11)
print(get_seq2,22)
if __name__ == '__main__':
pytest.main()运行结果：
collecting ... collected 9 items
fixture_test04.py::test_1[4-1]
fixture_test04.py::test_1[4-2]
fixture_test04.py::test_1[4-3]
fixture_test04.py::test_1[5-1]
fixture_test04.py::test_1[5-2]
fixture_test04.py::test_1[5-3]
fixture_test04.py::test_1[6-1]
fixture_test04.py::test_1[6-2]
fixture_test04.py::test_1[6-3]
============================== 9 passed in 0.07s ==============================
Process finished with exit code 0
seq1: 1
seq2: 4
PASSED [ 11%]1 11
4 22
seq1: 2
seq2: 4
PASSED [ 22%]2 11
4 22
seq1: 3
seq2: 4
PASSED [ 33%]3 11
4 22
seq1: 1
seq2: 5
PASSED [ 44%]1 11
5 22
seq1: 2
seq2: 5
PASSED [ 55%]2 11
5 22
seq1: 3
seq2: 5
PASSED [ 66%]3 11
5 22
seq1: 1
seq2: 6
PASSED [ 77%]1 11
6 22
seq1: 2
seq2: 6
PASSED [ 88%]2 11
6 22
seq1: 3
seq2: 6
PASSED [100%]3 11
6 22如果使用类，也是一样的，如图：
seq1和seq2参数都是3个，所以总的用例数就是9个。
单独使用@pytest.mark.parametrize参数组合
总结
pytest与unittest的区别之一参数化已经讲完，希望可以帮助你学习pytest框架。
如果对你有帮助或喜欢自动化测试开发的朋友，可以加入右下方QQ交流群学习与探索，更多干货与你分享。
腾讯云开发者
