def _test_helper(self, *args, **kwargs):
with testfixtures.TempDirectory() as d:
for image, label in zip(self.test_images, self.test_labels):
d.write(os.path.join(label, image), b'any')
f = tempfile.NamedTemporaryFile(mode='r+t', suffix='.csv')
dataset.gen_csv_from_images(d.path, f.name, *args, **kwargs)
f.seek(0)
reader = csv.reader(f)
return reader
def test_input_dir_does_not_exist(self):
test_dir = '/path/to/non/existent/dir'
test_csv = 'test.csv'
dataset.gen_csv_from_images(test_dir, test_csv)
self.assertFalse(os.path.isfile(test_csv))