def test_rd_pandas_mode(self):
X, y, c, w = make_multilabel_classification(n_labels=5,
return_distributions=True,
pandas_mode=True)
assert isinstance(c, pd.Series)
assert np.array_equal(y.columns, c.index)
assert c.name == 'prob_of_class'
assert isinstance(w, pd.DataFrame)
assert np.array_equal(X.columns, w.index)
assert np.array_equal(y.columns, w.columns)
def test_sparse_pandas_mode(self):
with pytest.raises(NotImplementedError):
make_multilabel_classification(pandas_mode=True, sparse=True)
def test_pandas_mode(self):
X, Y = make_multilabel_classification(n_labels=5, pandas_mode=True)
assert isinstance(X, pd.DataFrame)
assert isinstance(Y, pd.DataFrame)
assert np.array_equal(X.index, Y.index)
def test_normal_mode(self):
X, y = make_multilabel_classification()
assert isinstance(X, np.ndarray)
assert isinstance(y, np.ndarray)
