chromium/components/history_embeddings/vector_database_unittest.cc

// Copyright 2024 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/history_embeddings/vector_database.h"

#include <atomic>
#include <memory>

#include "base/logging.h"
#include "base/memory/weak_ptr.h"
#include "base/rand_util.h"
#include "base/time/time.h"
#include "base/timer/elapsed_timer.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace history_embeddings {

namespace {

Embedding RandomEmbedding() { … }

Embedding DeterministicEmbedding(float value) { … }

}  // namespace

TEST(HistoryEmbeddingsVectorDatabaseTest, Constructs) { … }

TEST(HistoryEmbeddingsVectorDatabaseTest, EmbeddingOperations) { … }

TEST(HistoryEmbeddingsVectorDatabaseTest, FindNearest) { … }

TEST(HistoryEmbeddingsVectorDatabaseTest, SearchCanBeHaltedEarly) { … }

TEST(HistoryEmbeddingsVectorDatabaseTest, TimeRangeNarrowsSearchResult) { … }

// Note: Disabled by default so as to not burden the bots. Enable when needed.
TEST(HistoryEmbeddingsVectorDatabaseTest, DISABLED_ManyVectorsAreFastEnough) { … }

}  // namespace history_embeddings