@@ -122,16 +122,17 @@ def test_basic_indexing(data: st.DataObject) -> None:
122122 assert_array_equal (nparray , zarray [:])
123123
124124
125+ @pytest .mark .asyncio
125126@settings (deadline = None )
126127@pytest .mark .filterwarnings ("ignore::zarr.core.dtype.common.UnstableSpecificationWarning" )
127128@given (data = st .data ())
128- def test_basic_indexing_async (data : st .DataObject ) -> None :
129+ async def test_basic_indexing_async (data : st .DataObject ) -> None :
129130 zarray = data .draw (simple_arrays ())
130131 nparray = zarray [:]
131132 indexer = data .draw (basic_indices (shape = nparray .shape ))
132133 async_zarray = zarray ._async_array
133134
134- actual = asyncio . run ( async_zarray .getitem (indexer ) )
135+ actual = await async_zarray .getitem (indexer )
135136 assert_array_equal (nparray [indexer ], actual )
136137
137138 # TODO test async setitem
@@ -164,16 +165,17 @@ def test_oindex(data: st.DataObject) -> None:
164165 assert_array_equal (nparray , zarray [:])
165166
166167
168+ @pytest .mark .asyncio
167169@given (data = st .data ())
168170@pytest .mark .filterwarnings ("ignore::zarr.core.dtype.common.UnstableSpecificationWarning" )
169- def test_oindex_async (data : st .DataObject ) -> None :
171+ async def test_oindex_async (data : st .DataObject ) -> None :
170172 # integer_array_indices can't handle 0-size dimensions.
171173 zarray = data .draw (simple_arrays (shapes = npst .array_shapes (max_dims = 4 , min_side = 1 )))
172174 nparray = zarray [:]
173175 async_zarray = zarray ._async_array
174176
175177 zindexer , npindexer = data .draw (orthogonal_indices (shape = nparray .shape ))
176- actual = asyncio . run ( async_zarray .oindex .getitem (zindexer ) )
178+ actual = await async_zarray .oindex .getitem (zindexer )
177179 assert_array_equal (nparray [npindexer ], actual )
178180
179181 # note: async oindex setting not yet implemented
@@ -203,9 +205,10 @@ def test_vindex(data: st.DataObject) -> None:
203205 # assert_array_equal(nparray, zarray[:])
204206
205207
208+ @pytest .mark .asyncio
206209@given (data = st .data ())
207210@pytest .mark .filterwarnings ("ignore::zarr.core.dtype.common.UnstableSpecificationWarning" )
208- def test_vindex_async (data : st .DataObject ) -> None :
211+ async def test_vindex_async (data : st .DataObject ) -> None :
209212 # integer_array_indices can't handle 0-size dimensions.
210213 zarray = data .draw (simple_arrays (shapes = npst .array_shapes (max_dims = 4 , min_side = 1 )))
211214 nparray = zarray [:]
@@ -216,7 +219,7 @@ def test_vindex_async(data: st.DataObject) -> None:
216219 shape = nparray .shape , result_shape = npst .array_shapes (min_side = 1 , max_dims = None )
217220 )
218221 )
219- actual = asyncio . run ( async_zarray .vindex .getitem (indexer ) )
222+ actual = await async_zarray .vindex .getitem (indexer )
220223 assert_array_equal (nparray [indexer ], actual )
221224
222225
0 commit comments