@pytest.mark.parametrize 允许在测试函数或类中定义多组参数和 fixtures。
参数化场景:
只有测试数据和预期结果不一样,但操作步骤是一样的测试用例是可以用上参数化的。
创建test_parametrize.py文件
示例一:未参数化
1、脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
# 未参数化
def test_case1():
assert 3 + 4 == 9
def test_case2():
assert 2 + 5 == 7
def test_case3():
assert 6 * 9 == 48
可以看到,三个用例都是先计算,然后断言某个值,重复写三个类似的用例有些冗余。
2、运行结果:
示例二:参数化(优化代码)
1、脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
# 参数化(优化代码)
@pytest.mark.parametrize("test_input, expected", [("3+4", 9), ("2+5", 7), ("6*9", 48)])
def test_eval(test_input, expected):
print(f"测试数据{test_input}, 预期结果{expected}")
assert eval(test_input) == expected
2、运行结果:
可以看到,只有一条用例,但是利用参数化输入三组不同的测试数据和预期结果,最终执行的测试用例数还是3条,可以节省很多代码。
1、参数
def parametrize(self, argnames, argvalues, indirect=False, ids=None, scope=None):
主要参数:
(1)argnames:参数名,是个字符串,如中间用逗号分隔则表示为多个参数名。
1、创建test_parametrize2.py文件
参数名也可以是list或者tuple里的字符串。
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
@pytest.mark.parametrize(["name", "pwd"], [("AllTests", "123456"), ("qq", "85135506")])
def test_case1(name, pwd):
print(name, pwd)
@pytest.mark.parametrize(("name", "pwd"), [("wangmcn", "567890"), ("admin", "123qwe")])
def test_case2(name, pwd):
print(name, pwd)
@pytest.mark.parametrize("name, pwd", [("root", "root"), ("guest", "guest")])
def test_case3(name, pwd):
print(name, pwd)
2、运行结果:
(2)argvalues:参数值,参数组成的列表,列表中有几个元素,就会生成几条用例。
1、创建test_parametrize3.py文件
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
# 如果只有一个参数,里面则是值的列表
@pytest.mark.parametrize("name", ["AllTests", "wangmcn", "admin"])
def test_case1(name):
print(name)
# 如果有多个参数,list包含tuple
@pytest.mark.parametrize(["name", "pwd"], [("AllTests", "123456"), ("qq", "85135506")])
def test_case2(name, pwd):
print(name, pwd)
# 如果有多个参数,tuple包含list
@pytest.mark.parametrize(("name", "pwd"), (["wangmcn", "567890"], ["admin", "123qwe"]))
def test_case3(name, pwd):
print(name, pwd)
# 如果有多个参数,list包含list
@pytest.mark.parametrize("name, pwd", [["root", "root"], ["guest", "guest"]])
def test_case4(name, pwd):
print(name, pwd)
2、运行结果:
(3)ids:用例的ID。传一个字符串列表,可以标识每一个测试用例,自定义测试数据结果的显示,为了增加可读性。
注:ids的长度需要与测试数据列表的长度一致。
1、创建test_parametrize8.py文件
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
my_data = [(1, 2, 3), (4, 5, 9)]
ids = ["a:{} + b:{} = expect:{}".format(a, b, expect) for a, b, expect in my_data]
@pytest.mark.parametrize("a, b, expect", my_data, ids=ids)
def test_parametrize_case(a, b, expect):
print("测试数据为{}-{}".format(a, b))
assert a + b == expect
2、运行结果:
2、装饰测试类
1、创建test_parametrize2.py文件
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
my_data = [(1, 2, 3), (4, 5, 9)]
@pytest.mark.parametrize("a, b, expect", my_data)
class TestParametrize:
def test_parametrize_case1(self, a, b, expect):
print("\n测试用例case1 测试数据为{}+{}".format(a, b))
assert a + b == expect
def test_parametrize_case2(self, a, b, expect):
print("\n测试用例case2 测试数据为{}+{}".format(a, b))
assert a + b == expect
2、运行结果:
当装饰器 @pytest.mark.parametrize 装饰测试类时,会将数据集合传递给类的所有测试用例方法。
3、多个参数化装饰器
一个函数或一个类可以装饰多个 @pytest.mark.parametrize
当参数化装饰器有很多个的时候,用例数等于n(个)*n(个)*n(个)*n(个)*....
1、创建test_parametrize5.py文件。
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
my_data_1 = [1, 2, 3]
my_data_2 = ["a", "b"]
@pytest.mark.parametrize("a", my_data_1)
@pytest.mark.parametrize("b", my_data_2)
def test_parametrize_case(a, b):
print(f"测试数据为:{a},{b}")
2、运行结果:
参数a的数据有3个,参数b的数据有2个,所以最终的用例数有3*2=6条。
4、参数化(传入字典数据)
1、创建test_parametrize6.py文件
数据类型为字典
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
my_data = (
{
"user": "AllTests",
"pwd": "123456"
},
{
"user": "wangmcn",
"pwd": "567890"
}
)
@pytest.mark.parametrize('dic', my_data)
def test_parametrize_case(dic):
print(f'测试数据为\n{dic}')
print(f'user:{dic["user"]}, pwd:{dic["pwd"]}')
2、运行结果:
5、标记参数化
pytest.param 可以传三种参数:
- param values - 参数集值的变量 args,按顺序排列。
- keyword marks - marks 关键字标记,要应用于此参数集的单个标记或标记列表。
- keyword str id - 参数集的属性 id。
示例一:
1、创建test_parametrize7.py文件。
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
@pytest.mark.parametrize("test_input, expected", [
("3+5", 8),
("2+4", 6),
pytest.param("6*9", 45, marks=pytest.mark.xfail),
pytest.param("6*6", 50, marks=pytest.mark.skip)
])
def test_case(test_input, expected):
assert eval(test_input) == expected
2、运行结果:
示例二:
1、创建test_parametrize7_2.py文件。
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
@pytest.mark.parametrize("test_input, expected", [
pytest.param("3+5", 8, id="case1"),
pytest.param("2+4", 6, id="case2"),
pytest.param("6*9", 45, marks=pytest.mark.xfail, id="case3"),
pytest.param("6*6", 50, marks=pytest.mark.skip, id="case4")
])
def test_case(test_input, expected):
assert eval(test_input) == expected
2、运行结果:
6、解决unicode编码问题
使用 @pytest.mark.parametrize 参数化的时候,加 ids 参数用例描述有中文时,在控制台输出会显示 unicode 编码问题,中文不能正常显示。
解决方法:使用 pytest_collection_modifyitems 钩子函数,对输出的 item.name 和 item.nodeid 重新编码即可。
1、创建test_parametrize9.py文件
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
import pytest
def login(username, password):
"登陆操作"
return {"code": 0, "msg": "success!"}
my_data = [
({"username": "AllTests", "password": "123456"}, "success!"),
({"username": "Admin123", "password": "123456"}, "success!"),
({"username": "Admin456", "password": "123456"}, "success!"),
]
ids = [
"输入正确a账号,正确密码,登录成功",
"输入正确b账号,正确密码,登录成功",
"输入正确c账号,正确密码,登录成功",
]
@pytest.mark.parametrize("input_data, expected", my_data, ids=ids)
def test_login(input_data, expected):
"测试用例-登陆"
result = login(input_data["username"], input_data["password"])
assert result["msg"] == expected
2、打开命令行,输入执行命令
pytest -v test_parametrize9.py
运行结果:
控制台输出中文时显示unicode编码问题。
3、解决ids参数用例描述为中文时,控制台输出显示unicode编码问题。
在项目的根目录或与用例同级目录下,创建conftest.py文件
脚本代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
微信公众号:AllTests软件测试
"""
def pytest_collection_modifyitems(items):
for item in items:
item.name = item.name.encode("utf-8").decode("unicode_escape")
# print(item.nodeid)
item._nodeid = item.nodeid.encode("utf-8").decode("unicode_escape")
重新执行用例,运行结果:
控制台输出中文时显示正常。