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
