Simiplify buffer equality check

anjakefala · Sep 6, 2024 · a64d54d · a64d54d
1 parent 129b2f7
commit a64d54d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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