unittest中的数据驱动ddt

unittest中的ddt（‌Data-Driven Tests，‌数据驱动测试）‌的作用是实现测试数据与测试脚本的分离，‌通过ddt将测试数据加载到脚本中，‌使得一组数据对应一个测试用例，‌用例自动加载生成。‌

ddt是一种数据驱动测试框架，‌它允许开发者将测试数据与测试脚本分离，‌通过这种方式，‌可以更灵活地管理和运行测试用例。‌使用ddt的好处在于，‌它能够提高测试用例的可维护性和可读性，‌同时减少重复代码，‌使得测试数据和测试逻辑更加清晰。‌通过ddt和unittest框架的结合，‌可以实现测试用例脚本的编写、‌测试执行控制以及测试数据的批量加载，‌从而完成不同接口测试用例的批量执行和覆盖不同的测试场景。‌

@ddt：类的装饰器，继承的是TestCase类
@data()：@data装饰符可以把参数当成测试数据，参数可以是单个值、列表、元祖、字典这些类型，用于输入测试数据
@unpack：分解数据标志，主要是把元祖和列表解析成多个参数
@file_data()：输入文件，如json或者yaml类型文件

ddt的安装

pip intsall ddt

下面我们以百度搜索为例，来看看ddt的用法

import unittest
from time import sleep
from selenium import webdriver
from ddt import ddt,data,file_data,unpack


@ddt
class TestBadidu(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "http://www.baidu.com"

    def baidu_search(self, search_key):
        self.driver.get(self.base_url)
        self.driver.find_element(by='id', value="kw").send_keys(search_key)
        self.driver.find_element(by='id', value="su").click()
        sleep(10)

    #参数化使用方法一
    @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):
        cls.driver.quit()



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

执行结果如下：

使用DDT需要注意以下几点。
首先，测试类需要通过@ddt装饰器进行装饰
其次，DDT提供了不同形式的参数化。这里列举 了三组参数胡，第一组是列表，第二组是元祖，第三组是字典。需要注意的是key与测试方法的参数要保持一致