Adds astype config for Delta in Zarr 3 wrapper (#664)

* fixes #663

* changelog

Author: normanrz

docs/release.rst

@@ -13,6 +13,19 @@ Release notes
 
 .. _unreleased:
 
+Unreleased
+----------
+
+Fixes
+~~~~~
+* Fixes issue with ``Delta`` Zarr 3 codec not working with ``astype``.
+  By :user:`Norman Rzepka <normanrz>`, :issue:`664`
+
+
+Improvements
+~~~~~~~~~~~~
+
+
 0.14.1
 ------
 

docs/zarr3.rst

@@ -1,3 +1,5 @@
+.. _Zarr 3 codecs:
+
 Zarr 3 codecs
 =============
 .. automodule:: numcodecs.zarr3

numcodecs/tests/test_zarr3.py

@@ -244,3 +244,24 @@ def test_generic_bytes_codec(store: StorePath, codec_class: type[numcodecs.zarr3
 
     a[:, :] = data.copy()
     np.testing.assert_array_equal(data, a[:, :])
+
+
+def test_delta_astype(store: StorePath):
+    data = np.linspace(0, 10, 256, dtype="i8").reshape((16, 16))
+
+    with pytest.warns(UserWarning, match=EXPECTED_WARNING_STR):
+        a = Array.create(
+            store / "generic",
+            shape=data.shape,
+            chunk_shape=(16, 16),
+            dtype=data.dtype,
+            fill_value=0,
+            codecs=[
+                numcodecs.zarr3.Delta(dtype="i8", astype="i2"),  # type: ignore[arg-type]
+                BytesCodec(),
+            ],
+        )
+
+        a[:, :] = data.copy()
+        a = Array.open(store / "generic")
+    np.testing.assert_array_equal(data, a[:, :])

numcodecs/zarr3.py

@@ -266,7 +266,19 @@ def evolve_from_array_spec(self, array_spec: ArraySpec) -> Shuffle:
 
 
 # array-to-array codecs ("filters")
-Delta = _add_docstring(_make_array_array_codec("delta", "Delta"), "numcodecs.delta.Delta")
+@_add_docstring_wrapper("numcodecs.delta.Delta")
+class Delta(_NumcodecsArrayArrayCodec):
+    codec_name = f"{CODEC_PREFIX}delta"
+
+    def __init__(self, **codec_config: dict[str, JSON]) -> None:
+        super().__init__(**codec_config)
+
+    def resolve_metadata(self, chunk_spec: ArraySpec) -> ArraySpec:
+        if astype := self.codec_config.get("astype"):
+            return replace(chunk_spec, dtype=np.dtype(astype))  # type: ignore[arg-type]
+        return chunk_spec
+
+
 BitRound = _add_docstring(
     _make_array_array_codec("bitround", "BitRound"), "numcodecs.bitround.BitRound"
 )