def test_pandas_mode(self): y, X, w = make_sparse_coded_signal(1, 512, 100, 17, pandas_mode=True) assert isinstance(y, pd.DataFrame) assert isinstance(X, pd.DataFrame) assert isinstance(w, pd.DataFrame) assert np.array_equal(y.index, X.index) assert np.array_equal(y.columns, w.columns) assert np.array_equal(X.columns, w.index)
def test_normal_mode(self): y, X, w = make_sparse_coded_signal(1, 512, 100, 17) assert isinstance(y, np.ndarray) assert isinstance(X, np.ndarray) assert isinstance(w, np.ndarray)