Skip to content

Commit 1ac02ea

Browse files
authored
test and fix indexing for scalar arrays (zarr-developers#2583)
* test and fix indexing for scalar arrays * fix mypy
1 parent fb11810 commit 1ac02ea

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/zarr/api/asynchronous.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -579,7 +579,7 @@ async def array(
579579
z = await create(**kwargs)
580580

581581
# fill with data
582-
await z.setitem(slice(None), data)
582+
await z.setitem(Ellipsis, data)
583583

584584
return z
585585

tests/test_array.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -881,3 +881,10 @@ async def test_nbytes(
881881
assert arr._async_array.nbytes == np.prod(arr.shape) * arr.dtype.itemsize
882882
else:
883883
assert arr.nbytes == np.prod(arr.shape) * arr.dtype.itemsize
884+
885+
886+
async def test_scalar_array() -> None:
887+
arr = zarr.array(1.5)
888+
assert arr[...] == 1.5
889+
assert arr[()] == 1.5
890+
assert arr.shape == ()

0 commit comments

Comments
 (0)