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)
def _make_test_message(self, sections): raw_message = mock.Mock(sections=sections, _message_id=self.message_id) file_ref = mock.Mock(open_file=self.grib_fh) return GribMessage(raw_message, None, file_ref=file_ref)