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"]))
