def _test(self, scanning_mode): def make_raw_message(): sections = {3: {'sourceOfGridDefinition': 0, 'numberOfOctectsForNumberOfPoints': 0, 'interpretationOfNumberOfPoints': 0, 'gridDefinitionTemplateNumber': 0, 'scanningMode': scanning_mode, 'Nj': 3, 'Ni': 4}, 7: {'codedValues': np.arange(12)}} raw_message = mock.Mock(sections=sections) return raw_message message = _GribMessage(make_raw_message(), make_raw_message, False) data = message.data self.assertIsInstance(data, biggus.Array) self.assertEqual(data.shape, (3, 4)) self.assertEqual(data.dtype, np.floating) self.assertIs(data.fill_value, np.nan) self.assertArrayEqual(data.ndarray(), np.arange(12).reshape(3, 4))
def _make_test_message(sections): return _GribMessage(mock.Mock(sections=sections), None, None)
def _make_test_message(sections): raw_message = mock.Mock(sections=sections) recreate_raw = mock.Mock(return_value=raw_message) return _GribMessage(raw_message, recreate_raw)