Skip to content

Commit 842bf95

Browse files
committed
Revert "fix mypy in tests"
This reverts commit 75d6cdf.
1 parent d57e1f2 commit 842bf95

File tree

2 files changed

+13
-12
lines changed

2 files changed

+13
-12
lines changed

src/zarr/core/indexing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929

3030
if TYPE_CHECKING:
3131
from zarr.core.array import Array
32+
from zarr.core.buffer import NDArrayLike
3233
from zarr.core.chunk_grids import ChunkGrid
3334
from zarr.core.common import ChunkCoords
3435

tests/test_array.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -649,15 +649,15 @@ def test_resize_1d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
649649
a = np.arange(105, dtype="i4")
650650
z[:] = a
651651
assert (105,) == z.shape
652-
assert (105,) == z[:].shape # type: ignore[union-attr]
652+
assert (105,) == z[:].shape # type: ignore
653653
assert np.dtype("i4") == z.dtype
654-
assert np.dtype("i4") == z[:].dtype # type: ignore[union-attr]
654+
assert np.dtype("i4") == z[:].dtype # type: ignore
655655
assert (10,) == z.chunks
656656
np.testing.assert_array_equal(a, z[:])
657657

658658
z.resize(205)
659659
assert (205,) == z.shape
660-
assert (205,) == z[:].shape # type: ignore[union-attr]
660+
assert (205,) == z[:].shape # type: ignore
661661
assert np.dtype("i4") == z.dtype
662662
assert np.dtype("i4") == z[:].dtype
663663
assert (10,) == z.chunks
@@ -666,17 +666,17 @@ def test_resize_1d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
666666

667667
z.resize(55)
668668
assert (55,) == z.shape
669-
assert (55,) == z[:].shape # type: ignore[union-attr]
669+
assert (55,) == z[:].shape
670670
assert np.dtype("i4") == z.dtype
671-
assert np.dtype("i4") == z[:].dtype # type: ignore[union-attr]
671+
assert np.dtype("i4") == z[:].dtype
672672
assert (10,) == z.chunks
673673
np.testing.assert_array_equal(a[:55], z[:])
674674

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

681681

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

701701
z.resize((205, 205))
702702
assert (205, 205) == z.shape
703-
assert (205, 205) == z[:].shape # type: ignore[union-attr]
703+
assert (205, 205) == z[:].shape
704704
assert np.dtype("i4") == z.dtype
705705
assert np.dtype("i4") == z[:].dtype
706706
assert (10, 10) == z.chunks
@@ -710,23 +710,23 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
710710

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

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

727727
z.resize((1, 55))
728728
assert (1, 55) == z.shape
729-
assert (1, 55) == z[:].shape # type: ignore[union-attr]
729+
assert (1, 55) == z[:].shape
730730
assert np.dtype("i4") == z.dtype
731731
assert np.dtype("i4") == z[:].dtype
732732
assert (10, 10) == z.chunks
@@ -737,7 +737,7 @@ def test_resize_2d(store: MemoryStore, zarr_format: ZarrFormat) -> None:
737737
new_shape = (105, 105)
738738
z.shape = new_shape
739739
assert new_shape == z.shape
740-
assert new_shape == z[:].shape # type: ignore[union-attr]
740+
assert new_shape == z[:].shape
741741

742742

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

0 commit comments

Comments
 (0)