fix earthkit error in tests

ecmwf · Nov 22, 2024 · d6eb20b · d6eb20b
1 parent c253d0f
commit d6eb20b
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -25,3 +25,4 @@ polytope_venv_latest
 new_updated_numpy_venv
 newest-polytope-venv
 serializedTree
+new_polytope_venv
diff --git a/tests/test_datacube_axes_init.py b/tests/test_datacube_axes_init.py
@@ -14,8 +14,7 @@ def setup_method(self, method):
         download_test_data(nexus_url, "foo.grib")
 
         ds = data.from_source("file", "./tests/data/foo.grib")
-        # latlon_array = ds.to_xarray().isel(step=0).isel(number=0).isel(surface=0).isel(time=0)
-        latlon_array = ds.to_xarray()
+        latlon_array = ds.to_xarray(engine="cfgrib").isel(step=0).isel(number=0).isel(surface=0).isel(time=0)
         latlon_array = latlon_array.t2m
         self.options = {
             "axis_config": [

diff --git a/tests/test_healpix_mapper.py b/tests/test_healpix_mapper.py
@@ -16,7 +16,7 @@ def setup_method(self, method):
         download_test_data(nexus_url, "healpix.grib")
 
         ds = data.from_source("file", "./tests/data/healpix.grib")
-        self.latlon_array = ds.to_xarray().isel(step=0).isel(time=0).isel(isobaricInhPa=0).z
+        self.latlon_array = ds.to_xarray(engine="cfgrib").isel(step=0).isel(time=0).isel(isobaricInhPa=0).z
         self.options = {
             "axis_config": [
                 {

diff --git a/tests/test_merge_octahedral_one_axis.py b/tests/test_merge_octahedral_one_axis.py
@@ -13,7 +13,7 @@ def setup_method(self, method):
         download_test_data(nexus_url, "foo.grib")
 
         ds = data.from_source("file", "./tests/data/foo.grib")
-        self.latlon_array = ds.to_xarray().isel(step=0).isel(number=0).isel(surface=0).isel(time=0)
+        self.latlon_array = ds.to_xarray(engine="cfgrib").isel(step=0).isel(number=0).isel(surface=0).isel(time=0)
         self.latlon_array = self.latlon_array.t2m
         self.options = {
             "axis_config": [

diff --git a/tests/test_octahedral_grid.py b/tests/test_octahedral_grid.py
@@ -13,7 +13,7 @@ def setup_method(self, method):
         download_test_data(nexus_url, "foo.grib")
 
         ds = data.from_source("file", "./tests/data/foo.grib")
-        self.latlon_array = ds.to_xarray().isel(step=0).isel(number=0).isel(surface=0).isel(time=0)
+        self.latlon_array = ds.to_xarray(engine="cfgrib").isel(step=0).isel(number=0).isel(surface=0).isel(time=0)
         self.latlon_array = self.latlon_array.t2m
         self.options = {
             "axis_config": [

diff --git a/tests/test_slicer_era5.py b/tests/test_slicer_era5.py
@@ -14,7 +14,7 @@ def setup_method(self, method):
         download_test_data(nexus_url, "era5-levels-members.grib")
 
         ds = data.from_source("file", "./tests/data/era5-levels-members.grib")
-        array = ds.to_xarray().isel(step=0).t
+        array = ds.to_xarray(engine="cfgrib").isel(step=0).t
         self.slicer = HullSlicer()
         options = {
             "axis_config": [{"axis_name": "latitude", "transformations": [{"name": "reverse", "is_reverse": True}]}],

diff --git a/tests/test_snapping_real_data.py b/tests/test_snapping_real_data.py
@@ -17,7 +17,7 @@ def setup_method(self, method):
         download_test_data(nexus_url, "era5-levels-members.grib")
 
         ds = data.from_source("file", "./tests/data/era5-levels-members.grib")
-        array = ds.to_xarray().isel(step=0).t
+        array = ds.to_xarray(engine="cfgrib").isel(step=0).t
         self.slicer = HullSlicer()
         options = {
             "axis_config": [