def test_classify_instance_all(competences, expected): query = np.array([-1, 1]) pool_classifiers = create_pool_classifiers() dcs_test = DCS(pool_classifiers, selection_method='all') dcs_test.estimate_competence = MagicMock(return_value=competences) predicted_label = dcs_test.classify_instance(query) assert predicted_label == expected
def test_classify_instance(): query = np.array([-1, 1]) pool_classifiers = create_pool_classifiers() dcs_test = DCS(pool_classifiers) competences = np.random.rand(dcs_test.n_classifiers) dcs_test.estimate_competence = MagicMock(return_value=competences) expected = pool_classifiers[np.argmax(competences)].predict(query)[0] predicted_label = dcs_test.classify_instance(query) assert predicted_label == expected