开发笔记:pytest参数化实现DDT：读取CSV数据

1.DDT思想

DDT(Data-Drive test)&＃xff1a;将测试数据提取出来&＃xff0c;与逻辑分离&＃xff0c;通过参数传递不同的测试数据来驱动用例运行。pytest中的pytest.mark.parametrize()便可实现参数化。

2.csv文件

csv&＃xff1a;逗号分隔的文本文件&＃xff0c;其文件以纯文本形式存储表格数据。当然分隔符也不一定是逗号&＃xff0c;最常见的是逗号或制表符。例如&＃xff1a;一个命名为test.csv的文件中的数据如下所示&＃xff1a;

a,b,c
1,2,3
3,4,5


3.读取csv数据&＃xff0c;实现DDT
创建test_csv.py,代码如下&＃xff1a;

import pytest
import csv
def get_data():
with open(&＃39;test.csv&＃39;) as f:
lst &＃61; csv.reader(f)
my_data &＃61; []
for row in lst:
print(row)
my_data.extend(row)
print(my_data)
return my_data
&＃64;pytest.mark.parametrize(&＃39;name&＃39;,get_data())
def test01(name):
print(name)
if __name__ &＃61;&＃61; &＃39;__main__&＃39;:
#get_data()
pytest.main([&＃39;-sv&＃39;, &＃39;test_csv.py&＃39;])


读取csv文件需要导入csv模块。查看reader方法可知&＃xff1a;
csv.reader(f)返回一个csv_reader对象&＃xff0c;再通过for循环去遍历&＃xff0c;每次返回一行。

csv_reader &＃61; reader(iterable [, dialect&＃61;&＃39;excel&＃39;]
[optional keyword args])
for row in csv_reader:
process(row)


我们在与test_csv.py同目录下创建一个test.csv&＃xff0c;数据如下&＃xff1a;

collecting ... [&＃39;a&＃39;, &＃39;b&＃39;, &＃39;c&＃39;]
[&＃39;1&＃39;, &＃39;2&＃39;, &＃39;3&＃39;]
[&＃39;3&＃39;, &＃39;4&＃39;, &＃39;5&＃39;]
[&＃39;a&＃39;, &＃39;b&＃39;, &＃39;c&＃39;, &＃39;1&＃39;, &＃39;2&＃39;, &＃39;3&＃39;, &＃39;3&＃39;, &＃39;4&＃39;, &＃39;5&＃39;]
collected 9 items
test_csv.py::test01[a] a
PASSED
test_csv.py::test01[b] b
PASSED
test_csv.py::test01[c] c
PASSED
test_csv.py::test01[1] 1
PASSED
test_csv.py::test01[2] 2
PASSED
test_csv.py::test01[30] 3
PASSED
test_csv.py::test01[31] 3
PASSED
test_csv.py::test01[4] 4
PASSED
test_csv.py::test01[5] 5
PASSED