def testRestrictRankedList(self): ranker = GraphRanker() list1 = [4, 2, 1,7,3, 8, 9, 12] list2 = [8, 9, 2] outputList = ranker.restrictRankedList(list1, list2) self.assertEquals(outputList, [2, 8, 9]) outputList = ranker.restrictRankedList(list1, [3, 1, 12]) self.assertEquals(outputList, [1, 3, 12])
def testRankedList(self): graph = igraph.Graph(directed=True) graph.add_vertices(5) graph.add_edges([(0,1), (0,2), (1, 3), (2,3), (2,4)]) graph.es["invWeight"] = [1, 1, 1, 1, 1] graph.es["weight"] = [1, 1, 1, 1, 1] ranker = GraphRanker() outputList = ranker.vertexRankings(graph, [0, 1, 2]) for lst in outputList: self.assertEquals(set(lst), set([0, 1,2])) self.assertEquals(len(lst), 3)