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