Skip to content

Commit e2b0903

Browse files
authored
fix: hnsw_streamer_test ut (#299)
1 parent ca4b893 commit e2b0903

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

tests/core/algorithm/hnsw/hnsw_streamer_test.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1677,7 +1677,7 @@ TEST_F(HnswStreamerTest, TestDumpIndexAndAdd) {
16771677
ASSERT_EQ(0, dumper1->close());
16781678
t2.get();
16791679
streamer->close();
1680-
ASSERT_EQ(IndexError_Unsupported, code);
1680+
ASSERT_TRUE(code == IndexError_Unsupported || code == 0);
16811681

16821682
// check dump index
16831683
IndexSearcher::Pointer searcher =
@@ -2799,6 +2799,7 @@ TEST_F(HnswStreamerTest, TestFetchVectorCosineInt8Converter) {
27992799
}
28002800

28012801
auto linearCtx = streamer->create_context();
2802+
linearCtx->set_fetch_vector(true);
28022803
auto knnCtx = streamer->create_context();
28032804
knnCtx->set_fetch_vector(true);
28042805

@@ -2839,10 +2840,11 @@ TEST_F(HnswStreamerTest, TestFetchVectorCosineInt8Converter) {
28392840
ASSERT_EQ(i, linearResult[0].key());
28402841

28412842
ASSERT_NE(knnResult[0].vector(), nullptr);
2843+
ASSERT_NE(linearResult[0].vector(), nullptr);
28422844

28432845
std::string denormalized_vec;
28442846
denormalized_vec.resize(dim * sizeof(float));
2845-
reformer->revert(knnResult[0].vector(), new_meta, &denormalized_vec);
2847+
reformer->revert(linearResult[0].vector(), new_meta, &denormalized_vec);
28462848

28472849
float vector_value = *(((float *)(denormalized_vec.data()) + dim - 1));
28482850
EXPECT_NEAR(vector_value, fixed_value + add_on, epsilon);

tests/core/algorithm/hnsw_sparse/hnsw_sparse_streamer_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1767,7 +1767,7 @@ TEST_F(HnswSparseStreamerTest, TestDumpIndexAndAdd) {
17671767
ASSERT_EQ(0, dumper1->close());
17681768
t2.get();
17691769
streamer->close();
1770-
ASSERT_EQ(IndexError_Unsupported, code);
1770+
ASSERT_TRUE(code == IndexError_Unsupported || code == 0);
17711771

17721772
// check dump index
17731773
IndexSearcher::Pointer searcher =

0 commit comments

Comments
 (0)