Function PlanFromTestCase

cpp/src/arrow/engine/substrait/function_test.cc:108–136

}

Result<std::shared_ptr<acero::ExecPlan>> PlanFromTestCase(
    const FunctionTestCase& test_case, std::shared_ptr<Table>* output_table) {
  ARROW_ASSIGN_OR_RAISE(std::shared_ptr<Table> input_table,
                        GetInputTable(test_case.arguments, test_case.data_types));
  ARROW_ASSIGN_OR_RAISE(
      std::shared_ptr<Buffer> substrait,
      internal::CreateScanProjectSubstrait(
          test_case.function_id, input_table, test_case.arguments, test_case.options,
          test_case.data_types, *test_case.expected_output_type));
  std::shared_ptr<acero::SinkNodeConsumer> consumer =
      std::make_shared<acero::TableSinkNodeConsumer>(output_table, default_memory_pool());

  // Mock table provider that ignores the table name and returns input_table
  NamedTableProvider table_provider = [input_table](const std::vector<std::string>&,
                                                    const Schema&) {
    std::shared_ptr<acero::ExecNodeOptions> options =
        std::make_shared<acero::TableSourceNodeOptions>(input_table);
    return acero::Declaration("table_source", {}, options, "mock_source");
  };

  ConversionOptions conversion_options;
  conversion_options.named_table_provider = std::move(table_provider);

  ARROW_ASSIGN_OR_RAISE(
      std::shared_ptr<acero::ExecPlan> plan,
      DeserializePlan(*substrait, std::move(consumer), default_extension_id_registry(),
                      /*ext_set_out=*/nullptr, conversion_options));
  return plan;
}

void CheckValidTestCases(const std::vector<FunctionTestCase>& valid_cases) {
  for (const FunctionTestCase& test_case : valid_cases) {