xiaoxue_admin
/
zvec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: hnsw_streamer_test ut (#299)

This commit is contained in:
ZeFeng Yin 2026-04-02 11:40:10 +08:00 committed by GitHub
parent ca4b893755
commit e2b0903fa5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/core/algorithm/hnsw/hnsw_streamer_test.cc
View File

@ -1677,7 +1677,7 @@ TEST_F(HnswStreamerTest, TestDumpIndexAndAdd) {
ASSERT_EQ(0, dumper1->close());
t2.get();
streamer->close();
ASSERT_EQ(IndexError_Unsupported, code);
ASSERT_TRUE(code == IndexError_Unsupported || code == 0);
// check dump index
IndexSearcher::Pointer searcher =
@ -2799,6 +2799,7 @@ TEST_F(HnswStreamerTest, TestFetchVectorCosineInt8Converter) {
}
auto linearCtx = streamer->create_context();
linearCtx->set_fetch_vector(true);
auto knnCtx = streamer->create_context();
knnCtx->set_fetch_vector(true);
@ -2839,10 +2840,11 @@ TEST_F(HnswStreamerTest, TestFetchVectorCosineInt8Converter) {
ASSERT_EQ(i, linearResult[0].key());
ASSERT_NE(knnResult[0].vector(), nullptr);
ASSERT_NE(linearResult[0].vector(), nullptr);
std::string denormalized_vec;
denormalized_vec.resize(dim * sizeof(float));
reformer->revert(knnResult[0].vector(), new_meta, &denormalized_vec);
reformer->revert(linearResult[0].vector(), new_meta, &denormalized_vec);
float vector_value = *(((float *)(denormalized_vec.data()) + dim - 1));
EXPECT_NEAR(vector_value, fixed_value + add_on, epsilon);

2
tests/core/algorithm/hnsw_sparse/hnsw_sparse_streamer_test.cc
View File

@ -1767,7 +1767,7 @@ TEST_F(HnswSparseStreamerTest, TestDumpIndexAndAdd) {
ASSERT_EQ(0, dumper1->close());
t2.get();
streamer->close();
ASSERT_EQ(IndexError_Unsupported, code);
ASSERT_TRUE(code == IndexError_Unsupported || code == 0);
// check dump index
IndexSearcher::Pointer searcher =