def test_filter(self): myFilter = ClusteringFilter({ "maximum_noise": 15, "maximum_clusters": 30, "minimum_clusters": 5, "minimum_cluster_size": 50 }, MatrixHandlerMock(1000)) clustering_info ={"clustering 1":{ "clustering":ClusteringMock(number_of_clusters = 50, number_of_elements = 800) }, "clustering 2":{ "clustering":ClusteringMock(number_of_clusters = 25, number_of_elements = 900) }, "clustering 3":{ "clustering":ClusteringMock(number_of_clusters = 25, number_of_elements = 900) }, "clustering 4":{ "clustering":ClusteringMock(number_of_clusters = 31, number_of_elements = 900) } } selected, not_selected = myFilter.filter(clustering_info) self.assert_(len(selected) == 1 and len(not_selected) == 3) self.assertItemsEqual(selected.keys() , [ "clustering 3"]) self.assertItemsEqual(not_selected.keys() , ["clustering 1","clustering 2","clustering 4",])