test coverage

Author: TomAugspurger

src/zarr/codecs/gpu.py

@@ -72,8 +72,6 @@ def to_dict(self) -> dict[str, JSON]:
 
     @cached_property
     def _zstd_codec(self) -> nvcomp.Codec:
-        # config_dict = {algorithm = "Zstd", "level": self.level, "checksum": self.checksum}
-        # return Zstd.from_config(config_dict)
         device = cp.cuda.Device()  # Select the current default device
         stream = cp.cuda.get_current_stream()  # Use the current default stream
         return nvcomp.Codec(

tests/test_codecs/test_nvcomp.py

@@ -32,3 +32,14 @@ def test_nvcomp_zstd(store: Store, checksum: bool) -> None:
 
         a[:, :] = data
         cp.testing.assert_array_equal(data, a[:, :])
+
+
+def test_invalid_raises() -> None:
+    with pytest.raises(ValueError):
+        NvcompZstdCodec(level=100, checksum=False)
+
+    with pytest.raises(TypeError):
+        NvcompZstdCodec(level="100", checksum=False)  # type: ignore[arg-type]
+
+    with pytest.raises(TypeError):
+        NvcompZstdCodec(checksum="False")  # type: ignore[arg-type]