Run test with array's associated device context
@@ -66,6 +66,7 @@ class TestUnownedMemory(unittest.TestCase):
# Delete the source object
del src_mem_ptr
+ with device.Device(device_id):
arr[:] = 2
assert (arr == 2).all()
