excel单元测试---步骤2:excel中存储测试数据,读取excel数据

读取excel数据

from openpyxl import load_workbook  # 打开excel

class DoExcel:

    def __init__(self,file_name,sheet_name):
        self.file_name = file_name
        self.sheet_name = sheet_name
        self.sheet_obj = load_workbook(self.file_name)[self.sheet_name]
        self.max_row = self.sheet_obj.max_row


    def get_datas(self):
        """根据嵌套循环读取数据"""
        wb = load_workbook(self.file_name)
        sheet = wb[self.sheet_name]
        test_data = []
        for i in range(2, sheet.max_row + 1):
            sub_data = {}
            sub_data['method'] = sheet.cell(i, 4).value
            sub_data['url'] = sheet.cell(i, 5).value
            sub_data['data'] = sheet.cell(i, 6).value
            sub_data['expected'] = sheet.cell(i, 7).value
            test_data.append(sub_data)  # 每条测试用例作为子元素传入test_data[]中
        return test_data

猜你喜欢

转载自www.cnblogs.com/kite123/p/12598222.html