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)