fix gefs tests (blaylockbk#100)

earthobservations · Jan 6, 2023 · 14d154e · 14d154e
1 parent e7eb30a
commit 14d154e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/herbie/models/gefs.py b/herbie/models/gefs.py
@@ -26,6 +26,10 @@ def template(self):
             "chem.25": "Chemistry fields on 0.25 degree grid",
         }
 
+        if self.product is None:
+            # Just select the first PRODUCT as default
+            self.product = list(self.PRODUCTS)[0]
+
         if self.product == "wave":
             if self.member == "spr":
                 self.member = "spread"

diff --git a/tests/test_gefs.py b/tests/test_gefs.py
@@ -12,9 +12,22 @@
 
 def test_gefs():
     H = Herbie(
-        "2017-03-14",
+        "2023-01-01 06:00",
         model="gefs",
         fxx=12,
+        member="mean",
+        save_dir=save_dir,
+    )
+
+    assert H.grib, "GEFS grib2 file not found"
+    assert H.idx, "GEFS index file not found"
+
+
+def test_gefs_reforecast():
+    H = Herbie(
+        "2017-03-14",
+        model="gefs_reforecast",
+        fxx=12,
         member=0,
         variable_level="tmp_2m",
         save_dir=save_dir,