Function test_eval_dask_np_functions

xarray/tests/test_eval.py:539–552

Test that numpy functions via np namespace preserve dask.

()

@requires_dask
def test_eval_dask_np_functions() -> None:
    """Test that numpy functions via np namespace preserve dask."""
    from xarray.core.utils import is_duck_dask_array

    ds = Dataset({"a": ("x", np.arange(1.0, 11.0))}).chunk({"x": 5})

    with raise_if_dask_computes():
        result = ds.eval("np.sqrt(a)")

    assert is_duck_dask_array(result.data)

    # Verify correctness
    expected = np.sqrt(ds["a"])
    assert_equal(result, expected)


@requires_dask