Add tests

Author: dcherian

xarray/tests/test_async.py

@@ -182,6 +182,7 @@ async def test_concurrent_load_multiple_objects(self, xr_obj) -> None:
         "method,indexer",
         [
             ("sel", {"x": 2}),
+            ("sel", {"x": slice(2, 4)}),
             ("sel", {"x": [2, 3]}),
             (
                 "sel",
@@ -191,7 +192,7 @@ async def test_concurrent_load_multiple_objects(self, xr_obj) -> None:
                 },
             ),
         ],
-        ids=["basic", "outer", "vectorized"],
+        ids=["basic-int", "basic-slice", "outer", "vectorized"],
     )
     async def test_indexing(self, memorystore, method, indexer) -> None:
         # TODO we don't need a LatencyStore for this test

xarray/tests/test_indexing.py

@@ -490,6 +490,23 @@ def test_sub_array(self) -> None:
         assert isinstance(child.array, indexing.NumpyIndexingAdapter)
         assert isinstance(wrapped.array, indexing.LazilyIndexedArray)
 
+    async def test_async_wrapper(self) -> None:
+        original = indexing.LazilyIndexedArray(np.arange(10))
+        wrapped = indexing.MemoryCachedArray(original)
+        await wrapped.async_get_duck_array()
+        assert_array_equal(wrapped, np.arange(10))
+        assert isinstance(wrapped.array, indexing.NumpyIndexingAdapter)
+
+    async def test_async_sub_array(self) -> None:
+        original = indexing.LazilyIndexedArray(np.arange(10))
+        wrapped = indexing.MemoryCachedArray(original)
+        child = wrapped[B[:5]]
+        assert isinstance(child, indexing.MemoryCachedArray)
+        await child.async_get_duck_array()
+        assert_array_equal(child, np.arange(5))
+        assert isinstance(child.array, indexing.NumpyIndexingAdapter)
+        assert isinstance(wrapped.array, indexing.LazilyIndexedArray)
+
     def test_setitem(self) -> None:
         original = np.arange(10)
         wrapped = indexing.MemoryCachedArray(original)