Function ASSERT_OK_AND_ASSIGN

cpp/src/arrow/compute/expression_test.cc:1142–1226

  ASSERT_OK_AND_ASSIGN(Datum actual, ExecuteScalarExpression(simplify_expr, input));
  compute::ExecContext* exec_context = default_exec_context();
  ASSERT_OK_AND_ASSIGN(auto function,
                       exec_context->func_registry()->GetFunction("random"));
  ASSERT_OK_AND_ASSIGN(Datum expected,
                       function->Execute(input, &random_options, exec_context));
  AssertDatumsEqual(actual, expected, /*verbose=*/true);

  EXPECT_EQ(actual.length(), kCount);
}

TEST(Expression, ExecuteChunkedArray) {
  // GH-41923: compute should generate the right result if input
  // ExecBatch is `chunked_array`.
  auto input_schema = struct_({field("a", struct_({
                                              field("a", float64()),
                                              field("b", float64()),
                                          }))});

  auto chunked_array_input = ChunkedArrayFromJSON(input_schema, {R"([
    {"a": {"a": 6.125, "b": 3.375}},
    {"a": {"a": 0.0,   "b": 1}}
  ])",
                                                                 R"([
    {"a": {"a": -1,    "b": 4.75}}
  ])"});

  ASSERT_OK_AND_ASSIGN(auto table_input,
                       Table::FromChunkedStructArray(chunked_array_input));

  auto expr = add(field_ref(FieldRef("a", "a")), field_ref(FieldRef("a", "b")));

  ASSERT_OK_AND_ASSIGN(expr, expr.Bind(input_schema));
  std::vector<Datum> inputs{table_input->column(0)};
  ExecBatch batch{inputs, 3};

  ASSERT_OK_AND_ASSIGN(Datum res, ExecuteScalarExpression(expr, batch));
  ASSERT_TRUE(res.is_chunked_array());

  AssertDatumsEqual(res, ChunkedArrayFromJSON(float64(), {R"([
    9.5,
    1,
    3.75
  ])"}));
}

TEST(Expression, ExecuteDictionaryTransparent) {
  ExpectExecute(
      equal(field_ref("a"), field_ref("b")),
      ArrayFromJSON(
          struct_({field("a", dictionary(int32(), utf8())), field("b", utf8())}), R"([
    {"a": "hi", "b": "hi"},
    {"a": "",   "b": ""},
    {"a": "hi", "b": "hello"}
  ])"));

  ASSERT_OK_AND_ASSIGN(
      auto expr, project({field_ref("i32"), field_ref("dict_str")}, {"i32", "dict_str"})
                     .Bind(*kBoringSchema));