def test_node_collection(): graph = Graph() a = graph.add_node('a') b = graph.add_node('b') c = graph.add_node('c') actual = graph.get_nodes() assert actual == [a, b, c]
def test_empty_graph(): graph = Graph() actual = graph.get_nodes() assert actual == None