def test_delegate_list_node_default_option(): tool_1 = Tool(tool_1_def) tool_2 = Tool(tool_1_def) tool_3 = Tool(tool_1_def) p = Pipeline() node_1 = p.add(tool_1) node_2 = p.add(tool_2) node_3 = p.add(tool_3) node_3.input = [node_1, node_2] assert len(p._edges) == 2 edge = p.get_edge(node_1, node_3) assert edge is not None assert len(edge._links) == 1 edge_2 = p.get_edge(node_2, node_3) assert edge_2 is not None assert len(edge_2._links) == 1
def test_delegate_singleton_node_default_option(): tool_1 = Tool(tool_1_def) tool_2 = Tool(tool_1_def) p = Pipeline() node_1 = p.add(tool_1) node_2 = p.add(tool_2) node_2.input = node_1 assert len(p._nodes) == 2 assert len(p._edges) == 1 edge = p.get_edge(node_1, node_2) assert edge is not None assert len(edge._links) == 1