def test_select_best_features(rfe_raw_results):
pp = PostProcessor(1)
avg_scores = pp._compute_score_curve(rfe_raw_results)
selected_feats = pp._select_best_outer_features(rfe_raw_results,
avg_scores)
assert sorted(selected_feats["min"]) == [2, 4]
assert sorted(selected_feats["mid"]) == [2, 3, 4]
assert sorted(selected_feats["max"]) == [1, 2, 3, 4]
def test_compute_score_curve(rfe_raw_results):
pp = PostProcessor(0.05)
avg_scores = pp._compute_score_curve(rfe_raw_results)
assert len(avg_scores) == 3
assert 2 in avg_scores
assert 3 in avg_scores
assert 4 in avg_scores
assert avg_scores[4] < avg_scores[3]