@@ -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);
0 commit comments