Function test_eval_dask_chained_comparisons

xarray/tests/test_eval.py:594–607

Test that chained comparisons preserve dask backing.

()

@requires_dask
def test_eval_dask_chained_comparisons() -> None:
    """Test that chained comparisons preserve dask backing."""
    from xarray.core.utils import is_duck_dask_array

    ds = Dataset({"x": ("dim", np.arange(10.0))}).chunk({"dim": 5})

    with raise_if_dask_computes():
        result = ds.eval("2 < x < 7")

    assert is_duck_dask_array(result.data)

    # Verify correctness
    expected = (ds["x"] > 2) & (ds["x"] < 7)
    assert_equal(result, expected)