add zstd to ZarrPythonTests.testWriteV2

Author: brokkoli71

src/main/java/dev/zarr/zarrjava/v2/ArrayMetadataBuilder.java

@@ -8,6 +8,7 @@
 import dev.zarr.zarrjava.v2.codec.Codec;
 import dev.zarr.zarrjava.v2.codec.core.BloscCodec;
 import dev.zarr.zarrjava.v2.codec.core.ZlibCodec;
+import dev.zarr.zarrjava.v2.codec.core.ZstdCodec;
 
 public class ArrayMetadataBuilder {
     long[] shape = null;
@@ -129,6 +130,23 @@ public ArrayMetadataBuilder withZlibCompressor() {
         return withZlibCompressor(5);
     }
 
+    public ArrayMetadataBuilder withZstdCompressor(int level, boolean checksum) {
+        try {
+            this.compressor = new ZstdCodec(level, checksum);
+        } catch (ZarrException e) {
+            throw new RuntimeException(e);
+        }
+        return this;
+    }
+
+    public  ArrayMetadataBuilder withZstdCompressor(int level) {
+        return withZstdCompressor(level, ZstdCodec.DEFAULT_CHECKSUM);
+    }
+
+    public  ArrayMetadataBuilder withZstdCompressor() {
+        return withZstdCompressor(ZstdCodec.DEFAULT_LEVEL);
+    }
+
     public ArrayMetadataBuilder putAttribute(String key, Object value) {
         this.attributes.put(key, value);
         return this;

src/main/java/dev/zarr/zarrjava/v2/codec/core/ZstdCodec.java

@@ -11,15 +11,17 @@
 
 public class ZstdCodec extends dev.zarr.zarrjava.core.codec.core.ZstdCodec implements Codec {
 
+    public static final int DEFAULT_LEVEL = 0;
+    public static final boolean DEFAULT_CHECKSUM = false;
     @JsonIgnore
     public final String id = "zstd";
     public final int level;
     public final boolean checksum;
 
     @JsonCreator(mode = JsonCreator.Mode.PROPERTIES)
     public ZstdCodec(
-            @JsonProperty(value = "level", defaultValue = "0") int level,
-            @JsonProperty(value = "checksum", defaultValue = "false") boolean checksum) throws ZarrException {
+            @JsonProperty(value = "level", defaultValue = "" + DEFAULT_LEVEL) int level,
+            @JsonProperty(value = "checksum", defaultValue = "" + DEFAULT_CHECKSUM) boolean checksum) throws ZarrException {
         if (level < -131072 || level > 22) {
             throw new ZarrException("'level' needs to be between -131072 and 22.");
         }

src/test/java/dev/zarr/zarrjava/ZarrPythonTests.java

@@ -244,6 +244,9 @@ public void testWriteV2(String compressor, String compressorParam, dev.zarr.zarr
             case "zlib":
                 builder = builder.withZlibCompressor(Integer.parseInt(compressorParam));
                 break;
+            case "zstd":
+                builder = builder.withZstdCompressor(Integer.parseInt(compressorParam.split("_")[0]), Boolean.parseBoolean(compressorParam.split("_")[1]));
+                break;
             default:
                 throw new IllegalArgumentException("Invalid compressor: " + compressor);
         }