185       std::unique_ptr<EmbeddingIndex> embedding_index,  in TEST_F()
190   ICING_ASSERT_OK(embedding_index->BufferEmbedding(  in TEST_F()
193   ICING_ASSERT_OK(embedding_index->CommitBufferToIndex());  in TEST_F()
194   embedding_index->set_last_added_document_id(0);  in TEST_F()
197       GetEmbeddingHitsFromIndex(embedding_index.get(), /*dimension=*/3,  in TEST_F()
201   EXPECT_THAT(GetRawEmbeddingDataFromIndex(embedding_index.get()),  in TEST_F()
203   EXPECT_EQ(embedding_index->last_added_document_id(), 0);  in TEST_F()
205   ICING_EXPECT_OK(embedding_index->GetChecksum());  in TEST_F()
213   embedding_index.reset();  in TEST_F()
222       std::unique_ptr<EmbeddingIndex> embedding_index,  in TEST_F()
227   ICING_ASSERT_OK(embedding_index->BufferEmbedding(  in TEST_F()
230   ICING_ASSERT_OK(embedding_index->CommitBufferToIndex());  in TEST_F()
231   embedding_index->set_last_added_document_id(0);  in TEST_F()
234       GetEmbeddingHitsFromIndex(embedding_index.get(), /*dimension=*/3,  in TEST_F()
238   EXPECT_THAT(GetRawEmbeddingDataFromIndex(embedding_index.get()),  in TEST_F()
240   EXPECT_EQ(embedding_index->last_added_document_id(), 0);  in TEST_F()
243   ICING_ASSERT_OK_AND_ASSIGN(Crc32 crc, embedding_index->GetChecksum());  in TEST_F()
244   EXPECT_THAT(embedding_index->UpdateChecksums(), IsOkAndHolds(Eq(crc)));  in TEST_F()
245   EXPECT_THAT(embedding_index->GetChecksum(), IsOkAndHolds(Eq(crc)));  in TEST_F()
262   embedding_index.reset();  in TEST_F()
272       std::unique_ptr<EmbeddingIndex> embedding_index,  in TEST_F()
277   ICING_ASSERT_OK(embedding_index->BufferEmbedding(  in TEST_F()
280   ICING_ASSERT_OK(embedding_index->CommitBufferToIndex());  in TEST_F()
281   embedding_index->set_last_added_document_id(0);  in TEST_F()
284       GetEmbeddingHitsFromIndex(embedding_index.get(), /*dimension=*/3,  in TEST_F()
288   EXPECT_THAT(GetRawEmbeddingDataFromIndex(embedding_index.get()),  in TEST_F()
290   EXPECT_EQ(embedding_index->last_added_document_id(), 0);  in TEST_F()
293   ICING_EXPECT_OK(embedding_index->PersistToDisk());  in TEST_F()
310   embedding_index.reset();  in TEST_F()