From c12e99c3fb3af6167df9c7158a8483389f6b7691 Mon Sep 17 00:00:00 2001
From: zhiru <zhiru.zhu@zilliz.com>
Date: Sun, 13 Oct 2019 17:30:05 +0800
Subject: [PATCH] update

Former-commit-id: 661cc9057b3c52fb09bef6adfec050e909ca4c3c
---
 cpp/src/core/unittest/test_ivf.cpp | 2 +-
 cpp/unittest/db/test_db.cpp        | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/cpp/src/core/unittest/test_ivf.cpp b/cpp/src/core/unittest/test_ivf.cpp
index 987692deb5..b25f230c0f 100644
--- a/cpp/src/core/unittest/test_ivf.cpp
+++ b/cpp/src/core/unittest/test_ivf.cpp
@@ -702,7 +702,7 @@ TEST_F(GPURESTEST, copyandsearch) {
 }
 
 TEST_F(GPURESTEST, TrainAndSearch) {
-    index_type = "GPUIVFSQ";
+    index_type = "GPUIVF";
     index_ = IndexFactory(index_type);
 
     auto conf = std::make_shared<knowhere::IVFSQCfg>();
diff --git a/cpp/unittest/db/test_db.cpp b/cpp/unittest/db/test_db.cpp
index 9e80afbc09..aaf0b7752c 100644
--- a/cpp/unittest/db/test_db.cpp
+++ b/cpp/unittest/db/test_db.cpp
@@ -297,6 +297,7 @@ TEST_F(DBTest, SEARCH_TEST) {
         ASSERT_TRUE(stat.ok());
     }
 
+#ifdef CUSTOMIZATION
     //test FAISS_IVFSQ8H optimizer
     index.engine_type_ = (int)milvus::engine::EngineType::FAISS_IVFSQ8H;
     db_->CreateIndex(TABLE_NAME, index); // wait until build index finish
@@ -314,9 +315,7 @@ TEST_F(DBTest, SEARCH_TEST) {
         stat = db_->Query(TABLE_NAME, file_ids, k, nq, 10, xq.data(), dates, results);
         ASSERT_TRUE(stat.ok());
     }
-
-
-    // TODO(lxj): add groundTruth assert
+#endif-
 }
 
 TEST_F(DBTest, PRELOADTABLE_TEST) {