test: update zarr v3 conversion expectations

Author: wietzesuijker

tests/test_s2_multiscale.py

@@ -1,15 +1,13 @@
 """
 Tests for S2 multiscale pyramid creation with xy-aligned sharding.
 """
-
-from typing import TypedDict
-
 import numpy as np
 import pytest
 import xarray as xr
 import zarr
 from pydantic_zarr.experimental.v3 import ArraySpec, GroupSpec
 from structlog.testing import capture_logs
+from typing_extensions import TypedDict
 
 from eopf_geozarr.s2_optimization.s2_multiscale import (
     calculate_aligned_chunk_size,

tests/test_zarrio.py

@@ -256,5 +256,22 @@ def custom_array_encoder(key: str, metadata: ArrayV2Metadata) -> ArrayV3Metadata
     assert new_node.attrs.asdict() == old_node.attrs.asdict()
 
 
-if __name__ == "__main__":
-    pytest.main([__file__])
+def test_reencode_group_omit_nodes_normalizes_leading_slash() -> None:
+    group_v2 = zarr.create_group({}, zarr_format=2)
+    group_v2.create_array("data", shape=(5,), dtype="int32")
+
+    group_v3 = reencode_group(group_v2, {}, "", omit_nodes=["/data"])
+
+    assert "data" not in group_v3
+
+
+def test_reencode_group_omit_nodes_is_prefix_safe() -> None:
+    group_v2 = zarr.create_group({}, zarr_format=2)
+    group_v2.create_array("foobar", shape=(5,), dtype="int32")
+    group_v2.create_group("foo").create_array("child", shape=(5,), dtype="int32")
+
+    group_v3 = reencode_group(group_v2, {}, "", omit_nodes=["foo"])
+
+    assert "foo" not in group_v3
+    assert "foo/child" not in group_v3
+    assert "foobar" in group_v3