From 6eaaf39bb4d90e811a654168f2a11dc7c95ec9d9 Mon Sep 17 00:00:00 2001 From: Pascal Bourgault Date: Wed, 7 Aug 2024 13:12:55 -0400 Subject: [PATCH] Rewrite test explicitly --- tests/test_sdba/test_adjustment.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_sdba/test_adjustment.py b/tests/test_sdba/test_adjustment.py index 4dc1a9735..0fdb354d2 100644 --- a/tests/test_sdba/test_adjustment.py +++ b/tests/test_sdba/test_adjustment.py @@ -549,7 +549,7 @@ def test_mon_U(self, mon_series, series, mon_triangular, kind, name, random): def test_add_dims(self, use_dask, open_dataset): with set_options(sdba_encode_cf=use_dask): if use_dask: - chunks = {"location": 1} + chunks = {"location": -1} else: chunks = None @@ -571,9 +571,9 @@ def test_add_dims(self, use_dask, open_dataset): .tasmax ) ref = convert_units_to(ref, "K") - ref = ref.isel(location=1, drop=True).expand_dims(location=["Amos"]) - # The idea is to have the same ref on all locations - ref, hist = xr.align(ref, hist, join="outer") + # The idea is to have ref defined only over 1 location + # But sdba needs the same dimensions on ref and hist for Grouping with add_dims + ref = ref.where(ref.location == "Amos") # With add_dims, "does it run" test group = Grouper("time.dayofyear", window=5, add_dims=["location"])