Function TEST

cpp/src/arrow/compute/kernels/codegen_internal_test.cc:30–70

namespace internal {

TEST(TestDispatchBest, CastBinaryDecimalArgs) {
  std::vector<TypeHolder> args;
  std::vector<DecimalPromotion> modes = {
      DecimalPromotion::kAdd, DecimalPromotion::kMultiply, DecimalPromotion::kDivide};

  // Any float -> all float
  for (auto mode : modes) {
    args = {decimal128(3, 2), float32(), float64()};
    ASSERT_OK(CastBinaryDecimalArgs(mode, &args));
    AssertTypeEqual(*args[0], *float64());
    AssertTypeEqual(*args[1], *float64());
  }

  // Integer -> decimal with common scale
  args = {decimal128(1, 0), int64()};
  ASSERT_OK(CastBinaryDecimalArgs(DecimalPromotion::kAdd, &args));
  AssertTypeEqual(*args[0], *decimal128(1, 0));
  AssertTypeEqual(*args[1], *decimal128(19, 0));

  // Add: rescale so all have common scale
  args = {decimal128(3, 2), decimal128(3, -2)};
  EXPECT_RAISES_WITH_MESSAGE_THAT(
      NotImplemented, ::testing::HasSubstr("Decimals with negative scales not supported"),
      CastBinaryDecimalArgs(DecimalPromotion::kAdd, &args));

  // Non-castable -> unchanged
  for (const auto promotion :
       {DecimalPromotion::kAdd, DecimalPromotion::kMultiply, DecimalPromotion::kDivide}) {
    for (const auto& args : std::vector<std::vector<TypeHolder>>{
             {decimal128(3, 2), boolean()},
             {boolean(), decimal128(3, 2)},
             {decimal128(3, 2), utf8()},
             {utf8(), decimal128(3, 2)},
         }) {
      auto args_copy = args;
      ASSERT_OK(CastBinaryDecimalArgs(promotion, &args_copy));
      AssertTypeEqual(*args_copy[0], *args[0]);
      AssertTypeEqual(*args_copy[1], *args[1]);
    }
  }
}

TEST(TestDispatchBest, CastDecimalArgs) {
  std::vector<TypeHolder> args;