milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add cpp coverage (#10801) 

Signed-off-by: yhmo <yihua.mo@zilliz.com>
pull/10842/head
groot 2021-10-28 18:54:35 +08:00 committed by GitHub
parent d447f3b584
commit 57783827fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
internal/core/unittest/test_index_wrapper.cpp
View File

@ -19,6 +19,7 @@
#include "indexbuilder/utils.h"
#include "index/knowhere/knowhere/index/vector_index/helpers/IndexParameter.h"
#include "index/knowhere/knowhere/index/vector_index/adapter/VectorAdapter.h"
#include "index/knowhere/knowhere/archive/KnowhereConfig.h"
#include "pb/index_cgo_msg.pb.h"
#include "test_utils/DataGen.h"
#include "test_utils/indexbuilder_test_utils.h"
@ -32,6 +33,8 @@ class IndexWrapperTest : public ::testing::TestWithParam<Param> {
protected:
void
SetUp() override {
milvus::engine::KnowhereConfig::SetStatisticsLevel(3);
auto param = GetParam();
index_type = param.first;
metric_type = param.second;
@ -124,6 +127,8 @@ TEST(IVFFLATNM, Build) {
}
TEST(IVFFLATNM, Query) {
milvus::engine::KnowhereConfig::SetStatisticsLevel(3);
auto index_type = milvus::knowhere::IndexEnum::INDEX_FAISS_IVFFLAT;
auto metric_type = milvus::knowhere::Metric::L2;
auto conf = generate_conf(index_type, metric_type);
@ -142,6 +147,13 @@ TEST(IVFFLATNM, Query) {
auto xq_data = dataset.get_col<float>(0);
auto xq_dataset = milvus::knowhere::GenDataset(NQ, DIM, xq_data.data());
auto result = index->Query(xq_dataset, conf, nullptr);
index->UpdateIndexSize();
ASSERT_GT(index->IndexSize(), 0);
auto stats = index->GetStatistics();
ASSERT_TRUE(stats != nullptr);
index->ClearStatistics();
}
TEST(NSG, Query) {
@ -190,6 +202,8 @@ print_query_result(const std::unique_ptr<milvus::indexbuilder::IndexWrapper::Que
// test for: https://github.com/milvus-io/milvus/issues/6569
TEST(BinIVFFlat, Build_and_Query) {
milvus::engine::KnowhereConfig::SetStatisticsLevel(2);
auto index_type = milvus::knowhere::IndexEnum::INDEX_FAISS_BIN_IVFFLAT;
auto metric_type = milvus::knowhere::Metric::TANIMOTO;
auto conf = generate_conf(index_type, metric_type);
@ -222,6 +236,13 @@ TEST(BinIVFFlat, Build_and_Query) {
memcpy(query_res->distances.data(), distances, sizeof(float) * nq * topk);
print_query_result(query_res);
index->UpdateIndexSize();
ASSERT_GT(index->IndexSize(), 0);
auto stats = index->GetStatistics();
ASSERT_TRUE(stats != nullptr);
index->ClearStatistics();
}
TEST(BINIDMAP, Build) {