def test_bouquet():
g = Graph()
apple = g.add_node("apple")
g.add_edge(apple, apple, 10)
neighbors = g.get_neighbors(apple)
assert len(neighbors) == 1
assert neighbors[0].vertex.value == "apple"
assert neighbors[0].weight == 10
def test_add_edge():
g = Graph()
apple = g.add_node("apple")
banana = g.add_node("banana")
g.add_edge(apple, banana, 5)
neighbors = g.get_neighbors(apple)
assert len(neighbors) == 1
assert neighbors[0].vertex.value == "banana"
assert neighbors[0].weight == 5
def test_get_neighbors():
graph = Graph()
banana = graph.add_node("banana")
apple = graph.add_node("apple")
graph.add_edge(apple, banana, 44)
neighbors = graph.get_neighbors(apple)
assert len(neighbors) == 1
neighbor_edge = neighbors[0]
assert neighbor_edge.vertex.value == "banana"
assert neighbor_edge.weight == 44
