Function test_range_index_sel

xarray/tests/test_range_index.py:225–264

()

def test_range_index_sel() -> None:
    ds = create_dataset_arange(0.0, 1.0, 0.1)

    # start-stop slice
    actual = ds.sel(x=slice(0.12, 0.28), method="nearest")
    expected = create_dataset_arange(0.1, 0.3, 0.1)
    assert_identical(actual, expected, check_default_indexes=False, check_indexes=True)

    # start-stop-step slice
    actual = ds.sel(x=slice(0.0, 1.0, 0.2), method="nearest")
    expected = ds.isel(x=range(0, 10, 2))
    assert_identical(actual, expected, check_default_indexes=False, check_indexes=True)

    # basic indexing
    actual = ds.sel(x=0.52, method="nearest")
    expected = xr.Dataset(coords={"x": 0.5})
    assert_allclose(actual, expected)

    actual = ds.sel(x=0.58, method="nearest")
    expected = xr.Dataset(coords={"x": 0.6})
    assert_allclose(actual, expected)

    # 1-d array indexing
    actual = ds.sel(x=[0.52, 0.58], method="nearest")
    expected = xr.Dataset(coords={"x": [0.5, 0.6]})
    assert_allclose(actual, expected)

    actual = ds.sel(x=xr.Variable("y", [0.52, 0.58]), method="nearest")
    expected = xr.Dataset(coords={"x": ("y", [0.5, 0.6])}).set_xindex("x")
    assert_allclose(actual, expected, check_default_indexes=False)

    actual = ds.sel(x=xr.DataArray([0.52, 0.58], dims="y"), method="nearest")
    expected = xr.Dataset(coords={"x": ("y", [0.5, 0.6])}).set_xindex("x")
    assert_allclose(actual, expected, check_default_indexes=False)

    with pytest.raises(ValueError, match=r"RangeIndex only supports.*method.*nearest"):
        ds.sel(x=0.1)

    with pytest.raises(ValueError, match=r"RangeIndex doesn't support.*tolerance"):
        ds.sel(x=0.1, method="nearest", tolerance=1e-3)


def test_range_index_to_pandas_index() -> None: