def test_multi_fw_complex(self):
dest1 = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'inputs.txt')
dest2 = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'temp_file.txt')
self._teardown([dest1, dest2])
try:
# create the Firework consisting of multiple tasks
firetask1 = TemplateWriterTask({'context': {'opt1': 5.0, 'opt2': 'fast method'}, 'template_file': 'simple_template.txt', 'output_file': dest1})
firetask2 = FileTransferTask({'files': [{'src': dest1, 'dest': dest2}], 'mode': 'copy'})
fw = Firework([firetask1, firetask2])
# store workflow and launch it locally, single shot
self.lp.add_wf(fw)
launch_rocket(self.lp, self.fworker)
# read inputs.txt, words.txt, dest
for d in [dest1, dest2]:
with open(d) as f:
self.assertEqual(f.read(), 'option1 = 5.0\noption2 = fast method')
finally:
self._teardown([dest1, dest2])
def test_multi_fw_complex(self):
dest1 = os.path.join(os.path.dirname(os.path.abspath(__file__)),
"inputs.txt")
dest2 = os.path.join(os.path.dirname(os.path.abspath(__file__)),
"temp_file.txt")
self._teardown([dest1, dest2])
try:
# create the Firework consisting of multiple tasks
firetask1 = TemplateWriterTask({
"context": {
"opt1": 5.0,
"opt2": "fast method"
},
"template_file": "simple_template.txt",
"output_file": dest1,
})
firetask2 = FileTransferTask({
"files": [{
"src": dest1,
"dest": dest2
}],
"mode":
"copy"
})
fw = Firework([firetask1, firetask2])
# store workflow and launch it locally, single shot
self.lp.add_wf(fw)
launch_rocket(self.lp, self.fworker)
# read inputs.txt, words.txt, dest
for d in [dest1, dest2]:
with open(d) as f:
self.assertEqual(f.read(),
"option1 = 5.0\noption2 = fast method")
finally:
self._teardown([dest1, dest2])