Add HnswIdentityMapping.

1 files changed, 6 insertions, 6 deletions

diff --git a/searchlib/src/tests/tensor/hnsw_saver/hnsw_save_load_test.cpp b/searchlib/src/tests/tensor/hnsw_saver/hnsw_save_load_test.cpp
index ea22aaabaff..a94b09621df 100644
--- a/searchlib/src/tests/tensor/hnsw_saver/hnsw_save_load_test.cpp
+++ b/searchlib/src/tests/tensor/hnsw_saver/hnsw_save_load_test.cpp
@@ -52,12 +52,12 @@ using V = std::vector<uint32_t>;
 
 void populate(HnswGraph &graph) {
     // no 0
-    graph.make_node(1, 1);
-    auto er = graph.make_node(2, 2);
+    graph.make_node(1, 1, 0, 1);
+    auto er = graph.make_node(2, 2, 0, 2);
     // no 3
-    graph.make_node(4, 2);
-    graph.make_node(5, 0);
-    graph.make_node(6, 1);
+    graph.make_node(4, 4, 0, 2);
+    graph.make_node(5, 5, 0, 0);
+    graph.make_node(6, 6, 0, 1);
 
     graph.set_link_array(1, 0, V{2, 4, 6});
     graph.set_link_array(2, 0, V{1, 4, 6});
@@ -71,7 +71,7 @@ void populate(HnswGraph &graph) {
 void modify(HnswGraph &graph) {
     graph.remove_node(2);
     graph.remove_node(6);
-    graph.make_node(7, 2);
+    graph.make_node(7, 7, 0, 2);
 
     graph.set_link_array(1, 0, V{7, 4});
     graph.set_link_array(4, 0, V{7, 2});