def test_hef(): grid = salem.grids.local_mercator_grid(center_ll=(10.76, 46.798444), extent=(10000, 7000)) c = Map(grid, countries=False) c.set_lonlat_countours(interval=10) c.set_shapefile(get_demo_file('Hintereisferner_UTM.shp')) c.set_topography(get_demo_file('hef_srtm.tif'), interp='linear') c.visualize(addcbar=False, title='linear') c.set_topography(get_demo_file('hef_srtm.tif'), interp='spline', ks=2) c.visualize(addcbar=False, title='spline deg 2') c.set_topography(get_demo_file('hef_srtm.tif')) c.visualize(addcbar=False, title='Default: spline deg 3') h = c.set_topography(get_demo_file('hef_srtm.tif'), interp='spline', ks=5) c.visualize(addcbar=False, title='spline deg 5') dem = salem.GeoTiff(get_demo_file('hef_srtm.tif')) mytopo = dem.get_vardata() c.set_topography(mytopo, crs=dem.grid, interp='spline') c.visualize(addcbar=False, title='From array') c.set_lonlat_countours() c.set_cmap(cleo.get_cm('topo')) c.set_plot_params(nlevels=256) c.set_data(h) c.visualize()
def test_gmap(): g = GoogleCenterMap(center_ll=(10.762660, 46.794221), zoom=13, size_x=640, size_y=640) m = Map(g.grid, countries=False, nx=640) m.set_lonlat_countours(interval=0.025) m.set_shapefile(get_demo_file('Hintereisferner.shp'), linewidths=2, edgecolor='darkred') m.set_rgb(g.get_vardata()) m.visualize(addcbar=False) dem = salem.GeoTiff(get_demo_file('hef_srtm.tif')) dem.set_subset(margin=-100) dem = salem.grids.local_mercator_grid(center_ll=(10.76, 46.798444), extent=(10000, 7000)) i, j = dem.ij_coordinates g = GoogleVisibleMap(x=i, y=j, src=dem, size_x=500, size_y=400) img = g.get_vardata() m = Map(dem, countries=False) assert_raises(ValueError, m.set_data, img) m.set_lonlat_countours(interval=0.025) m.set_shapefile(get_demo_file('Hintereisferner.shp'), linewidths=2, edgecolor='darkred') m.set_rgb(img, g.grid) m.visualize(addcbar=False)
def test_oceans(): f = os.path.join(get_demo_file('wrf_tip_d1.nc')) grid = GeoNetcdf(f).grid m = Map(grid, countries=False) m.set_shapefile(rivers=True, linewidths=2) m.set_shapefile(oceans=True, edgecolor='k', linewidth=3) fig, ax = plt.subplots(1, 1) m.visualize(ax=ax, addcbar=False) plt.tight_layout()