fix mypy in test_api.py

Author: brokkoli71

src/zarr/api/asynchronous.py

@@ -11,7 +11,7 @@
 
 from zarr.core.array import Array, AsyncArray, create_array, get_array_metadata
 from zarr.core.array_spec import ArrayConfig, ArrayConfigLike
-from zarr.core.buffer import NDArrayOrScalarLike
+from zarr.core.buffer.core import NDArrayLike
 from zarr.core.common import (
     JSON,
     AccessModeLiteral,
@@ -33,7 +33,7 @@
     from collections.abc import Iterable
 
     from zarr.abc.codec import Codec
-    from zarr.core.buffer.core import NDArrayLike
+    from zarr.core.buffer import NDArrayOrScalarLike
     from zarr.core.chunk_key_encodings import ChunkKeyEncoding
     from zarr.storage import StoreLike
 
@@ -481,9 +481,7 @@ async def save_group(
             )
     for k, v in kwargs.items():
         if not isinstance(v, NDArrayLike):
-            raise TypeError(
-                f"Keyword argument '{k}' must be a numpy or other NDArrayLike array"
-            )
+            raise TypeError(f"Keyword argument '{k}' must be a numpy or other NDArrayLike array")
 
     if len(args) == 0 and len(kwargs) == 0:
         raise ValueError("at least one array must be provided")

tests/test_api.py

@@ -23,6 +23,7 @@
     save_array,
     save_group,
 )
+from zarr.core.buffer.core import NDArrayLike
 from zarr.core.common import JSON, MemoryOrder, ZarrFormat
 from zarr.errors import MetadataValidationError
 from zarr.storage import MemoryStore
@@ -235,6 +236,7 @@ def test_open_with_mode_r(tmp_path: pathlib.Path) -> None:
     z2 = zarr.open(store=tmp_path, mode="r")
     assert isinstance(z2, Array)
     assert z2.fill_value == 1
+    assert isinstance(z2[:], NDArrayLike)
     assert (z2[:] == 1).all()
     with pytest.raises(ValueError):
         z2[:] = 3

tests/test_array.py

@@ -1394,13 +1394,6 @@ def test_roundtrip_numcodecs() -> None:
     assert metadata["codecs"] == expected
 
 
-def test_tmp():
-    a = np.array(1)
-    print(type(a[()]))
-    z = zarr.array(a, store=MemoryStore())
-    print(type(z[()]))
-
-
 def _index_array(arr: Array, index: Any) -> Any:
     return arr[index]
 

tests/test_buffer.py

@@ -11,7 +11,8 @@
 from zarr.codecs.gzip import GzipCodec
 from zarr.codecs.transpose import TransposeCodec
 from zarr.codecs.zstd import ZstdCodec
-from zarr.core.buffer import ArrayLike, BufferPrototype, NDArrayLike, cpu, gpu
+from zarr.core.buffer import ArrayLike, BufferPrototype, cpu, gpu
+from zarr.core.buffer.core import NDArrayLike
 from zarr.storage import MemoryStore, StorePath
 from zarr.testing.buffer import (
     NDBufferUsingTestNDArrayLike,