wavpack-numcodecs 0.3.0__tar.gz → 0.3.1__tar.gz

{wavpack_numcodecs-0.3.0 → wavpack_numcodecs-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavpack-numcodecs
-Version: 0.3.0
+Version: 0.3.1
 Summary: Numcodecs implementation of WavPack audio codec.
 Author-email: Alessio Buccino <alessiop.buccino@gmail.com>, David Bryant <david@wavpack.com>
 License: MIT

{wavpack_numcodecs-0.3.0 → wavpack_numcodecs-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wavpack-numcodecs"
-version = "0.3.0"
+version = "0.3.1"
 description = "Numcodecs implementation of WavPack audio codec."
 authors = [
     {name = "Alessio Buccino", email = "alessiop.buccino@gmail.com"},

{wavpack_numcodecs-0.3.0 → wavpack_numcodecs-0.3.1}/src/wavpack_numcodecs/wavpackv3.py

@@ -1,5 +1,5 @@
 from zarr.abc.codec import ArrayBytesCodec
-from zarr.core.buffer import Buffer, BufferPrototype
+from zarr.core.buffer import Buffer, BufferPrototype, NDBuffer
 from zarr.core.common import BytesLike
 from wavpack_numcodecs.wavpack import WavPack as WavPackV2
 import numpy as np
@@ -53,7 +53,7 @@ class WavPack(ArrayBytesCodec):
         self,
         chunk_bytes: Buffer,
         chunk_spec: "ArraySpec",
-    ) -> np.ndarray:
+    ) -> NDBuffer:
         """Decode a single chunk."""
         decoded = await asyncio.to_thread(self._codec.decode, chunk_bytes.to_bytes())
 
@@ -64,7 +64,8 @@ class WavPack(ArrayBytesCodec):
 
         # Ensure it's a numpy array with correct shape
         if isinstance(decoded, np.ndarray):
-            return decoded.reshape(chunk_spec.shape)
+            # zarr merges partial chunk writes into an NDBuffer, not a plain ndarray
+            return chunk_spec.prototype.nd_buffer.from_ndarray_like(decoded.reshape(chunk_spec.shape))
         else:
             raise TypeError(f"Expected numpy array from decode, got {type(decoded)}")
 

{wavpack_numcodecs-0.3.0 → wavpack_numcodecs-0.3.1}/src/wavpack_numcodecs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavpack-numcodecs
-Version: 0.3.0
+Version: 0.3.1
 Summary: Numcodecs implementation of WavPack audio codec.
 Author-email: Alessio Buccino <alessiop.buccino@gmail.com>, David Bryant <david@wavpack.com>
 License: MIT

{wavpack_numcodecs-0.3.0 → wavpack_numcodecs-0.3.1}/tests/test_wavpack_codec.py

@@ -301,6 +301,26 @@ def test_wavpack_zarr_v3(generate_test_data, bps, dtype):
                 np.testing.assert_array_equal(z[:], test_sig)
 
 
+
+@pytest.mark.parametrize("shards", [None, (300, 4)])
+@pytest.mark.skipif(not ZARR_V3, reason="These are tests for Zarr V3")
+@pytest.mark.zarr
+def test_wavpack_zarr_v3_partial_write(shards):
+    # partial writes into an existing chunk decode it and merge the new values
+    z = zarr.create_array(
+        store=zarr.storage.MemoryStore(),
+        shape=(250, 4),
+        chunks=(100, 4),
+        shards=shards,
+        dtype="int16",
+        serializer=WavPack(),
+    )
+    expected = np.zeros((250, 4), dtype="int16")
+    for start, stop, value in [(0, 100, 1), (100, 150, 2), (150, 250, 3), (120, 130, 4)]:
+        z[start:stop] = value
+        expected[start:stop] = value
+    np.testing.assert_array_equal(z[:], expected)
+
 if __name__ == "__main__":
     test_wavpack_numcodecs()
     test_wavpack_multi_threading_enabled()