From e48f3e1659f46aea8d34c49390eb8bab91c33c67 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Mon, 16 May 2022 15:40:56 +0200
Subject: [PATCH] update test
---
test/test_helpers/test_statistics.py | 15 +++++++++++----
1 file changed, 11 insertions(+), 4 deletions(-)
diff --git a/test/test_helpers/test_statistics.py b/test/test_helpers/test_statistics.py
index a12e6dcd..6f1952a9 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:
--
GitLab