def test_coef_multi_pandas_mode(self):
X, Y, coef = make_regression(n_targets=5, coef=True, pandas_mode=True)
assert isinstance(coef, pd.DataFrame)
assert np.array_equal(X.columns, coef.index)
assert np.array_equal(Y.columns, coef.columns)
def test_coef_single_pandas_mode(self):
X, _, coef = make_regression(coef=True, pandas_mode=True)
assert isinstance(coef, pd.Series)
assert np.array_equal(X.columns, coef.index)
assert coef.name == 'coefs'
def test_coef_normal_mode(self):
X, y, coef = make_regression(coef=True)
assert isinstance(X, np.ndarray)
assert isinstance(y, np.ndarray)
assert isinstance(coef, np.ndarray)
def test_multi_pandas_mode(self):
X, Y = make_regression(n_targets=5, pandas_mode=True)
assert isinstance(X, pd.DataFrame)
assert isinstance(Y, pd.DataFrame)
assert np.array_equal(X.index, Y.index)
def test_single_pandas_mode(self):
X, y = make_regression(pandas_mode=True)
assert isinstance(X, pd.DataFrame)
assert isinstance(y, pd.Series)
assert np.array_equal(X.index, y.index)
def test_normal_mode(self):
X, y = make_regression()
assert isinstance(X, np.ndarray)
assert isinstance(y, np.ndarray)
