一、上篇文章我们使用了 unittest + ddt 驱动 json 文件做数据驱动测试，本篇文章我们采用 unittest + ddt 驱动 yaml/yml 文件来实现数据驱动测试，话不多说上源码。。。
注意：如果文件是以 “.yml”或者".yaml" 结尾，ddt 会作为 yaml 类型处理，其他文件都会作为 json 文件处理。
如果文件是列表，列表的值会作为测试用例参数，同时，会作为测试用例方法名后缀显示。
二、安装 yaml 模块
import yaml f = open("ddt_data.yaml", encoding="utf8") print(yaml.load(f)) f.close() # 运行结果如下 """ [{ ‘url‘: ‘http://cms.duoceshi.cn/xxx/xxxx/xxxxx‘, ‘method‘: ‘post‘, ‘header‘: {‘Content-Type‘: ‘application/x-www-form-urlencoded‘}, ‘params‘: {‘userAccount‘: ‘admin‘, ‘loginPwd‘: 123456} }] """
import requests import unittest from ddt import ddt, file_data @ddt class CmsLogin(unittest.TestCase): @file_data("ddt_data.yaml") def testcase(self, method, url, header, params): res = requests.request(method, url, headers=header, data=params) print(res.text) if __name__ == ‘__main__‘: unittest.main() # 运行结果如下 """ Ran 2 tests in 0.215s .. {"code":"200","msg":"登录成功！","model":{}} {"code":"400","msg":"登录帐号不存在！","model":{}} ---------------------------------------------------------------------- """
import yaml from ddt import ddt, data, unpack def get_yml_data(yml_file): with open(yml_file, encoding="utf8") as f: return yaml.load(f) @ddt class CmsLogin(unittest.TestCase): @data(*get_yml_data("ddt_data.yml")) @unpack def testcase(self, name, age): print(name + "----" + str(age)) if __name__ == ‘__main__‘: unittest.main() # 运行结果如下 """ Ran 3 tests in 0.000s ... Evan----19 Lvan----20 Alex----21 """
原文：https://www.cnblogs.com/ZhengYing0813/p/14099918.html