def test_edge_error(): """delete non-existant edge""" g = SimpleGraph() g.add_node('a') g.add_node('b') g.add_node('c') g.add_edge('a', 'c') g.add_edge('a', 'b') with pytest.raises(KeyError): g.del_edge('a', 'z')
def test_del_edge(): """other edges preserved when one deleted""" g = SimpleGraph() g.add_node('a') g.add_node('b') g.add_node('c') g.add_edge('a', 'c') g.add_edge('a', 'b') g.del_edge('a', 'c') assert g.neighbors('a') == {'b': 0}
def test_weighted_edges_with_edge_delete(): g = SimpleGraph() g.add_edge('a', 'b', 5) g.add_edge('a', 'c', 2) g.del_edge('a', 'c') assert g.dict_graph['a'] == {'b': 5}