pytest结合csv模块实现数据驱动测试详解

Mia

发布时间：2026-10-15| 3人浏览过

原创

本文介绍如何使用pytest结合Python标准库csv模块实现数据驱动测试（DDT）。通过读取CSV文件中的多组数据，利用@pytest.mark.parametrize装饰器进行参数化迭代，从而用单一测试用例验证多种输入场景，提升测试效率与代码复用性。

一、什么是数据驱动测试

DDT即Data-Driven Testing（数据驱动测试）。其核心思想是准备一组测试数据，编写一个测试用例，通过迭代不同的数据来执行该用例，从而覆盖多种输入情况。

二、csv文件承载数据

CSV全称为Comma Separated Values（逗号分隔值）。

CSV文件本质上是使用逗号作为分隔符的纯文本文件。在Python中，可以使用内置的csv模块来解析和处理CSV文件。结合pytest的参数化功能，可以实现基于CSV文件的数据驱动测试。

三、csv数据驱动示例

首先，需要创建一个包含测试数据的CSV文件。

zhangsan,lisi,wangwu
tom,rose,lili

接着，创建名为test_csv.py的Python测试模块。

# -*- coding: utf-8 -*-
# @Time: 2022/5/29 4:20 下午
# @Author: wcystart
# @File: test_csv.py
# @description:
import pytest
import csv
def get_data():
    with open("test.csv") as f:
        lst = csv.reader(f)
        my_data = []
        for row in lst:
            my_data.extend(row) 
        return my_data
@pytest.mark.parametrize('name', get_data())
def test01(name):
    print(name)
if __name__ == '__main__':
   pytest.main(['-vs', 'test_csv.py'])
运行结果；
test_csv.py::test01[zhangsan] zhangsan
PASSED
test_csv.py::test01[lisi] lisi
PASSED
test_csv.py::test01[wangwu] wangwu
PASSED
test_csv.py::test01[tom] tom
PASSED
test_csv.py::test01[rose] rose
PASSED
test_csv.py::test01[lili] lili
PASSED

至此，关于pytest结合csv模块实现CSV格式数据驱动的介绍完成。如需了解更多相关pytest与CSV数据驱动的内容，建议查阅本站往期文章或继续浏览后续相关教程，感谢支持。