def test_filter_functions(self):
"""Make sure our filter functions are doing what we want!"""
m = plot.MapPlot(sector='iowa')
self.assertTrue(plot.state_filter(m, 'IAC001', dict()))
self.assertTrue(not plot.state_filter(m, 'MNC001', dict()))
m = plot.MapPlot(cwa='DMX')
self.assertTrue(plot.state_filter(m, 'IAC001', dict()))
def test_conus(self):
"""See if we can plot albers"""
mp = plot.MapPlot(sector='custom',
title='EPSG: 5070 Albers',
north=reference.CONUS_NORTH + 1,
east=reference.CONUS_EAST - 12,
west=reference.CONUS_WEST + 14,
south=reference.CONUS_SOUTH,
projection=reference.EPSG[5070],
continentalcolor='white')
# mp.ax.gridlines(xlocs=[-134, -60], ylocs=[24.5, 49.5])
mp.postprocess(filename='/tmp/test_plot_epsg5070.png')
mp.close()
mp = plot.MapPlot(sector='custom',
title='EPSG: 2163 Lambert Azimuthal Equal Area',
north=reference.CONUS_NORTH + 1,
east=reference.CONUS_EAST - 12,
west=reference.CONUS_WEST + 14,
south=reference.CONUS_SOUTH,
projection=reference.EPSG[2163],
continentalcolor='white')
# mp.ax.gridlines(xlocs=[-134, -60], ylocs=[24.5, 49.5])
mp.postprocess(filename='/tmp/test_plot_epsg2163.png')
mp.close()
def test_colorbar(self):
"""Run tests against the colorbar algorithm"""
mp = plot.MapPlot(sector='iowa', debug=True)
cmap = plot.maue()
clevs = list(range(0, 101, 10))
norm = mpcolors.BoundaryNorm(clevs, cmap.N)
mp.drawcities()
mp.draw_colorbar(clevs, cmap, norm)
mp.postprocess(filename='/tmp/test_plot_colorbar1.png')
mp.close()
mp = plot.MapPlot(sector='iowa')
cmap = plot.maue()
clevs = list(range(0, 101, 10))
clevlabels = [
"One", "Three", "Blahh", "Longest", "Five", "Six", "Ten", "Fourty",
100000, "Hi\nHo", 100
]
norm = mpcolors.BoundaryNorm(clevs, cmap.N)
mp.draw_colorbar(clevs, cmap, norm, clevlabels=clevlabels)
mp.postprocess(filename='/tmp/test_plot_colorbar2.png')
mp.close()
mp = plot.MapPlot(sector='iowa')
cmap = plot.maue()
clevs = list(range(0, 101, 1))
norm = mpcolors.BoundaryNorm(clevs, cmap.N)
mp.draw_colorbar(clevs,
cmap,
norm,
clevstride=10,
title="Erosion $kg/m^2$")
mp.postprocess(filename='/tmp/test_plot_colorbar3.png')
mp.close()
def test_plot2(self):
""" Exercise NWS plot API """
mp = plot.MapPlot(sector='nws', continentalcolor='white')
mp.fill_cwas({
'DMX': 80,
'MKX': 5,
'SJU': 30,
'AJK': 40,
'HFO': 50
},
units='NWS Something or Another',
ilabel=True)
mp.postprocess(filename='/tmp/test_plot_us.png')
mp.close()
mp = plot.MapPlot(sector='iowa', continentalcolor='white')
mp.fill_cwas({
'DMX': 80,
'MKX': 5,
'SJU': 30,
'AJK': 40,
'HFO': 50
},
units='NWS Something or Another')
mp.postprocess(filename='/tmp/test_plot_iowa.png')
mp.close()
def test_drawugcs(self):
"""test drawing of UGCS"""
m = plot.MapPlot(sector='conus', title='Counties, 3 filled in Iowa')
m.fill_ugcs({"IAC001": 10, "IAC003": 20, "IAC005": 30})
m.postprocess(filename='/tmp/test_plot_ugcs_counties.png')
m.close()
m = plot.MapPlot(sector='iowa', title='Zones, 3 filled in Iowa, label')
m.fill_ugcs({"IAZ001": 10, "IAZ003": 20, "IAZ005": 30}, ilabel=True)
m.postprocess(filename='/tmp/test_plot_ugcs_zones.png')
m.close()
def test_textplot(self):
''' Can we plot text and place labels on them '''
m = plot.MapPlot(sector='iowa')
m.plot_values(np.arange(-99, -94), np.arange(40, 45), np.arange(5))
m.postprocess(filename='/tmp/test_plot_1.png')
m = plot.MapPlot(sector='iowa')
m.plot_values(np.arange(-99, -94),
np.arange(40, 45),
np.arange(5),
labels=range(5, 11))
m.postprocess(filename='/tmp/test_plot_2.png')
m.close()
def test_climdiv(self):
"""Run tests agains the fill_climdiv"""
m = plot.MapPlot(sector='conus', title="Climate Divisions")
data = {'IAC001': 10, 'MNC001': 20, 'NMC001': 30}
m.fill_climdiv(data, ilabel=True)
m.postprocess(filename='/tmp/test_plot_climdiv.png')
m.close()
def test_fillstates(self):
"""Can we fill states"""
data = {'AK': 10, 'HI': 30, 'IA': 40, 'NY': 80}
m = plot.MapPlot(sector='nws', title='Fill States')
m.fill_states(data, ilabel=True)
m.postprocess(filename='/tmp/test_plot_states.png')
m.close()
def test_scatter(self):
""" Test scatter plots """
m = plot.MapPlot(sector='midwest')
m.scatter(np.linspace(-99, -94, 100), np.linspace(40, 45, 100),
np.arange(100), np.arange(0, 101, 10))
m.postprocess(filename='/tmp/test_plot_scatter.png')
m.close()
def plot_daily(date, interval, plotvar, mc, mckey):
""" Generate the plot, please """
opts = PLOT_OPS[plotvar]
offset = iemre.daily_offset(date)
nc = netCDF4.Dataset("/mesonet/data/iemre/%s_mw_daily.nc" % (date.year, ))
data = nc.variables[opts['ncvar_daily']][offset] / 25.4 # inches
lons = nc.variables['lon'][:]
lats = nc.variables['lat'][:]
extra = lons[-1] + (lons[-1] - lons[-2])
lons = np.concatenate([lons, [
extra,
]])
extra = lats[-1] + (lats[-1] - lats[-2])
lats = np.concatenate([lats, [
extra,
]])
x, y = np.meshgrid(lons, lats)
nc.close()
p = plot.MapPlot(
sector='midwest',
title='%s IEM Reanalysis %s [%s]' %
(date.strftime("%-d %b %Y"), opts['title'], opts['units']))
p.pcolormesh(x, y, data, opts['clevs'], units=opts['units'])
p.postprocess(web=True, memcache=mc, memcachekey=mckey, memcacheexpire=0)
def test_drawcities(self):
"""Fill the Iowa WFOs"""
m = plot.MapPlot(title='Fill and Draw Cities',
continentalcolor='blue',
sector='iowa')
m.drawcities()
m.postprocess(filename='/tmp/test_plot_drawcities.png')
m.close()
def test_barbs(self):
"""Testing the plotting of wind barbs"""
m = plot.MapPlot(continentalcolor='white')
data = [
dict(lat=41.5, lon=-96, tmpf=50, dwpf=30, sknt=10, drct=100),
dict(lat=42.0, lon=-95.5, tmpf=50, dwpf=30, sknt=20, drct=200),
]
m.plot_station(data)
m.postprocess(filename='/tmp/test_plot_barbs.png')
m.close()
def test_overlap(self):
""" Do some checking of our overlaps logic """
m = plot.MapPlot(sector='midwest', continentalcolor='white')
lons = np.linspace(-99, -90, 100)
lats = np.linspace(38, 44, 100)
vals = lats
labels = ['%.2f' % (s, ) for s in lats]
m.plot_values(lons, lats, vals, fmt='%.2f', labels=labels)
m.postprocess(filename='/tmp/test_plot_overlap.png')
m.close()
def test_contourf(self):
''' Test the contourf plot with labels specified '''
m = plot.MapPlot(sector='iowa')
m.contourf(np.arange(-94, -89),
np.arange(40, 45),
np.arange(5),
np.arange(5),
clevlabels=['a', 'b', 'c', 'd', 'e'])
m.postprocess(filename='/tmp/test_plot_contourf.png')
m.close()
def test_drawiowawfo(self):
"""Fill the Iowa WFOs"""
m = plot.MapPlot(sector='iowawfo', title='Fill Iowa WFOs')
m.contourf(np.arange(-94, -85),
np.arange(36, 45),
np.arange(9),
np.arange(9),
clevlabels=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i'])
m.postprocess(filename='/tmp/test_plot_iowawfo.png')
m.close()
def test_michigan(self):
"""See what we do with Michigan"""
m = plot.MapPlot(sector='state', state='MI')
m.contourf(np.arange(-84, -75),
np.arange(36, 45),
np.arange(9),
np.arange(9),
clevlabels=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i'])
m.postprocess(filename='/tmp/test_plot_michigan.png')
m.close()
def test_albers(self):
"""See if we can plot albers"""
m = plot.MapPlot(sector='custom',
north=43,
east=-80,
west=-96,
south=38,
projection=ccrs.AlbersEqualArea(),
continentalcolor='white')
m.postprocess(filename='/tmp/test_plot_albers.png')
m.close()
def test_drawrandomtext(self):
"""See if we can handle the fun that is drawing random text"""
m = plot.MapPlot(sector='iowa',
title='Fun Text, here and there',
continentalcolor='white',
debug=True)
m.plot_values([-94, -92, -91, -92], [42, 41, 43, 42.4], [
'One', 'Two\nTwo', 'Three\nThree\nThree', 'Four\nFour\nFour\nFour'
],
showmarker=True)
m.postprocess(filename='/tmp/test_plot_drawrandom.png')
m.close()
def test_plot3(self):
""" Exercise climdiv plot API """
mp = plot.MapPlot(sector='iowa')
mp.fill_climdiv({
'IAC001': 80,
'AKC003': 5,
'HIC003': 30,
'AJK': 40,
'HFO': 50
})
mp.postprocess(filename='/tmp/test_plot_iowa_climdiv.png')
mp.close()
def test_plot(self):
""" Exercise the API """
m = plot.MapPlot(sector='midwest')
m.fill_cwas({
'DMX': 80,
'MKX': 5,
'SJU': 30,
'AJK': 40
},
units='no units')
m.postprocess(filename='/tmp/test_plot_midwest.png')
m.close()
def test_pcolormesh(self):
"""See if we can do pcolormesh OKish"""
mp = plot.MapPlot(sector='custom',
north=43,
east=-80,
west=-96,
south=38,
projection=reference.EPSG[2163],
continentalcolor='white')
lons = np.arange(-100, -80, 0.25)
lats = np.arange(40, 50, 0.25)
vals = np.random.rand(lats.shape[0], lons.shape[0])
lons, lats = np.meshgrid(lons, lats)
mp.pcolormesh(lons, lats, vals, np.arange(0, 1, 0.1))
mp.postprocess(filename='/tmp/test_plot_pcolormesh.png')
mp.close()
def test_cwa(self):
"""Exercise the cwa plotting routines"""
m = plot.MapPlot(sector='cwa', cwa='MKX')
m.postprocess(filename='/tmp/test_plot_MKX.png')
m.close()
self.assertEquals(m.cwa, 'MKX')
offset = iemre.hourly_offset(now)
p01m = nc.variables['p01m'][offset]
# 0.05in is 1.27 mm
this = numpy.where(p01m > 1.27, 1, 0)
running = numpy.where(this == 1, 0, running + 1)
maxval = numpy.where(running > maxval, running, maxval)
print now, running[j,i], maxval[j,i]
now += interval
nc2 = netCDF4.Dataset("/mesonet/data/iemre/state_weights.nc")
domain = nc2.variables['domain'][:]
nc2.close()
#maxval = numpy.where(domain == 1, maxval, 1.e20)
m = plot.MapPlot(sector='midwest',
title='1 May - 28 June 2015 Max Period between hourly 0.05+ inch Precip',
subtitle='based on NCEP Stage IV data')
extra = lons[-1] + (lons[-1] - lons[-2])
#lons = numpy.concatenate([lons, [extra,]])
extra = lats[-1] + (lats[-1] - lats[-2])
lats[-1] = extra
#lats = numpy.concatenate([lats, [extra,]])
x, y = numpy.meshgrid(lons, lats)
#m.pcolormesh(x, y, maxval / 24.0, numpy.arange(0,25,1), units='days')
m.contourf(x, y, maxval / 24.0, numpy.arange(0,25,2), units='days')
m.postprocess(filename='test.png')
def test_illinois(self):
"""Produce a plot that doesn't suck"""
mp = plot.MapPlot(sector='state', state='IL')
mp.postprocess(filename="/tmp/test_plot_illinois.png")
mp.close()
def test_drawcounties(self):
"""draw counties on the map"""
m = plot.MapPlot(sector='midwest', title='Counties')
m.drawcounties()
m.postprocess(filename='/tmp/test_plot_counties.png')
m.close()
def test_usdm(self):
"""Can we plot the current USDM"""
mp = plot.MapPlot(sector='conus')
mp.draw_usdm(hatched=True, filled=False)
mp.postprocess(filename='/tmp/test_plot_usdm.png')
mp.close()
def test_states(self):
"""Exercise the state plotting routines"""
m = plot.MapPlot(sector='state', state='CA')
m.postprocess(filename='/tmp/test_plot_CA.png')
m.close()
self.assertEquals(m.state, 'CA')
