From b1b086a03f1eda8947cfeb43960289d548f0273a Mon Sep 17 00:00:00 2001 From: Yukikaze-CZR Date: Mon, 2 Dec 2019 17:47:17 +0800 Subject: [PATCH] remove change of db_test --- core/unittest/db/test_db.cpp | 21 ++++----------------- 1 file changed, 4 insertions(+), 17 deletions(-) diff --git a/core/unittest/db/test_db.cpp b/core/unittest/db/test_db.cpp index 8d05ae9923..0a47ac1b9b 100644 --- a/core/unittest/db/test_db.cpp +++ b/core/unittest/db/test_db.cpp @@ -295,15 +295,6 @@ TEST_F(DBTest, SEARCH_TEST) { ASSERT_TRUE(stat.ok()); } - { // search by specify index file - milvus::engine::meta::DatesT dates; - std::vector file_ids = {"1", "2", "3", "4", "5", "6"}; - milvus::engine::ResultIds result_ids; - milvus::engine::ResultDistances result_distances; - stat = db_->QueryByFileID(TABLE_NAME, file_ids, k, nq, 10, xq.data(), dates, result_ids, result_distances); - ASSERT_TRUE(stat.ok()); - } - index.engine_type_ = (int)milvus::engine::EngineType::FAISS_IVFFLAT; db_->CreateIndex(TABLE_NAME, index); // wait until build index finish @@ -345,16 +336,12 @@ TEST_F(DBTest, SEARCH_TEST) { } #endif - index.engine_type_ = (int)milvus::engine::EngineType::FAISS_PQ; - db_->CreateIndex(TABLE_NAME, index); // wait until build index finish - - { - std::vector tags; + { // search by specify index file + milvus::engine::meta::DatesT dates; + std::vector file_ids = {"1", "2", "3", "4", "5", "6"}; milvus::engine::ResultIds result_ids; milvus::engine::ResultDistances result_distances; - stat = db_->Query(TABLE_NAME, tags, k, nq, 10, xq.data(), result_ids, result_distances); - ASSERT_TRUE(stat.ok()); - stat = db_->Query(TABLE_NAME, tags, k, 1100, 10, xq.data(), result_ids, result_distances); + stat = db_->QueryByFileID(TABLE_NAME, file_ids, k, nq, 10, xq.data(), dates, result_ids, result_distances); ASSERT_TRUE(stat.ok()); }