Function test_eval_dask_boolean_operators

xarray/tests/test_eval.py:575–590

Test that bitwise boolean operators preserve dask.

()

@requires_dask
def test_eval_dask_boolean_operators() -> None:
    """Test that bitwise boolean operators preserve dask."""
    from xarray.core.utils import is_duck_dask_array

    ds = Dataset(
        {"a": ("x", np.arange(10.0)), "b": ("x", np.arange(10.0)[::-1])}
    ).chunk({"x": 5})

    with raise_if_dask_computes():
        result = ds.eval("(a > 3) & (b < 7)")

    assert is_duck_dask_array(result.data)

    # Verify correctness
    expected = (ds["a"] > 3) & (ds["b"] < 7)
    assert_equal(result, expected)


@requires_dask