Merge pull request #139 from ACCESS-NRI/137/capture-unrecognized-unit…

…-warning Capture filename key warning in test
ACCESS-NRI · Nov 7, 2024 · b55e1b1 · b55e1b1
2 parents 0ca826c + 21aeb32
commit b55e1b1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/test/test_conversion_driver_esm1p5.py b/test/test_conversion_driver_esm1p5.py
@@ -80,11 +80,12 @@ def test_get_nc_write_path_unrecognized_unit():
     nc_write_dir = Path("netCDF")
     expected_nc_write_path = nc_write_dir / f"aiihca.p{unknown_key}-005007.nc"
 
-    nc_write_path = esm1p5_convert.get_nc_write_path(
-                        Path(ff_name),
-                        nc_write_dir,
-                        (ff_year, ff_month, 1)
-                    )
+    with pytest.warns(RuntimeWarning):
+        nc_write_path = esm1p5_convert.get_nc_write_path(
+                            Path(ff_name),
+                            nc_write_dir,
+                            (ff_year, ff_month, 1)
+                        )
 
     assert nc_write_path == expected_nc_write_path
 

diff --git a/umpost/conversion_driver_esm1p5.py b/umpost/conversion_driver_esm1p5.py
@@ -153,7 +153,7 @@ def get_nc_filename(fields_file_name, date=None):
         warnings.warn(
             f"Unit code '{unit}' from filename f{fields_file_name} "
             "not recognized. Frequency information will not be added "
-            "to the netCDF filename."
+            "to the netCDF filename.", RuntimeWarning
         )
         suffix = ""