Function ValidateTakeXAImpl

cpp/src/arrow/compute/kernels/vector_selection_test.cc:1382–1415

template <typename ValuesType, typename IndexType>
void ValidateTakeXAImpl(const std::shared_ptr<Array>& values,
                        const std::shared_ptr<Array>& indices,
                        const std::shared_ptr<Array>& result) {
  using ValuesArrayType = typename TypeTraits<ValuesType>::ArrayType;
  using IndexArrayType = typename TypeTraits<IndexType>::ArrayType;
  auto typed_values = checked_pointer_cast<ValuesArrayType>(values);
  auto typed_result = checked_pointer_cast<ValuesArrayType>(result);
  auto typed_indices = checked_pointer_cast<IndexArrayType>(indices);
  for (int64_t i = 0; i < indices->length(); ++i) {
    if (typed_indices->IsNull(i) || typed_values->IsNull(typed_indices->Value(i))) {
      ASSERT_TRUE(result->IsNull(i)) << i;
      // The value of a null element is undefined, but right
      // out of the Take kernel it is expected to be 0.
      if constexpr (is_primitive(ValuesType::type_id)) {
        if constexpr (ValuesType::type_id == Type::BOOL) {
          ASSERT_EQ(typed_result->Value(i), false);
        } else {
          ASSERT_EQ(typed_result->Value(i), 0);
        }
      }
    } else {
      ASSERT_FALSE(result->IsNull(i)) << i;
      ASSERT_EQ(typed_result->GetView(i), typed_values->GetView(typed_indices->Value(i)))
          << i;
    }
  }
  // DoCheckTakeCACWithArrays transforms the indices which has a risk of
  // overflow, so we only call it if the index type is not too wide.
  if (indices->type()->byte_width() <= 4) {
    auto cast_options = CastOptions::Safe(TypeHolder{int64()});
    ASSERT_OK_AND_ASSIGN(auto indices64, Cast(indices, cast_options));
    DoCheckTakeCACWithArrays(values, indices64.make_array(), /*expected=*/result);
  }
}

template <typename ValuesType>
void ValidateTakeXA(const std::shared_ptr<Array>& values,