def test_allow_missing_items(self):
result_zero_given = {'h1': ['1', '3', '5'], 'h2': ['2', '4', '0']}
result_nothing_given = {'h1': ['1', '3', '5'], 'h2': ['2', '4', None]}
self.assertEqual(
csv_columns(StringIO('h1,h2\r\n1,2\r\n3,4\r\n5\r\n'), missing='0'),
result_zero_given
)
self.assertEqual(
csv_columns(StringIO('h1,h2\r\n1,2\r\n3,4\r\n5\r\n')),
result_nothing_given
)
def test_key_word_headers(self):
result_with_given_headers = {
'header1': ['1', '3'],
'header2': ['2', '4'],
}
result_with_default = {'1': ['3'], '2': ['4']}
csv_file = StringIO('1,2\r\n3,4\r\n')
self.assertEqual(
csv_columns(csv_file, headers=['header1', 'header2']),
result_with_given_headers
)
csv_file.seek(0)
self.assertEqual(csv_columns(csv_file), result_with_default)
def test_ordered_dictionary(self):
csv_data = dedent("""
Car 1,Car 2,Car 3,Car 4,Car 5
2012,2009,1965,2005,1995
Lexus,GMC,Ford,Hyundai,Mercedes-Benz
LFA,Yukon XL 1500,Mustang,Sonata,C-Class
""").lstrip()
result = [
('Car 1', ['2012', 'Lexus', 'LFA']),
('Car 2', ['2009', 'GMC', 'Yukon XL 1500']),
('Car 3', ['1965', 'Ford', 'Mustang']),
('Car 4', ['2005', 'Hyundai', 'Sonata']),
('Car 5', ['1995', 'Mercedes-Benz', 'C-Class']),
]
self.assertEqual(list(csv_columns(StringIO(csv_data)).items()), result)
def test_three_columns(self):
csv_data = dedent("""
Year,Make,Model
2012,Lexus,LFA
2009,GMC,Yukon XL 1500
1965,Ford,Mustang
2005,Hyundai,Sonata
1995,Mercedes-Benz,C-Class
""").lstrip()
result = {
'Year': ['2012', '2009', '1965', '2005', '1995'],
'Make': ['Lexus', 'GMC', 'Ford', 'Hyundai', 'Mercedes-Benz'],
'Model': ['LFA', 'Yukon XL 1500', 'Mustang', 'Sonata', 'C-Class'],
}
self.assertEqual(csv_columns(StringIO(csv_data)), result)
def test_file_is_left_open(self):
"""Make sure csv_columns doesn't close the file given to it."""
csv_file = StringIO('h1\r\n1\r\n2\r\n')
csv_columns(csv_file) # Read it once
csv_file.seek(0) # Seek to the beginning of the still-open file
self.assertEqual(csv_columns(csv_file), {'h1': ['1', '2']})
def test_commas_in_data(self):
result = {'header, 1': ['1', '3,4'], 'h2': ['2', '4']}
self.assertEqual(
csv_columns(StringIO('"header, 1",h2\r\n1,2\r\n"3,4",4\r\n')),
result
)
def test_two_columns(self):
result = {'h1': ['1', '3'], 'h2': ['2', '4']}
self.assertEqual(
csv_columns(StringIO('h1,h2\r\n1,2\r\n3,4\r\n')),
result
)
def test_single_column(self):
result = {'h1': ['1', '2']}
self.assertEqual(
csv_columns(StringIO('h1\r\n1\r\n2\r\n')),
result
)