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):
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))
