@pytest.mark. parametrize装饰器可以实现对测试用例的参数化,方便测试数据的获取。
@pytest.mark. parametrize的基本使用:
方便测试函数对测试数据的获取。 方法: parametrize(argnames, argvalues, indirect=False, ids=None, scope=None) 常用参数: argnames:参数名 argvalues:参数对应值,类型必须为list 当参数为一个时格式:[value] 当参数个数大于一个时,格式为:[(param_value1,param_value2.....),(param_value1,param_value2.....)] 使用方法: @pytest.mark.parametrize(argnames,argvalues)
单个参数
# file_name: test_parametrize.py import pytest class Test_D: @pytest.mark.parametrize("a", [1, 2, 3]) # 参数a被赋予3个值,test_a将会运行3遍 def test_a(self, a): # 参数必须和parametrize里面的参数一致 print('\n------------------> test_a has ran, and a = {}'.format(a)) assert 1 == a if __name__ == '__main__': pytest.main(['-s', 'test_parametrize.py'])
运行结果:
从运行结果中可以看到test_a方法被执行了3遍,说明参数a参数化成功。
多个参数
# file_name:test_parametrize.py import pytest class Test_D: @pytest.mark.parametrize("a,b", [(1, 2), (2, 3), (3, 4)]) # 参数a,b均被赋予三个值,函数会运行三遍 def test_b(self, a, b): # 参数必须和parametrize里面的参数一致 print('\n------------------> test_b has ran, and a = {}, b = {}'.format(a, b)) assert 1 if __name__ == '__main__': pytest.main(['-s', 'test_parametrize.py'])
运行结果:
从运行结果中可以看到test_b方法被执行了三遍,说明参数a,b都已经被参数化。
利用函数的返回值进行参数化
# file_name: test_parametrize.py import pytest # 定义返回参数值的函数 def return_test_data(): return [(1, 2), (2, 3), (3, 4)] class Test_D: @pytest.mark.parametrize("a,b", return_test_data()) # 使用函数返回值的方式传入参数值 def test_c(self, a, b): print('\n------------------> test_c has ran, and a = {}, b = {}'.format(a, b)) assert 1 if __name__ == '__main__': pytest.main(['-s', 'test_parametrize.py'])
运行结果:
从运行结果中可以看到test_c方法被执行了三遍,这说明使用函数的返回值同样可以做到参数化。
参数组合
获得多个参数化参数的所有组合,可以堆叠参数化装饰器:
# file_name: test_parametrize.py import pytest class Test_D: @pytest.mark.parametrize("x", [1, 2]) @pytest.mark.parametrize("y", [3, 4]) def test_d(self, x, y): print("\n------------------> test_d has ran, and x={}, y={}".format(x, y)) if __name__ == '__main__': pytest.main(['-s', 'test_parametrize.py'])
运行结果:
从运行结果中可以看到test_d方法被执行了四遍,这是因为参数x和参数y分别被赋予了2个值,组合起来就是2*2=4次,说明装饰器叠加可以获得所有参数的组合。