Function CheckBuffersOnDevice

cpp/src/arrow/flight/test_definitions.cc:1199–1211

namespace {

Status CheckBuffersOnDevice(const Array& array, const Device& device) {
  if (array.num_fields() != 0) {
    return Status::NotImplemented("Nested arrays");
  }
  for (const auto& buffer : array.data()->buffers) {
    if (!buffer) continue;
    if (!buffer->device()->Equals(device)) {
      return Status::Invalid("Expected buffer on device: ", device.ToString(),
                             ". Was allocated on device: ", buffer->device()->ToString());
    }
  }
  return Status::OK();
}

Status CheckBuffersOnDevice(const RecordBatch& batch, const Device& device) {
  for (const auto& column : batch.columns()) {