def test_universal_sink(self): vertices = [self.vertex, self.vertex2, self.vertex3] edges = [self.edge, self.edge2] graph = Graph(vertices, edges, True) vertex = graph.find_universal_sink() self.assertIsNone(vertex) edges.append(Edge(self.vertex, self.vertex3)) graph = Graph(vertices, edges, True) vertex = graph.find_universal_sink() self.assertEquals(vertex, self.vertex3)
def test_universal_sink_matrix(self): vertices = [self.vertex, self.vertex2, self.vertex3] edges = [self.edge, self.edge2] graph = Graph(vertices, edges, True, Graph.ADJACENCY_MATRIX) vertex = graph.find_universal_sink() self.assertIsNone(vertex) edges.append(Edge(self.vertex, self.vertex3)) graph = Graph(vertices, edges, True, Graph.ADJACENCY_MATRIX) vertex = graph.find_universal_sink() self.assertEquals(vertex, self.vertex3) edges = [Edge(self.vertex2, self.vertex3)] graph = Graph(vertices, edges, True, Graph.ADJACENCY_MATRIX) vertex = graph.find_universal_sink() self.assertIsNone(vertex)