|
@@ 212-219 (lines=8) @@
|
| 209 |
|
y_interp = log_interp(x_interp, x_log, y_log, axis=1) |
| 210 |
|
assert_array_almost_equal(y_interp[0, :, 0], y_interp_truth, 7) |
| 211 |
|
|
| 212 |
|
|
| 213 |
|
def test_log_interp_4d(): |
| 214 |
|
"""Test interpolating with log x-scale 4 dimensions.""" |
| 215 |
|
x_log = np.ones((2, 2, 3, 4)) * np.array([1e3, 1e4, 1e5, 1e6]) |
| 216 |
|
y_log = np.log(x_log) * 2 + 3 |
| 217 |
|
x_interp = np.array([5e3, 5e4, 5e5]) |
| 218 |
|
y_interp_truth = np.array([20.0343863828, 24.6395565688, 29.2447267548]) |
| 219 |
|
y_interp = log_interp(x_interp, x_log, y_log, axis=3) |
| 220 |
|
assert_array_almost_equal(y_interp[0, 0, 0, :], y_interp_truth, 7) |
| 221 |
|
|
| 222 |
|
|
|
@@ 202-209 (lines=8) @@
|
| 199 |
|
y_interp = log_interp(x_interp, x_log, y_log, axis=1) |
| 200 |
|
assert_array_almost_equal(y_interp[1], y_interp_truth, 7) |
| 201 |
|
|
| 202 |
|
|
| 203 |
|
def test_log_interp_3d(): |
| 204 |
|
"""Test interpolating with log x-scale 3 dimensions along second axis.""" |
| 205 |
|
x_log = np.ones((3, 4, 3)) * np.array([1e3, 1e4, 1e5, 1e6]).reshape(-1, 1) |
| 206 |
|
y_log = np.log(x_log) * 2 + 3 |
| 207 |
|
x_interp = np.array([5e3, 5e4, 5e5]) |
| 208 |
|
y_interp_truth = np.array([20.0343863828, 24.6395565688, 29.2447267548]) |
| 209 |
|
y_interp = log_interp(x_interp, x_log, y_log, axis=1) |
| 210 |
|
assert_array_almost_equal(y_interp[0, :, 0], y_interp_truth, 7) |
| 211 |
|
|
| 212 |
|
|
Unidata /
MetPy
