Python Clustering.gen_class_list Examples

Example #1

File: TestClustering.py Project: ztypaker/pyProCT

    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)

Example #2

File: TestClustering.py Project: lowks/pyProCT

 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)