add test case for scipy version UserWarning

EmilyXinyi · EmilyXinyi · commit 8579f6ffbde2 · 2024-07-10T18:09:57.000+02:00
diff --git a/sklearn/utils/tests/test_array_api.py b/sklearn/utils/tests/test_array_api.py
@@ -1,3 +1,4 @@
+import os
 import re
 from functools import partial
 
@@ -76,7 +77,7 @@ def test_get_namespace_ndarray_with_dispatch():
 
 
 @skip_if_array_api_compat_not_configured
-def test_get_namespace_array_api():
+def test_get_namespace_array_api(monkeypatch):
     """Test get_namespace for ArrayAPI arrays."""
     xp = pytest.importorskip("array_api_strict")
 
@@ -89,6 +90,19 @@ def test_get_namespace_array_api():
         with pytest.raises(TypeError):
             xp_out, is_array_api_compliant = get_namespace(X_xp, X_np)
 
+        def mock_getenv(key):
+            if key == "SCIPY_ARRAY_API":
+                return "0"
+            return None
+
+        monkeypatch.setattr("os.environ.get", mock_getenv)
+        assert os.environ.get("SCIPY_ARRAY_API") != "1"
+        with pytest.warns(
+            UserWarning,
+            match="enabling SciPy's own support for array API to function properly. ",
+        ):
+            xp_out, is_array_api_compliant = get_namespace(X_xp)
+
 
 class _AdjustableNameAPITestWrapper(_ArrayAPIWrapper):
     """API wrapper that has an adjustable name. Used for testing."""
