def test_valid_csv_with_header_names_specified(self):
"""Tests valid CSV with header, and header names given (override)."""
f = _make_csv_tempfile([self.header] + self.sample_data)
actual = client.read_csv(f.name, names=self.header, header=0)
self.assertEqual(list(actual.columns), self.header)
self.assertEqual(actual.values.tolist(), self.sample_data)
def test_valid_csv_no_header_no_names_specified(self):
"""Tests a valid CSV without a header and no header names given."""
f = _make_csv_tempfile(self.sample_data)
actual = client.read_csv(f.name, header=None)
self.assertEqual(list(actual.columns),
list(schema.image_csv_schema.keys()))
self.assertEqual(actual.values.tolist(), self.sample_data)
def test_valid_csv_no_header_names_specified(self):
"""Tests valid CSV without a header, but header names are given."""
f = _make_csv_tempfile(self.sample_data)
actual = client.read_csv(f.name, header=None, names=self.header)
self.assertEqual(list(actual.columns), self.header)
self.assertEqual(actual.values.tolist(), self.sample_data)
def test_valid_csv_no_header_no_names_specified(self):
"""Tests a valid CSV without a header and no header names given."""
f = _make_csv_tempfile(self.sample_data)
actual = client.read_csv(f.name, header=None)
self.assertEqual(list(actual.columns), constants.IMAGE_CSV_COLUMNS)
self.assertEqual(actual.values.tolist(), self.sample_data)
