def test_gz_csvFile(self):
reader = FileReader(self.__GZIP_FILE, is_csv=True, is_gzip=True)
data = []
for d in reader:
data.append(d)
self.assertEqual(data, self.__DATA_CSV)
def test_gzip_file(self):
reader = FileReader(self.__GZIP_FILE, is_gzip=True)
data = []
for d in reader:
data.append(d.strip())
self.assertEqual(data, self.__DATA)
def test_simple_file(self):
reader = FileReader(self.__CLEAR_FILE)
data = []
for d in reader:
data.append(d.strip())
self.assertEqual(data, self.__DATA)
def test_csv_file(self):
reader = FileReader(self.__CLEAR_FILE, is_csv=True)
data = []
for d in reader:
data.append(d)
self.assertEqual(data, self.__DATA_CSV)
def get_reader(file, kwargs):
"""Get reader class for file"""
reader = FileReader(file, **kwargs)
return reader