Esempio n. 1
0
    def test_filter_repeated(self):
        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)
                                       }
                          }

        myFilter = ClusteringFilter({},MatrixHandlerMock(1000))
        sel, not_sel = myFilter.filter_repeated(clustering_info,{})
        self.assertItemsEqual(sel.keys(),["clustering 1","clustering 3"])
        self.assertItemsEqual(not_sel.keys(),["clustering 2"])
        self.assertDictEqual(not_sel["clustering 2"]["reasons"][0],
                             {'reason': 'EQUAL_TO_OTHER_CLUSTERING', 'data': {'id': 'clustering 3'}})