chromium/components/optimization_guide/core/bert_model_executor_unittest.cc

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/optimization_guide/core/bert_model_handler.h"

#include "base/path_service.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "build/buildflag.h"
#include "components/optimization_guide/core/optimization_guide_features.h"
#include "components/optimization_guide/core/test_model_info_builder.h"
#include "components/optimization_guide/core/test_optimization_guide_model_provider.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace optimization_guide {

class BertModelExecutorTest : public testing::Test { … };

// TODO(crbug.com/40848529): Running the model is slow and times out tests on
// many platforms. Ideally, we can schedule this to run infrequently but for
// now we will only load the model.
TEST_F(BertModelExecutorTest, ValidBertModel) { … }

TEST_F(BertModelExecutorTest, InvalidBertModel) { … }

}  // namespace optimization_guide