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)