def test_can_remove_edge(): graph = Graph() graph.addNode("start") graph.addNode("end") graph.addEdge("start", "end") output = graph.removeEdge("start", "end") assert output == True
def test_can_find_edge(): graph = Graph() graph.addNode("start") graph.addNode("end") graph.addEdge("start", "end") output = graph.findEdge("start", "end") assert output == True