Skip to content

Commit aae3c49

Browse files
committed
fix(test): update MockSearchGraph to override search_intern after refactor
1 parent a29019b commit aae3c49

4 files changed

Lines changed: 8 additions & 10 deletions

File tree

‎cpp/test/src/regression/graph/test_readonly_graph_regression.cpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ TEST(ReadOnlyGraphRegression, SearchAndExplore_FP32_L2)
8787
uint32_t total_explore_gt = 0;
8888
auto t_start_explore = std::chrono::high_resolution_clock::now();
8989

90-
for (int run = 0; run < benchmark_runs; ++run) {
90+
for (int run = 0; run < benchmark_runs/10; ++run) {
9191
for (size_t i = 0; i < explore_count_per_run; ++i) {
9292
uint32_t entry_node = static_cast<uint32_t>((run * 13 + i) % base_count);
9393
auto results = graph.explore(entry_node, search_k, explore_max_calcs);
@@ -202,7 +202,7 @@ TEST(ReadOnlyGraphRegression, SearchAndExplore_FP32_InnerProduct)
202202
uint32_t correct_explore = 0;
203203
uint32_t total_explore_gt = 0;
204204
auto t_start_explore = std::chrono::high_resolution_clock::now();
205-
for (int run = 0; run < benchmark_runs; ++run) {
205+
for (int run = 0; run < benchmark_runs/10; ++run) {
206206
for (size_t i = 0; i < explore_count_per_run; ++i) {
207207
uint32_t entry_node = static_cast<uint32_t>((run * 13 + i) % base_count);
208208
auto results = graph.explore(entry_node, search_k, explore_max_calcs);

‎cpp/test/src/regression/graph/test_sizebounded_graph_regression.cpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ TEST(SizeBoundedGraphRegression, SearchAndExplore_FP32_L2)
8484
uint32_t total_explore_gt = 0;
8585
auto t_start_explore = std::chrono::high_resolution_clock::now();
8686

87-
for (int run = 0; run < benchmark_runs; ++run) {
87+
for (int run = 0; run < benchmark_runs/10; ++run) {
8888
for (size_t i = 0; i < explore_count_per_run; ++i) {
8989
uint32_t entry_node = static_cast<uint32_t>((run * 13 + i) % base_count);
9090
auto results = graph.explore(entry_node, search_k, explore_max_calcs);
@@ -196,7 +196,7 @@ TEST(SizeBoundedGraphRegression, SearchAndExplore_FP32_InnerProduct)
196196
uint32_t correct_explore = 0;
197197
uint32_t total_explore_gt = 0;
198198
auto t_start_explore = std::chrono::high_resolution_clock::now();
199-
for (int run = 0; run < benchmark_runs; ++run) {
199+
for (int run = 0; run < benchmark_runs/10; ++run) {
200200
for (size_t i = 0; i < explore_count_per_run; ++i) {
201201
uint32_t entry_node = static_cast<uint32_t>((run * 13 + i) % base_count);
202202
auto results = graph.explore(entry_node, search_k, explore_max_calcs);

‎cpp/test/src/unit/graph/test_readonly_graph.cpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ TEST(ReadOnlyGraph, Explore) {
201201
EXPECT_TRUE(graph.hasEdge(0, 2));
202202
EXPECT_TRUE(graph.hasEdge(0, 3));
203203
// explore from vertex 0, find 3 nearest neighbors (excluding entry)
204-
auto results = graph.explore(0, 3, 3, 0.0f, /*include_entry=*/false);
204+
auto results = graph.explore(0, 3, 0, 0.0f, /*include_entry=*/false);
205205

206206
// all 3 other vertices should be found (distance 1.0 each)
207207
EXPECT_EQ(results.size(), 3u);

‎cpp/test/src/unit/test_search.cpp‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -180,11 +180,9 @@ class MockSearchGraph : public deglib::search::SearchGraph {
180180
std::vector<deglib::search::ObjectDistance> hasPath(const std::vector<uint32_t>&, uint32_t, float, uint32_t) const override {
181181
return {};
182182
}
183-
deglib::search::ResultSet search(const std::vector<uint32_t>&, const std::byte*, float, uint32_t,
184-
const deglib::graph::Filter*, uint32_t) const override {
185-
return deglib::search::ResultSet();
186-
}
187-
deglib::search::ResultSet explore(uint32_t, uint32_t, bool, uint32_t) const override {
183+
deglib::search::ResultSet search_intern(const std::vector<uint32_t>&, const std::byte*,
184+
const uint32_t, const float = 0.0f, const bool = true,
185+
const deglib::graph::Filter* = nullptr, const uint32_t = 0) const override {
188186
return deglib::search::ResultSet();
189187
}
190188

0 commit comments

Comments
 (0)