Unidata / MetPy

    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.zeros((6)) * units.knots

    v = np.array([0, 10, 20, 30, 40, 50]) * units.knots

    heights = np.array([0, 1000, 2000, 3000, 4000, 5000]) * units.m

    bounds = np.array([500, 1500, 2500, 3500, 4500]) * units.m

    colors = ['r', 'g', 'b', 'r']

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

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

    h = Hodograph(ax1)

    h.add_grid(increment=10)

    h.plot_colormapped(u, v, heights, colors=colors, bounds=bounds)

    ax1.set_xlim(-50, 50)

    ax1.set_ylim(-5, 50)

    return fig

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

    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.zeros((6)) * units.knots

    v = np.array([0, 10, 20, 30, 40, 50]) * units.knots

    heights = np.array([0, 1, 2, 3, 4, 5]) * units.km

    bounds = np.array([500, 1500, 2500, 3500, 4500]) * units.m

    colors = ['r', 'g', 'b', 'r']

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

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

    h = Hodograph(ax1)

    h.add_grid(increment=10)

    h.plot_colormapped(u, v, heights, colors=colors, bounds=bounds)

    ax1.set_xlim(-50, 50)

    ax1.set_ylim(-5, 50)

    return fig

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