def test_header_row(self): with open('noquotes.csv') as csv_file: csv_reader = CsvReader(header_row=True, file=csv_file) rows = csv_reader.get_rows() self.assertEqual(rows, [{'a': 'c', 'b': 'd'}])
def test_reading_easy_csv_file(self): with open('noquotes.csv') as csv_file: csv_reader = CsvReader(header_row=False, file=csv_file) self.assertEqual(csv_reader.get_rows(), [['a', 'b'], ['c', 'd']])