Ejemplo n.º 1
0
	def test_indegree_matrix(self):
		vertex4 = Vertex()
		edge3 = Edge(self.vertex, vertex4)
		vertices = [self.vertex, self.vertex2, self.vertex3, vertex4]
		edges = [self.edge, self.edge2, edge3]
		graph = Graph(vertices, edges, True, Graph.ADJACENCY_MATRIX)
		self.assertEqual(graph.indegree(self.vertex), 0)
		self.assertEqual(graph.indegree(self.vertex2), 1)
		self.assertEqual(graph.indegree(self.vertex3), 1)
		self.assertEqual(graph.indegree(vertex4), 1)
		self.assertEqual(graph.indegree(Vertex()), 0)
Ejemplo n.º 2
0
	def test_indegree_list(self):
		vertex4 = Vertex()
		edge3 = Edge(self.vertex, vertex4)
		vertices = [self.vertex, self.vertex2, self.vertex3, vertex4]
		edges = [self.edge, self.edge2, edge3]
		graph = Graph(vertices, edges)
		self.assertEqual(graph.indegree(self.vertex), 0)
		self.assertEqual(graph.indegree(self.vertex2), 1)
		self.assertEqual(graph.indegree(self.vertex3), 1)
		self.assertEqual(graph.indegree(vertex4), 1)
		self.assertEqual(graph.indegree(Vertex()), 0)