From 8cbf4e1543c0118e97dcfdc72d64e258f73ebb51 Mon Sep 17 00:00:00 2001
From: Norman Rzepka <code@normanrz.com>
Date: Tue, 1 Aug 2023 17:17:37 +0200
Subject: [PATCH] fixes transpose codec

---
 tests/test_v3.py  | 8 ++++----
 zarrita/codecs.py | 4 +++-
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/tests/test_v3.py b/tests/test_v3.py
index dd518d7..43866b1 100644
--- a/tests/test_v3.py
+++ b/tests/test_v3.py
@@ -185,12 +185,12 @@ async def test_order(
     runtime_read_order: Literal["F", "C"],
     with_sharding: bool,
 ):
-    data = np.arange(0, 256, dtype="uint16").reshape((16, 16), order=input_order)
+    data = np.arange(0, 256, dtype="uint16").reshape((32, 8), order=input_order)
 
     codecs_: List[CodecMetadata] = (
         [
             codecs.sharding_codec(
-                (8, 8),
+                (16, 8),
                 codecs=[codecs.transpose_codec(store_order), codecs.endian_codec()],
             )
         ]
@@ -201,7 +201,7 @@ async def test_order(
     a = await Array.create_async(
         store / "order",
         shape=data.shape,
-        chunk_shape=(16, 16),
+        chunk_shape=(32, 8),
         dtype=data.dtype,
         fill_value=0,
         chunk_key_encoding=("v2", "."),
@@ -231,7 +231,7 @@ async def test_order(
         # Compare with zarr-python
         z = zarr.create(
             shape=data.shape,
-            chunks=(16, 16),
+            chunks=(32, 8),
             dtype="<u2",
             order=store_order,
             compressor=None,
diff --git a/zarrita/codecs.py b/zarrita/codecs.py
index dacbbb4..9bbec69 100644
--- a/zarrita/codecs.py
+++ b/zarrita/codecs.py
@@ -372,7 +372,9 @@ async def decode(
         if isinstance(new_order, tuple):
             chunk_array = chunk_array.transpose(new_order)
         elif new_order == "F":
-            chunk_array = chunk_array.T
+            chunk_array = chunk_array.ravel().reshape(
+                self.array_metadata.chunk_shape, order="F"
+            )
         return chunk_array
 
     async def encode(