diff --git a/mdtf_test_data/tests/test_synthetic_data.py b/mdtf_test_data/tests/test_synthetic_data.py index bdfa32f..a2851f7 100644 --- a/mdtf_test_data/tests/test_synthetic_data.py +++ b/mdtf_test_data/tests/test_synthetic_data.py @@ -20,6 +20,7 @@ from mdtf_test_data.synthetic.vertical import ncar_hybrid_coord from mdtf_test_data.synthetic.vertical import mom6_z_coord from mdtf_test_data.synthetic.vertical import mom6_rho2_coord +from mdtf_test_data.synthetic.vertical import cmip_vertical_coord __all__ = [ "test_xr_times_from_tuples_ncar", @@ -27,13 +28,14 @@ "test_ncar_hybrid_coord", "test_gfdl_plev19_vertical_coord", "test_gfdl_vertical_coord", + "test_cmip_vertical_coord", "test_mom6_z_coord", "test_generate_random_array", "test_generate_daily_time_axis", "test_generate_hourly_time_axis", "test_generate_monthly_time_axis", "test_generate_synthetic_dataset", - "test_dataset_stats", + "test_dataset_stats" ] @@ -44,7 +46,8 @@ def pytest_namespace(): "ncar_hybrid": None, "gfdl_plev19": None, "gfdl_vert": None, - "dummy_dset": None, + "cmip_vert": None, + "dummy_dset": None } @@ -107,6 +110,13 @@ def test_gfdl_vertical_coord(): pytest.gfdl_vert = result +def test_cmip_vertical_coord(): + result = cmip_vertical_coord() + assert isinstance(result, xr.Dataset) + assert np.allclose(float(result.plev.sum()), 15030.0) + pytest.cmip_vert = result + + def test_mom6_z_coord(): result = mom6_z_coord() assert isinstance(result, xr.Dataset)