def test_setTasksNotValid(self):
mp = MacroPipeline()
pip = Pipeline()
with self.assertRaises(TypeError):
mp.tasks = 0
with self.assertRaises(TypeError):
mp.tasks = pip
with self.assertRaises(AssertionError):
mp.tasks = ['load']
def test_runInvalidPipeline(self):
a, b = ShortIOPipeline(), ShortIOPipeline()
file = os.path.join(self._test_dir, 'file_a.txt')
text = 'hello world'
a.input = {'open': file,
'write': [text],
'close': {}}
b.input = {'open': self._test_dir,
'write': [text],
'close': {}}
mp = MacroPipeline()
mp.tasks = [a, b]
mp.setup_cluster(cluster=self.cluster)
mp.run()
self.assertListEqual(mp.get_failed_pipelines(), [b])
self.assertIs(mp.errors[0], None)
self.assertTrue(mp.errors[1][0], IsADirectoryError)
mp.print_outcome(to_file=self._outcome_file_path)
self.assertTrue(os.path.isfile(self._outcome_file_path))
with open(self._outcome_file_path, 'r') as f:
res = [line.split()[-1] for line in f.readlines()]
self.assertEqual(res[0], 'finished')
self.assertNotEqual(res[1], 'finished')
def test_runValidPipelines(self):
a, b = ShortIOPipeline(), ShortIOPipeline()
file_a, file_b = [os.path.join(self._test_dir, 'file_{}.txt'.format(s))
for s in 'ab']
text = 'hello world'
a.input = {'open': file_a,
'write': [text],
'close': {}}
b.input = {'open': file_b,
'write': [text],
'close': {}}
mp = MacroPipeline()
mp.tasks = [a, b]
mp.setup_cluster(cluster=self.cluster)
mp.run()
self.assertTrue(all([os.path.isfile(f) for f in [file_a, file_b]]))
lines_a, lines_b = [open(f).readlines() for f in [file_a, file_b]]
self.assertEqual(lines_a, lines_b)
self.assertListEqual(mp.get_failed_pipelines(), [])
mp.print_outcome(to_file=self._outcome_file_path)
self.assertTrue(os.path.isfile(self._outcome_file_path))
with open(self._outcome_file_path, 'r') as f:
res = [line.split()[-1] for line in f.readlines()]
self.assertListEqual(res, ['finished']*2)
def test_setLabels(self):
mp = MacroPipeline()
mp.tasks = [Pipeline(), Pipeline()]
labels = ['a', 'b']
mp.set_labels(labels)
self.assertListEqual(labels, [task.label for task in mp.tasks])