def test_unweighted_remove_node_node(): graph = Graph() nodes = [["a","b"],["b","c"],["a","c"],["c","a"]] graph.make_unweighted_from_list(nodes) graph.remove_node(Node("a")) test_graph = {Node("b"):[Edge("b","c")], Node("c"):[]} assert compare_graphs(test_graph, graph.graph) == True
def test_weighted_remove_node(): graph = Graph() nodes = [["a","b",1],["b","b",99],["c","b",2],["d","b",3],["e","b",4],["f","z","a"]] graph.make_weighted_from_list(nodes) graph.remove_node(Node("b")) test_graph = { Node("a"):[], Node("c"):[], Node("d"):[], Node("e"):[], Node("f"):[Edge("f","z","a")], Node("z"): [] } assert compare_graphs(test_graph, graph.graph) == True