def test_visualization_plot_precip_field(
source, type, bbox, colorscale, probthr, title, colorbar, axis,
):
if type == "intensity":
field, metadata = get_precipitation_fields(0, 0, True, True, None, source)
field = field.squeeze()
field, metadata = conversion.to_rainrate(field, metadata)
elif type == "depth":
field, metadata = get_precipitation_fields(0, 0, True, True, None, source)
field = field.squeeze()
field, metadata = conversion.to_raindepth(field, metadata)
elif type == "prob":
field, metadata = get_precipitation_fields(0, 10, True, True, None, source)
field, metadata = conversion.to_rainrate(field, metadata)
field = ensemblestats.excprob(field, probthr)
ax = plot_precip_field(
field,
type=type,
bbox=bbox,
geodata=metadata,
colorscale=colorscale,
probthr=probthr,
units=metadata["unit"],
title=title,
colorbar=colorbar,
axis=axis,
)
pl.close()
def test_visualization_plot_precip_field(
source,
plot_type,
bbox,
colorscale,
probthr,
title,
colorbar,
axis,
):
if plot_type == "intensity":
field, metadata = get_precipitation_fields(0, 0, True, True, None,
source)
field = field.squeeze()
field, metadata = conversion.to_rainrate(field, metadata)
elif plot_type == "depth":
field, metadata = get_precipitation_fields(0, 0, True, True, None,
source)
field = field.squeeze()
field, metadata = conversion.to_raindepth(field, metadata)
elif plot_type == "prob":
field, metadata = get_precipitation_fields(0, 10, True, True, None,
source)
field, metadata = conversion.to_rainrate(field, metadata)
field = ensemblestats.excprob(field, probthr)
field_orig = field.copy()
ax = plot_precip_field(
field.copy(),
type=plot_type,
bbox=bbox,
geodata=None,
colorscale=colorscale,
probthr=probthr,
units=metadata["unit"],
title=title,
colorbar=colorbar,
axis=axis,
)
# Check that plot_precip_field does not modify the input data
field_orig = np.ma.masked_invalid(field_orig)
field_orig.data[field_orig.mask] = -100
field = np.ma.masked_invalid(field)
field.data[field.mask] = -100
assert np.array_equal(field_orig.data, field.data)
def test_to_raindepth(R, metadata, expected):
"""Test the to_raindepth."""
assert_array_almost_equal(conversion.to_raindepth(R, metadata)[0], expected)