Unverified Commit 7f3abf1c authored by Zhiru Zhu's avatar Zhiru Zhu
Browse files

update 



Signed-off-by: default avatarZhiru Zhu <zzhu@fandm.edu>
parent b3bba221

core/unittest/db/test_delete.cpp

+18 −13
Original line number Diff line number Diff line
@@ -124,7 +124,8 @@ TEST_F(DeleteTest, delete_in_mem) { 
        std::vector<std::string> tags;

        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids,

                          result_distances);

        ASSERT_NE(result_ids[0], pair.first);

        //        ASSERT_LT(result_distances[0], 1e-4);

        ASSERT_GT(result_distances[0], 1);
@@ -192,7 +193,8 @@ TEST_F(DeleteTest, delete_on_disk) { 
        std::vector<std::string> tags;

        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids,

                          result_distances);

        ASSERT_NE(result_ids[0], pair.first);

        //        ASSERT_LT(result_distances[0], 1e-4);

        ASSERT_GT(result_distances[0], 1);
@@ -254,7 +256,8 @@ TEST_F(DeleteTest, delete_multiple_times) { 
        std::vector<std::string> tags;

        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids,

                          result_distances);

        ASSERT_NE(result_ids[0], pair.first);

        //        ASSERT_LT(result_distances[0], 1e-4);

        ASSERT_GT(result_distances[0], 1);
@@ -330,7 +333,8 @@ TEST_F(DeleteTest, delete_with_index) { 
        std::vector<std::string> tags;

        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids,

                          result_distances);

        ASSERT_NE(result_ids[0], pair.first);

        //        ASSERT_LT(result_distances[0], 1e-4);

        ASSERT_GT(result_distances[0], 1);
@@ -406,7 +410,8 @@ TEST_F(DeleteTest, delete_multiple_times_with_index) { 
        std::vector<std::string> tags;

        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, search, result_ids,

                          result_distances);

        ASSERT_TRUE(stat.ok());

        ASSERT_NE(result_ids[0], pair.first);

        //        ASSERT_LT(result_distances[0], 1e-4);
@@ -523,8 +528,8 @@ TEST_F(DeleteTest, delete_add_create_index) { 


    result_ids.clear();

    result_distances.clear();

    stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, ids_to_delete.front(), result_ids,

                          result_distances);

    stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, ids_to_delete.front(),

                          result_ids, result_distances);

    ASSERT_EQ(result_ids[0], -1);

    ASSERT_EQ(result_distances[0], std::numeric_limits<float>::max());

}
@@ -594,8 +599,8 @@ TEST_F(DeleteTest, delete_add_auto_flush) { 


    result_ids.clear();

    result_distances.clear();

    stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, ids_to_delete.front(), result_ids,

                          result_distances);

    stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, {{"nprobe", nprobe}}, ids_to_delete.front(),

                          result_ids, result_distances);

    ASSERT_EQ(result_ids[0], -1);

    ASSERT_EQ(result_distances[0], std::numeric_limits<float>::max());

}
@@ -646,8 +651,8 @@ TEST_F(CompactTest, compact_basic) { 
    milvus::engine::VectorsData qb = xb;



    for (auto& id : ids_to_delete) {

        stat =

            db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, id, result_ids, result_distances);

        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, id, result_ids,

                              result_distances);

        ASSERT_EQ(result_ids[0], -1);

        ASSERT_EQ(result_distances[0], std::numeric_limits<float>::max());

    }
@@ -736,8 +741,8 @@ TEST_F(CompactTest, compact_with_index) { 
        std::vector<std::string> tags;

        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;

        stat =

            db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, search, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, search, result_ids,

                          result_distances);

        ASSERT_NE(result_ids[0], pair.first);

        //        ASSERT_LT(result_distances[0], 1e-4);

        ASSERT_GT(result_distances[0], 1);

core/unittest/db/test_search_by_id.cpp

+14 −11
Original line number Diff line number Diff line
@@ -108,7 +108,8 @@ TEST_F(SearchByIdTest, basic) { 
        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;



        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, i, result_ids, result_distances);

        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, i, result_ids,

                              result_distances);

        ASSERT_EQ(result_ids[0], i);

        ASSERT_LT(result_distances[0], 1e-4);

    }
@@ -165,7 +166,8 @@ TEST_F(SearchByIdTest, with_index) { 
        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;



        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, i, result_ids, result_distances);

        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, i, result_ids,

                              result_distances);

        ASSERT_EQ(result_ids[0], i);

        ASSERT_LT(result_distances[0], 1e-3);

    }
@@ -225,7 +227,8 @@ TEST_F(SearchByIdTest, with_delete) { 
        milvus::engine::ResultIds result_ids;

        milvus::engine::ResultDistances result_distances;



        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, i, result_ids, result_distances);

        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, i, result_ids,

                              result_distances);

        ASSERT_EQ(result_ids[0], -1);

        ASSERT_EQ(result_distances[0], std::numeric_limits<float>::max());

    }
@@ -280,8 +283,8 @@ TEST_F(GetVectorByIdTest, basic) { 
        stat = db_->GetVectorByID(table_info.table_id_, id, vector);

        ASSERT_TRUE(stat.ok());



        stat =

            db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, vector, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, vector, result_ids,

                          result_distances);

        ASSERT_TRUE(stat.ok());

        ASSERT_EQ(result_ids[0], id);

        ASSERT_LT(result_distances[0], 1e-4);
@@ -343,8 +346,8 @@ TEST_F(GetVectorByIdTest, with_index) { 
        stat = db_->GetVectorByID(table_info.table_id_, id, vector);

        ASSERT_TRUE(stat.ok());



        stat =

            db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, vector, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, vector, result_ids,

                          result_distances);

        ASSERT_EQ(result_ids[0], id);

        ASSERT_LT(result_distances[0], 1e-3);

    }
@@ -483,8 +486,8 @@ TEST_F(SearchByIdTest, BINARY) { 
        ASSERT_TRUE(stat.ok());

        ASSERT_EQ(vector.vector_count_, 1);



        stat =

            db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, vector, result_ids, result_distances);

        stat = db_->Query(dummy_context_, table_info.table_id_, tags, topk, json_params, vector, result_ids,

                          result_distances);

        ASSERT_TRUE(stat.ok());

        ASSERT_EQ(result_ids[0], id);

        ASSERT_LT(result_distances[0], 1e-4);
@@ -493,8 +496,8 @@ TEST_F(SearchByIdTest, BINARY) { 
        result_ids.clear();

        result_distances.clear();



        stat =

            db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, id, result_ids, result_distances);

        stat = db_->QueryByID(dummy_context_, table_info.table_id_, tags, topk, json_params, id, result_ids,

                              result_distances);

        ASSERT_TRUE(stat.ok());

        ASSERT_EQ(result_ids[0], id);

        ASSERT_LT(result_distances[0], 1e-4);