Unidata / MetPy

def test_log_interp_2d():

    """Test interpolating with log x-scale in 2 dimensions."""

    x_log = np.array([[1e3, 1e4, 1e5, 1e6], [1e3, 1e4, 1e5, 1e6]])

    y_log = np.log(x_log) * 2 + 3

    x_interp = np.array([5e3, 5e4, 5e5])

    y_interp_truth = np.array([20.0343863828, 24.6395565688, 29.2447267548])

    y_interp = log_interp(x_interp, x_log, y_log, axis=1)

    assert_array_almost_equal(y_interp[1], y_interp_truth, 7)

def test_log_interp_3d():

    """Test interpolating with log x-scale 3 dimensions along second axis."""

    x_log = np.ones((3, 4, 3)) * np.array([1e3, 1e4, 1e5, 1e6]).reshape(-1, 1)

     (np.array([794.85264282, 700., 600., 540.01696548]) * units.hPa,

      np.array([7.93049516, 0., -8.33333333, -13.14758845]) * units.degC))

])

def test_get_layer(pressure, variable, heights, bottom, depth, interp, expected):

    """Tests get_layer functionality."""

    p_layer, y_layer = get_layer(pressure, variable, heights=heights, bottom=bottom,

                                 depth=depth, interpolate=interp)

    assert_array_almost_equal(p_layer, expected[0], 5)

    assert_array_almost_equal(y_layer, expected[1], 5)

def test_log_interp_2d():

    """Test interpolating with log x-scale in 2 dimensions."""

    x_log = np.array([[1e3, 1e4, 1e5, 1e6], [1e3, 1e4, 1e5, 1e6]])

def test_log_interp_3d():

    """Test interpolating with log x-scale 3 dimensions along second axis."""

    x_log = np.ones((3, 4, 3)) * np.array([1e3, 1e4, 1e5, 1e6]).reshape(-1, 1)

    y_log = np.log(x_log) * 2 + 3

    x_interp = np.array([5e3, 5e4, 5e5])

    y_interp_truth = np.array([20.0343863828, 24.6395565688, 29.2447267548])

    y_interp = log_interp(x_interp, x_log, y_log, axis=1)

    assert_array_almost_equal(y_interp[0, :, 0], y_interp_truth, 7)

def test_log_interp_4d():

    """Test interpolating with log x-scale 4 dimensions."""

    x_log = np.ones((2, 2, 3, 4)) * np.array([1e3, 1e4, 1e5, 1e6])

    y_log = np.log(x_log) * 2 + 3

    x_interp = np.array([5e3, 5e4, 5e5])