Function test_slice_broadcast

xarray/tests/test_accessor_str.py:2179–2187

(dtype)

def test_slice_broadcast(dtype) -> None:
    arr = xr.DataArray(["aafootwo", "aabartwo", "aabazqux"]).astype(dtype)
    start = xr.DataArray([1, 2, 3])
    stop = 5

    result = arr.str.slice(start=start, stop=stop)
    exp = xr.DataArray(["afoo", "bar", "az"]).astype(dtype)
    assert result.dtype == exp.dtype
    assert_equal(result, exp)


def test_slice_replace(dtype) -> None: