def test_get_model_heights(fake_model_file):
obj = model.Model(str(fake_model_file), ALT_SITE)
assert_array_equal(
obj.model_heights,
ALT_SITE
+ np.array([[1000, 2000, 3000, 4000], [1000, 2000, 3000, 4000], [1000, 2000, 3000, 4000]]),
)
def test_interpolate_to_common_height(fake_model_file):
obj = model.Model(str(fake_model_file), ALT_SITE)
radar_wl_band = 0
obj.interpolate_to_common_height(radar_wl_band)
for key in ('uwind', 'vwind', 'q', 'temperature', 'pressure', 'rh',
'gas_atten', 'specific_gas_atten',
'specific_saturated_gas_atten', 'specific_liquid_atten'):
assert key in obj.data_sparse
def test_interpolate_to_grid(fake_model_file):
obj = model.Model(str(fake_model_file), ALT_SITE)
radar_wl_band = 0
obj.interpolate_to_common_height(radar_wl_band)
time_grid = np.array([1, 3])
height_grid = np.array([1, 3])
obj.interpolate_to_grid(time_grid, height_grid)
assert_array_equal(obj.height, height_grid)
assert hasattr(obj, 'data_dense')
def test_interpolate_to_common_height(fake_model_file):
obj = model.Model(str(fake_model_file), ALT_SITE)
radar_wl_band = 0
obj.interpolate_to_common_height(radar_wl_band)
for key in (
"uwind",
"vwind",
"q",
"temperature",
"pressure",
"rh",
"gas_atten",
"specific_gas_atten",
"specific_saturated_gas_atten",
"specific_liquid_atten",
):
assert key in obj.data_sparse
def test_mean_height(fake_model_file):
obj = model.Model(str(fake_model_file), ALT_SITE)
assert_array_equal(obj.mean_height,
ALT_SITE + np.array([1000, 2000, 3000, 4000]))