自动化测试：yaml结合ddt实现数据驱动

在python+unittest+selenium+ddt的框架中，数据驱动常见有以下几种方式实现：

Csv/txt

Excel

YAML

本文主要给大家介绍测试数据存储在YAML文件中的使用场景。首先先来简单介绍一下YAML。

1. 什么是YAML

一种标记语言类似YAML，它实质上是一种通用的数据串行化格式，由于其可读性高，用来表达数据序列化的格式，非常简洁和强大。

它的基本语法规则如下：

大小写敏感

使用缩进表示层级关系

缩进时不允许使用Tab键，只允许使用空格。

缩进的空格数目不重要，只要相同层级的元素左侧对齐即可

YAML 支持的数据结构有三种：

对象：键值对的集合，即python中的字典

数组：一组按次序排列的值，即列表

纯量：单个的、不可再分的值

如列表中数据[1, 2, ’a’, ’b’]，在YAMLl中这样写， 横线+空格+数据：

字典数据这样写: key+冒号+空格+value

复杂数据结构，如列表嵌套字典，写成这样：

以上是关于YAML数据的基本表示方式，更多细节内容这里不一一详述。

2. 结合ddt实现数据驱动

假设对登录操作实现数据驱动。

▲ 首先创建YAML格式文件，添加多组测试数据。

在此添加三组，分别对应登录成功、账号错误及密码错误的三种场景：

▲ 第二步，对自动化代码做如下改动：

A. 对登录测试方法改写，定义传入参数，并在执行过程中使用改数据

B. 对测试方法添加file_data装饰器, 从指定yaml文件读取数据

C. 对测试类添加装饰器ddt, 启动运行

主要代码实现如下：

以上实现，非常简便: YAML管理数据非常方便，数据读取时不需要额外的代码进行读取，直接使用file_data装饰器即可。

操作过程有问题，欢迎讨论。