def test_check_num_clusters_in_range(self):
myFilter = ClusteringFilter({
"minimum_clusters": 5,
"maximum_clusters": 30,
},
MatrixHandlerMock(1000))
self.assertItemsEqual(myFilter.check_num_clusters_in_range(ClusteringMock(number_of_clusters = 10, number_of_elements = 1000)), [])
self.assertItemsEqual( myFilter.check_num_clusters_in_range(ClusteringMock(number_of_clusters = 2, number_of_elements = 1000)),
[
{
'reason': 'TOO_FEW_CLUSTERS',
'data': {'current': 2,
'minimum': 5}
}
]
)
self.assertItemsEqual( myFilter.check_num_clusters_in_range(ClusteringMock(number_of_clusters = 35, number_of_elements = 1000)),
[
{
'reason': 'TOO_MUCH_CLUSTERS',
'data': {
'current': 35,
'maximum': 30
}
}
])
