def test_csv(self): for (example, ExampleDataset) in examples.examples: dataset = ExampleDataset(Project(None)) csv_filename = write_tmp_csv_file(dataset.data(), example.sli_manifest) f = open(csv_filename, 'r') content = f.read() f.close() os.remove(csv_filename) self.assertEqual(csv_to_list(example.data_csv), csv_to_list(content))
def test_keep_csv(self): csv_file = os.path.join(os.path.abspath('./'), 'tmp.csv') for (example, ExampleDataset) in examples.examples: filename = create_archive( example.data_csv, example.sli_manifest, [], [], keep_csv=True, csv_file=csv_file ) with open(csv_file, 'r') as f: content = f.read() os.remove(csv_file) self.assertListEqual(csv_to_list(example.data_csv), csv_to_list(content)) self.assertRaises( TypeError, create_archive, example.data_csv, example.sli_manifest, [], [], keep_csv=True )
def test_csv_to_list(self): for (example, ExampleDataset) in examples.examples: data_list = csv_to_list(example.data_csv) self.assertEqual(ExampleDataset(Project(None)).data(), data_list)