[fix][test] Fix test_vector_index_ivf_pq.cc bug.The reason is that there

are duplicate IDs.
yuhaijun999 · Apr 16, 2024 · 1c1f2ef · 1c1f2ef
1 parent 1e70686
commit 1c1f2ef
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/unit_test/test_vector_index_ivf_pq.cc b/test/unit_test/test_vector_index_ivf_pq.cc
@@ -1890,11 +1890,11 @@ TEST_F(VectorIndexIvfPqTest, AddForIvfPq) {
 
     vector_with_ids.push_back(vector_with_id);
 
-    ok = vector_index_ivf_pq_l2->Add(vector_with_ids);
+    ok = vector_index_ivf_pq_l2->Upsert(vector_with_ids);
     EXPECT_EQ(ok.error_code(), pb::error::Errno::OK);
-    ok = vector_index_ivf_pq_ip->Add(vector_with_ids);
+    ok = vector_index_ivf_pq_ip->Upsert(vector_with_ids);
     EXPECT_EQ(ok.error_code(), pb::error::Errno::OK);
-    ok = vector_index_ivf_pq_cosine->Add(vector_with_ids);
+    ok = vector_index_ivf_pq_cosine->Upsert(vector_with_ids);
     EXPECT_EQ(ok.error_code(), pb::error::Errno::OK);
   }
 
@@ -1913,11 +1913,11 @@ TEST_F(VectorIndexIvfPqTest, AddForIvfPq) {
       vector_with_ids.push_back(vector_with_id);
     }
 
-    ok = vector_index_ivf_pq_l2->Add(vector_with_ids);
+    ok = vector_index_ivf_pq_l2->Upsert(vector_with_ids);
     EXPECT_EQ(ok.error_code(), pb::error::Errno::OK);
-    ok = vector_index_ivf_pq_ip->Add(vector_with_ids);
+    ok = vector_index_ivf_pq_ip->Upsert(vector_with_ids);
     EXPECT_EQ(ok.error_code(), pb::error::Errno::OK);
-    ok = vector_index_ivf_pq_cosine->Add(vector_with_ids);
+    ok = vector_index_ivf_pq_cosine->Upsert(vector_with_ids);
     EXPECT_EQ(ok.error_code(), pb::error::Errno::OK);
   }
 }