enforce dtype int

Author: Carreau

zarr/indexing.py

@@ -472,7 +472,7 @@ def __iter__(self):
             yield ChunkDimProjection(dim_chunk_ix, dim_chunk_sel, dim_out_sel)
 
 
-def slice_to_range(s, l):  # noqa: E741
+def slice_to_range(s: slice, l: int):  # noqa: E741
     return range(*s.indices(l))
 
 
@@ -916,8 +916,9 @@ def __init__(self, selection, arr_shape):
 
         chunk_loc_slices = []
         last_dim_slice = None if selection[-1].step > 1 else selection.pop()
-        for arr_shape_i, sl in zip(selection, arr_shape):
+        for arr_shape_i, sl in zip(arr_shape, selection):
             dim_chunk_loc_slices = []
+            assert isinstance(sl, slice)
             for x in slice_to_range(sl, arr_shape_i):
                 dim_chunk_loc_slices.append(slice(x, x + 1, 1))
             chunk_loc_slices.append(dim_chunk_loc_slices)
@@ -928,12 +929,10 @@ def __init__(self, selection, arr_shape):
     def __iter__(self):
         chunk1 = self.chunk_loc_slices[0]
         nitems = (chunk1[-1].stop - chunk1[-1].start) * np.prod(
-            self.arr_shape[len(chunk1) :]
+            self.arr_shape[len(chunk1) :], dtype=int
         )
         for chunk_selection in self.chunk_loc_slices:
             start = 0
             for i, sl in enumerate(chunk_selection):
-                start += sl.start * np.prod(self.arr_shape[i + 1 :])
-            assert isinstance(start, int)
-            assert isinstance(nitems, int)
+                start += sl.start * np.prod(self.arr_shape[i + 1 :], dtype=int)
             yield start, nitems, chunk_selection