diff --git a/test/test_array.py b/test/test_array.py index 4c41890303278dc1490528e74f1a0e147d1bbbd5..2f3def6f3a75b01520c37d07e0a16a408bcc0bf1 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 + + # }}}