Skip to content

Commit 102e411

Browse files
committed
make tests of async into async functions
1 parent a7e9e43 commit 102e411

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

tests/test_properties.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)