def test_sigma_to_pressure():
"""Test sigma_to_pressure."""
surface_pressure = 1000. * units.hPa
model_top_pressure = 0. * units.hPa
sigma = np.arange(0., 1.1, 0.1)
expected = np.arange(0., 1100., 100.) * units.hPa
pressure = sigma_to_pressure(sigma, surface_pressure, model_top_pressure)
assert_array_almost_equal(pressure, expected, 5)
def test_sigma_to_pressure():
"""Test sigma_to_pressure."""
surface_pressure = 1000. * units.hPa
model_top_pressure = 0. * units.hPa
sigma = np.arange(0., 1.1, 0.1)
expected = np.arange(0., 1100., 100.) * units.hPa
pressure = sigma_to_pressure(sigma, surface_pressure, model_top_pressure)
assert_array_almost_equal(pressure, expected, 5)
def test_sigma_to_pressure(array_type):
"""Test sigma_to_pressure."""
surface_pressure = 1000. * units.hPa
model_top_pressure = 0. * units.hPa
sigma_values = np.arange(0., 1.1, 0.1)
mask = np.zeros_like(sigma_values)[::2] = 1
sigma = array_type(sigma_values, '', mask=mask)
expected = array_type(np.arange(0., 1100., 100.), 'hPa', mask=mask)
pressure = sigma_to_pressure(sigma, surface_pressure, model_top_pressure)
assert_array_almost_equal(pressure, expected, 5)
