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)
