DDT（Data-Driven Tests）是针对 unittest 单元测试框架设计的扩展库。允许使用不同
的测试数据来运行一个测试用例，并将其展示为多个测试用例。
GitHub 地址：https://github.com/datadriventests/ddt。
DDT 支持 pip 安装。
安装ddt:pip install ddt
import unittest
from time import sleep
from selenium import webdriver
from ddt import ddt,data,file_data,unpack
@ddt
class TestBaidu(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver = webdriver.Chrome()
cls.base_url = "https://www.baidu.com"
def baidu_search(self,search_key):
self.driver.get(self.base_url)
self.driver.find_element_by_id("kw").send_keys(search_key)
self.driver.find_element_by_id("su").click()
sleep(3)
#参数化使用方式一--列表
@data(["case1","selenium"],["case2","ddt"],["case3","python"])
@unpack
def test_search1(self,case,search_key):
print("第一组测试用例：", case)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key + "_百度搜索")
#参数化使用方式二--元组
@data(("case1","selenium"),("case2","ddt"),("case3","python"))
@unpack
def test_search2(self,case,search_key):
print("第二组测试用例：",case)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key + "_百度搜索")
#参数化使用方式三--字典
@data({"search_key":"selenium"},{"search_key":"ddt"},{"search_key":"python"})
@unpack
def test_search3(self,search_key):
print("第三组测试用例：",search_key)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
@classmethod
def tearDownClass(cls):
sleep(10)
cls.driver.quit()
if __name__ == '__main__':
unittest.main(verbosity=2)
使用 DDT 需要注意以下几点。
首先，测试类需要通过@ddt 装饰器进行装饰。
其次，DDT 提供了不同形式的参数化。这里列举了三组参数化，第一组为列表，第二
组为元组，第三组为字典。需要注意的是，字典的 key 与测试方法的参数要保持一致。
执行结果如下。
image.png
DDT 同样支持数据文件的参数化。它封装了数据文件的读取，让我们更专注于数据文
件中的内容，以及在测试用例中的使用，而不需要关心数据文件是如何被读取进来的。
1、使用json参数化，脚本同级目录下新建ddt_data_file.json文件
{
"case1":{"search_key":"python"},
"case2":{"search_key":"ddt"},
"case3":{"search_key":"selenium"}
}
脚本中使用方式为：
#参数化使用方式四-json数据文件参数化
@file_data('ddt_data_file.json')
def test_search4(self,search_key):
print("第四组测试用例：",search_key)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
2、使用yaml，脚本同级目录下新建ddt_data_file.yaml文件
case1:
- search_key : "python"
case2:
- search_key : "ddt"
case3:
- search_key : "selenium"
脚本中使用方式为：
#参数化使用方式五-yaml数据文件参数化
@file_data('ddt_data_file.yaml')
def test_search5(self,case):
search_key = case[0]["search_key"]
print("第五组测试用例：",search_key)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
这里的取值与上面的 JSON 文件有所不同，因为每一条用例都被解析为[{'search_key':
'python'}]，所以要想取到搜索关键字，则需要通过 case[0]["search_key"]的方式获取。
所以完整的脚本为：
import unittest
from time import sleep
from selenium import webdriver
from ddt import ddt,data,file_data,unpack
@ddt
class TestBaidu(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver = webdriver.Chrome()
cls.base_url = "https://www.baidu.com"
def baidu_search(self,search_key):
self.driver.get(self.base_url)
self.driver.find_element_by_id("kw").send_keys(search_key)
self.driver.find_element_by_id("su").click()
sleep(3)
#参数化使用方式一--列表
@data(["case1","selenium"],["case2","ddt"],["case3","python"])
@unpack
def test_search1(self,case,search_key):
print("第一组测试用例：", case)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key + "_百度搜索")
#参数化使用方式二--元组
@data(("case1","selenium"),("case2","ddt"),("case3","python"))
@unpack
def test_search2(self,case,search_key):
print("第二组测试用例：",case)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key + "_百度搜索")
#参数化使用方式三--字典
@data({"search_key":"selenium"},{"search_key":"ddt"},{"search_key":"python"})
@unpack
def test_search3(self,search_key):
print("第三组测试用例：",search_key)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
#参数化使用方式四-json数据文件参数化
@file_data('ddt_data_file.json')
def test_search4(self,search_key):
print("第四组测试用例：",search_key)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
#参数化使用方式五-yaml数据文件参数化
@file_data('ddt_data_file.yaml')
def test_search5(self,case):
search_key = case[0]["search_key"]
print("第五组测试用例：",search_key)
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
@classmethod
def tearDownClass(cls):
sleep(10)
cls.driver.quit()
if __name__ == '__main__':
unittest.main(verbosity=2)
--来源于虫师的《Selenium3自动化测试实战——基于Python语言》
