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)