From a64d54dfd1c1a50fc34ea85bfae7d0529e7be26a Mon Sep 17 00:00:00 2001 From: anjakefala Date: Fri, 6 Sep 2024 18:01:47 +0000 Subject: [PATCH] Simiplify buffer equality check --- python/pyarrow/tests/test_cuda.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/pyarrow/tests/test_cuda.py b/python/pyarrow/tests/test_cuda.py index 29b322e695a57..c85a36e4de4b6 100644 --- a/python/pyarrow/tests/test_cuda.py +++ b/python/pyarrow/tests/test_cuda.py @@ -132,8 +132,7 @@ def test_copy_from_buffer(): assert not cudabuf2.is_cpu assert cudabuf2.device_type == pa.DeviceAllocationType.CUDA - arr2 = np.frombuffer(cudabuf2.copy_to_host(), dtype=np.uint8) - np.testing.assert_equal(arr, arr2) + assert cudabuf2.copy_to_host().equals(buf) assert cudabuf2.device == mm2.device