put tests back

blaylockbk · Jan 14, 2025 · 2df9fa9 · 2df9fa9
1 parent 246c2c1
commit 2df9fa9
Show file tree

Hide file tree

Showing 15 changed files with 16 additions and 11 deletions.
diff --git a/herbie/core.py b/herbie/core.py
@@ -1142,16 +1142,21 @@ def xarray(
         # Get CF grid projection information with pygrib and pyproj because
         # this is something cfgrib doesn't do (https://github.com/ecmwf/cfgrib/issues/251)
         # NOTE: Assumes the projection is the same for all variables
-        #        with pygrib.open(str(local_file)) as grb:
-        #            msg = grb.message(1)
-        #            cf_params = CRS(msg.projparams).to_cf()
-        #
-        #        # Funny stuff with polar stereographic (https://github.com/pyproj4/pyproj/issues/856)
-        #        # TODO: Is there a better way to handle this? What about south pole?
-        #        if cf_params["grid_mapping_name"] == "polar_stereographic":
-        #            cf_params["latitude_of_projection_origin"] = cf_params.get(
-        #                "latitude_of_projection_origin", 90
-        #            )
+
+        use_pygrib = False
+        if use_pygrib:
+            with pygrib.open(str(local_file)) as grb:
+                msg = grb.message(1)
+                cf_params = CRS(msg.projparams).to_cf()
+
+            # Funny stuff with polar stereographic (https://github.com/pyproj4/pyproj/issues/856)
+            # TODO: Is there a better way to handle this? What about south pole?
+            if cf_params["grid_mapping_name"] == "polar_stereographic":
+                cf_params["latitude_of_projection_origin"] = cf_params.get(
+                    "latitude_of_projection_origin", 90
+                )
+        else:
+            cf_params = {}
 
         # Here I'm looping over each dataset in the list returned by cfgrib
         for ds in Hxr:
@@ -1171,7 +1176,7 @@ def xarray(
             # ----------------------
             # http://cfconventions.org/Data/cf-conventions/cf-conventions-1.8/cf-conventions.html#appendix-grid-mappings
             ds.coords["gribfile_projection"] = None
-            #            ds.coords["gribfile_projection"].attrs = cf_params
+            ds.coords["gribfile_projection"].attrs = cf_params
             ds.coords["gribfile_projection"].attrs["long_name"] = (
                 f"{self.model.upper()} model grid projection"
             )

diff --git a/tttests/test_core.py → tests/test_core.py b/tttests/test_core.py → tests/test_core.py
diff --git a/tttests/test_ecmwf.py → tests/test_ecmwf.py b/tttests/test_ecmwf.py → tests/test_ecmwf.py
diff --git a/tttests/test_fast.py → tests/test_fast.py b/tttests/test_fast.py → tests/test_fast.py
diff --git a/tttests/test_first.py → tests/test_first.py b/tttests/test_first.py → tests/test_first.py
diff --git a/tttests/test_gefs.py → tests/test_gefs.py b/tttests/test_gefs.py → tests/test_gefs.py
diff --git a/tttests/test_gfs.py → tests/test_gfs.py b/tttests/test_gfs.py → tests/test_gfs.py
diff --git a/tttests/test_hrrr.py → tests/test_hrrr.py b/tttests/test_hrrr.py → tests/test_hrrr.py
diff --git a/tttests/test_init.py → tests/test_init.py b/tttests/test_init.py → tests/test_init.py
diff --git a/tttests/test_latest.py → tests/test_latest.py b/tttests/test_latest.py → tests/test_latest.py
diff --git a/tttests/test_paint.py → tests/test_paint.py b/tttests/test_paint.py → tests/test_paint.py
diff --git a/tttests/test_pick_points.py → tests/test_pick_points.py b/tttests/test_pick_points.py → tests/test_pick_points.py
diff --git a/tttests/test_rap.py → tests/test_rap.py b/tttests/test_rap.py → tests/test_rap.py
diff --git a/tttests/test_toolbox.py → tests/test_toolbox.py b/tttests/test_toolbox.py → tests/test_toolbox.py
diff --git a/tttests/util.py → tests/util.py b/tttests/util.py → tests/util.py