def test_radar2mat_with_multiple_radars_returns_expected_shape():
radars = [pyart.io.read(f) for f in CANADA_IRIS_DOPVOL_FILES]
data, _, _, _, _ = radar2mat(radars,
coords='cartesian',
dim=200,
r_max=150000)
assert data.shape == (3, 3, 200, 200)
def test_radar2mat_returns_expected_shape_legacy(legacy_radar):
data, fields, _, _, _ = radar2mat(legacy_radar,
coords='cartesian',
elevs=[0.5, 1.5, 2.5, 3.5, 4.5],
dim=100,
r_max=150000)
assert data.shape == (3, 5, 100, 100)
def test_volume_mosaic_returns_expected_shape(dualpol_radar):
data, fields, _, _, _ = radar2mat(dualpol_radar,
coords='cartesian',
elevs=[0.5, 1.5, 2.5, 3.5, 4.5],
dim=100,
r_max=150000)
mosaic_im = volume_mosaic(data, fields)
assert mosaic_im.shape == (500, 600, 4)
def test_radar2mat_value_error_if_bad_coords(dualpol_radar):
with pytest.raises(ValueError):
f = radar2mat(dualpol_radar, coords='foo')
def test_radar2mat_warns_if_field_unavailable(legacy_radar):
with pytest.warns(UserWarning):
radar2mat(legacy_radar,
fields=['differential_reflectivity', 'velocity'])
def test_radar2mat_raises_value_error_for_bad_field(dualpol_radar):
with pytest.raises(ValueError):
f = radar2mat(dualpol_radar, fields=['reflectivity', 'foo'])