def test_gen_class_list(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) class_list = clustering.gen_class_list() expected_class_list = [ 2, 2, 2, 2, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 3 ] self.assertItemsEqual(class_list, expected_class_list) clusters = (Cluster(0, [0, 1, 2, 3]), Cluster(9, [9, 10, 11, 12, 13, 14, 15])) clustering = Clustering(clusters) class_list = clustering.gen_class_list() expected_class_list = [ 1, 1, 1, 1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0, 0, 0 ] self.assertItemsEqual(class_list, expected_class_list)
def test_gen_class_list(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) class_list = clusterization.gen_class_list() expected_class_list = [2, 2, 2, 2, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 3] self.assertItemsEqual(class_list, expected_class_list) clusters =( Cluster(0,[0,1,2,3]), Cluster(9,[9,10,11,12,13,14,15]) ) clusterization = Clustering(clusters) class_list = clusterization.gen_class_list() expected_class_list = [1, 1, 1, 1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0, 0, 0] self.assertItemsEqual(class_list, expected_class_list)