python 读取txt, csv 文件

在做自动化测试时，需要准备一些数据，每次去执行时读取数据，如果写在代码中，不好看也不好维护，直接上代码：

def duqu_txt(self):
    '''读取文件方法,读取整个文件,返回一个数组.去掉回车'''
    with open('data_name.txt','r') as f:
        line = f.read()
        li = line.split('
')
        return li

def duqu_csv(self):
    '''读取csv文件'''
    with open('yiyuan_name.csv','r') as f:
        line = f.read()
        li = line.split()
        return li


 1、定义一个读取的函数
 2、通过with循环打开当前目录下的文件，以只读的方式打开
 3、定义一个列表line,使用读取全部文件方式赋值给line  
 4、定义一个变量li 使用split方法，赋值给li
 5、返回，li