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数据截图如下: