From 5ad2d31e124b538e00879295b966247898b1ed44 Mon Sep 17 00:00:00 2001 From: anjakefala Date: Fri, 6 Sep 2024 18:57:12 +0000 Subject: [PATCH] Add missing assert --- python/pyarrow/tests/test_cuda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pyarrow/tests/test_cuda.py b/python/pyarrow/tests/test_cuda.py index c85a36e4de4b6..9e9d543ee1cea 100644 --- a/python/pyarrow/tests/test_cuda.py +++ b/python/pyarrow/tests/test_cuda.py @@ -128,7 +128,7 @@ def test_copy_from_buffer(): buf2 = cudabuf.copy(dest) assert buf2.device_type == pa.DeviceAllocationType.CUDA cudabuf2 = cuda.CudaBuffer.from_buffer(buf2) - cudabuf2.size == cudabuf.size + assert cudabuf2.size == cudabuf.size assert not cudabuf2.is_cpu assert cudabuf2.device_type == pa.DeviceAllocationType.CUDA