python-unittests数据驱动测试(ddt)

2022-07-21 14:49:21 浏览数 (1)

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

一、环境准备

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

二、ddt数据驱动测试原理

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

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

代码语言:javascript复制
#!/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个参数传递

代码语言:javascript复制
#!/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、还可以传递列表、字典等复杂的数据结构,如下

代码语言:javascript复制
#!/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就可以完成了,数据驱动测试了。

0 人点赞