chromium/components/omnibox/browser/on_device_tail_model_executor_unittest.cc

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

#include <memory>
#include <string>
#include <vector>

#include "base/containers/contains.h"
#include "base/files/file_path.h"
#include "base/path_service.h"
#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "components/omnibox/browser/on_device_tail_model_executor.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

ElementsAreArray;

namespace {

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

base::FilePath GetTestFilePath(const std::string& filename) { … }

}  // namespace

class OnDeviceTailModelExecutorPublic : public OnDeviceTailModelExecutor { … };

// Note: suggestions which contain unprintable tokens (see definition at
// OnDeviceTailTokenizer::IsTokenPrintable) will not be outputted by the
// executor and be dropped in the private member function
// OnDeviceTailModelExecutor::InsertBeamNodeToCandidateQueue.
class OnDeviceTailModelExecutorTest : public ::testing::Test { … };

TEST_F(OnDeviceTailModelExecutorTest, TestEncodePreviousQuery) { … }

TEST_F(OnDeviceTailModelExecutorTest, TestRunRnnStep) { … }

TEST_F(OnDeviceTailModelExecutorTest, TestCreateNewBeams) { … }

TEST_F(OnDeviceTailModelExecutorTest, TestInsertBeamNodeToCandidateQueue) { … }

TEST_F(OnDeviceTailModelExecutorTest, TestGetRootBeamNode) { … }

TEST_F(OnDeviceTailModelExecutorTest, TestGenerateSuggestionsForPrefix) { … }

TEST_F(OnDeviceTailModelExecutorTest,
       TestGenerateSuggestionsWithBadwordFilter) { … }