Commit a764aa57 authored by xiaojun.lin's avatar xiaojun.lin
Browse files

MS-648 add ut v4 


Former-commit-id: c221f1e53356ff6761550fb94b230ea6a784f792
parent 99105997

core/src/index/unittest/test_gpuresource.cpp

+24 −26
Original line number Diff line number Diff line
@@ -35,8 +35,8 @@ 
#include "knowhere/index/vector_index/IndexIVFSQHybrid.h"

#include "knowhere/index/vector_index/helpers/Cloner.h"



#include "unittest/utils.h"

#include "unittest/Helper.h"

#include "unittest/utils.h"



class GPURESTEST : public DataGen, public TestGpuIndexBase {

 protected:
@@ -307,5 +307,3 @@ TEST_F(GPURESTEST, gpuivfsq) { 
    }

}

#endif

core/src/index/unittest/test_idmap.cpp

+5 −5
Original line number Diff line number Diff line
@@ -23,8 +23,8 @@ 
#include "knowhere/index/vector_index/IndexIDMAP.h"

#include "knowhere/index/vector_index/helpers/Cloner.h"



#include "unittest/utils.h"

#include "Helper.h"

#include "unittest/utils.h"



class IDMAPTest : public DataGen, public TestGpuIndexBase {

 protected:

core/src/index/unittest/test_ivf.cpp

+14 −16
Original line number Diff line number Diff line
@@ -35,8 +35,8 @@ 
#include "knowhere/index/vector_index/IndexIVFSQHybrid.h"

#include "knowhere/index/vector_index/helpers/Cloner.h"



#include "unittest/utils.h"

#include "unittest/Helper.h"

#include "unittest/utils.h"



using ::testing::Combine;

using ::testing::TestWithParam;
@@ -78,8 +78,7 @@ class IVFTest : public DataGen, public TestWithParam<::std::tuple<std::string, P 
};



INSTANTIATE_TEST_CASE_P(IVFParameters, IVFTest,

                        Values(

                            std::make_tuple("IVF", ParameterType::ivf),

                        Values(std::make_tuple("IVF", ParameterType::ivf),

                               std::make_tuple("GPUIVF", ParameterType::ivf),

                               std::make_tuple("IVFPQ", ParameterType::ivfpq),

                               std::make_tuple("GPUIVFPQ", ParameterType::ivfpq),
@@ -244,8 +243,7 @@ TEST_P(IVFTest, clone_test) { 
        auto finder = std::find(support_idx_vec.cbegin(), support_idx_vec.cend(), index_type);

        if (finder != support_idx_vec.cend()) {

            EXPECT_NO_THROW({

                                auto

                                    clone_index = knowhere::cloner::CopyCpuToGpu(index_, DEVICEID, knowhere::Config());

                auto clone_index = knowhere::cloner::CopyCpuToGpu(index_, DEVICEID, knowhere::Config());

                auto clone_result = clone_index->Search(query_dataset, conf);

                AssertEqual(result, clone_result);

                std::cout << "clone C <=> G [" << index_type << "] success" << std::endl;

core/src/index/unittest/test_nsg/test_nsg.cpp

+4 −6
Original line number Diff line number Diff line
@@ -84,9 +84,7 @@ TEST_F(NSGInterfaceTest, basic_test) { 


    ASSERT_EQ(index_->Count(), nb);

    ASSERT_EQ(index_->Dimension(), dim);

    ASSERT_THROW({

                     index_->Clone();

                 }, knowhere::KnowhereException);

    ASSERT_THROW({ index_->Clone(); }, knowhere::KnowhereException);

    ASSERT_NO_THROW({

        index_->Add(base_dataset, knowhere::Config());

        index_->Seal();

core/src/index/unittest/test_customized_index.cpp

+8 −8

File changed.

Contains only whitespace changes.