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

MS-648 add ut v3 


Former-commit-id: dfead97cf9f18f3965b4b8ffc8c88c9da6b81544
parent e50adbe4

core/src/index/unittest/Helper.h

+2 −1
Original line number Diff line number Diff line
@@ -16,6 +16,7 @@ 
// under the License.



#include <memory>

#include <string>



#include "knowhere/index/vector_index/IndexGPUIVF.h"

#include "knowhere/index/vector_index/IndexGPUIVFPQ.h"

core/src/index/unittest/test_customized_index.cpp

+15 −1
Original line number Diff line number Diff line 
// Licensed to the Apache Software Foundation (ASF) under one

// or more contributor license agreements.  See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership.  The ASF licenses this file

// to you under the Apache License, Version 2.0 (the

// "License"); you may not use this file except in compliance

// with the License.  You may obtain a copy of the License at

//

// Created by link on 2019/10/17.

//   http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing,

// software distributed under the License is distributed on an

// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

// KIND, either express or implied.  See the License for the

// specific language governing permissions and limitations

// under the License.



#include <gtest/gtest.h>



#include "unittest/Helper.h"

core/unittest/wrapper/test_hybrid_index.cpp

+2 −2
Original line number Diff line number Diff line
@@ -38,7 +38,6 @@ class KnowhereHybrid 
        nq = 100;

        k = 100;

        GenData(dim, nb, nq, xb, xq, ids, k, gt_ids, gt_dis);



    }



    void TearDown() override {
@@ -130,4 +129,5 @@ TEST_F(KnowhereHybrid, test_interface) { 
        }

    }

}



#endif

core/unittest/wrapper/utils.cpp

+0 −1
Original line number Diff line number Diff line
@@ -86,5 +86,4 @@ DataGenBase::AssertResult(const std::vector<int64_t>& ids, const std::vector<flo 
                  << ", match: " << match

                  << ", total: " << nq * k

                  << std::endl;



}

core/unittest/wrapper/utils.h

+0 −7
Original line number Diff line number Diff line
@@ -70,13 +70,6 @@ class DataGenBase { 
    std::vector<float> gt_dis;

};



namespace {



namespace ms = milvus::engine;

namespace kw = knowhere;



} // namespace



class ParamGenerator {

 public:

    static ParamGenerator& GetInstance() {