def test_declarative_overlay_projections():
"""Test making a contour plot."""
data = xr.open_dataset(get_test_data('NAM_test.nc', as_file_obj=False))
data2 = xr.open_dataset(get_test_data('GFS_test.nc', as_file_obj=False))
contour = ContourPlot()
contour.data = data
contour.field = 'Geopotential_height_isobaric'
contour.level = 300 * units.hPa
contour.linecolor = 'red'
contour.linestyle = '-'
contour.linewidth = 2
contour.contours = np.arange(0, 20000, 120).tolist()
contour2 = ContourPlot()
contour2.data = data2
contour2.field = 'Geopotential_height_isobaric'
contour2.level = 300 * units.hPa
contour2.linecolor = 'blue'
contour2.linestyle = '-'
contour2.linewidth = 2
contour2.contours = np.arange(0, 20000, 120).tolist()
panel = MapPanel()
panel.area = (-124, -72, 20, 53)
panel.projection = 'lcc'
panel.layers = ['coastline', 'borders', 'usstates']
panel.plots = [contour, contour2]
pc = PanelContainer()
pc.size = (8, 8)
pc.panels = [panel]
pc.draw()
return pc.figure
def test_declarative_barb_earth_relative():
"""Test making a contour plot."""
data = xr.open_dataset(get_test_data('NAM_test.nc', as_file_obj=False))
contour = ContourPlot()
contour.data = data
contour.field = 'Geopotential_height_isobaric'
contour.level = 300 * units.hPa
contour.linecolor = 'red'
contour.linestyle = '-'
contour.linewidth = 2
contour.contours = range(0, 20000, 120)
barb = BarbPlot()
barb.data = data
barb.level = 300 * units.hPa
barb.time = datetime(2016, 10, 31, 12)
barb.field = ['u-component_of_wind_isobaric', 'v-component_of_wind_isobaric']
barb.skip = (5, 5)
barb.color = 'black'
barb.barblength = 6.5
barb.earth_relative = False
panel = MapPanel()
panel.area = (-124, -72, 20, 53)
panel.projection = 'lcc'
panel.layers = ['coastline', 'borders', 'usstates']
panel.plots = [contour, barb]
pc = PanelContainer()
pc.size = (8, 8)
pc.panels = [panel]
pc.draw()
return pc.figure
def test_declarative_contour_convert_units():
"""Test making a contour plot."""
data = xr.open_dataset(get_test_data('narr_example.nc', as_file_obj=False))
contour = ContourPlot()
contour.data = data
contour.field = 'Temperature'
contour.level = 700 * units.hPa
contour.contours = 30
contour.linewidth = 1
contour.linecolor = 'red'
contour.linestyle = 'dashed'
contour.clabels = True
contour.plot_units = 'degC'
panel = MapPanel()
panel.area = 'us'
panel.projection = 'lcc'
panel.layers = ['coastline', 'borders', 'usstates']
panel.plots = [contour]
pc = PanelContainer()
pc.size = (8, 8)
pc.panels = [panel]
pc.draw()
return pc.figure
