Function test_interpolate_methods

xarray/tests/test_missing.py:333–344

()

@requires_scipy
def test_interpolate_methods():
    for method in ["linear", "nearest", "zero", "slinear", "quadratic", "cubic"]:
        kwargs: dict[str, Any] = {}
        da = xr.DataArray(
            np.array([0, 1, 2, np.nan, np.nan, np.nan, 6, 7, 8], dtype=np.float64),
            dims="x",
        )
        actual = da.interpolate_na("x", method=method, **kwargs)  # type: ignore[arg-type]
        assert actual.isnull().sum() == 0

        actual = da.interpolate_na("x", method=method, limit=2, **kwargs)  # type: ignore[arg-type]
        assert actual.isnull().sum() == 1


@requires_scipy