fix mypy in test_api.py and test_array.py

Author: brokkoli71

tests/test_api.py

@@ -237,7 +237,7 @@ def test_open_with_mode_r(tmp_path: pathlib.Path) -> None:
     assert isinstance(z2, Array)
     assert z2.fill_value == 1
     assert isinstance(z2[:], NDArrayLike)
-    assert (z2[:] == 1).all() # type: ignore [operator]
+    assert (z2[:] == 1).all() # type: ignore [union-attr]
     with pytest.raises(ValueError):
         z2[:] = 3
 
@@ -250,7 +250,7 @@ def test_open_with_mode_r_plus(tmp_path: pathlib.Path) -> None:
     z2 = zarr.open(store=tmp_path, mode="r+")
     assert isinstance(z2, Array)
     assert isinstance(z2[:], NDArrayLike)
-    assert (z2[:] == 1).all() # type: ignore [operator]
+    assert (z2[:] == 1).all() # type: ignore [union-attr]
     z2[:] = 3
 
 
@@ -267,7 +267,7 @@ async def test_open_with_mode_a(tmp_path: pathlib.Path) -> None:
     z2 = zarr.open(store=tmp_path, mode="a")
     assert isinstance(z2, Array)
     assert isinstance(z2[:], NDArrayLike)
-    assert (z2[:] == 1).all() # type: ignore [operator]
+    assert (z2[:] == 1).all() # type: ignore [union-attr]
     z2[:] = 3
 
 
@@ -280,7 +280,7 @@ def test_open_with_mode_w(tmp_path: pathlib.Path) -> None:
     z2 = zarr.open(store=tmp_path, mode="w", shape=(3, 3))
     assert isinstance(z2, Array)
     assert isinstance(z2[:], NDArrayLike)
-    assert (z2[:] == 3).all() # type: ignore [operator]
+    assert (z2[:] == 3).all() # type: ignore [union-attr]
     z2[:] = 3
 
 

tests/test_array.py

@@ -38,6 +38,7 @@
     create_array,
 )
 from zarr.core.buffer import default_buffer_prototype
+from zarr.core.buffer.core import NDArrayLike
 from zarr.core.buffer.cpu import NDBuffer
 from zarr.core.chunk_grids import _auto_partition
 from zarr.core.common import JSON, MemoryOrder, ZarrFormat
@@ -655,6 +656,7 @@ def test_resize_1d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
     a = np.arange(105, dtype="i4")
     z[:] = a
     assert (105,) == z.shape
+    assert isinstance(z[:], NDArrayLike)
     assert (105,) == z[:].shape
     assert np.dtype("i4") == z.dtype
     assert np.dtype("i4") == z[:].dtype