Function AssertAppendScalar

cpp/src/arrow/array/array_test.cc:617–664

}

void AssertAppendScalar(MemoryPool* pool, const std::shared_ptr<Scalar>& scalar) {
  std::unique_ptr<arrow::ArrayBuilder> builder;
  auto null_scalar = MakeNullScalar(scalar->type);
  ASSERT_OK(MakeBuilderExactIndex(pool, scalar->type, &builder));
  ASSERT_OK(builder->AppendScalar(*scalar));                 // [0] = scalar
  ASSERT_OK(builder->AppendScalar(*scalar));                 // [1] = scalar
  ASSERT_OK(builder->AppendScalar(*null_scalar));            // [2] = NULL
  ASSERT_OK(builder->AppendScalars({scalar, null_scalar}));  // [3, 4] = {scalar, NULL}
  ASSERT_OK(
      builder->AppendScalar(*scalar, /*n_repeats=*/2));  // [5, 6] = {scalar, scalar}
  ASSERT_OK(
      builder->AppendScalar(*null_scalar, /*n_repeats=*/2));  // [7, 8] = {NULL, NULL}

  std::shared_ptr<Array> out;
  FinishAndCheckPadding(builder.get(), &out);
  ASSERT_OK(out->ValidateFull());
  AssertTypeEqual(scalar->type, out->type());
  ASSERT_EQ(out->length(), 9);

  auto out_type_id = out->type()->id();
  const bool can_check_nulls = internal::may_have_validity_bitmap(out_type_id);
  // For a dictionary builder, the output dictionary won't necessarily be the same
  const bool can_check_values = !is_dictionary(out_type_id);

  if (can_check_nulls) {
    ASSERT_EQ(out->null_count(), 4);
  } else {
    ASSERT_EQ(out->null_count(), 0);
  }
  if (scalar->is_valid) {
    ASSERT_EQ(out->ComputeLogicalNullCount(), 4);
  } else {
    ASSERT_EQ(out->ComputeLogicalNullCount(), 9);
  }

  for (const auto index : {0, 1, 3, 5, 6}) {
    ASSERT_NE(out->IsNull(index), scalar->is_valid);
    ASSERT_OK_AND_ASSIGN(auto scalar_i, out->GetScalar(index));
    ASSERT_OK(scalar_i->ValidateFull());
    if (can_check_values) AssertScalarsEqual(*scalar, *scalar_i, /*verbose=*/true);
  }
  for (const auto index : {2, 4, 7, 8}) {
    ASSERT_TRUE(out->IsNull(index));
    ASSERT_OK_AND_ASSIGN(auto scalar_i, out->GetScalar(index));
    ASSERT_OK(scalar_i->ValidateFull());
    AssertScalarsEqual(*null_scalar, *scalar_i, /*verbose=*/true);
  }
}

static ScalarVector GetScalars() {
  auto hello = Buffer::FromString("hello");