试想,一个项目的测试用例,对应多个csv文件。但是这种方法只能读取到指定的一个Address.csv文件,怎么办?
--------把Address.csv做成一个变量。接受一个文件名,通过文件名找文件。
完整代码:
#coding=utf-8
import csv
import os
def read(filename):
base_path=os.path.dirname(__file__) # 获取当前文件目录
path=base_path.replace('py_file','data/'+filename)# 目录替换
with open(path) as file:
data=csv.reader(file)
list=[]
i=0
for row in data:
if i==0:
pass
else:
list.append(row)
i=i+1
return list
if __name__ == '__main__':# main以下的代码只会在当前文件中执行,不可被其他的方法调用
list=read("Address.csv")
print(list)