def test_transpose_matrix(self): vertices = [self.vertex, self.vertex2, self.vertex3] edges = [self.edge, self.edge2] graph = Graph(vertices, edges, True, Graph.ADJACENCY_MATRIX) graph.transpose() adjacencies = graph.adjacencies self.assertEqual(adjacencies, [[None, None, None], [self.vertex, None, None], [None, self.vertex2, None]])
def test_transpose(self): vertices = [self.vertex, self.vertex2, self.vertex3] edges = [self.edge, self.edge2] graph = Graph(vertices, edges, False) adjacencies = graph.adjacencies graph.transpose() adjacencies2 = graph.adjacencies self.assertEqual(adjacencies, adjacencies2) graph = Graph(vertices, edges) graph.transpose() adjacencies = graph.adjacencies self.assertEqual(adjacencies, [[], [self.vertex], [self.vertex2]])