chromium/components/omnibox/browser/on_device_tail_model_service_unittest.cc

// Copyright 2023 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/omnibox/browser/on_device_tail_model_service.h"

#include "base/containers/flat_set.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/bind.h"
#include "base/path_service.h"
#include "base/test/task_environment.h"
#include "components/memory_pressure/fake_memory_pressure_monitor.h"
#include "components/omnibox/browser/on_device_tail_model_executor.h"
#include "components/optimization_guide/core/test_model_info_builder.h"
#include "components/optimization_guide/core/test_optimization_guide_model_provider.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

ElementsAreArray;

namespace {

static const char kTailModelFilename[] = …;
static const char kVocabFilename[] = …;

constexpr int kNumLayer = …;
constexpr int kStateSize = …;
constexpr int kEmbeddingDim = …;
constexpr int kMaxNumSteps = …;
constexpr float kProbabilityThreshold = …;

}  // namespace

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

TEST_F(OnDeviceTailModelServiceTest, OnModelUpdated) { … }

TEST_F(OnDeviceTailModelServiceTest, GetPredictionsForInput) { … }

TEST_F(OnDeviceTailModelServiceTest, NullModelUpdate) { … }

TEST_F(OnDeviceTailModelServiceTest, MemoryPressureLevel) { … }