前言
数据驱动测试是一种有效的测试方法，可以使用不同的输入数据运行相同的测试用例。结合pytest和Excel表格可以方便地实现数据驱动测试。在这篇文章中，我们将介绍如何使用pytest和Excel实现数据驱动测试。
环境准备
在实现Excel的数据驱动之前，我们首先要安装Python操作excel的第三方库，我们这里使用openpyxl来实现对Excel的操作，安装命令如下：
pip install openpyxl
创建测试文件和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
pytest将会读取test_code.py文件中的测试用例，并根据Excel文件中提供的数据对multiply函数进行测试。每个测试用例都会使用Excel文件中的数据进行数据驱动测试。
总结
结合pytest和Excel表格，我们成功实现了数据驱动测试，对同一个函数在不同输入下进行了多组测试。这种方法不仅使得测试用例易于维护，而且可以轻松地扩展更多测试数据，确保代码的稳定性和正确性。数据驱动测试的优势在于能够更全面地覆盖不同的边界情况，提高测试的可靠性。
