Python unittest数据处理ddt

1.装饰器

#装饰器的官方定义：
装饰器本质上是一个Python函数(其实就是闭包)，它可以让其他函数在不需要做任何代码变动的前提下增加额外功能，装饰器的返回值也是一个函数对象。装饰器用于有以下场景，比如:插入日志、性能测试、事务处理、缓存、权限校验等场景。

2.ddt

import unittest
#引入ddt库
from ddt import ddt,data,unpack
test_data = [{"name":"张三"},2]

@ddt#装饰测试类
class DemoCase(unittest.TestCase):
    @data(test_data)#装饰函数
    def test_jia(self,item):
        print(item)
    def test_jian(self):
        pass

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

运行结果：

plugins: html-3.2.0, metadata-2.0.4collected 2 items demo.py .[{'name': '张三'}, 2] . [100%] ============================== 2 passed in 0.04s ============================== Process finished with exit code 0

3.ddt脱外套

import unittest
#引入ddt库
from ddt import ddt,data,unpack
test_data = [{"name":"张三"},2]

@ddt#装饰测试类
class DemoCase(unittest.TestCase):
    @data(*test_data)#装饰函数，*test_data脱外套，根据脱外套后的数据执行次数
    def test_jia(self,item):
        print(item)

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

运行结果：
plugins: html-3.2.0, metadata-2.0.4collected 2 items
demo.py .{'name': '张三'}
.2                                                              [100%]
============================== 2 passed in 0.04s ==============================
Process finished with exit code 0

4.ddt中的@unpack

import unittest
#引入ddt库
from ddt import ddt,data,unpack
test_data = [[2,6],[9,1000]]

@ddt#装饰测试类
class DemoCase(unittest.TestCase):

    @data(*test_data)#装饰函数，*test_data脱外套，根据脱外套后的数据执行次数
    @unpack
    def test_jia(self,a,b):
        print("a: ",a)
        print("b: ",b)

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

运行结果：
plugins: html-3.2.0, metadata-2.0.4collected 2 items
demo.py 
.a:  2
b:  6
.a:  9
b:  1000                                                              [100%]
============================== 2 passed in 0.04s ==============================

 

5.ddt+openpyxl参数化实现

import unittest
#引入ddt库
from ddt import ddt,data,unpack
from test01.doexcel import DoExcel

#引入仓库
from openpyxl import load_workbook

class DoExcel():
    def __init__(self,file,sheet):
        self.file=file
        self.sheet=sheet

    def return_excel_value(self):
        wb = load_workbook(self.file)#打开excel
        sheet_content = wb[self.sheet]#定位sheet工作博
        data_list = []#列表用于存储测试数据
        for n in range(2,sheet_content.max_row+1):#行,第一行是标题，所以从第二行开始
            data_dict = {}#字典用于存储每组测试数据
            for m in range(2,sheet_content.max_column+1):
                data_dict["method"]=sheet_content.cell(n,2).value
                data_dict["url"] = sheet_content.cell(n, 3).value
                data_dict["data"] = eval(sheet_content.cell(n, 4).value)#eval()将数据类型还原
                data_dict["expect"] = sheet_content.cell(n, 5).value
            data_list.append(data_dict)#将字典存储到list
        return data_list

DoExcel类代码

test_data = DoExcel("C:\Users\Administrator\Desktop\testdemo.xlsx","s1").return_excel_value()#[{},{},{}]

@ddt#装饰测试类
class DemoCase(unittest.TestCase):

    @data(*test_data)#装饰函数，*test_data脱外套，根据脱外套后的数据执行次数
    def test_jia(self,item):
        print(item["method"],item["url"],item["data"],item["expect"])

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

执行结果：
plugins: html-3.2.0, metadata-2.0.4collected 2 items
demo.py .post http://www.qabujiaban.com/user/login {'username': 'uuuu222都44', 'password': 'WJHasb124*1'} 0000
.get http://www.qabujiaban.com/user/login {'username': 'uuuu222都44', 'password': 'WJHasb124*1'} 0000
                                                               [100%]
============================== 2 passed in 0.31s ==============================
Process finished with exit code 0