Merge pull request #27862 from dstansby/kernel-array

QuLogic · web-flow · commit 2d5e29329ab9 · 2024-03-07T01:48:03.000-05:00
Add dtype/copy args to internal testing class
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):
