unittest的基本使用（二）数据驱动ddt的使用

古城微笑少年丶 2023-07-12 11:45 22阅读 0赞

ddt 是 python unittest自带的数据驱动功能。

### 使用方法： ###

**dd.ddt：**  
装饰类，也就是继承自TestCase的类。

**ddt.data：**  
装饰测试方法。参数是一系列的值。

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

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

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

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

**ddt.unpack：**  
传递的是复杂的数据结构时使用。比如使用元组或者列表，添加unpack之后，ddt会自动把元组或者列表对应到多个参数上。字典也可以这样处理。参见下面的示例2.

**测试用例方法名生成规则**  
使用ddt后，会产生一个新的测试用例方法名：之前的测试用例方法名\_ordinal\_data  
之前的测试用例方法名：即定义的测试用例方法名。比如def test\_large()，这里就是test\_large

ordinal：整数，从1开始递加。  
data：如果传递过来的数据存在\_\_name\_\_属性，则这里就是该数据的\_\_name\_\_值。如果未定义\_\_name\_\_属性，ddt会尽量将传递过来的数据转化为python标识符，作为data显示。比如（3,2）就转化为3\_2。需要注意的是，如果数据是字典，则这里就是字典的key。

### 使用实例： ###

#### @ddt，@data的使用 ####

下面是一个登录的测试用例，使用的 login\_check 方法中有两个参数： username 和 password

import unittest
    from ddt import data, ddt, unpack
    from homework.day15_ddt.loginfunction import login_check
    
    @ddt
    class TestLoginCase(unittest.TestCase):
        datas = [
            {'username': 'python27', 'password': 'lemonban'},
            {'username': 'python271', 'password': 'lemonban'},
            {'username': 'python27', 'password': 'lemonban1'},
            {'username': 'python271', 'password': 'lemonban1'},
            {'username': None, 'password': None},
            {'username': None, 'password': 'lemonban'},
            {'username': 'python27', 'password': None},
        ]
    
        expecteds = [
            {"code": 0, "msg": "登录成功"},
            {"code": 1, "msg": "账号或密码不正确"},
            {"code": 1, "msg": "账号或密码不正确"},
            {"code": 1, "msg": "账号或密码不正确"},
            {"code": 1, "msg": "所有的参数不能为空"},
            {"code": 1, "msg": "所有的参数不能为空"},
            {"code": 1, "msg": "所有的参数不能为空"}
        ]
    	
    	# 使用zip聚合打包方法，将参数和预期结果列表中的数据一一对应放在元组中，最后打包成一个列表
        data_expecteds = zip(datas, expecteds)
        
        # 使用了拆包，将列表拆包成元组，内容为：（data字典，期望结果）
        @data(*data_expecteds)
        def test_login_is_success(self, data_expected):
        	# 传入的元组能够通过索引获取相应的值
        	# 使用 ** 拆包字典中的数据，使用关键字传参将对应的username和password传入login_check方法
            res = login_check(**data_expected[0])
            self.assertEqual(res, data_expected[1])

执行结果都通过

#### @unpack的使用 ####

使用复杂的数据结构时，需要用到@unpack，同时测试方法的参数需要使用对应的多个，比如下面的frist\_value 以及 second\_value。

import unittest
    from ddt import ddt, data,unpack
    
    @ddt
    class FooTestCase(unittest.TestCase):
    
        @data((3, 2), (4, 3), (5, 3))
        @unpack
        def test_tuples_extracted_into_arguments(self, first_value, second_value):
            self.assertTrue(first_value > second_value)
    
        @data([3, 2], [4, 3], [5, 3])
        @unpack
        def test_list_extracted_into_arguments(self, first_value, second_value):
            self.assertTrue(first_value > second_value)
    
        @unpack
        @data({'first': 1, 'second': 3, 'third': 2},
              {'first': 4, 'second': 6, 'third': 5})
        def test_dicts_extracted_into_kwargs(self, first, second, third):
            self.assertTrue(first < third < second)

#### 使用json文件 ####

新建文件 test\_data\_list.json：

[
        "Hello",
        "Goodbye"
    ]

新建文件 test\_data\_dict.json：

{
        "unsorted_list": [ 10, 12, 15 ],
        "sorted_list": [ 15, 12, 50 ]
    }

新建测试脚本ddt\_test.py：

import unittest
    from ddt import ddt, file_data
    from ddt_demo.mycode import has_three_elements,is_a_greeting
    
    @ddt
    class FooTestCase(unittest.TestCase):
    
        @file_data('test_data_dict.json')
        def test_file_data_json_dict(self, value):
            self.assertTrue(has_three_elements(value))
    
        @file_data('test_data_list.json')
        def test_file_data_json_list(self, value):
            self.assertTrue(is_a_greeting(value))

#### 使用yaml文件 ####

新建文件 test\_data\_list.yaml:

- "Hello"
    - "Goodbye"

新建文件 test\_data\_dict.yaml:

unsorted_list:
      - 10
      - 15
      - 12
    
    sorted_list: [ 15, 12, 50 ]

新建测试脚本ddt\_test.py：

import unittest
    from ddt import ddt, file_data
    from ddt_demo.mycode import has_three_elements,is_a_greeting
    
    @ddt
    class FooTestCase(unittest.TestCase):
    
        @file_data('test_data_dict.yaml')
        def test_file_data_yaml_dict(self, value):
            self.assertTrue(has_three_elements(value))
    
        @file_data('test_data_list.yaml')
        def test_file_data_yaml_list(self, value):
            self.assertTrue(is_a_greeting(value))

最后总结一下：  
如果你对此文有任何疑问，如果你也需要接口项目实战，如果你对软件测试、接口测试、自动化测试、面试经验交流感兴趣欢迎加入：软件测试技术群：**593462778**，群里的免费资料都是笔者十多年测试生涯的精华。还有同行大神一起交流技术哦。

作者：暗潮汹涌  
原创不易，欢迎转载，但未经作者同意请保留此段声明，并在文章页面明显位置给出原文链接。