fix tests

Author: d-v-b

src/zarr/codecs/_numcodecs.py

@@ -40,7 +40,7 @@
 from zarr.core.common import JSON, parse_named_configuration, product
 from zarr.dtype import UInt8, ZDType, parse_dtype
 from zarr.errors import ZarrUserWarning
-from zarr.registry import get_numcodec
+from zarr.registry import get_numcodec, register_codec
 
 if TYPE_CHECKING:
     from zarr.abc.numcodec import Numcodec
@@ -326,6 +326,29 @@ class ZFPY(_NumcodecsArrayBytesCodec, codec_name="zfpy"):
     pass
 
 
+# TODO: move the codec registration outside this module
+register_codec("numcodecs.bz2", BZ2)
+register_codec("numcodecs.crc32", CRC32)
+register_codec("numcodecs.crc32c", CRC32C)
+register_codec("numcodecs.lz4", LZ4)
+register_codec("numcodecs.lzma", LZMA)
+register_codec("numcodecs.zfpy", ZFPY)
+register_codec("numcodecs.adler32", Adler32)
+register_codec("numcodecs.astype", AsType)
+register_codec("numcodecs.bitround", BitRound)
+register_codec("numcodecs.blosc", Blosc)
+register_codec("numcodecs.delta", Delta)
+register_codec("numcodecs.fixedscaleoffset", FixedScaleOffset)
+register_codec("numcodecs.fletcher32", Fletcher32)
+register_codec("numcodecs.gzip", GZip)
+register_codec("numcodecs.jenkins_lookup3", JenkinsLookup3)
+register_codec("numcodecs.pcodec", PCodec)
+register_codec("numcodecs.packbits", PackBits)
+register_codec("numcodecs.quantize", Quantize)
+register_codec("numcodecs.shuffle", Shuffle)
+register_codec("numcodecs.zlib", Zlib)
+register_codec("numcodecs.zstd", Zstd)
+
 __all__ = [
     "BZ2",
     "CRC32",

src/zarr/core/config.py

@@ -125,6 +125,27 @@ def enable_gpu(self) -> ConfigSet:
                 "transpose": "zarr.codecs.transpose.TransposeCodec",
                 "vlen-utf8": "zarr.codecs.vlen_utf8.VLenUTF8Codec",
                 "vlen-bytes": "zarr.codecs.vlen_utf8.VLenBytesCodec",
+                "numcodecs.bz2": "zarr.codecs._numcodecs.BZ2",
+                "numcodecs.crc32": "zarr.codecs._numcodecs.CRC32",
+                "numcodecs.crc32c": "zarr.codecs._numcodecs.CRC32C",
+                "numcodecs.lz4": "zarr.codecs._numcodecs.LZ4",
+                "numcodecs.lzma": "zarr.codecs._numcodecs.LZMA",
+                "numcodecs.zfpy": "zarr.codecs._numcodecs.ZFPY",
+                "numcodecs.adler32": "zarr.codecs._numcodecs.Adler32",
+                "numcodecs.astype": "zarr.codecs._numcodecs.AsType",
+                "numcodecs.bitround": "zarr.codecs._numcodecs.BitRound",
+                "numcodecs.blosc": "zarr.codecs._numcodecs.Blosc",
+                "numcodecs.delta": "zarr.codecs._numcodecs.Delta",
+                "numcodecs.fixedscaleoffset": "zarr.codecs._numcodecs.FixedScaleOffset",
+                "numcodecs.fletcher32": "zarr.codecs._numcodecs.Fletcher32",
+                "numcodecs.gZip": "zarr.codecs._numcodecs.GZip",
+                "numcodecs.jenkins_lookup3": "zarr.codecs._numcodecs.JenkinsLookup3",
+                "numcodecs.pcodec": "zarr.codecs._numcodecs.PCodec",
+                "numcodecs.packbits": "zarr.codecs._numcodecs.PackBits",
+                "numcodecs.shuffle": "zarr.codecs._numcodecs.Shuffle",
+                "numcodecs.quantize": "zarr.codecs._numcodecs.Quantize",
+                "numcodecs.zlib": "zarr.codecs._numcodecs.Zlib",
+                "numcodecs.zstd": "zarr.codecs._numcodecs.Zstd",
             },
             "buffer": "zarr.buffer.cpu.Buffer",
             "ndbuffer": "zarr.buffer.cpu.NDBuffer",

src/zarr/registry.py

@@ -155,7 +155,6 @@ def get_codec_class(key: str, reload_config: bool = False) -> type[Codec]:
     codec_classes = __codec_registries[key]
     if not codec_classes:
         raise KeyError(key)
-
     config_entry = config.get("codecs", {}).get(key)
     if config_entry is None:
         if len(codec_classes) == 1:

tests/test_array.py

@@ -1713,7 +1713,7 @@ def test_roundtrip_numcodecs() -> None:
     # Create the array with the correct codecs
     root = zarr.group(store)
     warn_msg = "Numcodecs codecs are not in the Zarr version 3 specification and may not be supported by other zarr implementations."
-    with pytest.warns(UserWarning, match=warn_msg):
+    with pytest.warns(ZarrUserWarning, match=warn_msg):
         root.create_array(
             "test",
             shape=(720, 1440),
@@ -1728,7 +1728,7 @@ def test_roundtrip_numcodecs() -> None:
     BYTES_CODEC = {"name": "bytes", "configuration": {"endian": "little"}}
     # Read in the array again and check compressor config
     root = zarr.open_group(store)
-    with pytest.warns(UserWarning, match=warn_msg):
+    with pytest.warns(ZarrUserWarning, match=warn_msg):
         metadata = root["test"].metadata.to_dict()
     expected = (*filters, BYTES_CODEC, *compressors)
     assert metadata["codecs"] == expected