def test_graph_add_vertex_fail():
grappy = Graph()
with pytest.raises(TypeError) as excinfo:
grappy.add_vertex()
actual = str(excinfo.value)
expected = "add_vertex() missing 1 required positional argument: 'value'"
assert actual == expected
def test_graph_get_nodes():
grappy = Graph()
vert_1 = grappy.add_vertex('A')
vert_2 = grappy.add_vertex('B')
actual = str(grappy.get_nodes())
expected = f'dict_keys({[vert_1, vert_2]})'
assert actual == expected
def test_graph_get_neighbors():
grappy = Graph()
vert_1 = grappy.add_vertex('A')
vert_2 = grappy.add_vertex('B')
grappy.add_edge(vert_1, vert_2)
actual = grappy.get_neighbors(vert_1)
expected = [(vert_2, 1)]
assert actual == expected
def my_graph():
g = Graph()
g.add_vertex('10')
g.add_vertex('5')
g.add_vertex('9')
g.add_vertex('44')
g.add_edge('10', '5', 3)
g.add_edge('10', '10', 6)
g.add_edge('5', '44', 2)
g.add_edge('44', '9', 4)
g.add_edge('44', '10', 1)
g.add_edge('5', '10', 7)
g.add_edge('9', '10', 8)
return g
def test_one_vertex():
g = Graph()
g.add_vertex('20')
actual = g.get_vertices()
expected = ['20']
assert actual == expected
def test_graph_add_edge_pass():
grappy = Graph()
vert_1 = grappy.add_vertex('A')
vert_2 = grappy.add_vertex('B')
grappy.add_edge(vert_1, vert_2)
assert True
def test_graph_add_vertex_pass():
grappy = Graph()
verty = grappy.add_vertex('A')
actual = verty.value
expected = 'A'
assert actual == expected
