def test_remove_noise(self): clusters = (Cluster(16, [16]), Cluster(4, [4, 5, 6, 7, 8]), Cluster(0, [0, 1, 2, 3]), Cluster(9, [9, 10, 11, 12, 13, 14, 15])) clustering = Clustering(clusters) clustering.eliminate_noise(5) self.assertEqual(len(clustering.clusters), 2)
def test_remove_noise(self): clusters =( Cluster(16,[16]), Cluster(4,[4,5,6,7,8]), Cluster(0,[0,1,2,3]), Cluster(9,[9,10,11,12,13,14,15]) ) clusterization = Clustering(clusters) clusterization.eliminate_noise(5) self.assertEqual(len(clusterization.clusters), 2)