有没有一种方法可以模拟DictReader
进行单元测试,而不必编写文件,然后重新打开它。
我的函数接受一个DictReader
实例,因此我可以很容易地向它们传递一个来测试它们的功能,但是如果不打开一个文件,似乎就无法获得一个。
目前我正在手动编写一个CSV文件,然后删除每个测试。
class TestRowsStuff(unittest.TestCase):
def write_csv(self, path, iterable):
with open(path, 'wb') as f:
writer = csv.DictWriter(f, [PP, SN, TN])
writer.writeheader()
writer.writerows(iterable)
def setUp(self):
...
self.test_file = os.path.join('test.csv')
self.write_csv(self.test_file, test_values)
def tearDown(self):
os.remove(self.test_file)
最佳答案
您可以使用内存中的StringIO对象来存储/读取Unicode/字符串:
In [10]: from StringIO import StringIO
In [11]: import csv
In [12]: csvfile = StringIO()
In [13]: csvfile.seek(0)
# sample taken from [here](https://docs.python.org/2/library/csv.html)
In [14]: fieldnames = ['first_name', 'last_name']
In [15]: writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
In [16]: writer.writeheader()
In [17]: writer.writerow({'first_name': 'Baked', 'last_name': 'Beans'})
In [18]: writer.writerow({'first_name': 'Lovely', 'last_name': 'Spam'})
In [19]: writer.writerow({'first_name': 'Wonderful', 'last_name': 'Spam'})
In [20]: csvfile.seek(0)
要读回:
In [21]: csvfile.readlines()
Out[21]:
['first_name,last_name\r\n',
'Baked,Beans\r\n',
'Lovely,Spam\r\n',
'Wonderful,Spam\r\n']
如果要使用内存缓冲区,也可以使用io.StringIO。