python-unittests数据驱动测试（ddt）

搁浅同学

发布于 2022-07-21 14:49:21

4600

发布于 2022-07-21 14:49:21

最近在做接口自动化测试，发现之前写好的框架，每写一条用例都要写一个test函数，这样子就会造成了很多冗余代码，对以后的用例维护也很不方便。对于一个接口测试用例有很多条，就会对应有很多组数据，目前的做法是一组数据，一个test函数。这样子代码很多重复冗余，最后我把它进行了升级，把测试数据全部都填写在Excel表格里面，然后读取处理一个list，再通过ddt传递给test函数，这样子一个test函数就可以处理很多组数，下面来说一下这个ddt。

一、环境准备

安装ddt模块，通过pip命令安装，打开cmd命令

二、ddt数据驱动测试原理

ddt主要通过@data（数据类型）,@upack这两个装饰器进行数据的传递给测试函数，废话不多说。下面进行举例子实战。

1、传1个参数，如测试1个数大于5，通过@data

#!/usr/bin/env python
#coding=utf-8
from ddt import data,unpack,ddt
import unittest

@ddt
class Sample(unittest.TestCase):
    def setUp(self):
        pass
    
    """
    测试1个数是否等于5
    """
    @data(1,2,5)
    def test_number(self,num):
        self.assertEqual(5,num,msg='fail')
    def tearDown(self):
        pass

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

运行结果如下：

总共运行了3条记录，失败2条，成功1条

2、传递多个参数，例如2个参数，通过@data,@unpack，其中@unpack是告诉每个元组里面的元素作为1个参数传递

#!/usr/bin/env python
#coding=utf-8
from ddt import ddt,data,unpack
import unittest

@ddt
class Sample(unittest.TestCase):
    def setUp(self):
        pass
    @data((1,2),(2,2),(3,3))
    @unpack
    def test_two_number(self,a,b):
        self.assertEqual(a,b,msg='Fail')
    def tearDown(self):
        pass

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

3、还可以传递列表、字典等复杂的数据结构，如下

#!/usr/bin/env python
#coding=utf-8
from ddt import ddt,data,unpack
import unittest

@ddt
class Sample(unittest.TestCase):
    def setUp(self):
        pass
    @data([1,2,3],[2,3,4,5,6,])
    def test_list_leght(self,list):
        self.assertEqual(5,len(list),msg='Fail')
    def tearDown(self):
        pass

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

ddt的使用大概就是这样，如果需要把接口测试数据填写在Excel表格里面，则需要通过xlrd等库读取出来作为1个list，然后通过ddt就可以完成了，数据驱动测试了。

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2019-08-20，如有侵权请联系 cloudcommunity@tencent.com 删除

uml