python单元测试框架

单元测试dome.py

class  Calc:

    def  add(self,a,b):
        return a + b

    def minus(self,a,b):
        return a - b

    def multi(self,a,b):
        return a * b
    
    def devision(self,a,b):
        return a / b

TestCalc.py

import time
from unittest import  TestCase
from ddt import  ddt
from ddt import data
from ddt import  unpack
from 单元测试dmeo import Calc
import xlrd
from xlutils.copy import  copy

# 任务一：xlrd，读取参数化数据
# da = xlrd.read()

def readData():
    li = []
    wb = xlrd.open_workbook(filename="calc.xls", encoding_override=True)
    st = wb.sheet_by_index(0)
    cols = st.ncols
    rows = st.nrows
    for i in range(1,rows):# 第一行属于表头数据，肯定不要，从第二行开始读取数据
        l = []
        for j in range(0,cols-3):
            l.append(st.cell_value(i,j))
        li.append(l)

    return li

def writeData(row,col,data):
    new_wb = copy(xlrd.open_workbook(filename="calc.xls",encoding_override=True))
    new_wb.get_sheet(0).write(row,col,data)
    new_wb.save("calc.xls")


@ddt
class TestCalc(TestCase):

    @data(*readData())
    @unpack
    def testAdd1(self,d,a,b,c):
        calc = Calc()
        s = calc.add(a,b)
        dt = time.localtime()
        ft = "%Y-%m-%d %H:%M:%S"
        nt = time.strftime(ft, dt)
        if s == c:
            writeData(d,4,"成功！")
        else:
            writeData(d,4,"失败！")
        self.assertEqual(s,c)
        writeData(d, 6, nt)