Make some tests depend on eccodes version (#437)

ecmwf · Sep 2, 2024 · 973c436 · 973c436
1 parent fb8a720
commit 973c436
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/src/earthkit/data/utils/message.py b/src/earthkit/data/utils/message.py
@@ -55,6 +55,9 @@ def check_clone_kwargs(self, **kwargs):
             kwargs.pop("headers_only", None)
         return kwargs
 
+    def has_Ni_Nj_in_geo_namespace(self):
+        return self._version >= (2, 37, 0)
+
     @property
     def versions(self):
         return f"ecCodes: {self._version} eccodes-python: {self._py_version}"

diff --git a/tests/array_fieldlist/test_numpy_fs_summary.py b/tests/array_fieldlist/test_numpy_fs_summary.py
@@ -121,6 +121,12 @@ def test_array_fl_dump():
         },
     ]
 
+    from earthkit.data.utils.message import ECC_FEATURES
+
+    if not ECC_FEATURES.has_Ni_Nj_in_geo_namespace():
+        ref[1]["data"].pop("Ni")
+        ref[1]["data"].pop("Nj")
+
     assert len(r) == len(namespaces)
     assert isinstance(r, list)
     for d in r:

diff --git a/tests/grib/test_grib_summary.py b/tests/grib/test_grib_summary.py
@@ -475,6 +475,12 @@ def test_grib_dump(fl_type, array_backend):
         },
     ]
 
+    from earthkit.data.utils.message import ECC_FEATURES
+
+    if not ECC_FEATURES.has_Ni_Nj_in_geo_namespace():
+        ref[1]["data"].pop("Ni")
+        ref[1]["data"].pop("Nj")
+
     assert len(r) == len(namespaces)
     assert isinstance(r, list)
     for d in r: