Unidata / MetPy

    p = np.arange(10) * units.hPa

    y = np.arange(9) * units.degC

    with pytest.raises(ValueError):

        get_layer(p, y)

def test_get_layer_invalid_depth_units():

    """Tests that error is raised when depth has invalid units."""

    p = np.arange(10) * units.hPa

    y = np.arange(9) * units.degC

    with pytest.raises(ValueError):

        get_layer(p, y, depth=400 * units.degC)

                                 1105.25854492, 1549.8079], dtype=np.float32) * units.meter

    p_layer, hgt_layer = get_layer(p[::-1], hgt[::-1], heights=hgt[::-1],

                                   depth=1000. * units.meter)

    assert_array_almost_equal(p_layer, true_p_layer, 4)

    assert_array_almost_equal(hgt_layer, true_hgt_layer, 4)

def test_get_layer_ragged_data():

    """Tests that error is raised for unequal length pressure and data arrays."""

    p = np.arange(10) * units.hPa

    y = np.arange(9) * units.degC

    with pytest.raises(ValueError):

        get_layer(p, y)

    with pytest.raises(ValueError):

        get_layer(p, y, depth=400 * units.degC)

@pytest.fixture

def layer_test_data():

    """Provide test data for testing of layer bounds."""

    pressure = np.arange(1000, 10, -100) * units.hPa

    temperature = np.linspace(25, -50, len(pressure)) * units.degC

    return pressure, temperature

@pytest.mark.parametrize('pressure, variable, heights, bottom, depth, interp, expected', [

    (layer_test_data()[0], layer_test_data()[1], None, None, 150 * units.hPa, True,

     (np.array([1000, 900, 850]) * units.hPa,

      np.array([25.0, 16.666666, 12.62262]) * units.degC)),