Pytest框架 — 11、Pytest的标记(二)（parametrize参数化）

目录

• 1、标记实现参数化
  • 示例1：单个参数
  • 示例2：多个参数
  • 示例3：多个参数化（笛卡尔积）
  • 示例4：ids参数给用例起别名
  • 示例5：使用indirect处理参数值
  • 示例6：标记数据

1、标记实现参数化

通过@pytest.mark.parametrize(argnames, argvalues, indirect=False, ids=None, scope=None)装饰器来实现参数化。

参数说明：

• argnames：参数名，使用逗号分隔的列表，或元祖，或字符串，表示一个或多个参数名，【常用】
• argvalues：参数值，可以是列表、元祖、字典列表、元祖列表，【常用】
• indirect：中文翻译为中间人，为True时可以对argvalues的参数值进行处理，默认False，【不常用】
  • indirect=True: 且argnames的值为fixture函数名，此时argnames的值变为可执行函数，会将argvalues的参数值当做参数传递给fixture函数进行处理，fixture函数返回处理结果给argnames
  • indirect=False: argnames仅为参数名
• ids：给用例起别名，字符串列表或数字列表，不设置会自动从测试数据中提取
• scope：待理解【不常用】

示例1：单个参数

import pytest


phone_list = [
    "13881118888",
    "13012034288",
    "13234324188",
    "13231423288"
]

@pytest.mark.parametrize(argnames="phone_num", argvalues=phone_list)
def test_phone_number(phone_num):  # 注意，这里的参数要和argnames参数名一致
    print(f"正在测试手机号{phone_num}")

"""
执行结果
mark/parametrize/single_param.py::test_phone_number[13881118888] 正在测试手机号[phone]
PASSED
mark/parametrize/single_param.py::test_phone_number[13012034288] 正在测试手机号[phone]
PASSED
mark/parametrize/single_param.py::test_phone_number[13234324188] 正在测试手机号[phone]
PASSED
mark/parametrize/single_param.py::test_phone_number[13231423288] 正在测试手机号[phone]
PASSED
"""

示例2：多个参数

import pytest

user_info = [
    ("张三", "18011111111"),
    ("李四", "18022222222"),
    ("王五", "18033333333")
]

@pytest.mark.parametrize(argnames="name,phonenum", argvalues=user_info)
def test_read_info(name, phonenum):
    print(f"正在读取用户{name}，手机号{phonenum}")

"""
执行结果
mark/parametrize/multiple_param.py::test_read_info[\u5f20\u4e09-18011111111] 正在读取用户张三，手机号[phone]
PASSED
mark/parametrize/multiple_param.py::test_read_info[\u674e\u56db-18022222222] 正在读取用户李四，手机号[phone]
PASSED
mark/parametrize/multiple_param.py::test_read_info[\u738b\u4e94-18033333333] 正在读取用户王五，手机号[phone]
PASSED
"""

由上面例子执行结果中可以看到中文参数值乱码了，解决办法如下
方法1. 在pytest.ini中加入disable_test_id_escaping_and_forfeit_all_rights_to_community_support=True
方法2. 在conftest.py中加入

# 收集每一个用例name和nodeid的中文显示，转化为utf-8形式
def pytest_collection_modifyitems(items):
    for item in items:
        item.name = item.name.encode("utf-8").decode("unicode_escape")
        item._nodeid = item.nodeid.encode("utf-8").decode("unicode_escape")

示例3：多个参数化（笛卡尔积）

import pytest


data1 = ['a', 'b', 'c']
data2 = [1, 2]


@pytest.mark.parametrize('test1', data1)
@pytest.mark.parametrize('test2', data2)
def test_param(test1, test2):
    print('\n测试数据：{}-{}'.format(test1, test2))
"""
执行结果
mark/parametrize/multiple_parameterization.py::test_param[1-a] 
测试数据：a-1
PASSED
mark/parametrize/multiple_parameterization.py::test_param[1-b] 
测试数据：b-1
PASSED
mark/parametrize/multiple_parameterization.py::test_param[1-c] 
测试数据：c-1
PASSED
mark/parametrize/multiple_parameterization.py::test_param[2-a] 
测试数据：a-2
PASSED
mark/parametrize/multiple_parameterization.py::test_param[2-b] 
测试数据：b-2
PASSED
mark/parametrize/multiple_parameterization.py::test_param[2-c] 
测试数据：c-2
PASSED
"""

示例4：ids参数给用例起别名

import pytest


user_info = [
    ("张三", "18011111111"),
    ("李四", "18022222222"),
    ("王五", "18033333333")
]

@pytest.mark.parametrize(argnames="name,phonenum", argvalues=user_info, ids=["用户1","用户2","用户3"])
def test_read_info(name, phonenum):
    print(f"正在读取用户{name}，手机号{phonenum}")
    
"""
执行结果
mark/parametrize/ids.py::test_read_info[用户1] 正在读取用户张三，手机号[phone]
PASSED
mark/parametrize/ids.py::test_read_info[用户2] 正在读取用户李四，手机号[phone]
PASSED
mark/parametrize/ids.py::test_read_info[用户3] 正在读取用户王五，手机号[phone]
PASSED
"""

示例5：使用indirect处理参数值

import pytest

@pytest.fixture()
def fixture_and_parametrize(request):  # request是关键字不能改变，用来接收参数
    print('邮箱账号为：{}'.format(request.param))
    return request.param + "@qq.com"


@pytest.mark.parametrize('fixture_and_parametrize', ['100203', '466238894', '23942423'],
                         indirect=True)
def test_fixture_and_parametrize_2(fixture_and_parametrize):
    print('拼接后邮箱为：{}'.format(fixture_and_parametrize))

"""
执行结果
mark/parametrize/indirect.py::test_fixture_and_parametrize_2[100203] 邮箱账号为：100203
拼接后邮箱为：[e-mail]
PASSED
mark/parametrize/indirect.py::test_fixture_and_parametrize_2[466238894] 邮箱账号为：466238894
拼接后邮箱为：[e-mail]
PASSED
mark/parametrize/indirect.py::test_fixture_and_parametrize_2[23942423] 邮箱账号为：23942423
拼接后邮箱为：[e-mail]
PASSED
"""

示例6：标记数据

在参数化的过程中也可以标记数据进行跳过等

import pytest


@pytest.mark.parametrize("test_input,expected", [
    ("3+9", 12), ("5+4", 9),
    pytest.param("7 * 9", 42, marks=pytest.mark.xfail),
    pytest.param("8 * 6", 42, marks=pytest.mark.skip)
])
def test_mark(test_input, expected):
    assert eval(test_input) == expected
"""
执行结果
mark/parametrize/mark_data.py::test_mark[3+9-12] PASSED
mark/parametrize/mark_data.py::test_mark[5+4-9] PASSED
mark/parametrize/mark_data.py::test_mark[7 * 9-42] XFAIL
mark/parametrize/mark_data.py::test_mark[8 * 6-42] SKIPPED (unconditional skip)
"""

参考
https://www.cnblogs.com/miki-peng/p/14736332.html
https://zhuanlan.zhihu.com/p/515377205