Self-review

Author: crusaderky

tests/conftest.py

@@ -131,11 +131,11 @@ def xp(
     return xp
 
 
-@pytest.fixture(params=[Backend.DASK])  # Select the test with `pytest -k dask`
+@pytest.fixture(params=[Backend.DASK])  # Can select the test with `pytest -k dask`
 def da(
     request: pytest.FixtureRequest, monkeypatch: pytest.MonkeyPatch
 ) -> ModuleType:  # numpydoc ignore=PR01,RT01
-    """Fixture that returns dask.array (wrapped by array-api-compat)."""
+    """Variant of the `xp` fixture that only yields dask.array."""
     xp = pytest.importorskip("dask.array")
     xp = array_namespace(xp.empty(0))
     patch_lazy_xp_functions(request, monkeypatch, xp=xp)

tests/test_lazy.py

@@ -95,11 +95,9 @@ def f(x: Array) -> tuple[Array, Array]:
     [
         pytest.param(
             False,
-            marks=[
-                pytest.mark.xfail_xp_backend(
-                    Backend.TORCH, reason="illegal dtype promotion"
-                ),
-            ],
+            marks=pytest.mark.xfail_xp_backend(
+                Backend.TORCH, reason="illegal dtype promotion"
+            ),
         ),
         pytest.param(
             True,
@@ -218,7 +216,8 @@ def f(x: Array) -> Array:
         return x
 
     x = x[x > 1]
-    return lazy_apply(f, x)
+    # Use explicit namespace to bypass monkey-patching by lazy_xp_function
+    return xpx.lazy_apply(f, x)
 
 
 lazy_xp_function(check_lazy_apply_none_shape_broadcast)