fix open-meteo kerchunk test

Author: terraputix

kerchunk/open_meteo.py

@@ -23,6 +23,7 @@
 
 try:
     import omfiles
+    import omfiles._numcodecs
 except ModuleNotFoundError:  # pragma: no cover
     raise ImportError(
         "omfiles is required for kerchunking Open-Meteo files. Please install with "
@@ -191,10 +192,10 @@ def __init__(
             fs, path = fsspec.core.url_to_fs(om_file, **(storage_options or {}))
             self.input_file = fs.open(path, "rb")
             url = om_file
-            self.reader = omfiles.OmFilePyReader(self.input_file)
+            self.reader = omfiles.OmFileReader(self.input_file)
         elif isinstance(om_file, io.IOBase):
             self.input_file = om_file
-            self.reader = omfiles.OmFilePyReader(self.input_file)
+            self.reader = omfiles.OmFileReader(self.input_file)
         else:
             raise ValueError("type of input `om_file` not recognized")
 
@@ -219,7 +220,7 @@ def translate(self):
         # 1. Extract metadata about shape, dtype, chunks, etc.
         shape = self.reader.shape
         dtype = self.reader.dtype
-        chunks = self.reader.chunk_dimensions
+        chunks = self.reader.chunks
         scale_factor = self.reader.scale_factor
         add_offset = self.reader.add_offset
         lut = self.reader.get_complete_lut()

pyproject.toml

@@ -50,7 +50,7 @@ dev = [
     "scipy",
     "netcdf4",
     "pytest-subtests",
-    "omfiles @ git+https://github.com/open-meteo/python-omfiles.git@codecs"
+    "omfiles @ git+https://github.com/open-meteo/python-omfiles.git@get-complete-lut"
 ]
 
 [project.urls]

tests/test_open_meteo.py

@@ -48,10 +48,10 @@ def test_single_om_to_zarr():
     print("z.chunks", z.chunks)
 
     # Verify basic metadata matches original file
-    reader = omfiles.OmFilePyReader(test_file)
-    assert list(z.shape) == reader.shape, f"Shape mismatch: {z.shape} vs {reader.shape}"
+    reader = omfiles.OmFileReader(test_file)
+    assert z.shape == reader.shape, f"Shape mismatch: {z.shape} vs {reader.shape}"
     assert str(z.dtype) == str(reader.dtype), f"Dtype mismatch: {z.dtype} vs {reader.dtype}"
-    assert list(z.chunks) == reader.chunk_dimensions, f"Chunks mismatch: {z.chunks} vs {reader.chunk_dimensions}"
+    assert z.chunks == reader.chunks, f"Chunks mismatch: {z.chunks} vs {reader.chunks}"
 
     # TODO: Using the following chunk_index leads to a double free / corruption error!
     # Even with a concurrency of 1: `zarr.config.config["async"]["concurrency"] = 1`
@@ -127,8 +127,8 @@ def test_multizarr_to_zarr():
     z = group["data"]
 
     # Open both original files for comparison
-    reader1 = omfiles.OmFilePyReader(file1)
-    reader2 = omfiles.OmFilePyReader(file2)
+    reader1 = omfiles.OmFileReader(file1)
+    reader2 = omfiles.OmFileReader(file2)
 
     # Check that the combined shape is the sum along the time axis
     expected_shape = list(reader1.shape)
@@ -165,7 +165,7 @@ def test_multizarr_to_zarr():
 #         ds = xr.open_zarr(store, consolidated=False)
 
 #         # Basic validation
-#         reader = omfiles.OmFilePyReader(test_file)
+#         reader = omfiles.OmFileReader(test_file)
 #         assert ds.dims == dict(zip(["time", "y", "x"], reader.shape))
 
 #         # Get some data to verify decompression pipeline