def test_create_graph(self): q = JobQueue(1) g = CommandGraph(q) g.add("n1", "n2") g.add("n1", "n4") g.add("n2", "n3") assert (set(g._graph.keys()) == set(["n1", "n2"])) g.remove("n1") assert (set(g._graph.keys()) == set(["n2"])) g.add("n1", "n4") g.remove("n4") assert (g._graph['n1'] == set([])) assert (g.get_adjacent('n1') == set([])) assert (g.get_adjacent('n2') == set(["n3"]))
def test_create_graph(self): q = JobQueue(1) g = CommandGraph(q) g.add("n1", "n2") g.add("n1", "n4") g.add("n2", "n3") assert (set(g._graph.keys())== set(["n1", "n2"])) g.remove("n1") assert (set(g._graph.keys()) == set(["n2"])) g.add("n1", "n4") g.remove("n4") assert (g._graph['n1'] == set([])) assert (g.get_adjacent('n1') == set([])) assert (g.get_adjacent('n2') == set(["n3"]))