Catch re-chunking warning

pnuu · Dec 12, 2023 · 5f00f09 · 5f00f09
1 parent a7f93eb
commit 5f00f09
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/satpy/tests/test_resample.py b/satpy/tests/test_resample.py
@@ -247,8 +247,12 @@ def test_expand_reduce_agg_rechunk(self):
         into that chunk size.
 
         """
+        from satpy.utils import PerformanceWarning
+
         d_arr = da.zeros((6, 20), chunks=3)
-        new_data = NativeResampler._expand_reduce(d_arr, {0: 0.5, 1: 0.5})
+        text = "Array chunk size is not divisible by aggregation factor. Re-chunking to continue native resampling."
+        with pytest.warns(PerformanceWarning, match=text):
+            new_data = NativeResampler._expand_reduce(d_arr, {0: 0.5, 1: 0.5})
         assert new_data.shape == (3, 10)
 
     def test_expand_reduce_numpy(self):