Exemplo n.º 1
0
def test_works_with_reordered_feats():
    v_t = VarianceThreshold(0)
    v_t.fit(DATA)
    res_df = v_t.transform(DATA[["non_zero_1", "zero_var", "non_zero_2"]])
    assert "zero_var" not in res_df.columns
Exemplo n.º 2
0
def test_variances_are_dfs():
    v_t = VarianceThreshold(0)
    v_t.fit(DATA)
    assert is_series(v_t.variances_)