check that errors in chunk are raised properly

pydata · Oct 24, 2024 · 8b35676 · 8b35676
1 parent 73b4466
commit 8b35676
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/xarray/tests/test_datatree.py b/xarray/tests/test_datatree.py
@@ -2296,3 +2296,12 @@ def test_chunk(self):
 
         assert_identical(actual, expected)
         assert actual.chunksizes == expected.chunksizes
+
+        with pytest.raises(TypeError, match="invalid type"):
+            tree.chunk(None)
+
+        with pytest.raises(TypeError, match="invalid type"):
+            tree.chunk((1, 2))
+
+        with pytest.raises(ValueError, match="not found in data dimensions"):
+            tree.chunk({"u": 2})