def test_year_filter_none(self): msgs = GribMessage.messages_from_filename(self.file_path) chosen_messages = [] for gmsg in msgs: if gmsg.sections[1]['year'] == 1958: chosen_messages.append(gmsg) cubes_msgs = list(load_pairs_from_fields(chosen_messages)) self.assertEqual(len(cubes_msgs), 0)
def test_as_pairs(self): messages = GribMessage.messages_from_filename(self.file_path) cubes = [] cube_msg_pairs = load_pairs_from_fields(messages) for cube, gmsg in cube_msg_pairs: if gmsg.sections[1]['year'] == 1998: cube.attributes['the year is'] = gmsg.sections[1]['year'] cubes.append(cube) self.assertEqual(len(cubes), 1) self.assertEqual(cubes[0].attributes['the year is'], 1998)