make sure unify_chunks does not compute.

dcherian · Oct 1, 2019 · f6dfb12 · f6dfb12
1 parent ae53b85
commit f6dfb12
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/xarray/tests/test_dask.py b/xarray/tests/test_dask.py
@@ -965,7 +965,8 @@ def test_unify_chunks(obj):
         ds_copy.chunks
 
     expected_chunks = {"x": (4, 4, 2), "y": (5, 5, 5, 5), "z": (4,)}
-    actual_chunks = ds_copy.unify_chunks().chunks
+    with raise_if_dask_computes():
+        actual_chunks = ds_copy.unify_chunks().chunks
     expected_chunks == actual_chunks
     assert_identical(map_ds, ds_copy.unify_chunks())