python 获取CSV数据(数据驱动+pytest参数化)




 1 import csv
 2 import pytest
 3 
 4 
 5 def get_csv_data():
 6     '''
 7     @return: csv_list 返回列表
 8     '''
 9     with open('test.csv') as f:
10         csv_list = []
11         lst = csv.reader(f)  # csv.reader返回的是列表,可以对列表进行遍历
12         for row in lst:
13             csv_list.append(row)
14     return csv_list
15 
16 
17 @pytest.mark.parametrize('name,age,six', get_csv_data())
18 def test_csv(name, age, six):
19     print(name, age, six)
20 
21 
22 if __name__ == '__main__':
23     # print(get_csv_data())
24     pytest.main(['-vs', 'test_csv.py'])
附上:test.csv数据截图如下: