Function MakeFloatBatchSized

cpp/src/arrow/ipc/test_common.cc:333–348

}

Status MakeFloatBatchSized(int length, std::shared_ptr<RecordBatch>* out, uint32_t seed) {
  // Make the schema
  auto f0 = field("f0", float16());
  auto f1 = field("f1", float32());
  auto f2 = field("f2", float64());
  auto schema = ::arrow::schema({f0, f1, f2});

  // Example data
  std::shared_ptr<Array> a0, a1, a2;
  MemoryPool* pool = default_memory_pool();
  RETURN_NOT_OK(MakeRandomArray<HalfFloatType>(length, false, pool, &a0, seed));
  RETURN_NOT_OK(MakeRandomArray<FloatType>(length, false, pool, &a1, seed + 1));
  RETURN_NOT_OK(MakeRandomArray<DoubleType>(length, true, pool, &a2, seed + 2));
  *out = RecordBatch::Make(schema, length, {a0, a1, a2});
  return Status::OK();
}

Status MakeFloatBatch(std::shared_ptr<RecordBatch>* out) {
  return MakeFloatBatchSized(10, out);