def test_reduced_raw(self): # Loading a GRIB message defined on a reduced grid without # interpolating to a regular grid. gribfile = tests.get_data_path( ("GRIB", "reduced", "reduced_gg.grib2")) grib_generator = load_cubes(gribfile, auto_regularise=False) self.assertCML(next(grib_generator))
def test(self): generator = iris.fileformats.grib._load_generate converter = iris.fileformats.grib._load_convert.convert files = mock.sentinel.FILES callback = mock.sentinel.CALLBACK expected_result = mock.sentinel.RESULT with mock.patch('iris.fileformats.rules.load_cubes') as rules_load: rules_load.return_value = expected_result result = load_cubes(files, callback) kwargs = {} loader = Loader(generator, kwargs, converter, None) rules_load.assert_called_once_with(files, callback, loader) self.assertIs(result, expected_result)
def _test(self, mode, generator, converter): # Ensure that `load_cubes` defers to # `iris.fileformats.rules.load_cubes`, passing a correctly # configured `Loader` instance. with iris.FUTURE.context(strict_grib_load=mode): with mock.patch('iris.fileformats.rules.load_cubes') as rules_load: rules_load.return_value = mock.sentinel.RESULT result = load_cubes(mock.sentinel.FILES, mock.sentinel.CALLBACK, mock.sentinel.REGULARISE) loader = iris.fileformats.rules.Loader( generator, {'auto_regularise': mock.sentinel.REGULARISE}, converter, None) rules_load.assert_called_once_with(mock.sentinel.FILES, mock.sentinel.CALLBACK, loader) self.assertIs(result, mock.sentinel.RESULT)
def _test(self, mode, generator, converter): # Ensure that `load_cubes` defers to # `iris.fileformats.rules.load_cubes`, passing a correctly # configured `Loader` instance. with iris.FUTURE.context(strict_grib_load=mode): with mock.patch('iris.fileformats.rules.load_cubes') as rules_load: rules_load.return_value = mock.sentinel.RESULT result = load_cubes(mock.sentinel.FILES, mock.sentinel.CALLBACK, mock.sentinel.REGULARISE) if mode: kw_args = {} else: kw_args = {'auto_regularise': mock.sentinel.REGULARISE} loader = iris.fileformats.rules.Loader(generator, kw_args, converter, None) rules_load.assert_called_once_with(mock.sentinel.FILES, mock.sentinel.CALLBACK, loader) self.assertIs(result, mock.sentinel.RESULT)
def test_no_forecast_period(self): # The stock cube has a non-compliant forecast_period. fname = tests.get_data_path(('GRIB', 'global_t', 'global.grib2')) [cube] = load_cubes(fname) cube.remove_coord("forecast_period") self._test(cube)
def test_reduced_raw(self): # Loading a GRIB message defined on a reduced grid without # interpolating to a regular grid. gribfile = tests.get_data_path(("GRIB", "reduced", "reduced_gg.grib2")) grib_generator = load_cubes(gribfile, auto_regularise=False) self.assertCML(next(grib_generator))