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

你在学习 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 的核心概念:

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

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

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