Add missing assert

anjakefala · Sep 6, 2024 · 5ad2d31 · 5ad2d31
1 parent a64d54d
commit 5ad2d31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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