From 511e3ad5cb98967183ff2cd9584264c6cd7faaa2 Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Fri, 23 Jun 2017 00:50:38 -0500
Subject: [PATCH] Add a test for Array.copy().

---
 test/test_array.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/test/test_array.py b/test/test_array.py
index 4c418903..2f3def6f 100644
--- a/test/test_array.py
+++ b/test/test_array.py
@@ -762,6 +762,32 @@ def test_diff(ctx_factory):
             (cl.array.diff(a_dev).get() - np.diff(a)))
     assert err < 1e-4
 
+
+def test_copy(ctx_factory):
+    context = ctx_factory()
+    queue1 = cl.CommandQueue(context)
+    queue2 = cl.CommandQueue(context)
+
+    # Test copy
+
+    arr = cl.array.zeros(queue1, 100, np.int32)
+    arr_copy = arr.copy()
+
+    assert (arr == arr_copy).all().get()
+    assert arr.data != arr_copy.data
+
+    # Test queue association
+
+    arr_copy = arr.copy(queue=queue2)
+    assert arr_copy.queue == queue2
+
+    arr_copy = arr.copy(queue=None)
+    assert arr_copy.queue == None
+
+    arr_copy = arr.with_queue(None).copy(queue=queue1)
+    assert arr_copy.queue == queue1
+
+
 # }}}
 
 
-- 
GitLab