Skip to content

Commit 75d6cdf

Browse files
committed
fix mypy in tests
1 parent 34bf260 commit 75d6cdf

File tree

2 files changed

+12
-13
lines changed

2 files changed

+12
-13
lines changed

src/zarr/core/indexing.py

-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@
2929

3030
if TYPE_CHECKING:
3131
from zarr.core.array import Array
32-
from zarr.core.buffer import NDArrayLike
3332
from zarr.core.chunk_grids import ChunkGrid
3433
from zarr.core.common import ChunkCoords
3534

tests/test_array.py

+12-12
Original file line numberDiff line numberDiff line change
@@ -648,15 +648,15 @@ def test_resize_1d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
648648
a = np.arange(105, dtype="i4")
649649
z[:] = a
650650
assert (105,) == z.shape
651-
assert (105,) == z[:].shape # type: ignore
651+
assert (105,) == z[:].shape # type: ignore[union-attr]
652652
assert np.dtype("i4") == z.dtype
653-
assert np.dtype("i4") == z[:].dtype # type: ignore
653+
assert np.dtype("i4") == z[:].dtype # type: ignore[union-attr]
654654
assert (10,) == z.chunks
655655
np.testing.assert_array_equal(a, z[:])
656656

657657
z.resize(205)
658658
assert (205,) == z.shape
659-
assert (205,) == z[:].shape # type: ignore
659+
assert (205,) == z[:].shape # type: ignore[union-attr]
660660
assert np.dtype("i4") == z.dtype
661661
assert np.dtype("i4") == z[:].dtype
662662
assert (10,) == z.chunks
@@ -665,17 +665,17 @@ def test_resize_1d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
665665

666666
z.resize(55)
667667
assert (55,) == z.shape
668-
assert (55,) == z[:].shape
668+
assert (55,) == z[:].shape # type: ignore[union-attr]
669669
assert np.dtype("i4") == z.dtype
670-
assert np.dtype("i4") == z[:].dtype
670+
assert np.dtype("i4") == z[:].dtype # type: ignore[union-attr]
671671
assert (10,) == z.chunks
672672
np.testing.assert_array_equal(a[:55], z[:])
673673

674674
# via shape setter
675675
new_shape = (105,)
676676
z.shape = new_shape
677677
assert new_shape == z.shape
678-
assert new_shape == z[:].shape
678+
assert new_shape == z[:].shape # type: ignore[union-attr]
679679

680680

681681
@pytest.mark.parametrize("store", ["memory"], indirect=True)
@@ -691,15 +691,15 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
691691
a = np.arange(105 * 105, dtype="i4").reshape((105, 105))
692692
z[:] = a
693693
assert (105, 105) == z.shape
694-
assert (105, 105) == z[:].shape
694+
assert (105, 105) == z[:].shape # type: ignore[union-attr]
695695
assert np.dtype("i4") == z.dtype
696696
assert np.dtype("i4") == z[:].dtype
697697
assert (10, 10) == z.chunks
698698
np.testing.assert_array_equal(a, z[:])
699699

700700
z.resize((205, 205))
701701
assert (205, 205) == z.shape
702-
assert (205, 205) == z[:].shape
702+
assert (205, 205) == z[:].shape # type: ignore[union-attr]
703703
assert np.dtype("i4") == z.dtype
704704
assert np.dtype("i4") == z[:].dtype
705705
assert (10, 10) == z.chunks
@@ -709,23 +709,23 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
709709

710710
z.resize((55, 55))
711711
assert (55, 55) == z.shape
712-
assert (55, 55) == z[:].shape
712+
assert (55, 55) == z[:].shape # type: ignore[union-attr]
713713
assert np.dtype("i4") == z.dtype
714714
assert np.dtype("i4") == z[:].dtype
715715
assert (10, 10) == z.chunks
716716
np.testing.assert_array_equal(a[:55, :55], z[:])
717717

718718
z.resize((55, 1))
719719
assert (55, 1) == z.shape
720-
assert (55, 1) == z[:].shape
720+
assert (55, 1) == z[:].shape # type: ignore[union-attr]
721721
assert np.dtype("i4") == z.dtype
722722
assert np.dtype("i4") == z[:].dtype
723723
assert (10, 10) == z.chunks
724724
np.testing.assert_array_equal(a[:55, :1], z[:])
725725

726726
z.resize((1, 55))
727727
assert (1, 55) == z.shape
728-
assert (1, 55) == z[:].shape
728+
assert (1, 55) == z[:].shape # type: ignore[union-attr]
729729
assert np.dtype("i4") == z.dtype
730730
assert np.dtype("i4") == z[:].dtype
731731
assert (10, 10) == z.chunks
@@ -736,7 +736,7 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
736736
new_shape = (105, 105)
737737
z.shape = new_shape
738738
assert new_shape == z.shape
739-
assert new_shape == z[:].shape
739+
assert new_shape == z[:].shape # type: ignore[union-attr]
740740

741741

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

0 commit comments

Comments
 (0)