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