Function GenericTestSlice

cpp/src/arrow/compute/light_array_test.cc:237–261

template <typename OffsetType>
void GenericTestSlice(const std::shared_ptr<DataType>& type, const char* json_data,
                      const std::vector<SliceTestCase>& testCases) {
  auto array = ArrayFromJSON(type, json_data);
  KeyColumnArray kc_array =
      ColumnArrayFromArrayData(array->data(), 0, array->length()).ValueOrDie();

  for (const auto& testCase : testCases) {
    ARROW_SCOPED_TRACE("Offset: ", testCase.offset, " Length: ", testCase.length);
    KeyColumnArray sliced = kc_array.Slice(testCase.offset, testCase.length);

    // Extract binary data from the sliced KeyColumnArray
    std::vector<std::string> sliced_data;
    const auto* offset_data = reinterpret_cast<const OffsetType*>(sliced.data(1));
    const auto* string_data = reinterpret_cast<const char*>(sliced.data(2));

    for (auto i = 0; i < testCase.length; ++i) {
      auto start = offset_data[i];
      auto end = offset_data[i + 1];
      sliced_data.push_back(std::string(string_data + start, string_data + end));
    }

    // Compare the sliced values to the expected string
    ASSERT_EQ(testCase.expected, sliced_data);
  }
}

TEST(KeyColumnArray, SliceBinaryTest) {
  const char* json_test_strings = R"(["Hello", "World", "Slice", "Binary", "Test"])";