few checks on tests

dbekaert · Feb 18, 2024 · 8005823 · 8005823
1 parent f818eac
commit 8005823
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/test/test_GUNW.py b/test/test_GUNW.py
@@ -37,7 +37,7 @@ def compute_transform(lats, lons):
 
 
 @pytest.mark.isce3
-@pytest.mark.parametrize('weather_model_name', ['MERRA2'])
+@pytest.mark.parametrize('weather_model_name', ['GMAO'])
 def test_GUNW_dataset_update(test_dir_path, test_gunw_path_factory, weather_model_name,
                              weather_model_dict_for_gunw_integration_test, mocker):
     """The GUNW from test gunw factor is:
@@ -277,7 +277,7 @@ def test_azimuth_timing_interp_against_center_time_interp(weather_model_name: st
             assert np.nanmax(abs_diff_mm) < 1
 
 
-@pytest.mark.parametrize('weather_model_name', ['MERRA2', 'HRRR', 'HRES', 'ERA5', 'ERA5T'])
+@pytest.mark.parametrize('weather_model_name', ['HRRR', 'HRES', 'ERA5', 'ERA5T'])
 def test_check_weather_model_availability(test_gunw_path_factory, weather_model_name, mocker):
     # Should be True for all weather models
     # S1-GUNW-D-R-071-tops-20200130_20200124-135156-34956N_32979N-PP-913f-v2_0_4.nc

diff --git a/tools/RAiDER/models/__init__.py b/tools/RAiDER/models/__init__.py
@@ -3,5 +3,7 @@
 from .gmao import GMAO
 from .hres import HRES
 from .hrrr import HRRR, HRRRAK
+from .merra2 import MERRA2
+from .ncmr import NCMR
 
 __all__ = ['HRRR', 'HRRRAK', 'GMAO', 'ERA5', 'ERA5T', 'HRES', 'MERRA2']