Add dtype/copy args to internal testing class

dstansby · dstansby · commit 5b2efbc795f8 · 2024-03-05T09:09:05.000Z
diff --git a/lib/matplotlib/tests/test_units.py b/lib/matplotlib/tests/test_units.py
@@ -282,8 +282,16 @@ class Kernel:
     def __init__(self, array):
         self._array = np.asanyarray(array)
 
-    def __array__(self):
-        return self._array
+    def __array__(self, dtype=None, copy=None):
+        if dtype is not None and dtype != self._array.dtype:
+            if copy is not None and not copy:
+                raise ValueError(
+                    f"Converting array from {self._array.dtype} to "
+                    f"{dtype} requires a copy"
+                )
+
+        arr = np.asarray(self._array, dtype=dtype)
+        return (arr if not copy else np.copy(arr))
 
     @property
     def shape(self):
