Function BM_ListComprehension

eval/tests/benchmark_test.cc:920–949

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

void BM_ListComprehension(benchmark::State& state) {
  google::protobuf::Arena arena;
  Activation activation;
  ASSERT_OK_AND_ASSIGN(ParsedExpr parsed_expr,
                       parser::Parse("list_var.map(x, x * 2)"));

  int len = state.range(0);
  std::vector<CelValue> list;
  list.reserve(len);
  for (int i = 0; i < len; i++) {
    list.push_back(CelValue::CreateInt64(1));
  }

  ContainerBackedListImpl cel_list(std::move(list));
  activation.InsertValue("list_var", CelValue::CreateList(&cel_list));
  InterpreterOptions options = GetOptions(arena);
  options.comprehension_max_iterations = 10000000;
  options.enable_comprehension_list_append = true;
  auto builder = CreateCelExpressionBuilder(options);
  ASSERT_OK(RegisterBuiltinFunctions(builder->GetRegistry(), options));
  ASSERT_OK_AND_ASSIGN(
      auto cel_expr, builder->CreateExpression(&(parsed_expr.expr()), nullptr));

  for (auto _ : state) {
    ASSERT_OK_AND_ASSIGN(CelValue result,
                         cel_expr->Evaluate(activation, &arena));
    ASSERT_TRUE(result.IsList());
    ASSERT_EQ(result.ListOrDie()->size(), len);
  }
}

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