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