def test_add_classifier(self): ccol = ClassifierCollection() self.assertEqual(ccol.size(), 0) c = DummyClassifier() ccol.add_classifier('label', c) self.assertEqual(ccol.size(), 1) self.assertEqual(ccol._label_to_classifier['label'], c)
def test_size_len(self): ccol = ClassifierCollection() self.assertEqual(ccol.size(), 0) self.assertEqual(len(ccol), 0) ccol = ClassifierCollection( a=DummyClassifier(), b=DummyClassifier(), ) self.assertEqual(ccol.size(), 2) self.assertEqual(len(ccol), 2)