在ui自动化中,经常会用到数据的存储和读取,这里主要介绍一下如何处理测试中经常用到的数据。
一般习惯把元素定位地址作为类变量存储,因为这样对于调试纠错比较方便,书写调用也较简单。
例如:Demo类中,在类下申请一个变量name,如果这个类中的方法要引用name,需要self.name这样的格式调用。
demo_class.py
class Demo():
name = "二狗子"
def print_name(self):
print(self.name)
如果其他class类要引用,那需要先import这个类,然后再使用Demo.name这种形式调用。
other_class.py
from demo_class import Demo
class Demo2():
print(Demo.name)
这里引用了pyyaml这个库,用来读取.yaml格式的文件。
如下所示:
account:
# 我是注释
user_1:
username:
"二狗子"
password:
meiyoumima
命令行安装:pip install pyyaml
代码导入库:import yaml
大致步骤是:
打开文件、载入文件、读取文件、关闭文件
需要注意的一点是:打开文件时,我设置了encoding格式,因为发现就算是加了引号的中文,部分字符仍然无法正确编码读取。(我猜是有些生僻字没有对应编码),所以这里设置了gb18030编码。
import yaml
# 打开文件
fr = open("D:/config.yaml", 'r', encoding='gb18030', errors='ignore')
# 载入文件格式
data = yaml.load(fr)
# 读取文件层级
get_user = data["account"]["user_1"]["username"]
print(get_user)
# 关闭文件
fr.close()
excel文件一般适合更加复杂的场景,如果只是简单的记录配置、数据,我认为还是类变量和yaml这种可以在python编辑器中读取编辑的类型更适合些。
引用库介绍:
安装:
pip install xlrd
pip install xlwt
引用:
import xlrd
import xlwt
文件的读取
xlrd.open_workbook(file_path)
获取excel文件的sheet名:
sheet_names()
根据sheet名/索引获取sheet表数据
sheet_by_index(index)
sheet_by_name(name)
获取sheet表中行数、列数
nrows # 行数
ncols # 列数
根据行、列索引获取行、列数据(索引从0开始)
获取某行数据
row_values(index)
获取某列数据
col_values(index)
根据坐标位置获取单元格数据(坐标是从0开始)
cell(row, col).value
基于上述的方法和功能,我在想,一般的读取excel表格,都是列名和数据对应的,所以写了如下方法,可以把每一行的数据作为一个字典,通过列名来对应,每行作为一个字典来存储在整体的列表中。
def excel_table_byname(file_path, colnameindex=0, sheet_name='Sheet1'):
"""
根据名称获取Excel表格中的数据
参数
file:Excel文件路径
colnameindex:表头列名所在行的索引
sheet_index:Sheet1名称
返回:列表,每个列表元素为列名对应的字典
"""
data = xlrd.open_workbook(file_path)
table = data.sheet_by_name(sheet_name)
nrows = table.nrows # 行数
ncols = table.ncols # 列数
# 列名行
colnames = table.row_values(colnameindex)
dic_list = []
# 从第一行到最后一行循环
for rown in range(1, nrows):
row = table.row_values(rown)
if row:
app = {}
# 单行数据组合为字典
for i in range(ncols):
app[colnames[i]] = row[i]
dic_list.append(app)
return dic_list
excel如图
实际演示示例:
import xlrd
# 读取载入文件
file = xlrd.open_workbook("F:/test.xlsx")
# 获取所有sheet名
print("sheet名称:%s" % file.sheet_names())
# 获取sheet
sheet1 = file.sheet_by_name("表1")
print(sheet1)
# 获取sheet中行列数
print("列数:%s" % sheet1.ncols)
print("行数:%s" % sheet1.nrows)
# 获取行、列数据
print("第1行数据:%s" % sheet1.row_values(0))
print("第1列数据:%s" % sheet1.col_values(0))
# 获取单元格数据
print("第2行、第2列数据:%s" % sheet1.cell(1, 1).value)
def excel_table_byname(file_path, colnameindex=0, sheet_name='Sheet1'):
"""
根据名称获取Excel表格中的数据
参数
file:Excel文件路径
colnameindex:表头列名所在行的索引
sheet_index:Sheet1名称
返回:列表,每个列表元素为列名对应的字典
"""
data = xlrd.open_workbook(file_path)
table = data.sheet_by_name(sheet_name)
nrows = table.nrows # 行数
ncols = table.ncols # 列数
# 列名行
colnames = table.row_values(colnameindex)
dic_list = []
# 从第一行到最后一行循环
for rown in range(1, nrows):
row = table.row_values(rown)
if row:
app = {}
# 单行数据组合为字典
for i in range(ncols):
app[colnames[i]] = row[i]
dic_list.append(app)
return dic_list
# 读取单个sheet表数据
print("读取excel数据....")
data_list = excel_table_byname(file_path="F:/test.xlsx",sheet_name="表1")
print(data_list)
for x in data_list:
print(x)
结果:
sheet名称:['表1']
列数:3
行数:6
第1行数据:['测试序号', '测试功能', '测试结果']
第1列数据:['测试序号', '1', '2', '3', '4', '5']
第2行、第2列数据:首页
读取excel数据。。。
[{'测试序号': '1', '测试结果': '成功', '测试功能': '首页'}, {'测试序号': '2', '测试结果': '失败', '测试功能': '流程'}, {'测试序号': '3', '测试结果': '成功', '测试功能': '资金'}, {'测试序号': '4', '测试结果': '成功', '测试功能': '账户'}, {'测试序号': '5', '测试结果': '失败', '测试功能': '服务'}]
{'测试序号': '1', '测试结果': '成功', '测试功能': '首页'}
{'测试序号': '2', '测试结果': '失败', '测试功能': '流程'}
{'测试序号': '3', '测试结果': '成功', '测试功能': '资金'}
{'测试序号': '4', '测试结果': '成功', '测试功能': '账户'}
{'测试序号': '5', '测试结果': '失败', '测试功能': '服务'}