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)