def test_predict_proba_instance_called(index):
query = np.atleast_2d([1, 1])
ds_test = DS(create_pool_classifiers(), with_IH=True, IH_rate=0.10)
ds_test.fit(X_dsel_ex1, y_dsel_ex1)
ds_test.neighbors = neighbors_ex1[index, :]
ds_test.distances = distances_ex1[index, :]
ds_test.predict_proba_instance = MagicMock(return_value=np.atleast_2d([0.25, 0.75]))
proba = ds_test.predict_proba(query)
assert np.isclose(proba, np.atleast_2d([0.25, 0.75])).all()
def test_predict_proba_DFP():
query = np.atleast_2d([1, 1])
ds_test = DS(create_pool_classifiers(), DFP=True, safe_k=3)
ds_test.fit(X_dsel_ex1, y_dsel_ex1)
# change the state of the system
ds_test.processed_dsel = dsel_processed_ex1
ds_test.DSEL_target = y_dsel_ex1
ds_test.DSEL_data = X_dsel_ex1
ds_test.neighbors = neighbors_ex1[0, :]
ds_test.distances = distances_ex1[0, :]
ds_test.predict_proba_instance = MagicMock(return_value=np.atleast_2d([0.25, 0.75]))
ds_test.predict_proba(query)
assert np.array_equal(ds_test.DFP_mask, np.array([1, 1, 0]))
