def test_weighted_graph(): g = WeightedGraph() g.insert_edge("a", {(9, "b"), (6, "c"), (1, "d")}) assert g.get_neighbors("a") == {(9, "b"), (6, "c"), (1, "d")} assert g.get_neighbors("b") == {(9, "a")} assert g.get_neighbors("c") == {(6, "a")} assert g.get_neighbors("d") == {(1, "a")} assert g.get_weight("a", "b") == 9 assert g.get_weight("a", "c") == 6 assert g.get_weight("a", "d") == 1