diff --git a/test/test_helpers/test_statistics.py b/test/test_helpers/test_statistics.py index a12e6dcd5d47c15ee1bc2241ce4be1fa444eee1e..6f1952a9c1df2b16a2e298b433d732d9e69200d4 100644 --- a/test/test_helpers/test_statistics.py +++ b/test/test_helpers/test_statistics.py @@ -250,13 +250,20 @@ class TestCreateBootstrapRealizations: def test_create_n_bootstrap_realizations(self, data): boot_data = create_n_bootstrap_realizations(data, dim_name_time='time', dim_name_model='model', n_boots=1000, dim_name_boots='boots') - assert isinstance(boot_data, xr.DataArray) - assert boot_data.shape == (1000, 2) + assert isinstance(boot_data, dict) + assert "" in boot_data.keys() + assert isinstance(boot_data[""], xr.DataArray) + assert boot_data[""].shape == (1000, 2) boot_data = create_n_bootstrap_realizations(data.sel(model='m1').squeeze(), dim_name_time='time', dim_name_model='model', n_boots=1000, dim_name_boots='boots') - assert isinstance(boot_data, xr.DataArray) - assert boot_data.shape == (1000,) + assert isinstance(boot_data[""], xr.DataArray) + assert boot_data[""].shape == (1000,) + + data["time"] = pd.date_range("2022-01", periods=10, freq="1m") + boot_data = create_n_bootstrap_realizations(data, dim_name_time='time', dim_name_model='model', + n_boots=100, dim_name_boots='boots', seasons=["JJA", "DJF"]) + assert sorted(list(boot_data.keys())) == sorted(["", "JJA", "DJF"]) class TestMeanSquaredError: