def test_clear_operation_links_end(self, graph, sum_op, square_op,
negative_op):
graph.add_operations(sum_op, square_op, negative_op)
graph.add_link(sum_op, square_op, "sum", "n")
graph.add_link(square_op, negative_op, "square", "num")
graph.clear_operation_links(negative_op)
assert graph.links() == [(sum_op, square_op, "sum", "n")]
def test_clear_operation_links_unlinked_operation(self, graph, sum_op,
square_op):
# TODO should this raise an exception?
graph.add_operations(sum_op, square_op)
graph.clear_operation_links(sum_op)
assert graph.links() == []
def test_clear_operation_links_bad_arg(self, graph):
with pytest.raises(TypeError):
graph.clear_operation_links("this should be an operation")
def test_clear_operation_links_empty(self, graph, sum_op):
graph.clear_operation_links(sum_op)
assert graph.links() == []
assert graph.get_outbound_links(sum_op) == {}
assert graph.get_inbound_links(sum_op) == {}
