写点什么

软件测试 / 测试开发全日制|Pytest 结合 CSV 实现测试的数据驱动

  • 2024-01-11
    北京
  • 本文字数:854 字

    阅读完需:约 3 分钟

前言

数据驱动测试是一种有效的测试方法,可以使用不同的输入数据运行相同的测试用例。结合 pytest 和 CSV 文件可以方便地实现数据驱动测试,相比于 yaml 以及 Excel,使用 CSV 实现数据驱动不需要借助其他的第三方库,可以直接使用 Python 读取数据。在这篇文章中,我们将介绍如何使用 pytest 和 CSV 实现数据驱动测试。

创建测试文件和 CSV 文件

假设我们要测试一个简单的函数,比如计算两个数的乘积:


# code.py
def multiply(a, b): return a * b
复制代码


接下来,创建一个 CSV 文件,用于存储测试数据:


operand_a,operand_b,expected_result2,3,6-1,5,-50,10,00.5,2,10.5,0.5,0.25
复制代码

编写测试用例

创建一个 pytest 测试文件,编写测试用例并读取 CSV 文件中的数据:


# test_code.py
import csvimport pytestfrom code import multiply
def load_test_data(): test_data = [] with open('test_data.csv', newline='') as csvfile: reader = csv.DictReader(csvfile) for row in reader: test_data.append(row) return test_data
@pytest.mark.parametrize("data", load_test_data())def test_multiply(data): operand_a = int(data['operand_a']) operand_b = int(data['operand_b']) expected_result = int(data['expected_result'])
result = multiply(operand_a, operand_b) assert result == expected_result, f"{operand_a} * {operand_b} 应该得到 {expected_result},实际得到 {result}"
复制代码

运行测试

现在,运行 pytest 命令来执行测试:


pytest test_code.py
复制代码


pytest将会读取test_code.py文件中的测试用例,并根据 CSV 文件中提供的数据对multiply函数进行测试。每个测试用例都会使用 CSV 文件中的数据进行数据驱动测试。

总结

结合pytestCSV文件,我们成功实现了数据驱动测试,对同一个函数在不同输入下进行了多组测试。这种方法使得测试用例易于管理,同时能够更全面地覆盖不同的测试场景,确保代码的稳定性和正确性。数据驱动测试的优势在于能够快速扩展测试数据,提高测试的覆盖率和可靠性。


用户头像

社区:ceshiren.com 微信:ceshiren2021 2019-10-23 加入

微信公众号:霍格沃兹测试开发 提供性能测试、自动化测试、测试开发等资料,实时更新一线互联网大厂测试岗位内推需求,共享测试行业动态及资讯,更可零距离接触众多业内大佬。

评论

发布
暂无评论
软件测试/测试开发全日制|Pytest结合CSV实现测试的数据驱动_霍格沃兹测试开发学社_InfoQ写作社区