Unidata / MetPy

        isentropic_interpolation(isentlev, lev, tmpk)

def test_isentropic_pressure_data_warning():

    """Test of isentropic pressure function, warning for non-meteorological data."""

    lev = [1000., 950., 900., 850.] * units.Pa

    tmp = np.ones((4, 5, 5))

    tmp[0, :] = 296.

    tmp[1, :] = 292.

    tmp[2, :] = 290.

    tmp[3, :] = 288.

    tmpk = tmp * units.kelvin

    isentlev = [296., 297.] * units.kelvin

    with pytest.warns(Warning):

        isentropic_interpolation(isentlev, lev, tmpk)

        isentropic_interpolation(isentlev, lev, tmpk)

def test_isentropic_pressure_data_bounds_error():

    """Test calculation of isentropic pressure function, error for data out of bounds."""

    lev = [100000., 95000., 90000., 85000.] * units.Pa

    tmp = np.ones((4, 5, 5))

    tmp[0, :] = 296.

    tmp[1, :] = 292.

    tmp[2, :] = 290.

    tmp[3, :] = 288.

    tmpk = tmp * units.kelvin

    isentlev = [296., 350.] * units.kelvin

    with pytest.raises(ValueError):

        isentropic_interpolation(isentlev, lev, tmpk)

def test_isentropic_pressure_data_warning():

    assert_almost_equal(isentprs[1][1], truetmp, 3)

def test_isentropic_pressure_3d_error():

    """Test calculation of isentropic pressure function, error for input data not 3-D."""

    lev = [100000., 95000., 90000., 85000.] * units.Pa

    tmp = np.ones((4, 5,))

    tmp[0, :] = 296.

    tmp[1, :] = 292.

    tmp[2, :] = 290.

    tmp[3, :] = 288.

    tmpk = tmp * units.kelvin

    isentlev = [296., 297.] * units.kelvin

    with pytest.raises(ValueError):

        isentropic_interpolation(isentlev, lev, tmpk)

def test_isentropic_pressure_data_bounds_error():