fixup

Author: TomAugspurger

src/zarr/core/array.py

@@ -1219,6 +1219,12 @@ def info(self) -> ArrayInfo:
 
     async def info_complete(self) -> ArrayInfo:
         # TODO: get the size of the object from the store.
+        extra = {
+            "count_chunks_initialized": self.nchunks_initialized,  # this should be async?
+            # count_bytes_stored isn't yet implemented.
+        }
+        return self._info(extra=extra)
+
         raise NotImplementedError
 
     def _info(self, extra: dict[str, int] | None = None) -> ArrayInfo:
@@ -1239,15 +1245,17 @@ def _info(self, extra: dict[str, int] | None = None) -> ArrayInfo:
             if isinstance(chunk_grid, RegularChunkGrid):
                 kwargs["chunk_shape"] = chunk_grid.chunk_shape
             else:
-                msg = f"'info' is not yet implemented for chunk grids of type {type(self.metadata.chunk_grid)}"
-                raise NotImplementedError(msg)
+                raise NotImplementedError(
+                    "'info' is not yet implemented for chunk grids of type {type(self.metadata.chunk_grid)}"
+                )
 
         return ArrayInfo(
             zarr_format=self.metadata.zarr_format,
             shape=self.shape,
             order=self.order,
             read_only=self.store_path.store.mode.readonly,
             store_type=type(self.store_path.store).__name__,
+            count_bytes=self.dtype.itemsize * self.size,
             **kwargs,
         )
 

tests/test_array.py

@@ -435,6 +435,7 @@ def test_info_v2(self) -> None:
             order="C",
             read_only=False,
             store_type="MemoryStore",
+            count_bytes=128,
         )
         assert result == expected
 
@@ -450,6 +451,7 @@ def test_info_v3(self) -> None:
             read_only=False,
             store_type="MemoryStore",
             codecs="[BytesCodec(endian=<Endian.little: 'little'>)]",
+            count_bytes=128,
         )
         assert result == expected