Unidata / MetPy

    assert_almost_equal(isentprs[0][1], trueprs, 3)

def test_isentropic_pressure_p_increase():

    """Test calculation of isentropic pressure function, p increasing order."""

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

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

    tmp[0, :] = 288.

    tmp[1, :] = 290.

    tmp[2, :] = 292.

    tmp[3, :] = 296.

    tmpk = tmp * units.kelvin

    isentlev = [296.] * units.kelvin

    isentprs = get_isentropic_pressure(isentlev, lev, tmpk)

    trueprs = 1000. * units.hPa

    assert_almost_equal(isentprs[0][1], trueprs, 3)

def test_isentropic_pressure_adition_args():

    assert_almost_equal(isentprs[1][0][1], truerh, 3)

def test_isentropic_pressure_tmp_out():

    """Test calculation of isentropic pressure function, temperature output."""

    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.] * units.kelvin

    isentprs = get_isentropic_pressure(isentlev, lev, tmpk, tmpk_out=True)

    truetmp = 296. * units.kelvin

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