Add __array__ method to pyopencl.Array

pyopencl/array.py

@@ -330,6 +330,7 @@ class Array:
     .. automethod :: get
     .. automethod :: get_async
     .. automethod :: copy
+    .. automethod :: __array__
 
     .. automethod :: __str__
     .. automethod :: __repr__
@@ -806,6 +807,21 @@ def copy(self, queue=_copy_queue):
 
         return result
 
+    def __array__(self, dtype=None):
+        """Return a numpy array copy of the array.
+
+        Equivalent to :meth:`get` with no arguments. If `dtype` is given, two
+        copies of the input array will be made: one for the conversion, and
+        another for the cast.
+
+        :arg dtype: a valid NumPy dtype, such as ``np.float64`` or
+                    ``'float32'``.
+
+        .. versionadded:: ...
+        """
+        out = self.get().astype(dtype)
+        return out
+
     def __str__(self):
         if self.queue is None:
             return (f"<cl.Array {self.shape} of {self.dtype} "

test/test_array.py

@@ -806,6 +806,13 @@ def test_random_int_in_range(ctx_factory, rng_class, dtype, plot_hist=False):
 
 # {{{ misc
 
+def test_array_method(ctx_factory):
+    ctx = ctx_factory()
+    queue = cl.CommandQueue(ctx)
+    arr = make_random_array(queue, np.float32, 1024)
+    np.testing.assert_array_equal(arr, arr.get())
+
+
 def test_numpy_integer_shape(ctx_factory):
     try:
         list(np.int32(17))