def test_select_single_sample():
des_mi = DESMI(pct_accuracy=0.7)
des_mi.N_ = 2
competences = np.array([0.7, 0.2, 1.0])
selected_clf = des_mi.select(competences)
expected = np.array([0, 2])
assert np.array_equal(np.unique(selected_clf), np.unique(expected))
def test_select_batch_samples():
n_samples = 10
des_mi = DESMI(pct_accuracy=0.7)
des_mi.N_ = 2
competences = np.tile(np.array([0.7, 0.2, 1.0]), (n_samples, 1))
selected_clf = des_mi.select(competences)
expected = np.tile(np.array([0, 2]), (n_samples, 1))
assert np.array_equal(np.unique(selected_clf), np.unique(expected))