def test_load_stash2standard_name(self): f = cf.read(self.ppfile)[0] self.assertEqual(f.identity(), "eastward_wind") self.assertEqual(f.Units, cf.Units("m s-1")) for merge in (True, False): cf.load_stash2standard_name(self.new_table, merge=merge) f = cf.read(self.ppfile)[0] self.assertEqual(f.identity(), "NEW_NAME") self.assertEqual(f.Units, cf.Units("Pa")) cf.load_stash2standard_name() f = cf.read(self.ppfile)[0] self.assertEqual(f.identity(), "eastward_wind") self.assertEqual(f.Units, cf.Units("m s-1")) cf.load_stash2standard_name()
def test_PP_load_stash2standard_name(self): f = cf.read(self.ppfilename)[0] self.assertEqual(f.identity(), 'surface_temperature') self.assertEqual(f.Units, cf.Units('K')) for merge in (True, False): cf.load_stash2standard_name(self.new_table, merge=merge) f = cf.read(self.ppfilename)[0] self.assertEqual(f.identity(), 'NEW_NAME') self.assertEqual(f.Units, cf.Units('Pa')) cf.load_stash2standard_name() f = cf.read(self.ppfilename)[0] self.assertEqual(f.identity(), 'surface_temperature') self.assertEqual(f.Units, cf.Units('K')) cf.load_stash2standard_name()
X = P.set_construct(cf.DomainAxis(2)) # Set the data for the field P.set_data(cf.Data(array), axes=[T, Y, X]) P print(P.data.array) P.data.get_compression_type() print(P.data.compressed_array) list_variable = P.data.get_list() list_variable print(list_variable.array) cf.write(P, 'P_gathered.nc') print("\n**PP and UM fields files**\n") pp = cf.read('umfile.pp') pp print(pp[0]) cf.write(pp, 'umfile1.nc') type(cf.read_write.um.umread.stash2standard_name) cf.read_write.um.umread.stash2standard_name[(1, 4)] cf.read_write.um.umread.stash2standard_name[(1, 2)] cf.read_write.um.umread.stash2standard_name[(1, 7)] (1, 999) in cf.read_write.um.umread.stash2standard_name with open('new_STASH.txt', 'w') as new: new.write('1!999!My STASH code!1!!!ultraviolet_index!!') _ = cf.load_stash2standard_name('new_STASH.txt', merge=True) cf.read_write.um.umread.stash2standard_name[(1, 999)]