Sending array rather than images to transfer method

mantidproject · Dec 1, 2020 · b144343 · b144343
1 parent a7eecda
commit b144343
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mantidimaging/core/gpu/test/gpu_test.py b/mantidimaging/core/gpu/test/gpu_test.py
@@ -154,7 +154,7 @@ def test_gpu_running_out_of_memory_causes_free_memory_to_be_called(self):
         with mock.patch("mantidimaging.core.gpu.utility._send_single_array_to_gpu",
                         side_effect=cp.cuda.memory.OutOfMemoryError(0, 0)):
             with mock.patch("mantidimaging.core.gpu.utility._free_memory_pool") as mock_free_gpu:
-                gpu._send_arrays_to_gpu_with_pinned_memory(images, [cp.cuda.Stream() for _ in range(n_images)])
+                gpu._send_arrays_to_gpu_with_pinned_memory(images.data, [cp.cuda.Stream() for _ in range(n_images)])
 
         mock_free_gpu.assert_called()