def test_predict_proba_instance_all(competences, expected): query = np.array([-1, 1]) pool_classifiers = create_pool_classifiers() dcs_test = DCS(pool_classifiers, selection_method='all') dcs_test.n_classes = 2 dcs_test.estimate_competence = MagicMock(return_value=competences) predicted_proba = dcs_test.predict_proba_instance(query) assert np.isclose(predicted_proba, expected).all()
def test_predict_proba_instance(): query = np.array([-1, 1]) pool_classifiers = create_pool_classifiers() dcs_test = DCS(pool_classifiers) dcs_test.n_classes = 2 competences = np.random.rand(dcs_test.n_classifiers) dcs_test.estimate_competence = MagicMock(return_value=competences) expected = pool_classifiers[np.argmax(competences)].predict_proba(query) predicted_proba = dcs_test.predict_proba_instance(query) assert np.array_equal(predicted_proba, expected)