def test_gempak_color_image_compare(): """Test creating a plot with all the GEMPAK colors.""" c = range(32) mplc = convert_gempak_color(c) delta = 0.025 x = y = np.arange(-3.0, 3.01, delta) xx, yy = np.meshgrid(x, y) z1 = np.exp(-xx**2 - yy**2) z2 = np.exp(-(xx - 1)**2 - (yy - 1)**2) z = (z1 - z2) * 2 fig = plt.figure(figsize=(9, 9)) cs = plt.contourf(xx, yy, z, levels=np.linspace(-1.8, 1.8, 33), colors=mplc) plt.colorbar(cs) return fig
def test_gempak_color_image_compare(): """Test creating a plot with all the GEMPAK colors.""" c = range(32) mplc = convert_gempak_color(c) delta = 0.025 x = y = np.arange(-3.0, 3.01, delta) xx, yy = np.meshgrid(x, y) z1 = np.exp(-xx**2 - yy**2) z2 = np.exp(-(xx - 1)**2 - (yy - 1)**2) z = (z1 - z2) * 2 fig = plt.figure(figsize=(9, 9)) cs = plt.contourf(xx, yy, z, levels=np.linspace(-1.8, 1.8, 33), colors=mplc) plt.colorbar(cs) return fig
def test_gempak_color_scalar(): """Test converting a single GEMPAK color.""" mplc = convert_gempak_color(6) truth = 'cyan' assert mplc == truth
def test_gempak_color_quirks(): """Test converting some unusual GEMPAK colors.""" c = [-5, 95, 101] mplc = convert_gempak_color(c) truth = ['white', 'bisque', 'white'] assert mplc == truth
def test_gempak_color_invalid_style(): """Test converting a GEMPAK color with an invalid style parameter.""" c = range(32) with pytest.raises(ValueError): convert_gempak_color(c, style='plt')
def test_gempak_color_scalar(): """Test converting a single GEMPAK color.""" mplc = convert_gempak_color(6) truth = 'cyan' assert mplc == truth
def test_gempak_color_quirks(): """Test converting some unusual GEMPAK colors.""" c = [-5, 95, 101] mplc = convert_gempak_color(c) truth = ['white', 'bisque', 'white'] assert mplc == truth
def test_gempak_color_invalid_style(): """Test converting a GEMPAK color with an invalid style parameter.""" c = range(32) with pytest.raises(ValueError): convert_gempak_color(c, style='plt')