class TestGraph(unittest.TestCase):
def setUp(self):
task1 = FooTask()
task2 = FooTask()
task3 = FooTask()
self.workflow = EOWorkflow(dependencies=[
Dependency(task=task1, inputs=[]),
Dependency(task=task2, inputs=[]),
Dependency(task=task3, inputs=[task1, task2])
])
def test_graph_nodes_and_edges(self):
dot = self.workflow.get_dot()
self.assertTrue(isinstance(dot, Digraph))
digraph = self.workflow.dependency_graph()
self.assertTrue(isinstance(digraph, Digraph))
class TestGraph(unittest.TestCase):
def setUp(self):
input_task1 = InputTask()
input_task2 = InputTask()
divide_task = DivideTask()
self.workflow = EOWorkflow(dependencies=[
Dependency(task=input_task1, inputs=[]),
Dependency(task=input_task2, inputs=[]),
Dependency(task=divide_task, inputs=[input_task1, input_task2])
])
def test_graph_nodes_and_edges(self):
dot = self.workflow.get_dot()
dot_file = StringIO()
dot_file.write(dot.source)
dot_file.seek(0)
digraph = self.workflow.dependency_graph()
class TestGraph(unittest.TestCase):
def setUp(self):
input_task1 = InputTask()
input_task2 = InputTask()
divide_task = DivideTask()
self.workflow = EOWorkflow(dependencies=[
Dependency(task=input_task1, inputs=[]),
Dependency(task=input_task2, inputs=[]),
Dependency(task=divide_task, inputs=[input_task1, input_task2])
])
def test_graph_nodes_and_edges(self):
dot = self.workflow.get_dot()
dot_file = StringIO()
dot_file.write(dot.source)
dot_file.seek(0)
graph = nx.drawing.nx_pydot.read_dot(dot_file)
self.assertEqual(graph.number_of_nodes(), 3)
self.assertEqual(graph.number_of_edges(), 2)
