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'])