Function test_tree_index_1d_coords

xarray/tests/test_nd_point_index.py:196–223

()

@requires_scipy
def test_tree_index_1d_coords() -> None:
    ds = xr.Dataset(
        {"temp": ("points", np.arange(20, dtype=float))},
        coords={
            "x": ("points", np.linspace(0, 10, 20)),
            "y": ("points", np.linspace(15, 8, 20)),
        },
    )
    ds_indexed = ds.set_xindex(("x", "y"), NDPointIndex)

    assert isinstance(ds_indexed.xindexes["x"], NDPointIndex)

    actual = ds_indexed.sel(x=5.0, y=11.5, method="nearest")
    assert actual.temp.values == 10.0

    actual = ds_indexed.sel(
        x=xr.Variable("query", [0.0, 5.0, 10.0]),
        y=xr.Variable("query", [15.0, 11.5, 8.0]),
        method="nearest",
    )
    expected = xr.Dataset(
        {"temp": ("query", [0.0, 10.0, 19.0])},
        coords={
            "x": ("query", [0.0, ds.x.values[10], 10.0]),
            "y": ("query", [15.0, ds.y.values[10], 8.0]),
        },
    )
    assert_identical(actual, expected)