def test_no_path(self): g = Graph() a = g.add_node(Node()) b = g.add_node(Node()) # no edges between them path = g.get_shortest_path(a, b) self.assertEqual(path, []) # FIXME: shouldn't this be None?
def make_trivial_graph(): """ Construct a trivial graph: a ─> b """ g = Graph() a = g.add_node(NamedNode('a')) b = g.add_node(NamedNode('b')) ab = g.add_edge(a, b) return g, a, b, ab
def add_node(self, supernode): Graph.add_node(self, supernode) # Keep track of mapping from stmtnode -> supernode self.supernode_for_stmtnode[supernode.innernode] = supernode return supernode