def test_smooth_goddard(fosi_3d, dple_3d):
"""Test whether goddard smoothing function reduces ntime."""
hindcast = HindcastEnsemble(dple_3d.isel(nlat=slice(1, None)))
hindcast.add_reference(fosi_3d.isel(nlat=slice(1, None)), 'reconstruction')
hindcast.add_uninitialized(fosi_3d.isel(nlat=slice(1, None)))
initialized_before = hindcast.initialized
hindcast.smooth(smooth_kws='goddard2013')
actual_initialized = hindcast.initialized
dim = 'lead'
assert actual_initialized[dim].size < initialized_before[dim].size
for dim in ['nlon', 'nlat']:
assert actual_initialized[dim[1:]].size < initialized_before[dim].size
def test_calendar_matching_uninitialized(hind_ds_initialized_1d,
hist_ds_uninitialized_1d):
"""Tests that error is thrown if calendars mismatch when adding uninitialized."""
hindcast = HindcastEnsemble(hind_ds_initialized_1d)
hist_ds_uninitialized_1d["time"] = xr.cftime_range(
start="1950",
periods=hist_ds_uninitialized_1d.time.size,
freq="MS",
calendar="all_leap",
)
with pytest.raises(ValueError, match="does not match"):
hindcast.add_uninitialized(hist_ds_uninitialized_1d)
def test_compute_uninitialized(initialized_ds, uninitialized_ds,
reconstruction_ds, observations_ds):
"""Test to see if compute_uninitialized can be frun from the HindcastEnsemble"""
hindcast = HindcastEnsemble(initialized_ds)
hindcast.add_reference(reconstruction_ds, 'reconstruction')
hindcast.add_uninitialized(uninitialized_ds)
# single reference, no declaration of name.
hindcast.compute_uninitialized()
hindcast.add_reference(observations_ds, 'observations')
# multiple references, no name declaration.
hindcast.compute_uninitialized()
# multiple references, call one.
hindcast.compute_uninitialized('reconstruction')
hindcast.compute_uninitialized(metric='rmse', comparison='m2r')
def test_inplace(hind_ds_initialized_1d, reconstruction_ds_1d,
hist_ds_uninitialized_1d):
"""Tests that inplace operations do not work."""
hindcast = HindcastEnsemble(hind_ds_initialized_1d)
# Adding observations.
hindcast.add_observations(reconstruction_ds_1d, 'FOSI')
with_obs = hindcast.add_observations(reconstruction_ds_1d, 'FOSI')
assert hindcast != with_obs
# Adding an uninitialized ensemble.
hindcast.add_uninitialized(hist_ds_uninitialized_1d)
with_uninit = hindcast.add_uninitialized(hist_ds_uninitialized_1d)
assert hindcast != with_uninit
# Applying arbitrary func.
hindcast.sum('init')
summed = hindcast.sum('init')
assert hindcast != summed
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 hindcast_hist_obs_1d(hind_ds_initialized_1d, hist_ds_uninitialized_1d,
observations_ds_1d):
"""HindcastEnsemble initialized with `initialized`, `uninitialzed` and `obs`."""
hindcast = HindcastEnsemble(hind_ds_initialized_1d)
hindcast = hindcast.add_uninitialized(hist_ds_uninitialized_1d)
hindcast = hindcast.add_observations(observations_ds_1d)
hindcast = hindcast - hindcast.sel(
time=slice("1964", "2014")).mean("time").sel(
init=slice("1964", "2014")).mean("init")
return hindcast
def test_smooth_coarsen(reconstruction_ds_3d, hind_ds_initialized_3d):
"""Test whether coarsening reduces dim.size."""
hindcast = HindcastEnsemble(hind_ds_initialized_3d)
hindcast = hindcast.add_observations(reconstruction_ds_3d,
'reconstruction')
hindcast = hindcast.add_uninitialized(reconstruction_ds_3d)
initialized_before = hindcast._datasets['initialized']
dim = 'nlon'
hindcast = hindcast.smooth(smooth_kws={dim: 2})
actual_initialized = hindcast._datasets['initialized']
assert initialized_before[dim].size // 2 == actual_initialized[dim].size
def test_smooth_goddard(reconstruction_ds_3d, hind_ds_initialized_3d):
"""Test whether goddard smoothing function reduces ntime."""
hindcast = HindcastEnsemble(
hind_ds_initialized_3d.isel(nlat=slice(1, None)))
hindcast = hindcast.add_observations(
reconstruction_ds_3d.isel(nlat=slice(1, None)), 'reconstruction')
hindcast = hindcast.add_uninitialized(
reconstruction_ds_3d.isel(nlat=slice(1, None)))
initialized_before = hindcast._datasets['initialized']
hindcast = hindcast.smooth(smooth_kws='goddard2013')
actual_initialized = hindcast._datasets['initialized']
dim = 'lead'
assert actual_initialized[dim].size < initialized_before[dim].size
for dim in ['nlon', 'nlat']:
assert actual_initialized[dim[1:]].size < initialized_before[dim].size
def test_HindcastEnsemble_plot(
hind_ds_initialized_1d,
hist_ds_uninitialized_1d,
reconstruction_ds_1d,
observations_ds_1d,
variable,
show_members,
cmap,
):
"""Test PredictionEnsemble.plot()."""
he = HindcastEnsemble(hind_ds_initialized_1d)
kws = {"cmap": cmap, "show_members": show_members, "variable": variable}
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)
def test_hindcastEnsemble_plus_hindcastEnsemble(hind_ds_initialized_1d,
hist_ds_uninitialized_1d,
observations_ds_1d, operator):
"""Test that HindcastEnsemble math operator (+-*/) HindcastEnsemble works
correctly."""
he = HindcastEnsemble(hind_ds_initialized_1d)
he = he.add_uninitialized(hist_ds_uninitialized_1d)
he = he.add_observations(observations_ds_1d)
other = he.mean("init")
operator = eval(operator)
he2 = operator(he, other)
for dataset in he._datasets:
if he._datasets[dataset]:
print("check", dataset)
assert_equal(
he2._datasets[dataset],
operator(he._datasets[dataset], other._datasets[dataset]),
)
# check same dims and data_vars as before
check_dataset_dims_and_data_vars(he, he2, dataset)
def test_add_hist_da_uninitialized_1d(hind_ds_initialized_1d,
hist_da_uninitialized_1d):
"""Test to see if da uninitialized ensemble can be added to the HindcastEnsemble"""
hindcast = HindcastEnsemble(hind_ds_initialized_1d)
hindcast = hindcast.add_uninitialized(hist_da_uninitialized_1d)
assert hindcast.get_uninitialized()
def test_get_uninitialized(hind_ds_initialized_1d, hist_ds_uninitialized_1d):
"""Test whether get_uninitialized method works."""
hindcast = HindcastEnsemble(hind_ds_initialized_1d)
hindcast = hindcast.add_uninitialized(hist_ds_uninitialized_1d)
uninit = hindcast.get_uninitialized()
assert uninit == hindcast._datasets['uninitialized']
def test_add_uninitialized(initialized_ds, uninitialized_ds):
"""Test to see if an uninitialized ensemble can be added to the HindcastEnsemble"""
hindcast = HindcastEnsemble(initialized_ds)
hindcast.add_uninitialized(uninitialized_ds)
