Unidata / MetPy

    assert_almost_equal(cin, 0.0 * units('joule / kilogram'), 6)

def test_find_append_zero_crossings():

    """Tests finding and appending zero crossings of an x, y series."""

    x = np.arange(11) * units.hPa

    y = np.array([3, 2, 1, -1, 2, 2, 0, 1, 0, -1, 2]) * units.degC

    x2, y2 = _find_append_zero_crossings(x, y)

    x_truth = np.array([0., 1., 2., 2.5, 3., 3.33333333, 4., 5.,

                        6., 7., 8., 9., 9.33333333, 10.]) * units.hPa

    y_truth = np.array([3, 2, 1, 0, -1, 0, 2, 2, 0, 1, 0, -1, 0, 2]) * units.degC

    assert_array_almost_equal(x2, x_truth, 6)

    assert_almost_equal(y2, y_truth, 6)

def test_most_unstable_parcel():

    """Tests calculating the most unstable parcel."""

    levels = np.array([1000., 959., 867.9]) * units.mbar

    temperatures = np.array([18.2, 22.2, 17.4]) * units.celsius

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

    isentprs = isentropic_interpolation(isentlev, lev, tmpk)

    trueprs = 936.18057 * units.hPa

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

def test_isentropic_pressure_adition_args_interp():

    """Test calculation of isentropic pressure function, additional args."""

    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.

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

    rh[0, :] = 100.

    rh[1, :] = 80.

    rh[2, :] = 40.

    rh[3, :] = 20.

    isentlev = [296.] * units.kelvin

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

    truetmp = 296. * units.kelvin

    assert_almost_equal(isentprs[1], truetmp, 3)

def test_isentropic_pressure_p_increase_rh_out():

    """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

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

    rh[0, :] = 20.

    rh[1, :] = 40.

    rh[2, :] = 80.

    isentlev = [296.] * units.kelvin

    isentprs = isentropic_interpolation(isentlev, lev, tmpk)

    trueprs = 1000. * units.hPa

    assert_almost_equal(isentprs[0], trueprs, 3)

def test_isentropic_pressure_adition_args():

    """Test calculation of isentropic pressure function, additional args."""

    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.

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

    rh[0, :] = 100.

    rh[1, :] = 80.

    rh[2, :] = 40.

    rh[3, :] = 20.

def test_most_unstable_parcel():

    """Tests calculating the most unstable parcel."""

    levels = np.array([1000., 959., 867.9]) * units.mbar

    temperatures = np.array([18.2, 22.2, 17.4]) * units.celsius

    dewpoints = np.array([19., 19., 14.3]) * units.celsius

    ret = most_unstable_parcel(levels, temperatures, dewpoints, depth=100 * units.hPa)

    assert_almost_equal(ret[0], 959.0 * units.hPa, 6)

    assert_almost_equal(ret[1], 22.2 * units.degC, 6)

    assert_almost_equal(ret[2], 19.0 * units.degC, 6)

def test_isentropic_pressure():

    """Test calculation of isentropic pressure function."""

    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.