def test_get_edges(self):
""" Ensures edges are returned correctly. """
vertex_one = Vertex(0)
vertex_two = Vertex(1)
edge = Edge(0, vertex_one, vertex_two)
g = Graph()
g._edges = [edge]
assert g.get_edges() == [edge]
def test_get_edge(self):
""" Ensure the correct edge is returned with given id. """
vertex_one = Vertex(0)
vertex_two = Vertex(1)
edge_one = Edge(0, vertex_one, vertex_two)
edge_two = Edge(500, vertex_one, vertex_two)
g = Graph()
g._edges = [edge_one, edge_two]
assert g.get_edge(500) is edge_two