读取适用于pytest的参数化数据

import xlrd

def open_file(filename,tablename):
    '''打开文件'''
    open_file = xlrd.open_workbook(filename) #打开表
    open_table = open_file.sheet_by_name(tablename) #打开Tab
    nrows = open_table.nrows  # 获取总行数
    ncols = open_table.ncols  # 获取总列数
    keys =open_table.row_values(0) #获取第一行作为Key
    ncols_ed =open_table.col_values(0)[1:] #获取第一列,作为ids值,供参数化时使用
    # print(ncols_ed)
    return open_table,nrows,ncols,keys,ncols_ed


def read_excel(open_table,nrows,ncols,keys,ncols_ed):
    '''读取excel内容'''

    j=1
    list1=[]
    for i in range(nrows-1):
        dict1 = {}
        values= open_table.row_values(j) #取一列进行遍历,将key和value对应
        for k in range(ncols):
            dict1[keys[k]]=values[k]
        j +=1
        list1.append(tuple(dict1.values())) #取字典的值,加入列表之前转换成元祖
    # 返回list1,ncols_ed组成的元祖
    return list1,ncols_ed


filename="/Users/liujing/PycharmProjects/untitled1/practice/test_datas/creat_special_subjest_dates.xlsx"
tablename="Sheet1"
a = open_file(filename,tablename)
# print(read_excel(*a))
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章