diff --git a/sklearn/utils/tests/test_extmath.py b/sklearn/utils/tests/test_extmath.py index 07a553c8cf09d..8fbc66e5e8e96 100644 --- a/sklearn/utils/tests/test_extmath.py +++ b/sklearn/utils/tests/test_extmath.py @@ -4,6 +4,7 @@ # # License: BSD 3 clause import numpy as np +import scipy from scipy import sparse from scipy import linalg from scipy import stats @@ -33,6 +34,9 @@ from sklearn.utils.extmath import stable_cumsum from sklearn.utils.extmath import safe_sparse_dot from sklearn.datasets import make_low_rank_matrix, make_sparse_spd_matrix +from sklearn.externals._packaging.version import parse as parse_version + +sp_version = parse_version(scipy.__version__) def test_density(): @@ -59,6 +63,11 @@ def test_uniform_weights(): mode, score = stats.mode(x, axis) mode2, score2 = weighted_mode(x, weights, axis=axis) + # See https://github.com/scipy/scipy/issues/16418 + if sp_version >= parse_version("1.9.0") and axis is not None: + mode = np.expand_dims(mode, axis=axis) + score = np.expand_dims(score, axis=axis) + assert_array_equal(mode, mode2) assert_array_equal(score, score2)