def test_PredictionEnsemble_plot(
hind_ds_initialized_1d,
hist_ds_uninitialized_1d,
reconstruction_ds_1d,
observations_ds_1d,
variable,
show_members,
x,
):
"""Test PredictionEnsemble.plot()."""
he = HindcastEnsemble(hind_ds_initialized_1d)
kws = {"show_members": show_members, "variable": variable, "x": x}
he.plot(**kws)
he = he.add_uninitialized(hist_ds_uninitialized_1d)
he.plot(**kws)
he = he.add_observations(reconstruction_ds_1d)
he.plot(**kws)
he = he.add_observations(observations_ds_1d)
he.plot(**kws)
if x == "time":
pm = PerfectModelEnsemble(hind_ds_initialized_1d)
pm.plot(**kws)
pm = pm.add_control(hist_ds_uninitialized_1d.isel(member=0, drop=True))
pm.plot(**kws)
def test_PerfectModelEnsemble_plot(PM_ds_initialized_1d, PM_ds_control_1d,
variable, show_members, cmap):
"""Test PredictionEnsemble.plot()."""
pm = PerfectModelEnsemble(PM_ds_initialized_1d)
kws = {"cmap": cmap, "show_members": show_members, "variable": variable}
pm.plot(**kws)
pm = pm.add_control(PM_ds_control_1d)
pm.plot(**kws)
pm = pm.generate_uninitialized()
pm.plot(**kws)
def test_PerfectModelEnsemble_plot_fails_3d(PM_ds_initialized_3d):
"""Test PredictionEnsemble.plot()."""
pm = PerfectModelEnsemble(PM_ds_initialized_3d)
with pytest.raises(DimensionError) as excinfo:
pm.plot()
assert "does not allow dimensions other" in str(excinfo.value)
