test and fix indexing for scalar arrays

scalableminds · Dec 22, 2024 · e314f54 · e314f54
1 parent f360fc6
commit e314f54
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/zarr/api/asynchronous.py b/src/zarr/api/asynchronous.py
@@ -574,7 +574,7 @@ async def array(
     z = await create(**kwargs)
 
     # fill with data
-    await z.setitem(slice(None), data)
+    await z.setitem(Ellipsis, data)
 
     return z
 

diff --git a/tests/test_array.py b/tests/test_array.py
@@ -782,3 +782,10 @@ async def test_special_complex_fill_values_roundtrip(fill_value: Any, expected:
     assert content is not None
     actual = json.loads(content.to_bytes())
     assert actual["fill_value"] == expected
+
+
+async def test_scalar_array():
+    arr = zarr.array(1.5)
+    assert arr[...] == 1.5
+    assert arr[()] == 1.5
+    assert arr.shape == ()