Update acquire-zarr tests (#320)

Update acquire-zarr OME v0.5 fixture / aqz test to reflect new config API and downsampling behavior

Co-authored-by: Alan Liddell <aliddell@chanzuckerberg.com>

Author: ziw-liu

tests/conftest.py

@@ -184,51 +184,55 @@ def aqz_ome_zarr_05(tmpdir):
     store_path = tmpdir / "ome_zarr_v0.5.zarr"
 
     settings = aqz.StreamSettings(
-        compression=aqz.CompressionSettings(
-            codec=aqz.CompressionCodec.BLOSC_LZ4,
-            compressor=aqz.Compressor.BLOSC1,
-            level=1,
-            shuffle=0,
-        ),
-        data_type=aqz.DataType.UINT16,
-        dimensions=[
-            aqz.Dimension(
-                name="t",
-                kind=aqz.DimensionType.TIME,
-                array_size_px=0,
-                chunk_size_px=16,
-                shard_size_chunks=1,
-            ),
-            aqz.Dimension(
-                name="c",
-                kind=aqz.DimensionType.CHANNEL,
-                array_size_px=4,
-                chunk_size_px=1,
-                shard_size_chunks=1,
-            ),
-            aqz.Dimension(
-                name="z",
-                kind=aqz.DimensionType.SPACE,
-                array_size_px=10,
-                chunk_size_px=10,
-                shard_size_chunks=1,
-            ),
-            aqz.Dimension(
-                name="y",
-                kind=aqz.DimensionType.SPACE,
-                array_size_px=48,
-                chunk_size_px=16,
-                shard_size_chunks=3,
-            ),
-            aqz.Dimension(
-                name="x",
-                kind=aqz.DimensionType.SPACE,
-                array_size_px=64,
-                chunk_size_px=16,
-                shard_size_chunks=2,
-            ),
+        arrays=[
+            aqz.ArraySettings(
+                data_type=np.uint16,
+                compression=aqz.CompressionSettings(
+                    codec=aqz.CompressionCodec.BLOSC_LZ4,
+                    compressor=aqz.Compressor.BLOSC1,
+                    level=1,
+                    shuffle=0,
+                ),
+                dimensions=[
+                    aqz.Dimension(
+                        name="t",
+                        kind=aqz.DimensionType.TIME,
+                        array_size_px=0,
+                        chunk_size_px=16,
+                        shard_size_chunks=1,
+                    ),
+                    aqz.Dimension(
+                        name="c",
+                        kind=aqz.DimensionType.CHANNEL,
+                        array_size_px=4,
+                        chunk_size_px=1,
+                        shard_size_chunks=1,
+                    ),
+                    aqz.Dimension(
+                        name="z",
+                        kind=aqz.DimensionType.SPACE,
+                        array_size_px=10,
+                        chunk_size_px=10,
+                        shard_size_chunks=1,
+                    ),
+                    aqz.Dimension(
+                        name="y",
+                        kind=aqz.DimensionType.SPACE,
+                        array_size_px=48,
+                        chunk_size_px=16,
+                        shard_size_chunks=3,
+                    ),
+                    aqz.Dimension(
+                        name="x",
+                        kind=aqz.DimensionType.SPACE,
+                        array_size_px=64,
+                        chunk_size_px=16,
+                        shard_size_chunks=2,
+                    ),
+                ],
+                downsampling_method=aqz.DownsamplingMethod.MEAN,
+            )
         ],
-        multiscale=True,
         store_path=str(store_path),
         version=aqz.ZarrVersion.V3,
         max_threads=1,

tests/ngff/test_ngff.py

@@ -1337,13 +1337,16 @@ def test_acquire_zarr_ome_zarr_05(aqz_ome_zarr_05):
         assert len(dataset.zattrs["ome"]["multiscales"]) == 1
 
         multiscale = dataset.zattrs["ome"]["multiscales"][0]
-        assert len(multiscale["datasets"]) == 2
+        assert len(multiscale["datasets"]) == 3
         assert multiscale["datasets"][0]["coordinateTransformations"][0][
             "scale"
         ] == [1.0, 1.0, 1.0, 1.0, 1.0]
         assert multiscale["datasets"][1]["coordinateTransformations"][0][
             "scale"
-        ] == [1.0, 1.0, 2.0, 2.0, 2.0]
+        ] == [1.0, 1.0, 1.0, 2.0, 2.0]
+        assert multiscale["datasets"][2]["coordinateTransformations"][0][
+            "scale"
+        ] == [1.0, 1.0, 1.0, 4.0, 4.0]
         assert 1 < dataset["0"].numpy().mean() < np.iinfo(np.uint16).max