Skip to content

Commit e19ab55

Browse files
committed
correct test to actually use vectorized indexing
1 parent 842a06c commit e19ab55

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

xarray/tests/test_async.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,8 @@ async def test_concurrent_load_multiple_objects(self, xr_obj) -> None:
179179
)
180180

181181
async def test_indexing(self, memorystore) -> None:
182-
latencystore = LatencyStore(memorystore, latency=self.LATENCY)
182+
# TODO we don't need a LatencyStore for this test
183+
latencystore = LatencyStore(memorystore, latency=0.0)
183184
ds = xr.open_zarr(latencystore, zarr_format=3, consolidated=False, chunks=None)
184185

185186
# test basic indexing
@@ -193,7 +194,6 @@ async def test_indexing(self, memorystore) -> None:
193194
xrt.assert_identical(result, ds.sel(indexer).load())
194195

195196
# test vectorized indexing
196-
# TODO this shouldn't pass! I haven't implemented async vectorized indexing yet...
197-
indexer = xr.DataArray([2, 3], dims=["x"])
198-
result = await ds.foo[indexer].load_async()
199-
xrt.assert_identical(result, ds.foo[indexer].load())
197+
indexer = {"x": xr.DataArray([2, 3], dims="points"), "y": xr.DataArray([2, 3], dims="points")}
198+
result = await ds.isel(indexer).load_async()
199+
xrt.assert_identical(result, ds.isel(indexer).load())

0 commit comments

Comments
 (0)