def test_basic(self):
"""Basic calculation of some saturated vapour pressures."""
data = np.array([[260.0, 270.0, 280.0]], dtype=np.float32)
plugin = SaturatedVapourPressureTable()
result = plugin.saturation_vapour_pressure_goff_gratch(data)
expected = 0.01 * np.array([[195.6419, 469.67078, 990.9421]])
self.assertArrayAlmostEqual(result, expected)
def test_coordinate_values(self):
"""Test that returned cube temperature coordinate has expected
values."""
t_min, t_max, t_increment = 183.15, 338.15, 10.0
expected = np.arange(t_min, t_max, t_increment)
result = SaturatedVapourPressureTable(
t_min=t_min, t_max=t_max, t_increment=t_increment).process()
self.assertArrayAlmostEqual(
result.coord("air_temperature").points, expected)
def test_cube_values(self):
"""Test that returned cube has expected values."""
t_min, t_max, t_increment = 183.15, 338.15, 10.0
expected = [
0.0096646,
0.0546844,
0.2613554,
1.0799927,
3.9333663,
12.8286096,
37.9714586,
103.1532749,
259.6617372,
610.6359361,
1227.0888425,
2337.0801979,
4242.7259947,
7377.3294046,
12338.9996048,
19925.4362844,
]
result = SaturatedVapourPressureTable(
t_min=t_min, t_max=t_max, t_increment=t_increment).process()
self.assertArrayAlmostEqual(result.data, expected)
def test_cube_attributes(self):
"""Test that returned cube has appropriate attributes."""
t_min, t_max, t_increment = 200.15, 220.15, 10.0
result = SaturatedVapourPressureTable(
t_min=t_min, t_max=t_max, t_increment=t_increment).process()
self.assertEqual(result.attributes["minimum_temperature"], t_min)
self.assertEqual(result.attributes["maximum_temperature"], t_max)
self.assertEqual(result.attributes["temperature_increment"],
t_increment)
self.assertEqual(result.units, Unit("Pa"))
def _svp_table():
"""
Calculate a saturated vapour pressure (SVP) lookup table.
The lru_cache decorator caches this table on first call to this function,
so that the table does not need to be re-calculated if used multiple times.
A value of SVP for any temperature between T_MIN and T_MAX (inclusive) can be
obtained by interpolating through the table, as is done in the _svp_from_lookup
function.
Returns:
numpy.ndarray:
Array of saturated vapour pressures (Pa).
"""
svp_data = SaturatedVapourPressureTable(
t_min=SVP_T_MIN, t_max=SVP_T_MAX,
t_increment=SVP_T_INCREMENT).process()
return svp_data.data
def check_svp_table(t_min, t_max, t_increment, expected):
"""Recreate part of table and compare with expected values."""
result = SaturatedVapourPressureTable(
t_min=t_min, t_max=t_max, t_increment=t_increment).process()
np.testing.assert_allclose(result.data, expected, rtol=1.e-5)
def test_basic(self):
"""Test that the __repr__ returns the expected string."""
result = str(SaturatedVapourPressureTable())
msg = ("<SaturatedVapourPressureTable: t_min: {}; t_max: {}; "
"t_increment: {}>".format(183.15, 338.25, 0.1))
self.assertEqual(result, msg)