爆肝整理，Python自动化测试-Pytest参数化实战封装，一篇打通...

网易测试开发猿2023-08-12 17:30

目录：导读

前言

参数化？

通俗点理解就是，定义一个测试类或测试函数，可以传入不同测试用例对应的参数，从而执行多个测试用例。

例如：

对登录接口进行测试，假设有3条用例，正确账号正确密码登录、正确账号错误密码登录、错误账号正确密码登录，那么我们只需要定义一个登陆测试函数test_login()，然后使用这3条用例对应的参数去调用test_login()即可。

在unittest中可以使用ddt进行参数化，而pytest中也提供非常方便的参数化方式，即使用装饰器@pytest.mark.parametrize()。

一般写为pytest.mark.parametrize("argnames", argvalues)。

其中：

argnames为参数名称，可以是单个或多个，多个写法为"argname1, argname2, ..."；

argvalues为参数值，类型必须为list（单个参数时可以为元组，多个参数时必须为list，所以最好统一）；

例如有下接口：

请求的登陆接口信息：

python 复制代码 
接口url：http://127.0.0.1:5000/login
请求方式：post
请求参数：
响应信息：

1、单个参数

只需要传入一个参数时，示例如下：

python 复制代码 
# 待测试函数
def sum(a):
    return a+1

# 单个参数
data = [1, 2, 3, 4]
@pytest.mark.parametrize("item", data)
def test_add(item):
    actual = sum(item)
    print("\n{}".format(actual))
    # assert actual == 3

if __name__ == '__main__':
    pytest.main()

注意：

@pytest.mark.parametrize()中的第一个参数，必须以字符串的形式来标识测试函数的入参，如上述示例中，定义的测试函数test_login()中传入的参数名为item，那么@pytest.mark.parametrize()的第一个参数则为"item"。

运行结果如下：

python 复制代码 
rootdir: E:\blog\python接口自动化\apiAutoTest, configfile: pytest.ini
plugins: html-2.1.1, metadata-1.10.0, ordering-0.6, rerunfailures-9.1.1
collecting ... collected 4 items

test_case_2.py::test_add[1] PASSED                                       [ 25%]
2

test_case_2.py::test_add[2] PASSED                                       [ 50%]
3

test_case_2.py::test_add[3] PASSED                                       [ 75%]
4

test_case_2.py::test_add[4] PASSED                                       [100%]
5

============================== 4 passed in 0.02s ==============================

从结果可以看到，测试函数分别传入了data中的参数，总共执行了5次。

2、多个参数

测试用例需传入多个参数时，@pytest.mark.parametrize() 的第一个参数同样是字符串， 对应用例的多个参数用逗号分隔。

示例：

python 复制代码 
import pytest
import requests
import json

# 列表嵌套元组
data = [("lilei", "123456"), ("hanmeimei", "888888")]
# 列表嵌套列表
# data = [["lilei", "123456"], ["hanmeimei", "888888"]]

@pytest.mark.parametrize("username, password", data)
def test_login(username, password):
    headers = {"Content-Type": "application/json;charset=utf8"}
    url = "http://127.0.0.1:5000/login"
    _data = {
        "username": username,
        "password": password
    }
    res = requests.post(url=url, headers=headers, json=_data).text
    res = json.loads(res)
    assert res['code'] == 1000


if __name__ == '__main__':
    pytest.main()

需要注意：

代码中data的格式，可以是列表嵌套列表，也可以是列表嵌套元组，列表中的每个列表或元组代表一组独立的请求参数。

"username, password"不能写成 "username", "password"。

运行结果如下：

从结果中我们还可以看到每次执行传入的参数，如下划线所示部分。

这里所举示例是2个参数，传入3个或更多参数时，写法也同样如此，一定要注意它们之间一一对应的关系，如下图：

3、对测试类参数化

上面所举示例都是对测试函数进行参数化，那么对测试类怎么进行参数化呢？

其实，对测试类的参数化，就是对测试类中的测试方法进行参数化。

@pytest.mark.parametrize()中标识的参数个数，必须与类中的测试方法的参数一致。示例如下：

python 复制代码 
# 将登陆接口请求单独进行了封装，仅仅只是为了方便下面的示例
def login(username, password):
    headers = {"Content-Type": "application/json;charset=utf8"}
    url = "http://127.0.0.1:5000/login"
    _data = {
        "username": username,
        "password": password
    }
    res = requests.post(url=url, headers=headers, json=_data).text
    res = json.loads(res)
    return res

# 测试类参数化
data = [
    ("lilei", "123456"), ("hanmeimei", "888888")
]
@pytest.mark.parametrize("username, password", data)
class TestLogin:
    def test_login_01(self, username, password):
        res = login(username, password)
        assert res['code'] == 1000

    def test_login_02(self, username, password):
        res = login(username, password)
        assert res['msg'] == "登录成功！"

if __name__ == '__main__':
    pytest.main(["-s"])

运行结果如下：

从结果中可以看出来，总共执行了4次，测试类中的每个测试方法都执行了2次，即每个测试方法都将data中的每一组参数都执行了一次。

注意：

这里还是要强调参数对应的关系，即@pytest.mark.parametrize()中的第一个参数，需要与测试类下面的测试方法的参数一一对应。

4、参数组合

在编写测试用例的过程中，有时候需要将参数组合进行接口请求，如示例的登录接口中username有 lilei、hanmeimei，password有 123456、888888，进行组合的话有下列四种情况：

json 复制代码 
{"username": "lilei", "password": "123456"}
{"username": "lilei", "password": "888888"}
{"username": "hanmeimei", "password": "123456"}
{"username": "hanmeimei", "password": "888888"}

在@pytest.mark.parametrize()也提供了这样的参数组合功能，编写格式示例如下：

python 复制代码 
import pytest
import requests
import json


username = ["lilei", "hanmeimei"]
password = ["123456", "888888"]

@pytest.mark.parametrize("password", password)
@pytest.mark.parametrize("username", username)
def test_login(username, password):
    headers = {"Content-Type": "application/json;charset=utf8"}
    url = "http://127.0.0.1:5000/login"
    _data = {
        "username": username,
        "password": password
    }
    res = requests.post(url=url, headers=headers, json=_data).text
    res = json.loads(res)
    assert res['code'] == 1000
	
	
if __name__ == '__main__':
    pytest.main()

运行结果如下：

python 复制代码 
rootdir: E:\blog\python接口自动化\apiAutoTest, configfile: pytest.ini
plugins: html-2.1.1, metadata-1.10.0, ordering-0.6, rerunfailures-9.1.1
collecting ... collected 4 items

test_case_5.py::test_login[lilei-123456] PASSED                          [ 25%]
test_case_5.py::test_login[lilei-888888] FAILED                          [ 50%]
test_case_5.py::test_login[hanmeimei-123456] FAILED                      [ 75%]
test_case_5.py::test_login[hanmeimei-888888] PASSED                      [100%]
=========================== short test summary info ===========================
FAILED test_case_5.py::test_login[lilei-888888] - assert 1001 == 1000
FAILED test_case_5.py::test_login[hanmeimei-123456] - assert 1001 == 1000
========================= 2 failed, 2 passed in 0.18s =========================

从结果可以看出来，2个username、2个password 有4中组合方式，总执行了4次。如果是3个username、2个password，那么就有6中参数组合方式，依此类推。

注意：

以上这些示例中的测试用例仅仅只是用于举例，实际项目中的登录接口测试脚本与测试数据会不一样。

5、增加测试结果可读性

从示例的运行结果中我们可以看到，为了区分参数化的运行结果，在结果中都会显示由参数组合而成的执行用例名称，很方便就能看出来执行了哪些参数组合的用例。

示例：

但这只是简单的展示，如果参数多且复杂的话，仅仅这样展示是不够清晰的，需要添加一些说明才能一目了然。

因此，在@pytest.mark.parametrize()中有两种方式来自定义上图中划线部分的显示结果，即使用@pytest.mark.parametrize()提供的参数 ids 自定义，或者使用pytest.param()中的参数id自定义。

ids（推荐）

ids使用方法示例如下：

python 复制代码 
import pytest
import requests
import json


data = [("lilei", "123456"), ("hanmeimei", "888888")]
ids = ["username:{}-password:{}".format(username, password) for username, password in data]
@pytest.mark.parametrize("username, password", data, ids=ids)
def test_login(username, password):
    headers = {"Content-Type": "application/json;charset=utf8"}
    url = "http://127.0.0.1:5000/login"
    _data = {
        "username": username,
        "password": password
    }
    res = requests.post(url=url, headers=headers, json=_data).text
    res = json.loads(res)
    assert res['code'] == 1000
    

if __name__ == '__main__':
    pytest.main()

从编写方式可以看出来，ids就是一个list，且它的长度与参数组合的分组数量一致。

运行结果如下：

比较上面个执行结果，我们能看出ids自定义执行结果与默认执行结果展示的区别。使用过程中，需要根据实际情况来自定义。

id

使用方式示例如下：

python 复制代码 
import pytest
import requests
import json


data = [
    pytest.param("lilei", "123456", id="correct username and correct password"),
    pytest.param("lilei", "111111", id="correct user name and wrong password")
]

@pytest.mark.parametrize("username, password", data)
def test_login(username, password):
    headers = {"Content-Type": "application/json;charset=utf8"}
    url = "http://127.0.0.1:5000/login"
    _data = {
        "username": username,
        "password": password
    }
    res = requests.post(url=url, headers=headers, json=_data).text
    res = json.loads(res)
    assert res['code'] == 1000


if __name__ == '__main__':
    pytest.main()

运行结果如下：

|-------------------------------------|
| 下面是我整理的2023年最全的软件测试工程师学习知识架构体系图 |

一、Python编程入门到精通

二、接口自动化项目实战

三、Web自动化项目实战

四、App自动化项目实战

五、一线大厂简历

六、测试开发DevOps体系

七、常用自动化测试工具

八、JMeter性能测试

九、总结（尾部小惊喜）

奋斗是一段漫长的旅程，痛苦与磨难只是通往成功的试炼。不惧困难，坚守信念，用汗水浇灌梦想的花朵。相信自己，勇往直前，你将创造属于自己的辉煌，留下无悔的足迹！

执着的火焰燃烧内心，不屈的勇气驱散黑暗。放下畏惧，迎接挑战，奋斗的脚步不停歇。每一次努力铸就坚韧，每一次拼搏开启新篇章。勇往直前，追逐梦想。

人生的舞台，唯有奋斗才能谱写出绚丽的乐章。不要畏惧困难，要勇敢地迎接挑战。坚持努力，永不放弃，相信自己的力量，你将创造出意想不到的精彩人生！

相关推荐
互联网杂货铺15 小时前
自动化测试基础知识总结
自动化测试·软件测试·python·selenium·测试工具·职场和发展·测试用例
qq_433716951 天前
Selenium+Pytest自动化测试框架 ------ 禅道实战
自动化测试·软件测试·selenium·单元测试·pytest·接口测试·压力测试
HKJ_numb12 天前
软件测试 —— 自动化基础
软件测试·selenium·测试工具·自动化
blues_C2 天前
Pytest-Bdd-Playwright 系列教程（11）：场景快捷方式
自动化测试·pytest·bdd·playwright
blues_C2 天前
Pytest-Bdd-Playwright 系列教程（12）：步骤参数 & parsers参数解析
自动化测试·pytest·bdd·playwright
霍格沃兹测试开发学社测试人社区3 天前
深入了解测试开发与DevOps体系
运维·软件测试·测试开发·devops
blues_C5 天前
Pytest-Bdd-Playwright 系列教程（10）：配置功能文件路径 & 优化场景定义
自动化测试·软件测试·pytest·bdd·playwright
热门推荐
01(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明02PyTorch机器学习实现液态神经网络03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04玄机平台应急响应—webshell查杀05Coze扣子平台完整体验和实践（附国内和国际版对比）06Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO07【目标跟踪】相机运动补偿08Unity中PICO实现 隔空取物 和 接触抓取物体09RAG 实践- Ollama+RagFlow 部署本地知识库10怎样让音频速度变慢？请跟随以下方法进行操作