Skip to content

Commit 9899db2

Browse files
committed
TST check error raise when sparse are not accepted
1 parent b71dc21 commit 9899db2

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

imblearn/tests/test_misc.py

+8
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,14 @@
2020
random_state=0)
2121

2222

23+
def test_function_sampler_reject_sparse():
24+
X_sparse = sparse.csr_matrix(X)
25+
sampler = FunctionSampler(accept_sparse=False)
26+
with pytest.raises(TypeError, message="A sparse matrix was passed, "
27+
"but dense data is required"):
28+
sampler.fit(X_sparse, y)
29+
30+
2331
@pytest.mark.parametrize("X,y", [(X, y),
2432
(sparse.csr_matrix(X), y),
2533
(sparse.csc_matrix(X), y)])

0 commit comments

Comments
 (0)