测试数据管理的这个神级工具,大佬们都舍不得告诉你

本贴最后更新于 1616 天前,其中的信息可能已经天翻地覆

你在学习 python 自动化测试吗?听过 requests 库吗?tablib 是 requests 库常年维护的一个可以操作 Excel 等多种文件格式,将他们变成一种通用数据集的第三方库。

tablib 支持的主要数据格式有:

也就是说,tablib 能把不同格式的数据转化成一种通用的关系型数据格式,然后再各个格式之间无缝切换。什么叫关系型数据格式呢?比如:

tablib 这种通用数据格式的特性解决了以前一些 Excel 操作库的一些问题。

操作 Excel 的几个常用库

这些库都是非常优秀的库,限制性也有:

  1. xlrd 的问题。他只能读,不能写。要写还得另外装库。
  2. openpyxl 的问题。只支持 xlsx 系列的读写。因为现在大多使用这种 Excel 格式,所以这个问题其实可以忽略,但如果你的 office 软件比较老,只能用 xls 格式,这个库就用不了了。

tablib 的安装

  1. 安装 pip install tablib
  2. 导入 import tablib

核心概念

tablib.Dataset() # 相当于 Excel 中的 sheet tablib.Databook() # 相当于 Excel 中的 workbook 工作簿

Dataset 快速入门

先看一个小例子:

url method expected
lemon.ke.qq.com get 成功
lemonban.com post 成功

url, method, expected 是一个表的表头 header,其实就是每一列的名字,在数据库里我们称为字段名。下面的行都是数据。

想要创建一个 Excel 非常的简单,只需要准备你需要的数据 data, 表头 header 和 表的名字 title:

import tablib # 表各列的标题 header headers = ['url', 'method', 'expected'] # 需要存到 excel 的数据 data_list = [ ['https://lemon.ke.qq.com', 'get', '成功'], ['https://lemonban.com', 'post', '成功'] ] # 生成数据 data = tablib.Dataset(*data_list, headers=headers,title='测试用例') print(data)

打印的结果是这样的:

image.png

保存为 excel

要保存为 excel 文档,只需要像操作普通的文件一样读写就可以了,写入 data.xls 或者 data.xlsx 属性里面的数据:

with open('demo.xls', 'wb') as f: f.write(a.xls) with open('demo.xlsx', 'wb') as f: f.write(a.xlsx)

注意:模式需要用二进制模式 wb

保存之后的结果是这样的:

image.png

Databook

Databook 就是类似工作簿的概念,所以 Databook 需要的参数实际上就是上文中的 DataSet。如果一个 Excel 里只有一个表,用 DataSet 就够了,如果有多个表:

book = tablib.Databook([data, data2]) with open('demo_book.xls', 'wb') as f: f.write(book.xls)

表格导入:import_set

说完 Excel 的写操作, 接下来是读操作, 读取一个 excel 文件也是和普通文件读写类似,调用 import_set 方法:

with open('demo.xls', 'rb') as f: # 接受 2 个参数,读出来的数据和读取的文件格式 data = tablib.import_set(f.read(), 'xls') print(data) # 或者采用第二种方式 with open('demo.xls', 'rb') as f: data = tablib.DataSet().load(f.read(), 'xls') print(data)

获取数据进行自动化测试

用一个例子来实战,获取 excel 文件里的数据,执行自动化测试程序:

def api_tester(url, method, expected): print("正在测试{},请求方法{}, 预期结果{}".format( url, method, expected)) with open('demo.xls', 'rb') as f: # 接受 2 个参数,读出来的数据和读取的文件格式 data = tablib.import_set(f.read(), 'xls') print(data) for i in data: api_tester(*i)

导入book

和 DataSet 的操作一样的:

with open('demo.xls', 'rb') as f: # 接受 2 个参数,读出来的数据和读取的文件格式 data = tablib.import_book(f.read(), 'xls') print(data) # 或者采用第二种方式 with open('demo.xls', 'rb') as f: data = tablib.DataBook().load(f.read(), 'xls') print(data)

总结

这篇我们熟悉了 tablib 的核心概念:

下篇我们将行列和数据操作,以及它的一些特性。

1 操作
yuze 在 2020-08-06 17:50:13 更新了该帖
1 回帖
请输入回帖内容 ...
  • xiaojian

    这个工具有意思,跟pandas有那么丢丢像