check_test.py 文件源码

def test_unusual_errors():
    dataset = load_breast_cancer()
    X, y = dataset['data'], dataset['target_names'].take(dataset['target'])

    # Classifier to be used in the metaheuristic
    clf = SVC()

    for metaclass in METACLASSES:
        meta = metaclass(classifier=clf, random_state=0, verbose=0,
                        make_logbook=True, repeat=1, number_gen=2, size_pop=2)
        print("Checking unusual error: ", meta._name)
        meta.fit(X, y, normalize=True)

        # Let's suppose you have a empty array 
        meta.best_mask_ = np.array([])
        assert_warns(UserWarning, meta.transform, X)
        assert_raises(ValueError, meta.safe_mask, X, meta.best_mask_)

    meta = metaclass(classifier=clf, random_state=0, verbose=0,
                        make_logbook=True, repeat=1, number_gen=2, size_pop=2)

    assert_raises(ValueError, meta.score_func_to_gridsearch, meta)

    for metaclass in [BRKGA, BRKGA2]:
            assert_raises(ValueError, metaclass,classifier=clf, random_state=0, verbose=0,
                        make_logbook=True, repeat=1, number_gen=2, size_pop=2,
                        elite_size=5)