def test_bcubed_calculation(self): ede = ClusteringEvaluator() documents_labels_clusters = [(0, 0), (0, 1), (0, 0), (0, 0), (0, 1), (0, 0), (1, 1), (1, 1), (1, 2), (1, 1), (1, 1), (1, 1), (2, 1), (2, 0), (2, 2), (2, 0), (2, 2), (2, 2)] precision ,recall, f = ede.calculate_bcubed_measures(documents_labels_clusters) self.assertAlmostEqual(0.532407407407 - precision, 0, places=7) self.assertAlmostEqual(0.555555555556 - recall, 0, places=7) self.assertAlmostEqual(0.543735224586 - f, 0, places=7)