def test_remove_missing_edge_raises_directed():
# given
g = DirectedGraph()
g.add_edge(1, 2)
g.add_edge(2, 3)
g.add_edge(3, 1)
# when
with pytest.raises(DoesNotExistError):
g.remove_edge(2, 4)
def test_remove_edge_directed():
# given
g = DirectedGraph()
g.add_edge(1, 2)
g.add_edge(2, 1)
g.add_edge(2, 3)
g.add_edge(3, 1)
# when
g.remove_edge(1, 2)
# then
assert (1, 2) not in g.edges
assert (2, 1) in g.edges
assert len(g.edges) == 3
assert len(g.nodes) == 3