自动化测试-ddt数据驱动yaml文件实战（详细）

最新推荐文章于 2025-11-03 09:13:29 发布

原创最新推荐文章于 2025-11-03 09:13:29 发布 · 752 阅读

19 ·

CC 4.0 BY-SA版权

文章标签：

#python #websocket #功能测试 #测试用例 #jmeter #selenium #单元测试

前言

ddt 驱动 yaml/yml 文件来实现数据驱动测试

ddt.file_data：装饰测试方法，参数是文件名。文件可以是 json 或者 yaml 类型。

注意：如果文件是以 “.yml”或者".yaml" 结尾，ddt 会作为 yaml 类型处理，其他文件都会作为 json 文件处理。

如果文件是列表，列表的值会作为测试用例参数，同时，会作为测试用例方法名后缀显示。

如果文件是字典，字典的 key 会作为测试用例方法的后缀显示，字典的 value 会作为测试用例参数。

1、yaml 文件基础格式要求

基础语法：
大小写敏感；
使用缩进表示层级关系；
缩进不允许使用 tab，只允许空格；
缩进的空格数不重要，只要相同层级的元素左对齐即可；
‘#’ 表示注释；

YAML 支持以下几种数据类型：
对象：键值对的集合，又称为映射（mapping）/ 哈希（hashes） / 字典（dictionary）　　
数组：一组按次序排列的值，又称为序列（sequence） / 列表（list）　　
纯量（scalars）：单个的、不可再分的值　　
yaml 列表结构
以 - 开头的行表示构成一个数组

# data.yaml 文件内容
-
data1
-
data2
-
data3

AI写代码

import yaml
f = open("data", encoding="utf8")
v = yaml.load(f)
print(v)
print(type(v))
f.close()
# 结果如下
['data1', 'data2', 'data3']
<class 'list'>

yaml 多维列表结构

# data.yaml 文件内容
-
- data1
- data2
-
data3

import yaml
f = open("data", encoding="utf8")
v = yaml.load(f)
print(v)
print(type(v))
f.close()
# 结果如下
[['data1', 'data2'], 'data3']
<class 'list'>

yaml 字典结构
字典结构没有 -
冒号前为键，冒号后为值
冒号后面一定要空格

# data.yaml 文件内容
username : admin
password : "123456"

import yaml
f = open("data", encoding="utf8")
v = yaml.load(f)
print(v)
print(type(v))
f.close()
# 结果如下
{'username': 'admin', 'password': '123456'}
<class 'dict'>

yaml 多维字典结构

# data.yaml 文件内容
data :
username: admin
password: "123456"
code : "0000"
msg : "登录成功！"

import yaml
f = open("data", encoding="utf8")
v = yaml.load(f)
print(v)
print(type(v))
f.close()
# 结果如下
{'data': {'username': 'admin', 'password': '123456'}, 'code': '0000', 'msg': '登录成功！'}
<class 'dict'>

列表 + 字典，复杂结构

# data.yaml 文件内容
-
username : admin
password : "123456"
-
username : Evan
password : xxxxxx

import yaml
f = open("data", encoding="utf8")
v = yaml.load(f)
print(v)
print(type(v))
f.close()
# 结果如下
[{'username': 'admin', 'password': '123456'}, {'username': 'Evan', 'password': 'xxxxxx'}]
<class 'list'>