Unidata / MetPy

    p = np.linspace(1000, 100, 10)

    u = np.linspace(-10, 10, 10)

    skew.plot_barbs(p, u, u, c=u)

    return fig

@pytest.mark.mpl_image_compare(tolerance=0, remove_text=True)

def test_hodograph_plot_layers():

    """Test hodograph colored height layers with interpolation."""

    u = np.arange(5, 65, 5) * units('knot')

    v = np.arange(-5, -65, -5) * units('knot')

    h = [178, 213, 610, 656, 721, 914, 1060,

         1219, 1372, 1412, 1512, 1524] * units('meter')

    colors = ['red', 'green']

    levels = [0, 500, 1000] * units('meter')

    fig = plt.figure(figsize=(9, 9))

    ax = fig.add_subplot(1, 1, 1)

    hodo = Hodograph(ax, component_range=80)

    hodo.add_grid(increment=20, color='k')

    hodo.plot_colormapped(u, v, h, bounds=levels, colors=colors)

    ax = fig.add_subplot(1, 1, 1)

    hodo = Hodograph(ax, component_range=80)

    hodo.add_grid(increment=20, color='k')

    hodo.plot_colormapped(u, v, h, bounds=levels, colors=colors)

    return fig

@pytest.mark.mpl_image_compare(tolerance=0, remove_text=True)

def test_hodograph_plot_layers_different_units():

    """Test hodograph colored height layers with interpolation and different units."""

    u = np.arange(5, 65, 5) * units('knot')

    v = np.arange(-5, -65, -5) * units('knot')

    h = [178, 213, 610, 656, 721, 914, 1060,

         1219, 1372, 1412, 1512, 1524] * units('meter')

    colors = ['red', 'green']

    levels = [0, 0.5, 1] * units.km

    fig = plt.figure(figsize=(9, 9))

    ax = fig.add_subplot(1, 1, 1)

    hodo = Hodograph(ax, component_range=80)

    hodo.add_grid(increment=20, color='k')