def test_edition_1(self):
# Test object with no '.sections', and '.edition' ==1.
field = mock.Mock(edition=1, spec=('edition'))
func = 'iris.fileformats.grib._load_convert.grib1_convert'
metadata = mock.sentinel.metadata
with mock.patch(func, return_value=metadata) as grib1_convert:
result = convert(field)
grib1_convert.assert_called_once_with(field)
self.assertEqual(result, metadata)
def test_call(self):
sections = [{'editionNumber': 2}]
field = _make_test_message(sections)
this = 'iris.fileformats.grib._load_convert.grib2_convert'
factory = mock.sentinel.factory
func = lambda field, metadata: metadata['factories'].append(factory)
with mock.patch(this, side_effect=func) as grib2_convert:
# The call being tested.
result = convert(field)
self.assertTrue(grib2_convert.called)
metadata = ([factory], [], None, None, None, {}, [], [], [])
self.assertEqual(result, metadata)
def test_call(self):
sections = [{'editionNumber': 2}]
field = _make_test_message(sections)
this = 'iris.fileformats.grib._load_convert.grib2_convert'
factory = mock.sentinel.factory
func = lambda field, metadata: metadata['factories'].append(factory)
with mock.patch(this, side_effect=func) as grib2_convert:
# The call being tested.
result = convert(field)
self.assertTrue(grib2_convert.called)
metadata = ([factory], [], None, None, None, {}, [], [], [])
self.assertEqual(result, metadata)
def test_edition_2_bad(self):
# Test object with no '.sections', and '.edition' ==2.
field = mock.Mock(edition=2, spec=('edition'))
emsg = 'edition 2 is not supported'
with self.assertRaisesRegexp(TranslationError, emsg):
convert(field)
def test_edition_1_bad(self):
sections = [{'editionNumber': 1}]
field = _make_test_message(sections)
emsg = 'edition 1 is not supported'
with self.assertRaisesRegexp(TranslationError, emsg):
convert(field)
