def test_dagflow_cut(self):
""" disconnected graph """
wfl = Workflow([self.fw1, self.fw2, self.fw3], {self.fw1: self.fw2})
msg = 'The workflow graph must be connected'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_cut(self):
""" disconnected graph """
wfl = Workflow([self.fw1, self.fw2, self.fw3], {self.fw1: self.fw2})
msg = 'The workflow graph must be connected'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_link(self):
""" wrong links """
wfl = Workflow([self.fw1, self.fw2, self.fw3],
{self.fw1: [self.fw2, self.fw3]})
msg = 'An input field must have exactly one source'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_cut(self):
"""disconnected graph"""
from fireworks.utilities.dagflow import DAGFlow
wfl = Workflow([self.fw1, self.fw2, self.fw3], {self.fw1: self.fw2})
msg = "The workflow graph must be connected"
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl).check()
self.assertTrue(msg in str(context.exception))
def test_dagflow_ok(self):
""" construct and replicate """
wfl = Workflow([self.fw1, self.fw2, self.fw3], {
self.fw1: [self.fw2],
self.fw2: [self.fw3],
self.fw3: []
})
dagf = DAGFlow.from_fireworks(wfl)
DAGFlow(**dagf.to_dict())
def test_dagflow_link(self):
""" wrong links """
from fireworks.utilities.dagflow import DAGFlow
wfl = Workflow([self.fw1, self.fw2, self.fw3],
{self.fw1: [self.fw2, self.fw3]})
msg = 'Every input in inputs list must have exactly one source.'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl).check()
self.assertTrue(msg in str(context.exception))
def test_dagflow_link(self):
""" wrong links """
wfl = Workflow(
[self.fw1, self.fw2, self.fw3],
{self.fw1: [self.fw2, self.fw3]}
)
msg = 'An input field must have exactly one source'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_loop(self):
""" loop in graph """
wfl = Workflow(
[self.fw1, self.fw2, self.fw3],
{self.fw1: self.fw2, self.fw2: self.fw3, self.fw3: self.fw1}
)
msg = 'The workflow graph must be a DAG.: found cycles:'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_loop(self):
""" loop in graph """
wfl = Workflow([self.fw1, self.fw2, self.fw3], {
self.fw1: self.fw2,
self.fw2: self.fw3,
self.fw3: self.fw1
})
msg = 'The workflow graph must be a DAG.: found cycles:'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_input(self):
""" missing input """
fw2 = Firework(PyTask(func='math.pow',
inputs=['first power', 'exponent'],
outputs=['second power']),
name='pow(pow(2, 3), 4)')
wfl = Workflow([self.fw1, fw2], {self.fw1: [fw2], fw2: []})
msg = (r"An input field must have exactly one source', 'step', "
r"'pow(pow(2, 3), 4)', 'entity', 'exponent', 'sources', []")
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_loop(self):
"""loop in graph"""
from fireworks.utilities.dagflow import DAGFlow
wfl = Workflow([self.fw1, self.fw2, self.fw3], {
self.fw1: self.fw2,
self.fw2: self.fw3,
self.fw3: self.fw1
})
msg = "The workflow graph must be a DAG.: found cycles:"
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl).check()
self.assertTrue(msg in str(context.exception))
def add_sf(args):
sp = get_sp(args)
if args.dir:
files = []
for f in args.sf_file:
files.extend([os.path.join(f, i) for i in os.listdir(f)])
else:
files = args.sf_file
for f in files:
fwf = SwarmFlow.from_file(f)
if args.check:
from fireworks.utilities.dagflow import DAGFlow
DAGFlow.from_fireworks(fwf)
sp.add_sf(fwf)
def test_dagflow_input(self):
""" missing input """
fw2 = Firework(
PyTask(
func='math.pow',
inputs=['first power', 'exponent'],
outputs=['second power']
),
name='pow(pow(2, 3), 4)'
)
wfl = Workflow([self.fw1, fw2], {self.fw1: [fw2], fw2: []})
msg = (r"An input field must have exactly one source', 'step', "
r"'pow(pow(2, 3), 4)', 'entity', 'exponent', 'sources', []")
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_missing_input(self):
"""missing input"""
from fireworks.utilities.dagflow import DAGFlow
fw2 = Firework(
PyTask(func="math.pow",
inputs=["first power", "exponent"],
outputs=["second power"]),
name="pow(pow(2, 3), 4)",
)
wfl = Workflow([self.fw1, fw2], {self.fw1: [fw2], fw2: []})
msg = (
r"Every input in inputs list must have exactly one source.', 'step', "
r"'pow(pow(2, 3), 4)', 'entity', 'exponent', 'sources', []")
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl).check()
self.assertTrue(msg in str(context.exception))
def test_dagflow_ok(self):
""" construct and replicate """
wfl = Workflow(
[self.fw1, self.fw2, self.fw3],
{self.fw1: [self.fw2], self.fw2: [self.fw3], self.fw3: []}
)
dagf = DAGFlow.from_fireworks(wfl)
DAGFlow(**dagf.to_dict())
def test_dagflow_race_condition(self):
""" two parent firework outputs overwrite each other """
from fireworks.utilities.dagflow import DAGFlow
task = PyTask(func='math.pow',
inputs=['base', 'exponent'],
outputs=['second power'])
fw1 = Firework(task, name='pow(2, 3)', spec={'base': 2, 'exponent': 3})
fw2 = Firework(task, name='pow(2, 3)', spec={'base': 2, 'exponent': 3})
wfl = Workflow([fw1, fw2, self.fw3], {
fw1: [self.fw3],
fw2: [self.fw3]
})
msg = (
r"'Every input in inputs list must have exactly one source.', 'step', "
r"'the third one', 'entity', 'second power', 'sources', [0, 1]")
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl).check()
self.assertTrue(msg in str(context.exception))
def test_dagflow_output(self):
""" clashing inputs """
from fireworks.utilities.dagflow import DAGFlow
fw2 = Firework(
PyTask(
func='math.pow',
inputs=['first power', 'exponent'],
outputs=['second power']
),
name='pow(pow(2, 3), 4)',
spec={'exponent': 4, 'first power': 8}
)
wfl = Workflow([self.fw1, fw2], {self.fw1: [fw2], fw2: []})
msg = (r"'An input field must have exactly one source', 'step', "
r"'pow(pow(2, 3), 4)', 'entity', 'first power', 'sources'")
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(wfl)
self.assertTrue(msg in str(context.exception))
def test_dagflow_non_dataflow_tasks(self):
""" non-dataflow tasks using outputs and inputs keys do not fail """
from fireworks.utilities.dagflow import DAGFlow
from fireworks.core.firework import FireTaskBase
class NonDataFlowTask(FireTaskBase):
""" a firetask class for testing """
_fw_name = 'NonDataFlowTask'
required_params = ['inputs', 'outputs']
def run_task(self, fw_spec):
pass
task = NonDataFlowTask(inputs=['first power', 'exponent'],
outputs=['second power'])
fw2 = Firework(task, spec={'exponent': 4, 'first power': 8})
wfl = Workflow([self.fw1, fw2], {self.fw1: [fw2], fw2: []})
DAGFlow.from_fireworks(wfl).check()
def test_dagflow_non_dataflow_tasks(self):
"""non-dataflow tasks using outputs and inputs keys do not fail"""
from fireworks.core.firework import FireTaskBase
from fireworks.utilities.dagflow import DAGFlow
class NonDataFlowTask(FireTaskBase):
"""a firetask class for testing"""
_fw_name = "NonDataFlowTask"
required_params = ["inputs", "outputs"]
def run_task(self, fw_spec):
pass
task = NonDataFlowTask(inputs=["first power", "exponent"],
outputs=["second power"])
fw2 = Firework(task, spec={"exponent": 4, "first power": 8})
wfl = Workflow([self.fw1, fw2], {self.fw1: [fw2], fw2: []})
DAGFlow.from_fireworks(wfl).check()
def test_dagflow_clashing_outputs(self):
""" subsequent task overwrites output of a task """
from fireworks.utilities.dagflow import DAGFlow
tasks = [
PyTask(func='math.pow',
inputs=['first power 1', 'exponent'],
outputs=['second power']),
PyTask(func='math.pow',
inputs=['first power 2', 'exponent'],
outputs=['second power'])
]
fwk = Firework(tasks,
spec={
'exponent': 4,
'first power 1': 8,
'first power 2': 4
})
msg = 'Several tasks may not use the same name in outputs list.'
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(Workflow([fwk], {})).check()
self.assertTrue(msg in str(context.exception))
def test_dagflow_clashing_outputs(self):
"""subsequent task overwrites output of a task"""
from fireworks.utilities.dagflow import DAGFlow
tasks = [
PyTask(func="math.pow",
inputs=["first power 1", "exponent"],
outputs=["second power"]),
PyTask(func="math.pow",
inputs=["first power 2", "exponent"],
outputs=["second power"]),
]
fwk = Firework(tasks,
spec={
"exponent": 4,
"first power 1": 8,
"first power 2": 4
})
msg = "Several tasks may not use the same name in outputs list."
with self.assertRaises(AssertionError) as context:
DAGFlow.from_fireworks(Workflow([fwk], {})).check()
self.assertTrue(msg in str(context.exception))
def test_dagflow_view(self):
""" visualize the workflow graph """
wfl = Workflow(
[self.fw1, self.fw2, self.fw3],
{self.fw1: [self.fw2], self.fw2: [self.fw3], self.fw3: []}
)
dagf = DAGFlow.from_fireworks(wfl)
dagf.add_step_labels()
filename = str(uuid.uuid4())
dagf.to_dot(filename, view='combined')
self.assertTrue(os.path.exists(filename))
os.remove(filename)
filename = str(uuid.uuid4())
dagf.to_dot(filename, view='dataflow')
self.assertTrue(os.path.exists(filename))
os.remove(filename)
filename = str(uuid.uuid4())
dagf.to_dot(filename, view='controlflow')
self.assertTrue(os.path.exists(filename))
os.remove(filename)
def test_dagflow_view(self):
""" visualize the workflow graph """
wfl = Workflow([self.fw1, self.fw2, self.fw3], {
self.fw1: [self.fw2],
self.fw2: [self.fw3],
self.fw3: []
})
dagf = DAGFlow.from_fireworks(wfl)
dagf.add_step_labels()
filename = str(uuid.uuid4())
dagf.to_dot(filename, view='combined')
self.assertTrue(os.path.exists(filename))
os.remove(filename)
filename = str(uuid.uuid4())
dagf.to_dot(filename, view='dataflow')
self.assertTrue(os.path.exists(filename))
os.remove(filename)
filename = str(uuid.uuid4())
dagf.to_dot(filename, view='controlflow')
self.assertTrue(os.path.exists(filename))
os.remove(filename)