Python操作Excel，并结合unittest单元测试框架

第一步：写Excel操作方法

excel_operate.py文件

from openpyxl import load_workbook  #引入模块

class MyExcel: def __init__(self,filepath,sheetname):  # 初始化函数在实例化对象时调用
        self.wb = load_workbook(filepath) self.sh = self.wb[sheetname] # 读取指定单元格的数据
    def read_data(self,row,column): return self.sh.cell(row,column).value # 读取所有数据
    def read_alldate(self): for index in range(1,self.sh.max_row+1): print('行号：',index) for i in range(1,self.sh.max_column+1): print('列号：',i,'内容：',self.sh.cell(row=index,column=1).value) # 写入数据
    def write_data(self,row,column,content): self.sh.cell(row,column).value = content # 保存数据
    def save_data(self,filepath): self.wb.save(filepath) # excel = MyExcel('testdata.xlsx','Sheet1') # print(excel.read_data(2,1)) # # excel.write_data(4,1,'12345678936') # excel.write_data(4,2,'a1234567') # excel.save_data('testdata.xlsx')

第二步：写测试用例

test_excel_operate.py文件

import unittest from excel_operate import MyExcel class TestExcel(unittest.TestCase): @classmethod def setUpClass(cls): cls.excel = MyExcel('testdata.xlsx','Sheet1') # 测试用例 读
    def test_read_data(self): pass

    # 测试用例 写
    def test_write_data(self): # 调用自己的写方法
        self.excel.write_data(5, 1, '12345678937') self.excel.write_data(5, 2, 'a1234567') self.excel.save_data('testdata.xlsx')

第三步：执行测试用例，生成HTML报告

TestSuite_Excel.py文件

import unittest from HTMLTestRunnerNew import HTMLTestRunner from test_excel_operate import TestExcel import time import os if __name__ == '__main__': s = unittest.TestSuite() s.addTests(unittest.TestLoader().loadTestsFromTestCase(TestExcel)) now = time.strftime('%Y%m%d_%H%M%S') filename = open(os.getcwd() + '/ExcelUnittestReport_' + now + '.html','wb') runner = HTMLTestRunner( stream = filename, title = 'Excel单元测试报告', description = 'Excel单元测试报告', tester = 'lixiaozhe' ) runner.run(s)