skip morton indices out of bound

Author: brokkoli71

src/zarr/core/indexing.py

@@ -1343,9 +1343,15 @@ def decode_morton(z: int, chunk_shape: ChunkCoords) -> ChunkCoords:
 
 
 def morton_order_iter(chunk_shape: ChunkCoords) -> Iterator[ChunkCoords]:
-    for i in range(product(chunk_shape)):
-        yield decode_morton(i, chunk_shape)
-
+    i = 0
+    order = []
+    while len(order)<product(chunk_shape):
+        m = decode_morton(i, chunk_shape)
+        if m not in order and all(x < y for x,y in zip(m, chunk_shape, strict=False)):
+            order.append(m)
+        i+=1
+    for j in range(product(chunk_shape)):
+        yield order[j]
 
 def c_order_iter(chunks_per_shard: ChunkCoords) -> Iterator[ChunkCoords]:
     return itertools.product(*(range(x) for x in chunks_per_shard))

tests/v3/test_codecs/test_codecs.py

@@ -211,8 +211,8 @@ def test_morton() -> None:
         [2, 2, 2],
         [5, 2],
         [2, 5],
-        [2, 2, 2, 2],
-        [1, 2, 0],
+        [2, 9, 2],
+        [3, 2, 12],
         [2, 5, 1],
         [4, 3, 6, 2, 7],
         [3, 2, 1, 6, 4, 5, 2],