pytest之测试用例参数化

背景

本文总结pytest的测试用例参数化。

说明

软件测试中，输入相应值，检查期望值，是常见测试方法。
在自动化测试中，一个测试用例对应一个测试点，通常一组测试数据无法完全覆盖测试范围，所以，需要参数化来传递多组数据。
pytest的测试用例参数化使用如下装饰器即可完成。

@pytest.mark.parametrize(argnames, argvalues)
# 参数：
# argnames:以逗号分隔的字符串
# argvaluse: 参数值列表，若有多个参数，一组参数以元组形式存在，包含多组参数的所有参数
# 以元组列表形式存在

示例：
参数化之一个参数。

# ./test_case/test_func.py
import pytest

@pytest.mark.parametrize("arg_1", [4399, 2012])
def test_add_by_func_aaa(arg_1):
	print(arg_1)
	
# ./run_test.py
import pytest

if __name__ == '__main__':
	pytest.main(['-v','-s'])
	
'''
============================= test session starts =============================
platform win32 -- Python 3.7.0, pytest-5.3.4, py-1.8.1, pluggy-0.13.1 -- D:\Python3.7\python.exe
cachedir: .pytest_cache
rootdir: D:\Python3.7\project\pytest, inifile: pytest.ini
plugins: allure-pytest-2.8.9, rerunfailures-8.0
collecting ... collected 2 items

test_case/test_func.py::test_add_by_func_aaa[4399] 4399
PASSED
test_case/test_func.py::test_add_by_func_aaa[2012] 2012
PASSED

============================== 2 passed in 0.04s ==============================
[Finished in 1.3s]
'''	
	

参数化之多个参数。

# ./test_case/test_func.py
import pytest		

@pytest.mark.parametrize("arg_1, arg_2", [(4399, 'AAAA'), (2012, 'BBBB')])
def test_add_by_func_aaa(arg_1,arg_2):
	print("arg_1:{}  arg_2:{}".format(arg_1, arg_2))

# ./run_test.py
import pytest

if __name__ == '__main__':
	pytest.main(['-v','-s'])
	
'''
============================= test session starts =============================
platform win32 -- Python 3.7.0, pytest-5.3.4, py-1.8.1, pluggy-0.13.1 -- D:\Python3.7\python.exe
cachedir: .pytest_cache
rootdir: D:\Python3.7\project\pytest, inifile: pytest.ini
plugins: allure-pytest-2.8.9, rerunfailures-8.0
collecting ... collected 2 items

test_case/test_func.py::test_add_by_func_aaa[4399-AAAA] arg_1:4399  arg_2:AAAA
PASSED
test_case/test_func.py::test_add_by_func_aaa[2012-BBBB] arg_1:2012  arg_2:BBBB
PASSED

============================== 2 passed in 0.05s ==============================
[Finished in 1.3s]
'''		

以上第2个示例，展现的是一个测试用例有两个参数，然后参数化了两组数据。
但在实际测试中，有的场景，比如多条件查询，比如有2个查询条件，每个条件有3个选项，如果要全部覆盖，则是3*3==9种情况。这种情景，人工测试一般是不会全部覆盖的，但在自动化测试中，只要你想，就可以做到。如下示例：
如下格式参数化，其测试结果为所有参数选项数量的乘积。

# ./test_case/test_func.py
import pytest
from func import *

'''
class TestFunc:

	# 正常测试用例
	def test_add_by_class(self):
		assert add(2,3) == 5


	def test_add_by_class_11(self):
		assert add(2,3) == 5
'''		

@pytest.mark.parametrize("arg_1", [4399,  2012, 1997])
@pytest.mark.parametrize("arg_2", ['AAAA', 'BBBB', 'CCCC'])
def test_add_by_func_aaa(arg_1,arg_2):
	print("arg_1:{}  arg_2:{}".format(arg_1, arg_2))
	

# ./run_test.py
import pytest

if __name__ == '__main__':
	pytest.main(['-v','-s'])
		
	
'''
============================= test session starts =============================
platform win32 -- Python 3.7.0, pytest-5.3.4, py-1.8.1, pluggy-0.13.1 -- D:\Python3.7\python.exe
cachedir: .pytest_cache
rootdir: D:\Python3.7\project\pytest, inifile: pytest.ini
plugins: allure-pytest-2.8.9, rerunfailures-8.0
collecting ... collected 9 items

test_case/test_func.py::test_add_by_func_aaa[AAAA-4399] arg_1:4399  arg_2:AAAA
PASSED
test_case/test_func.py::test_add_by_func_aaa[AAAA-2012] arg_1:2012  arg_2:AAAA
PASSED
test_case/test_func.py::test_add_by_func_aaa[AAAA-1997] arg_1:1997  arg_2:AAAA
PASSED
test_case/test_func.py::test_add_by_func_aaa[BBBB-4399] arg_1:4399  arg_2:BBBB
PASSED
test_case/test_func.py::test_add_by_func_aaa[BBBB-2012] arg_1:2012  arg_2:BBBB
PASSED
test_case/test_func.py::test_add_by_func_aaa[BBBB-1997] arg_1:1997  arg_2:BBBB
PASSED
test_case/test_func.py::test_add_by_func_aaa[CCCC-4399] arg_1:4399  arg_2:CCCC
PASSED
test_case/test_func.py::test_add_by_func_aaa[CCCC-2012] arg_1:2012  arg_2:CCCC
PASSED
test_case/test_func.py::test_add_by_func_aaa[CCCC-1997] arg_1:1997  arg_2:CCCC
PASSED

============================== 9 passed in 0.06s ==============================
[Finished in 1.4s]
'''
	

总结

以上，就是我们测试中使用的pytest测试用例参数化。
当然，如实际需要，你也可以把测试数据独立到文件里。然后读取出来，传递给@pytest.mark.parametrize(argnames, argvalues)装饰器。