Function BM_ListComprehension

eval/tests/modern_benchmark_test.cc:1018–1048

BENCHMARK(BM_NestedComprehension_Trace)->Range(1, 1 << 10);

void BM_ListComprehension(benchmark::State& state) {
  ASSERT_OK_AND_ASSIGN(ParsedExpr parsed_expr, Parse("list_var.map(x, x * 2)"));

  RuntimeOptions options = GetOptions();
  options.comprehension_max_iterations = 10000000;
  options.enable_comprehension_list_append = true;
  auto runtime = StandardRuntimeOrDie(options);

  ASSERT_OK_AND_ASSIGN(auto cel_expr, ProtobufRuntimeAdapter::CreateProgram(
                                          *runtime, parsed_expr));

  google::protobuf::Arena arena;
  Activation activation;

  auto list_builder = cel::NewListValueBuilder(&arena);

  int len = state.range(0);
  list_builder->Reserve(len);
  for (int i = 0; i < len; i++) {
    ASSERT_THAT(list_builder->Add(IntValue(1)), IsOk());
  }

  activation.InsertOrAssignValue("list_var", std::move(*list_builder).Build());

  for (auto _ : state) {
    ASSERT_OK_AND_ASSIGN(cel::Value result,
                         cel_expr->Evaluate(&arena, activation));
    ASSERT_TRUE(InstanceOf<ListValue>(result));
    ASSERT_THAT(Cast<ListValue>(result).Size(), IsOkAndHolds(len));
  }
}

BENCHMARK(BM_ListComprehension)->Range(1, 1 << 16);