Make test for tensorstore compression more robust (#574)

cubed-dev · Sep 17, 2024 · 8a406dc · 8a406dc
1 parent 4ff032b
commit 8a406dc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/cubed/tests/storage/test_zarr.py b/cubed/tests/storage/test_zarr.py
@@ -27,7 +27,11 @@ def test_lazy_zarr_array(tmp_path):
 )
 @pytest.mark.parametrize(
     "compressor",
-    [None, {"id": "zstd"}, {"id": "blosc", "cname": "lz4", "clevel": 2, "shuffle": -1}],
+    [
+        None,
+        {"id": "zstd", "level": 1},
+        {"id": "blosc", "cname": "lz4", "clevel": 2, "shuffle": -1},
+    ],
 )
 def test_compression(tmp_path, compressor):
     zarr_path = tmp_path / "lazy.zarr"

diff --git a/cubed/tests/test_core.py b/cubed/tests/test_core.py
@@ -352,7 +352,11 @@ def test_default_spec_config_override():
 
 @pytest.mark.parametrize(
     "compressor",
-    [None, {"id": "zstd"}, {"id": "blosc", "cname": "lz4", "clevel": 2, "shuffle": -1}],
+    [
+        None,
+        {"id": "zstd", "level": 1},
+        {"id": "blosc", "cname": "lz4", "clevel": 2, "shuffle": -1},
+    ],
 )
 def test_spec_compressor(tmp_path, compressor):
     spec = cubed.Spec(tmp_path, allowed_mem=100000, zarr_compressor=compressor)