#569: [pyclustering.cluster.birch] unit-tests are added.

annoviko · Nov 28, 2019 · 5bd10be · 5bd10be
1 parent 4b06385
commit 5bd10be
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pyclustering/cluster/tests/unit/ut_birch.py b/pyclustering/cluster/tests/unit/ut_birch.py
@@ -50,7 +50,14 @@ def templateClusterAllocation(self, path, cluster_sizes, number_clusters, branch
         birch_instance.process()
 
         clusters = birch_instance.get_clusters()
+        cf_clusters = birch_instance.get_cf_cluster()
+        cf_entries = birch_instance.get_cf_entries()
+
         self.assertEqual(birch_instance.get_cluster_encoding(), type_encoding.CLUSTER_INDEX_LIST_SEPARATION)
+        self.assertEqual(number_clusters, len(clusters))
+        self.assertEqual(number_clusters, len(cf_clusters))
+        self.assertGreater(len(cf_entries), 0)
+        self.assertLessEqual(len(cf_entries), entry_size_limit)
 
         obtained_cluster_sizes = [len(cluster) for cluster in clusters]