Add grib to geotiff test

ecmwf · Feb 21, 2025 · adfbe31 · adfbe31
1 parent 5f591e9
commit adfbe31
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/src/earthkit/data/encoders/geotiff.py b/src/earthkit/data/encoders/geotiff.py
@@ -16,7 +16,6 @@
 
 
 class GeoTIFFEncodedData(EncodedData):
-
     _GDAL_DRIVER = "GTiff"
 
     def __init__(self, ds):
@@ -42,7 +41,6 @@ def metadata(self, key):
 
 
 class GeoTIFFEncoder(Encoder):
-
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 

diff --git a/tests/targets/test_target_file.py b/tests/targets/test_target_file.py
@@ -242,6 +242,23 @@ def test_target_file_grib_to_netcdf():
         # assert np.allclose(ds1.values[:, :4], vals_ref)
 
 
+@pytest.mark.skipif(NO_RIOXARRAY, reason="rioxarray not available")
+@pytest.mark.with_proj
+def test_target_file_grib_to_geotiff():
+    ds = from_source("file", earthkit_examples_file("test.grib"))
+    vals_ref = ds.values[:, :4]
+
+    with temp_file() as path:
+        ds.to_target("file", path, encoder="geotiff")
+
+        ds1 = from_source("file", path)
+        assert len(ds1) == len(ds)
+        from earthkit.data.readers.geotiff import GeoTIFFField
+
+        assert isinstance(ds1[0], GeoTIFFField)
+        assert np.allclose(ds1.values[:, :4], vals_ref)
+
+
 @pytest.mark.skipif(NO_RIOXARRAY, reason="rioxarray not available")
 @pytest.mark.with_proj
 def test_target_file_geotiff():