fix mypy in tests

Author: brokkoli71

src/zarr/api/synchronous.py

@@ -118,7 +118,7 @@ def load(
     path: str | None = None,
     zarr_format: ZarrFormat | None = None,
     zarr_version: ZarrFormat | None = None,
-) -> NDArrayLike | dict[str, NDArrayLike]:
+) -> npt.ArrayLike | dict[str, npt.ArrayLike]:
     """Load data from an array or group into memory.
 
     Parameters

src/zarr/core/array.py

@@ -1246,7 +1246,7 @@ async def _get_selection(
                 drop_axes=indexer.drop_axes,
             )
         if indexer.shape == ():
-            return cast(out_buffer.as_numpy_array().item(), out_buffer.dtype)
+            return cast(out_buffer.as_numpy_array().item(), npt.ArrayLike)
         return out_buffer.as_ndarray_like()
 
     async def getitem(

tests/test_api.py

@@ -275,7 +275,7 @@ def test_open_with_mode_w(tmp_path: pathlib.Path) -> None:
     arr[...] = 3
     z2 = zarr.open(store=tmp_path, mode="w", shape=(3, 3))
     assert isinstance(z2, Array)
-    assert not (z2[:] == 3).all()
+    assert not np.array(z2[:] == 3).all()
     z2[:] = 3
 
 

tests/test_array.py

@@ -658,7 +658,7 @@ def test_resize_1d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
     assert (205,) == z.shape
     assert (205,) == z[:].shape  # type: ignore[union-attr]
     assert np.dtype("i4") == z.dtype
-    assert np.dtype("i4") == z[:].dtype
+    assert np.dtype("i4") == z[:].dtype  # type: ignore[union-attr]
     assert (10,) == z.chunks
     np.testing.assert_array_equal(a, z[:105])
     np.testing.assert_array_equal(np.zeros(100, dtype="i4"), z[105:])
@@ -693,15 +693,15 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
     assert (105, 105) == z.shape
     assert (105, 105) == z[:].shape  # type: ignore[union-attr]
     assert np.dtype("i4") == z.dtype
-    assert np.dtype("i4") == z[:].dtype
+    assert np.dtype("i4") == z[:].dtype  # type: ignore[union-attr]
     assert (10, 10) == z.chunks
     np.testing.assert_array_equal(a, z[:])
 
     z.resize((205, 205))
     assert (205, 205) == z.shape
     assert (205, 205) == z[:].shape  # type: ignore[union-attr]
     assert np.dtype("i4") == z.dtype
-    assert np.dtype("i4") == z[:].dtype
+    assert np.dtype("i4") == z[:].dtype  # type: ignore[union-attr]
     assert (10, 10) == z.chunks
     np.testing.assert_array_equal(a, z[:105, :105])
     np.testing.assert_array_equal(np.zeros((100, 205), dtype="i4"), z[105:, :])
@@ -711,23 +711,23 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
     assert (55, 55) == z.shape
     assert (55, 55) == z[:].shape  # type: ignore[union-attr]
     assert np.dtype("i4") == z.dtype
-    assert np.dtype("i4") == z[:].dtype
+    assert np.dtype("i4") == z[:].dtype  # type: ignore[union-attr]
     assert (10, 10) == z.chunks
     np.testing.assert_array_equal(a[:55, :55], z[:])
 
     z.resize((55, 1))
     assert (55, 1) == z.shape
     assert (55, 1) == z[:].shape  # type: ignore[union-attr]
     assert np.dtype("i4") == z.dtype
-    assert np.dtype("i4") == z[:].dtype
+    assert np.dtype("i4") == z[:].dtype  # type: ignore[union-attr]
     assert (10, 10) == z.chunks
     np.testing.assert_array_equal(a[:55, :1], z[:])
 
     z.resize((1, 55))
     assert (1, 55) == z.shape
     assert (1, 55) == z[:].shape  # type: ignore[union-attr]
     assert np.dtype("i4") == z.dtype
-    assert np.dtype("i4") == z[:].dtype
+    assert np.dtype("i4") == z[:].dtype  # type: ignore[union-attr]
     assert (10, 10) == z.chunks
     np.testing.assert_array_equal(a[:1, :10], z[:, :10])
     np.testing.assert_array_equal(np.zeros((1, 55 - 10), dtype="i4"), z[:, 10:55])