def test_SingleMap_Init(self): try: import cartopy.crs as ccrs except: return True #no testing if cartopy not installed # just test if things pass SM1 = mapping.SingleMap(self.D) SM2 = mapping.SingleMap(self.D, stat_type='sum') proj_prop = {'projection': 'robin'} SM3 = mapping.SingleMap(self.D, backend='basemap', stat_type='median') SM4 = mapping.SingleMap(self.D, backend='cartopy') SM1.plot(show_zonal=True) SM2.plot(show_zonal=True, colorbar_orientation='horizontal') SM3.plot(show_zonal=True, colorbar_orientation='horizontal', proj_prop=proj_prop) SM4.plot(show_zonal=True, colorbar_orientation='horizontal', proj_prop=proj_prop)
def test_SingleMap_Save(self): SM = mapping.SingleMap(self.D, savefile=self._tmpdir + os.sep + 'my_test_save_file.nc') SM.save(save_mean=True, save_all=True) self.assertTrue( os.path.exists(self._tmpdir + os.sep + 'my_test_save_file.nc_timmean.nc')) self.assertTrue( os.path.exists(self._tmpdir + os.sep + 'my_test_save_file.nc_timmean.nc_all.nc')) if os.path.exists(self._tmpdir + os.sep + 'my_test_save_file.nc_timmean.nc'): os.remove(self._tmpdir + os.sep + 'my_test_save_file.nc_timmean.nc') if os.path.exists(self._tmpdir + os.sep + 'my_test_save_file.nc_timmean.nc_all.nc'): os.remove(self._tmpdir + os.sep + 'my_test_save_file.nc_timmean.nc_all.nc')
def test_SingleMap_InitWithMissingProjectionProperties(self): # just test if things pass with self.assertRaises(ValueError): SM = mapping.SingleMap(self.D, backend='cartopy') SM.plot()
def test_SingleMap_InitWithInvalidBackend(self): # just test if things pass with self.assertRaises(ValueError): SM = mapping.SingleMap(self.D, backend='some_invalid_backend')
def test_SingleMap_WithPredefinedAxisButWhichIsNone(self): ax = None SM1 = mapping.SingleMap(self.D, ax=ax)
def test_SingleMap_WithPredefinedAxis(self): f = plt.figure() ax = f.add_subplot(2, 1, 1) SM1 = mapping.SingleMap(self.D, ax=ax)
def test_invalid_colorbar_orientation(self): SM = mapping.SingleMap(self.D) with self.assertRaises(ValueError): SM.plot(colorbar_orientation='something')
def test_SingleMap_WithoutColorbar(self): SM = mapping.SingleMap(self.D) SM.plot(show_colorbar=False)