Data type of patch must match.

bluesky · Nov 18, 2024 · 24b9027 · 24b9027
1 parent cb395d4
commit 24b9027
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/tiled/_tests/test_writing.py b/tiled/_tests/test_writing.py
@@ -169,6 +169,10 @@ def test_extend_array(tree):
         ac.patch(ones * 8, offset=8, extend=True)
         numpy.testing.assert_equal(ac[8:9], ones * 8)
 
+        # Data type must match.
+        with pytest.raises(ValueError):
+            ac.patch(ones.astype("uint8"), offset=9, extend=True)
+
 
 def test_write_dataframe_full(tree):
     with Context.from_app(

diff --git a/tiled/client/array.py b/tiled/client/array.py
@@ -233,6 +233,11 @@ def patch(self, array: NDArray, offset: Union[int, tuple[int, ...]], extend=Fals
                    [[0., 0.],
                     [0., 0.]]])
         """
+        if array.dtype != self.dtype:
+            raise ValueError(
+                f"Data given to patch has dtype {array.dtype} which does not "
+                f"match the dtype of this array {self.dtype}."
+            )
         array_ = numpy.ascontiguousarray(array)
         if isinstance(offset, int):
             offset = (offset,)