add additional assert

jorisvandenbossche · May 30, 2024 · b46a237 · b46a237
1 parent 9a1c54b
commit b46a237
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/pyarrow/tests/test_device.py b/python/pyarrow/tests/test_device.py
@@ -37,6 +37,7 @@ def test_buffer_device():
     assert buf.device_type == pa.DeviceAllocationType.CPU
     assert isinstance(buf.device, pa.Device)
     assert isinstance(buf.memory_manager, pa.MemoryManager)
+    assert buf.is_cpu
     assert buf.device.is_cpu
     assert buf.device == pa.default_cpu_memory_manager().device
     assert buf.memory_manager.is_cpu