()
| 99 | |
| 100 | |
| 101 | def test_range_index_isel() -> None: |
| 102 | ds = create_dataset_arange(0.0, 1.0, 0.1) |
| 103 | |
| 104 | # slicing |
| 105 | actual = ds.isel(x=slice(None)) |
| 106 | assert_identical(actual, ds, check_default_indexes=False, check_indexes=True) |
| 107 | |
| 108 | actual = ds.isel(x=slice(1, None)) |
| 109 | expected = create_dataset_arange(0.1, 1.0, 0.1) |
| 110 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 111 | |
| 112 | actual = ds.isel(x=slice(None, 2)) |
| 113 | expected = create_dataset_arange(0.0, 0.2, 0.1) |
| 114 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 115 | |
| 116 | actual = ds.isel(x=slice(1, 3)) |
| 117 | expected = create_dataset_arange(0.1, 0.3, 0.1) |
| 118 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 119 | |
| 120 | actual = ds.isel(x=slice(None, None, 2)) |
| 121 | expected = create_dataset_arange(0.0, 1.0, 0.2) |
| 122 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 123 | |
| 124 | actual = ds.isel(x=slice(None, None, -1)) |
| 125 | expected = create_dataset_arange(0.9, -0.1, -0.1) |
| 126 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 127 | |
| 128 | actual = ds.isel(x=slice(None, 4, -1)) |
| 129 | expected = create_dataset_arange(0.9, 0.4, -0.1) |
| 130 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 131 | |
| 132 | actual = ds.isel(x=slice(8, 4, -1)) |
| 133 | expected = create_dataset_arange(0.8, 0.4, -0.1) |
| 134 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 135 | |
| 136 | actual = ds.isel(x=slice(8, None, -1)) |
| 137 | expected = create_dataset_arange(0.8, -0.1, -0.1) |
| 138 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 139 | |
| 140 | # https://github.com/pydata/xarray/issues/10441 |
| 141 | ds2 = create_dataset_arange(0.0, 3.0, 0.1) |
| 142 | actual = ds2.isel(x=slice(4, None, 3)) |
| 143 | expected = create_dataset_arange(0.4, 3.0, 0.3) |
| 144 | assert_identical(actual, expected, check_default_indexes=False, check_indexes=True) |
| 145 | |
| 146 | # scalar |
| 147 | actual = ds.isel(x=0) |
| 148 | expected = xr.Dataset(coords={"x": 0.0}) |
| 149 | assert_identical(actual, expected) |
| 150 | |
| 151 | # outer indexing with arbitrary array values |
| 152 | actual = ds.isel(x=[0, 2]) |
| 153 | expected = xr.Dataset(coords={"x": [0.0, 0.2]}) |
| 154 | assert_identical(actual, expected) |
| 155 | assert isinstance(actual.xindexes["x"], PandasIndex) |
| 156 | |
| 157 | # fancy indexing with 1-d Variable |
| 158 | actual = ds.isel(x=xr.Variable("y", [0, 2])) |
nothing calls this directly
no test coverage detected
searching dependent graphs…