def test_get_vertices():

    graph = Graph()

    test1 = graph.add_vertex('test1')

    test2 = graph.add_vertex('test2')

    assert len(graph.get_vertices()) == 2
Esempio n. 2
0
def test_empty_graph():
    graph = Graph()
    assert graph.get_vertices() == None
    assert graph.get_neighbors("key") == None
    assert len(graph) == 0