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",])
