def test_desknn_proba():
pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers()
desknn = DESKNN(pool_classifiers, DFP=True)
desknn.fit(X_dsel, y_dsel)
probas = desknn.predict_proba(X_test)
expected = np.load('deslib/tests/expected_values/desknn_probas_DFP.npy')
assert np.allclose(probas, expected)
def test_desknn_proba(knn_methods):
pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers()
desknn = DESKNN(pool_classifiers, knn_classifier=knn_methods)
desknn.fit(X_dsel, y_dsel)
probas = desknn.predict_proba(X_test)
expected = np.load(
'deslib/tests/expected_values/desknn_proba_integration.npy')
assert np.allclose(probas, expected)
