python单元测试_读取excel表中的数据

from openpyxl import load_workbook
class do_excel:
    def __init__(self,file_name,sheet_name):
        self.file_name=file_name
        self.sheet_name=sheet_name
    def get_data(self):
        wb=load_workbook("test.xlsx")
        sheet=wb[self.sheet_name]
        test_data=[]
        for i in range(1,sheet.max_row+1):
            sub_data = {}
            sub_data['method']=sheet.cell(i,1).value
            sub_data['url']=sheet.cell(i,2).value
            sub_data['data']=sheet.cell(i,3).value
            sub_data['expected']=sheet.cell(i,4).value
            test_data.append(sub_data)
        return test_data

if __name__ == '__main__':
    print(do_excel("test.xlsx",'Sheet1').get_data())