Commit b1b086a0 authored by Yukikaze-CZR's avatar Yukikaze-CZR
Browse files

remove change of db_test

parent 36d84f37

core/unittest/db/test_db.cpp

+4 −17
Original line number Diff line number Diff line
@@ -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<std::string> 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<std::string> tags;

    {  // search by specify index file

        milvus::engine::meta::DatesT dates;

        std::vector<std::string> 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());

    }