Skip to content

Commit 1884eb7

Browse files
Tialoglemaitre
andauthored
TST Extend tests for scipy.sparse.*array in sklearn/metrics/tests/test_common.py (scikit-learn#27205)
Co-authored-by: Guillaume Lemaitre <g.lemaitre58@gmail.com>
1 parent df60c75 commit 1884eb7

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

sklearn/metrics/tests/test_common.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
import numpy as np
66
import pytest
7-
import scipy.sparse as sp
87

98
from sklearn._config import config_context
109
from sklearn.datasets import make_multilabel_classification
@@ -66,6 +65,7 @@
6665
assert_array_less,
6766
ignore_warnings,
6867
)
68+
from sklearn.utils.fixes import COO_CONTAINERS
6969
from sklearn.utils.multiclass import type_of_target
7070
from sklearn.utils.validation import _num_samples, check_random_state
7171

@@ -1033,7 +1033,8 @@ def test_multioutput_regression_invariance_to_dimension_shuffling(name):
10331033

10341034

10351035
@ignore_warnings
1036-
def test_multilabel_representation_invariance():
1036+
@pytest.mark.parametrize("coo_container", COO_CONTAINERS)
1037+
def test_multilabel_representation_invariance(coo_container):
10371038
# Generate some data
10381039
n_classes = 4
10391040
n_samples = 50
@@ -1057,8 +1058,8 @@ def test_multilabel_representation_invariance():
10571058
y1 = np.vstack([y1, [[0] * n_classes]])
10581059
y2 = np.vstack([y2, [[0] * n_classes]])
10591060

1060-
y1_sparse_indicator = sp.coo_matrix(y1)
1061-
y2_sparse_indicator = sp.coo_matrix(y2)
1061+
y1_sparse_indicator = coo_container(y1)
1062+
y2_sparse_indicator = coo_container(y2)
10621063

10631064
y1_list_array_indicator = list(y1)
10641065
y2_list_array_indicator = list(y2)

0 commit comments

Comments
 (0)