def test_invalid_input(): kbest = SelectKBest() with pytest.raises(TypeError): kbest.fit_transform({})
def test_fit_transform(): kbest = SelectKBest() params = {"score_func": f_regression, "k": 5, "X": X_reg, "y": y_reg} kbest.fit_transform(params) assert params["X_best"].shape[1] == 5
def test_fit_transform(params, split_size): kbest = SelectKBest() kbest.fit_transform(params) assert params["train_df"].shape == (split_size, params["k"] + 1) if "test_df" in params.keys(): assert params["test_df"].shape == (1000 - split_size, params["k"] + 1)
def test_fit_transform(self): kbest = SelectKBest(score_func=f_regression, k=5) X_new = kbest.fit_transform(X_reg, y_reg) assert X_new.shape[1] == 5