ddt 数据驱动/参数化

1.什么是ddt

痛点:测试代码和测试数据耦合了，修改测试或者增加测试都需要修改代码。

DDT：DATA DRIVER TEST （数据驱动测试/参数化）,是一个设计思想，不同的人有不同的理解。 它解决的问题:测试数据与测试用例代码分离，通过外部数据生成单元测试函数。

2.ddt 模块

pip install ddt

 

 

ddt模块的原理：本质上就是一个装饰器，总共高就百来行代码。在创建类的时候，根据用例数据，动态的创建测试函数。

使用

1. 从ddt模块中导入from ddt import ddt, data两个装饰器

2. 定义测试用例类时需要用

   @ddt
   class SomeTestCase(unittest.TestCase):
    pass

    

    

3. 定义单元测试函数时，需要

   @data(case1, case2, case3,...)
   def test_some(self, case):
    pass

    

    

4. ddt模块会自动的通过case1,case2,...在类里面创建 test_some_1,test_some_2,..单元测试函数

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2021/11/8 21:08
# @Author  : shisuiyi
# @File    : test_login.py
# @Software: win10 Tensorflow1.13.1 python3.9
import unittest
from testcode import login
from ddt import ddt, data
# from unittestreport import ddt, list_data
cases = [
    {"title": "登陆成功", "request": {"username": "python45", "password": "lemonban"},
     "expect": {"code": 0, "msg": "登录成功"}},
    {"title": "账号或密码不正确", "request": {"username": "python45", "password": "lemonban123"},
     "expect": {"code": 1, "msg": "账号或密码不正确"}},
    {"title": "账号或密码不正确", "request": {"username": "python45123", "password": "lemonban"},
     "expect": {"code": 1, "msg": "账号或密码不正确"}},
    {"title": "密码长度在6-18位之间", "request": {"username": "python45", "password": "lemo"},
     "expect": {"code": 1, "msg": "密码长度在6-18位之间"}},
    {"title": "密码长度在6-18位之间", "request": {"username": "python45", "password": "lemonbanlemonbanlemonbanlemonbanlemonban"},
     "expect": {"code": 1, "msg": "密码长度在6-18位之间"}},
]


@ddt
class TestLogin(unittest.TestCase):
    """
    测试登录函数
    """

    @classmethod
    def setUpClass(cls) -> None:
        """
        类级前置
        :return:
        """
        print('我是类级前置，在整个类的测试前执行')

    @classmethod
    def tearDownClass(cls) -> None:
        """
        类级后置
        :return:
        """
        print('我是类级后置，在整个测试类测试完成后执行')

    def setUp(self) -> None:
        """
        方法级前置
        :return:
        """
        print('======我会在每个测试执行前执行====')

    def tearDown(self) -> None:
        """
        方法级后置
        :return:
        """
        print('======我会在每个测试执行后执行======')

    @data(*cases)
    # @list_data(cases)
    def test_login(self, case):
        
        # 1. 准备测试数据
        print(case)
        print('{}开始测试'.format(case['title']))
        # 2. 测试步骤
        res = login(**case['request'])
        # 3. 断言
        self.assertEqual(res, case['expect'])


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

 

 

输出

Launching unittests with arguments python -m unittest D:/Lemon/py45/day17/test_ddt_login.py in D:\Lemon\py45\day17

我是类级前置，在整个类的测试前执行
======我会在每个测试执行前执行====
{'title': '登陆成功', 'request': {'username': 'python45', 'password': 'lemonban'}, 'expect': {'code': 0, 'msg': '登录成功'}}
登陆成功开始测试
======我会在每个测试执行后执行======
======我会在每个测试执行前执行====
{'title': '账号或密码不正确', 'request': {'username': 'python45', 'password': 'lemonban123'}, 'expect': {'code': 1, 'msg': '账号或密码不正确'}}
账号或密码不正确开始测试
======我会在每个测试执行后执行======
======我会在每个测试执行前执行====
{'title': '账号或密码不正确', 'request': {'username': 'python45123', 'password': 'lemonban'}, 'expect': {'code': 1, 'msg': '账号或密码不正确'}}
账号或密码不正确开始测试
======我会在每个测试执行后执行======
======我会在每个测试执行前执行====
{'title': '密码长度在6-18位之间', 'request': {'username': 'python45', 'password': 'lemo'}, 'expect': {'code': 1, 'msg': '密码长度在6-18位之间'}}
密码长度在6-18位之间开始测试
======我会在每个测试执行后执行======
======我会在每个测试执行前执行====
{'title': '密码长度在6-18位之间', 'request': {'username': 'python45', 'password': 'lemonbanlemonbanlemonbanlemonbanlemonban'}, 'expect': {'code': 1, 'msg': '密码长度在6-18位之间'}}
密码长度在6-18位之间开始测试
======我会在每个测试执行后执行======
我是类级后置，在整个测试类测试完成后执行


Ran 5 tests in 0.010s

OK

Process finished with exit code 0

 

 

ddt-data 数据

一组数据之间用逗号隔开
data的参数可以为如下几种

• 1.一组数据中，每个数据为单个值
• 2.一组数据中，每个数据为一个列表或者一个字典
• 3.文件对象：json,yaml

一组数据中的数据为列表或者字典：
@data([a,b],[c,d])

如何从以上数据中获取到字典中每一项值
@unpack
若变量A=[{a:b,a1:b1},{c:d,c1:d1}]
如何将变量A中的每一组元素作为测试数据
@ddt.data(*A)

包含知识点

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

• 当没有加unpack时，test_case方法的参数只能填一个；
• 当你加了unpack时，传递的数据量需要一致；如列表例子中，每个列表我都固定传了2个数据，当你多传或少传时会报错，而test_case方法的参数也要写2个，需要匹配上
• 当传的数据是字典类型时，要注意每个字典的key都要一致，test_case的参数的命名也要一致；如字典的例子，两个字典的key都是value1和value2，而方法的参数也是
• 当传的数据是通过变量的方式，变量前需要加上*

列表

import ddt
import unittest

@ddt.ddt
class test_PersonInfo(unittest.TestCase):
    @classmethod
    def setUp(self):
        print("==========开始测试==========")
    @classmethod
    def tearDown(self):
        print("==========结束测试==========")
 
    @ddt.data([1,2],[3,4])
    def test_print(self,a):
        print(a)

if __name__ == '__main__':
    unittest.main()
=====================输出结果====================

[1, 2]

[3, 4]

print(a[0])

1
3

 

 

列表拆包

    @ddt.data([1,2],[3,4])
    @ddt.unpack
    def test_print(self,a,b):
        print(a,b)
 
==========开始测试==========
1 2
==========结束测试==========
==========开始测试==========
3 4
==========结束测试==========

 

字典

@data({'value1': 1, 'value2': 2}, {'value1': 1, 'value2': 2})
    @unpack
    def test_dict(self, value1, value2):
        print("test_dict", value1, value2)

===============输出==============
test_dict 1 2
test_dict 1 2

 

 

元祖

tuples = ((1, 2, 3), (1, 2, 3))
 @data(*tuples)
    def test_tuples(self, n):
        print("test_tuples", n)
=============输出================
test_tuples (1, 2, 3)
test_tuples (1, 2, 3)

 

 

包含字典的列表

datas = [{'class': 'python2', 'name': 'xiaoshitou', 'id': '0001', 'sex': 'male'},
         {'class': 'python2', 'name': 'nuonuo', 'id': '0002', 'sex': 'female'},
         {'class': 'python2', 'name': 'fly', 'id': '0003', 'sex': 'male'},
         {'class': 'python2', 'name': 'haiyang', 'id': '0004', 'sex': 'male'},
         {'class': 'python2', 'name': 'shuangshuang', 'id': '0005', 'sex': 'female'}
]
@ddt.data(*datas)
    def test_print(self,datas):
        print(datas)

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

            
==========开始测试==========
{'class': 'python2', 'name': 'xiaoshitou', 'id': '0001', 'sex': 'male'}
==========结束测试==========
==========开始测试==========
{'class': 'python2', 'name': 'nuonuo', 'id': '0002', 'sex': 'female'}
==========结束测试==========
==========开始测试==========
{'class': 'python2', 'name': 'fly', 'id': '0003', 'sex': 'male'}
==========结束测试==========
==========开始测试==========
{'class': 'python2', 'name': 'haiyang', 'id': '0004', 'sex': 'male'}
==========结束测试==========
==========开始测试==========
{'class': 'python2', 'name': 'shuangshuang', 'id': '0005', 'sex': 'female'}
==========结束测试==========

 

 

传递json文件

dat-data数据
数据为文件对象：Json yaml
@ddt.file_data(jason文件路径) json格式里面必须为双引号“”，且必须为键值对，不存在纯列表格式，值作为测试数据
{"name":"nick","gender":"male","age":29}

JSON文件

{
  "first": [
    {
      "isRememberMe": "True",
      "password": "111111",
      "username": "root"
    },
    "200"
  ],
  "second": [
    "{'isRememberMe': True, 'password': '1111111', 'username': 'root'}",
    "406"
  ],
  "third": [
    1,
    2
  ],
  "four": "123123"
}

 

 

单元测试类

from ddt import *


# 在测试类前必须首先声明使用 ddt
@ddt
class imoocTest(unittest.TestCase):

    @file_data('F:/test/config/testddt.json')
    def test_json(self, data):
        print(data)
 
==========输出结果==========
[{'isRememberMe': 'True', 'password': '111111', 'username': 'root'}, '200']
["{'isRememberMe': True, 'password': '1111111', 'username': 'root'}", '406']
[1, 2]
123123
      

 

 

传递YAML文件

YAML文件

unsorted_list:
  - 10
  - 15
  - 12

sorted_list: [ 15, 12, 50 ]

 

 

单元测试类

from ddt import *


# 在测试类前必须首先声明使用 ddt
@ddt
class imoocTest(unittest.TestCase):

    @file_data('F:/test/config/testddt.yaml')
    def test4(self, data):
        print("yaml", data)
===============输出结果==================
yaml [10, 15, 12]
yaml [15, 12, 50]

 

 

3. unittestreport

这个也有ddt功能，使用方法同ddt

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2021/11/8 21:08
# @Author  : shisuiyi
# @File    : test_login.py
# @Software: win10 Tensorflow1.13.1 python3.9
import unittest
from testcode import login
# from ddt import ddt, data
from unittestreport import ddt, list_data
cases = [
    {"title": "登陆成功", "request": {"username": "python45", "password": "lemonban"},
     "expect": {"code": 0, "msg": "登录成功"}},
    {"title": "账号或密码不正确", "request": {"username": "python45", "password": "lemonban123"},
     "expect": {"code": 1, "msg": "账号或密码不正确"}},
    {"title": "账号或密码不正确", "request": {"username": "python45123", "password": "lemonban"},
     "expect": {"code": 1, "msg": "账号或密码不正确"}},
    {"title": "密码长度在6-18位之间", "request": {"username": "python45", "password": "lemo"},
     "expect": {"code": 1, "msg": "密码长度在6-18位之间"}},
    {"title": "密码长度在6-18位之间", "request": {"username": "python45", "password": "lemonbanlemonbanlemonbanlemonbanlemonban"},
     "expect": {"code": 1, "msg": "密码长度在6-18位之间"}},
]


@ddt
class TestLogin(unittest.TestCase):
    """
    测试登录函数
    """

    @classmethod
    def setUpClass(cls) -> None:
        """
        类级前置
        :return:
        """
        print('我是类级前置，在整个类的测试前执行')

    @classmethod
    def tearDownClass(cls) -> None:
        """
        类级后置
        :return:
        """
        print('我是类级后置，在整个测试类测试完成后执行')

    def setUp(self) -> None:
        """
        方法级前置
        :return:
        """
        print('======我会在每个测试执行前执行====')

    def tearDown(self) -> None:
        """
        方法级后置
        :return:
        """
        print('======我会在每个测试执行后执行======')

    # @data(*cases)
    @list_data(cases)
    def test_login(self, case):
        
        # 1. 准备测试数据
        print(case)
        print('{}开始测试'.format(case['title']))
        # 2. 测试步骤
        res = login(**case['request'])
        # 3. 断言
        self.assertEqual(res, case['expect'])


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