Function CheckStructToStructSubset

cpp/src/arrow/compute/kernels/scalar_cast_test.cc:3814–4227

}

static void CheckStructToStructSubset(
    const std::vector<std::shared_ptr<DataType>>& value_types) {
  for (const auto& src_value_type : value_types) {
    ARROW_SCOPED_TRACE("From type: ", src_value_type->ToString());
    for (const auto& dest_value_type : value_types) {
      ARROW_SCOPED_TRACE("To type: ", dest_value_type->ToString());

      std::vector<std::string> field_names = {"a", "b", "c", "d", "e"};

      std::shared_ptr<Array> a1, b1, c1, d1, e1;
      a1 = ArrayFromJSON(src_value_type, "[1, 2, 5]");
      b1 = ArrayFromJSON(src_value_type, "[3, 4, 7]");
      c1 = ArrayFromJSON(src_value_type, "[9, 11, 44]");
      d1 = ArrayFromJSON(src_value_type, "[6, 51, 49]");
      e1 = ArrayFromJSON(src_value_type, "[19, 17, 74]");

      std::shared_ptr<Array> a2, b2, c2, d2, e2;
      a2 = ArrayFromJSON(dest_value_type, "[1, 2, 5]");
      b2 = ArrayFromJSON(dest_value_type, "[3, 4, 7]");
      c2 = ArrayFromJSON(dest_value_type, "[9, 11, 44]");
      d2 = ArrayFromJSON(dest_value_type, "[6, 51, 49]");
      e2 = ArrayFromJSON(dest_value_type, "[19, 17, 74]");

      auto nulls = ArrayFromJSON(dest_value_type, "[null, null, null]");

      ASSERT_OK_AND_ASSIGN(auto src,
                           StructArray::Make({a1, b1, c1, d1, e1}, field_names));
      ASSERT_OK_AND_ASSIGN(auto dest1,
                           StructArray::Make({a2}, std::vector<std::string>{"a"}));
      CheckCast(src, dest1);

      ASSERT_OK_AND_ASSIGN(
          auto dest2, StructArray::Make({b2, c2}, std::vector<std::string>{"b", "c"}));
      CheckCast(src, dest2);

      ASSERT_OK_AND_ASSIGN(
          auto dest3,
          StructArray::Make({c2, d2, e2}, std::vector<std::string>{"c", "d", "e"}));
      CheckCast(src, dest3);

      ASSERT_OK_AND_ASSIGN(
          auto dest4, StructArray::Make({a2, b2, c2, e2},
                                        std::vector<std::string>{"a", "b", "c", "e"}));
      CheckCast(src, dest4);

      ASSERT_OK_AND_ASSIGN(
          auto dest5, StructArray::Make({a2, b2, c2, d2, e2}, {"a", "b", "c", "d", "e"}));
      CheckCast(src, dest5);

      // field does not exist
      ASSERT_OK_AND_ASSIGN(auto dest6,
                           StructArray::Make({a2, d2, nulls}, {"a", "d", "f"}));
      CheckCast(src, dest6);

      const auto dest7 = arrow::struct_(
          {std::make_shared<Field>("a", int8()), std::make_shared<Field>("d", int16()),
           std::make_shared<Field>("f", int64(), /*nullable=*/false)});
      const auto options7 = CastOptions::Safe(dest7);