def test_predict_df(self):
c = ChildClassifier(features=["x", "y"], targets=["c"])
res = c.predict(DATA_FRAME)
assert is_series(res)
assert np.array_equal(res.index, DATA_FRAME.index)
assert res.name == "c"
def test_predict_df(self):
c = ChildRegressor(features=['x', 'y'], targets=['c'])
res = c.predict(DATA_FRAME)
assert is_series(res)
assert np.array_equal(res.index, DATA_FRAME.index)
assert res.name == 'c'
def test_predict_arr_pdmode(self):
c = ChildClassifier(features=["x", "y"], targets=["a"])
with pytest.warns(CompatabilityWarning):
res = c.predict(TWO_D_ARRAY)
assert is_series(res)
assert np.array_equal(res.index, range(len(TWO_D_ARRAY)))
assert res.name == "a"
def test_predict_arr_pdmode(self):
c = ChildRegressor(features=['x', 'y'], targets=['a'])
with pytest.warns(CompatabilityWarning):
res = c.predict(TWO_D_ARRAY)
assert is_series(res)
assert np.array_equal(res.index, range(len(TWO_D_ARRAY)))
assert res.name == 'a'
def test_variances_are_dfs():
v_t = VarianceThreshold(0)
v_t.fit(DATA)
assert is_series(v_t.variances_)