Function CheckWholeAggregateCase

cpp/src/arrow/engine/substrait/function_test.cc:647–661

}

void CheckWholeAggregateCase(const AggregateTestCase& test_case) {
  std::shared_ptr<Table> output_table;
  std::shared_ptr<acero::ExecPlan> plan =
      PlanFromAggregateCase(test_case, &output_table, /*with_keys=*/false);

  plan->StartProducing();
  ASSERT_FINISHES_OK(plan->finished());

  ASSERT_OK_AND_ASSIGN(output_table,
                       output_table->SelectColumns({output_table->num_columns() - 1}));

  std::shared_ptr<Table> expected_output =
      GetOutputTableForAggregateCase(test_case.output_type, test_case.combined_output);
  AssertTablesEqual(*expected_output, *output_table, /*same_chunk_layout=*/false);
}

void CheckGroupedAggregateCase(const AggregateTestCase& test_case) {
  ARROW_SCOPED_TRACE("function = ", test_case.function_id.ToString());