def test_operation_links_multiple(self, sum_op, square_op, negative_op): # # (y) --> (n1) # my_func sum (n) --> (num) square # (x) --> (n2) # def my_func(x: int, y: int) -> (int, int): return y, x my_op = operation(my_func, output_names=("y", "x"))() workflow = Workflow() # workflow.add_operations(sum_op, square_op, negative_op, my_op) workflow.add_operations(my_op, sum_op, square_op, negative_op) link1 = (my_op, sum_op, "y", "n1") link2 = (my_op, sum_op, "x", "n2") link3 = (sum_op, square_op, "sum", "n") link4 = (square_op, negative_op, "square", "num") workflow.add_link(my_op, sum_op, "y", "n1") workflow.add_link(my_op, sum_op, "x", "n2") workflow.add_link(sum_op, square_op, "sum", "n") workflow.add_link(square_op, negative_op, "square", "num") # assert workflow.operation_links(my_op) == [link1, link2] # assert workflow.operation_links(sum_op) == [link3] # assert workflow.operation_links(square_op) == [link4] # assert workflow.operation_links(negative_op) == [] workflow._pretty_print() dask_graph, end_ids = workflow.as_dask_graph() # test execution results = workflow.execute_synchronous(x=3, y=5)
def test_two_copy_ops_to_one_op(self, double_and_triple_op, sum_op, square_op): # 1**2 + 2**2 => 5 workflow = Workflow() square_op.filled_values.update(n=1) square_op_2 = square_op.__class__() square_op_2.filled_values.update(n=2) square_op_2.output_names = ["square"] workflow.add_operations(sum_op, square_op, square_op_2) workflow.add_link(square_op, sum_op, "square", "n1") workflow.add_link(square_op_2, sum_op, "square", "n2") workflow._pretty_print() result = workflow.execute_synchronous() assert result == ({"sum": 5},)