def reader_once(self, string, filetype):
"""Helper function for only read once testcases"""
file_object = StringIO(string.strip())
reader = CSVReader(file_object, filetype)
instance = reader.next_entity()
return instance
def test_next_entity_end(self):
"""next_entity should return None when ending"""
file_object = StringIO()
reader = CSVReader(file_object, 'Testing')
self.assertIsNone(reader.next_entity())