def test_phenom_known_long_name(self, mock_set): grib = None cube = iris.cube.Cube(np.array([1.0]), long_name='cloud_mixing_ratio') grib_save_rules.set_discipline_and_parameter(cube, grib) mock_set.assert_any_call(grib, "discipline", 0) mock_set.assert_any_call(grib, "parameterCategory", 1) mock_set.assert_any_call(grib, "parameterNumber", 22)
def test_phenom_known_long_name(self, mock_set_long): grib = None cube = iris.cube.Cube(np.array([1.0]), long_name='cloud_mixing_ratio') grib_save_rules.set_discipline_and_parameter(cube, grib) mock_set_long.assert_any_call(grib, "discipline", 0) mock_set_long.assert_any_call(grib, "parameterCategory", 1) mock_set_long.assert_any_call(grib, "parameterNumber", 22)
def test_phenom_known_standard_name(self, mock_set_long): grib = None cube = iris.cube.Cube(np.array([1.0]), standard_name='sea_surface_temperature') grib_save_rules.set_discipline_and_parameter(cube, grib) mock_set_long.assert_any_call(grib, "discipline", 10) mock_set_long.assert_any_call(grib, "parameterCategory", 3) mock_set_long.assert_any_call(grib, "parameterNumber", 0)
def test_phenom_known_standard_name(self, mock_set): grib = None cube = iris.cube.Cube(np.array([1.0]), standard_name='sea_surface_temperature') grib_save_rules.set_discipline_and_parameter(cube, grib) mock_set.assert_any_call(grib, "discipline", 10) mock_set.assert_any_call(grib, "parameterCategory", 3) mock_set.assert_any_call(grib, "parameterNumber", 0)
def test_phenom_unknown(self, mock_set_long): grib = None cube = iris.cube.Cube(np.array([1.0])) # Force reset of warnings registry to avoid suppression of # repeated warnings. warnings.resetwarnings() does not do this. if hasattr(grib_save_rules, '__warningregistry__'): grib_save_rules.__warningregistry__.clear() with warnings.catch_warnings(): # This should issue a warning about unrecognised data warnings.simplefilter("error") with self.assertRaises(UserWarning): grib_save_rules.set_discipline_and_parameter(cube, grib) # do it all again, and this time check the results grib = None cube = iris.cube.Cube(np.array([1.0])) grib_save_rules.set_discipline_and_parameter(cube, grib) mock_set_long.assert_any_call(grib, "discipline", 255) mock_set_long.assert_any_call(grib, "parameterCategory", 255) mock_set_long.assert_any_call(grib, "parameterNumber", 255)
def test_phenom_unknown(self, mock_set): grib = None cube = iris.cube.Cube(np.array([1.0])) # Force reset of warnings registry to avoid suppression of # repeated warnings. warnings.resetwarnings() does not do this. if hasattr(grib_save_rules, '__warningregistry__'): grib_save_rules.__warningregistry__.clear() with warnings.catch_warnings(): # This should issue a warning about unrecognised data warnings.simplefilter("error") with self.assertRaises(UserWarning): grib_save_rules.set_discipline_and_parameter(cube, grib) # do it all again, and this time check the results grib = None cube = iris.cube.Cube(np.array([1.0])) grib_save_rules.set_discipline_and_parameter(cube, grib) mock_set.assert_any_call(grib, "discipline", 255) mock_set.assert_any_call(grib, "parameterCategory", 255) mock_set.assert_any_call(grib, "parameterNumber", 255)