def test_labels(self): ccol = ClassifierCollection( classifiers={ 'b': DummyClassifier(), }, a=DummyClassifier(), label2=DummyClassifier(), ) self.assertEqual(ccol.labels(), {'a', 'b', 'label2'})
def test_labels_empty(self): ccol = ClassifierCollection() self.assertEqual(ccol.labels(), set())