def test_DFP_is_used():
ds_test = BaseDS(create_pool_classifiers(), DFP=True, safe_k=3)
ds_test.fit(X_dsel_ex1, y_dsel_ex1)
ds_test.DSEL_processed_ = dsel_processed_ex1
ds_test.DSEL_target_ = y_dsel_ex1
ds_test.DSEL_data_ = X_dsel_ex1
DFP_mask = ds_test._frienemy_pruning(neighbors_ex1[0, :])
assert np.array_equal(DFP_mask, np.atleast_2d([1, 1, 0]))
def test_IH_is_used():
expected = [0, 0, 1]
query = np.ones((3, 2))
ds_test = BaseDS(create_pool_classifiers(), with_IH=True, IH_rate=0.5)
ds_test.fit(X_dsel_ex1, y_dsel_ex1)
ds_test.DSEL_processed_ = dsel_processed_ex1
ds_test.DSEL_target_ = y_dsel_ex1
ds_test.DSEL_data_ = X_dsel_ex1
ds_test._get_region_competence = MagicMock(return_value=(distances_ex1, neighbors_ex1))
predicted = ds_test.predict(query)
assert np.array_equal(predicted, expected)
