Merge pull request #446 from vieramercado/feature/ncei-historical-analysis

Added GFS NCEI historical analysis source

Author: blaylockbk

src/herbie/models/gfs.py

@@ -85,8 +85,9 @@ def template(self):
                 "ncei_analysis": f"https://www.ncei.noaa.gov/data/global-forecast-system/access/grid-{grid_num:03d}-{self.product}/analysis/{self.date:%Y%m/%Y%m%d}/gfs_{grid_num}_{self.date:%Y%m%d_%H%M}_{self.fxx:03d}.grb2",
                 "ncei_forecast": f"https://www.ncei.noaa.gov/data/global-forecast-system/access/grid-{grid_num:03d}-{self.product}/forecast/{self.date:%Y%m/%Y%m%d}/gfs_{grid_num}_{self.date:%Y%m%d_%H%M}_{self.fxx:03d}.grb2",
                 "ncar_rda": f"https://data.rda.ucar.edu/d084001/{self.date:%Y/%Y%m%d}/gfs.0p25.{self.date:%Y%m%d%H}.f{self.fxx:03d}.grib2",
+                "ncei_historical_analysis": f"https://www.ncei.noaa.gov/data/global-forecast-system/access/historical/analysis/{self.date:%Y%m/%Y%m%d}/gfsanl_{grid_num}_{self.date:%Y%m%d_%H%M}_{self.fxx:03d}.grb2",
             }
-            self.IDX_SUFFIX = [".grb2.inv", ".idx"]
+            self.IDX_SUFFIX = [".grb2.inv", ".idx", ".inv"]
         self.LOCALFILE = f"{self.get_remoteFileName}"
 
 

tests/test_gfs.py

@@ -53,4 +53,25 @@ def test_graphcast():
     assert H.get_localFilePath(filter).exists(), "File doesn't exist!"
     f.unlink()
 
-    H.xarray(filter)
+    H.xarray(filter)
+
+def test_gfs_ncei_historical_anl():
+    H = Herbie(
+        datetime(2015, 1, 15),
+        priority="ncei_historical_analysis",
+        product="0.5-degree",
+        model="gfs",
+        fxx=0,
+        save_dir=save_dir,
+        overwrite=True,
+    )
+
+    assert H.grib, "GFS NCEI Historical Analysis grib2 file not found"
+    assert H.idx, "GFS NCEI Historical Analysis index file not found"
+
+    filter = "TMP:900 mb"
+    f = H.download(filter)
+    assert H.get_localFilePath(filter).exists(), "File doesn't exist!"
+    f.unlink()
+
+    H.xarray(filter)