add hypothesis tests for supported async indexing

Author: TomNicholas

tests/test_properties.py

@@ -1,6 +1,7 @@
 import json
 import numbers
 from typing import Any
+import asyncio
 
 import numpy as np
 import pytest
@@ -121,6 +122,26 @@ def test_basic_indexing(data: st.DataObject) -> None:
     assert_array_equal(nparray, zarray[:])
 
 
+@settings(deadline=None)
+@pytest.mark.filterwarnings("ignore::zarr.core.dtype.common.UnstableSpecificationWarning")
+@given(data=st.data())
+def test_basic_indexing_async(data: st.DataObject) -> None:
+    zarray = data.draw(simple_arrays())
+    nparray = zarray[:]
+    indexer = data.draw(basic_indices(shape=nparray.shape))
+    async_zarray = zarray._async_array
+    
+    actual = asyncio.run(async_zarray.getitem(indexer))
+    assert_array_equal(nparray[indexer], actual)
+
+    # TODO test async setitem
+    # new_data = data.draw(numpy_arrays(shapes=st.just(actual.shape), dtype=nparray.dtype))
+    # asyncio.run(async_zarray.setitem(indexer, new_data))
+    # nparray[indexer] = new_data
+    # result = asyncio.run(async_zarray.getitem(indexer))
+    # assert_array_equal(nparray, result)
+
+
 @given(data=st.data())
 @pytest.mark.filterwarnings("ignore::zarr.core.dtype.common.UnstableSpecificationWarning")
 def test_oindex(data: st.DataObject) -> None:
@@ -143,6 +164,21 @@ def test_oindex(data: st.DataObject) -> None:
     assert_array_equal(nparray, zarray[:])
 
 
+@given(data=st.data())
+@pytest.mark.filterwarnings("ignore::zarr.core.dtype.common.UnstableSpecificationWarning")
+def test_oindex_async(data: st.DataObject) -> None:
+    # integer_array_indices can't handle 0-size dimensions.
+    zarray = data.draw(simple_arrays(shapes=npst.array_shapes(max_dims=4, min_side=1)))
+    nparray = zarray[:]
+    async_zarray = zarray._async_array
+
+    zindexer, npindexer = data.draw(orthogonal_indices(shape=nparray.shape))
+    actual = asyncio.run(async_zarray.oindex.getitem(zindexer))
+    assert_array_equal(nparray[npindexer], actual)
+
+    # note: async oindex setting not yet implemented
+
+
 @given(data=st.data())
 @pytest.mark.filterwarnings("ignore::zarr.core.dtype.common.UnstableSpecificationWarning")
 def test_vindex(data: st.DataObject) -> None:
@@ -167,6 +203,23 @@ def test_vindex(data: st.DataObject) -> None:
     # assert_array_equal(nparray, zarray[:])
 
 
+@given(data=st.data())
+@pytest.mark.filterwarnings("ignore::zarr.core.dtype.common.UnstableSpecificationWarning")
+def test_vindex_async(data: st.DataObject) -> None:
+    # integer_array_indices can't handle 0-size dimensions.
+    zarray = data.draw(simple_arrays(shapes=npst.array_shapes(max_dims=4, min_side=1)))
+    nparray = zarray[:]
+    async_zarray = zarray._async_array
+
+    indexer = data.draw(
+        npst.integer_array_indices(
+            shape=nparray.shape, result_shape=npst.array_shapes(min_side=1, max_dims=None)
+        )
+    )
+    actual = asyncio.run(async_zarray.vindex.getitem(indexer))
+    assert_array_equal(nparray[indexer], actual)
+
+
 @given(store=stores, meta=array_metadata())  # type: ignore[misc]
 @pytest.mark.filterwarnings("ignore::zarr.core.dtype.common.UnstableSpecificationWarning")
 async def test_roundtrip_array_metadata_from_store(