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] 正在测试手机号13881118888
PASSED
mark/parametrize/single_param.py::test_phone_number[13012034288] 正在测试手机号13012034288
PASSED
mark/parametrize/single_param.py::test_phone_number[13234324188] 正在测试手机号13234324188
PASSED
mark/parametrize/single_param.py::test_phone_number[13231423288] 正在测试手机号13231423288
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] 正在读取用户张三，手机号18011111111
PASSED
mark/parametrize/multiple_param.py::test_read_info[\\u674e\\u56db-18022222222] 正在读取用户李四，手机号18022222222
PASSED
mark/parametrize/multiple_param.py::test_read_info[\\u738b\\u4e94-18033333333] 正在读取用户王五，手机号18033333333
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] 正在读取用户张三，手机号18011111111
PASSED
mark/parametrize/ids.py::test_read_info[用户2] 正在读取用户李四，手机号18022222222
PASSED
mark/parametrize/ids.py::test_read_info[用户3] 正在读取用户王五，手机号18033333333
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
拼接后邮箱为：100203@qq.com
PASSED
mark/parametrize/indirect.py::test_fixture_and_parametrize_2[466238894] 邮箱账号为：466238894
拼接后邮箱为：466238894@qq.com
PASSED
mark/parametrize/indirect.py::test_fixture_and_parametrize_2[23942423] 邮箱账号为：23942423
拼接后邮箱为：23942423@qq.com
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

来源：https://www.cnblogs.com/qishuaiRisen/p/16601851.html
本站部分图文来源于网络，如有侵权请联系删除。