Function TEST_F

cpp/src/gandiva/tests/decimal_alignment_test.cc:113–156

Test that decimal operations work correctly with 8-byte aligned (but not 16-byte aligned) data

// Test that decimal operations work correctly with 8-byte aligned (but not 16-byte
// aligned) data
TEST_F(TestDecimalAlignment, TestMisalignedDecimalSubtract) {
  constexpr int32_t precision = 38;
  constexpr int32_t scale = 17;
  auto decimal_type = std::make_shared<arrow::Decimal128Type>(precision, scale);
  auto field_a = arrow::field("a", decimal_type);
  auto field_b = arrow::field("b", decimal_type);
  auto schema = arrow::schema({field_a, field_b});

  Decimal128TypePtr output_type;
  auto status = DecimalTypeUtil::GetResultType(
      DecimalTypeUtil::kOpSubtract, {decimal_type, decimal_type}, &output_type);
  ASSERT_OK(status);

  auto res = arrow::field("res", output_type);
  auto node_a = TreeExprBuilder::MakeField(field_a);
  auto node_b = TreeExprBuilder::MakeField(field_b);
  auto subtract =
      TreeExprBuilder::MakeFunction("subtract", {node_a, node_b}, output_type);
  auto expr = TreeExprBuilder::MakeExpression(subtract, res);

  std::shared_ptr<Projector> projector;
  status = Projector::Make(schema, {expr}, TestConfiguration(), &projector);
  ASSERT_OK(status);

  // Create test data
  std::vector<Decimal128> values_a = {Decimal128(100), Decimal128(200), Decimal128(300)};
  std::vector<Decimal128> values_b = {Decimal128(10), Decimal128(20), Decimal128(30)};

  // Create arrays with 8-byte alignment (but NOT 16-byte aligned)
  auto array_a = MakeMisalignedDecimalArray(decimal_type, values_a, 8);
  auto array_b = MakeMisalignedDecimalArray(decimal_type, values_b, 8);

  auto in_batch = arrow::RecordBatch::Make(schema, 3, {array_a, array_b});

  // This should NOT crash even with misaligned data
  arrow::ArrayVector outputs;
  status = projector->Evaluate(*in_batch, pool_, &outputs);
  ASSERT_OK(status);

  // Verify results: 100-10=90, 200-20=180, 300-30=270
  auto result = std::dynamic_pointer_cast<arrow::Decimal128Array>(outputs[0]);
  ASSERT_NE(result, nullptr);
  EXPECT_EQ(result->length(), 3);
}

// Create a misaligned output buffer for decimal128
std::shared_ptr<arrow::ArrayData> MakeMisalignedDecimalOutput(