def test_weights_zero(): query = np.atleast_2d([1, 1]) knop_test = KNOP(create_pool_classifiers()) knop_test.fit(X_dsel_ex1, y_dsel_ex1) knop_test.estimate_competence = MagicMock(return_value=np.zeros(3)) result = knop_test.select(query) assert np.array_equal(result, np.array([0, 1, 0]))
def test_estimate_competence(index, expected): query = np.atleast_2d([1, 1]) knop_test = KNOP(create_pool_classifiers()) knop_test.fit(X_dsel_ex1, y_dsel_ex1) knop_test.DFP_mask = np.ones(knop_test.n_classifiers) knop_test.neighbors = neighbors_ex1[index, :] knop_test.distances = distances_ex1[index, :] competences = knop_test.estimate_competence(query) assert np.isclose(competences, expected, atol=0.01).all()