Skip to content

Commit 17d4a9b

Browse files
committed
fix mypy in test_array.py
1 parent 363f06b commit 17d4a9b

File tree

1 file changed

+26
-14
lines changed

1 file changed

+26
-14
lines changed

tests/test_array.py

Lines changed: 26 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -706,53 +706,65 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
706706
)
707707
a = np.arange(105 * 105, dtype="i4").reshape((105, 105))
708708
z[:] = a
709+
result = z[:]
710+
assert isinstance(result, NDArrayLike)
709711
assert (105, 105) == z.shape
710-
assert (105, 105) == z[:].shape
712+
assert (105, 105) == result.shape
711713
assert np.dtype("i4") == z.dtype
712-
assert np.dtype("i4") == z[:].dtype
714+
assert np.dtype("i4") == result.dtype
713715
assert (10, 10) == z.chunks
714-
np.testing.assert_array_equal(a, z[:])
716+
np.testing.assert_array_equal(a, result)
715717

716718
z.resize((205, 205))
719+
result = z[:]
720+
assert isinstance(result, NDArrayLike)
717721
assert (205, 205) == z.shape
718-
assert (205, 205) == z[:].shape
722+
assert (205, 205) == result.shape
719723
assert np.dtype("i4") == z.dtype
720-
assert np.dtype("i4") == z[:].dtype
724+
assert np.dtype("i4") == result.dtype
721725
assert (10, 10) == z.chunks
722726
np.testing.assert_array_equal(a, z[:105, :105])
723727
np.testing.assert_array_equal(np.zeros((100, 205), dtype="i4"), z[105:, :])
724728
np.testing.assert_array_equal(np.zeros((205, 100), dtype="i4"), z[:, 105:])
725729

726730
z.resize((55, 55))
731+
result = z[:]
732+
assert isinstance(result, NDArrayLike)
727733
assert (55, 55) == z.shape
728-
assert (55, 55) == z[:].shape
734+
assert (55, 55) == result.shape
729735
assert np.dtype("i4") == z.dtype
730-
assert np.dtype("i4") == z[:].dtype
736+
assert np.dtype("i4") == result.dtype
731737
assert (10, 10) == z.chunks
732-
np.testing.assert_array_equal(a[:55, :55], z[:])
738+
np.testing.assert_array_equal(a[:55, :55], result)
733739

734740
z.resize((55, 1))
741+
result = z[:]
742+
assert isinstance(result, NDArrayLike)
735743
assert (55, 1) == z.shape
736-
assert (55, 1) == z[:].shape
744+
assert (55, 1) == result.shape
737745
assert np.dtype("i4") == z.dtype
738-
assert np.dtype("i4") == z[:].dtype
746+
assert np.dtype("i4") == result.dtype
739747
assert (10, 10) == z.chunks
740-
np.testing.assert_array_equal(a[:55, :1], z[:])
748+
np.testing.assert_array_equal(a[:55, :1], result)
741749

742750
z.resize((1, 55))
751+
result = z[:]
752+
assert isinstance(result, NDArrayLike)
743753
assert (1, 55) == z.shape
744-
assert (1, 55) == z[:].shape
754+
assert (1, 55) == result.shape
745755
assert np.dtype("i4") == z.dtype
746-
assert np.dtype("i4") == z[:].dtype
756+
assert np.dtype("i4") == result.dtype
747757
assert (10, 10) == z.chunks
748758
np.testing.assert_array_equal(a[:1, :10], z[:, :10])
749759
np.testing.assert_array_equal(np.zeros((1, 55 - 10), dtype="i4"), z[:, 10:55])
750760

751761
# via shape setter
752762
new_shape = (105, 105)
753763
z.shape = new_shape
764+
result = z[:]
765+
assert isinstance(result, NDArrayLike)
754766
assert new_shape == z.shape
755-
assert new_shape == z[:].shape
767+
assert new_shape == result.shape
756768

757769

758770
@pytest.mark.parametrize("store", ["memory"], indirect=True)

0 commit comments

Comments
 (0)