前言
数据驱动测试是一种有效的测试方法,可以使用不同的输入数据运行相同的测试用例。结合 pytest 和 Excel 表格可以方便地实现数据驱动测试。在这篇文章中,我们将介绍如何使用 pytest 和 Excel 实现数据驱动测试。
环境准备
在实现 Excel 的数据驱动之前,我们首先要安装 Python 操作 excel 的第三方库,我们这里使用openpyxl
来实现对 Excel 的操作,安装命令如下:
创建测试文件和 Excel 文件
首先,创建一个简单的函数需要进行测试:
# code.py
def multiply(a, b):
return a * b
复制代码
接着,创建一个 Excel 文件,用于存储测试数据:
| Test Case | Operand A | Operand B | Expected Result |
|-----------|------------|------------|------------------|
| Case 1 | 2 | 3 | 6 |
| Case 2 | -1 | 5 | -5 |
| Case 3 | 0 | 10 | 0 |
复制代码
编写测试用例
创建一个 pytest 测试文件,编写测试用例并读取 Excel 文件中的数据:
# test_code.py
import pytest
from openpyxl import load_workbook
from code import multiply
def load_test_data():
workbook = load_workbook('test_data.xlsx')
sheet = workbook.active
test_data = []
for row in sheet.iter_rows(min_row=2, values_only=True):
test_data.append(row)
return test_data
@pytest.mark.parametrize("test_case, operand_a, operand_b, expected_result", load_test_data())
def test_multiply(test_case, operand_a, operand_b, expected_result):
result = multiply(operand_a, operand_b)
assert result == expected_result, f"测试用例 {test_case} 失败:{operand_a} * {operand_b} 应该得到 {expected_result},实际得到 {result}"
复制代码
运行测试
我们可以点击 pycharm 里的文件的小三角来执行测试,也可以运行 pytest 命令来执行测试:
pytest
将会读取test_code.py
文件中的测试用例,并根据 Excel 文件中提供的数据对multiply
函数进行测试。每个测试用例都会使用 Excel 文件中的数据进行数据驱动测试。
总结
结合pytest
和 Excel 表格,我们成功实现了数据驱动测试,对同一个函数在不同输入下进行了多组测试。这种方法不仅使得测试用例易于维护,而且可以轻松地扩展更多测试数据,确保代码的稳定性和正确性。数据驱动测试的优势在于能够更全面地覆盖不同的边界情况,提高测试的可靠性。
评论