Ejemplo n.º 1
0
	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)
Ejemplo n.º 2
0
	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)