From 7bb2d98fe2d90a64ee7c4f5943e2ea3be5fd2431 Mon Sep 17 00:00:00 2001
From: starlord <chxiehust@gmail.com>
Date: Mon, 9 Sep 2019 19:22:53 +0800
Subject: [PATCH] reduce unittest run time

Former-commit-id: d643a9bdc49bc704cf6e9e6994776b22270476e7
---
 cpp/src/core/test/test_nsg/test_nsg.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cpp/src/core/test/test_nsg/test_nsg.cpp b/cpp/src/core/test/test_nsg/test_nsg.cpp
index 50eccab2..3c8d4777 100644
--- a/cpp/src/core/test/test_nsg/test_nsg.cpp
+++ b/cpp/src/core/test/test_nsg/test_nsg.cpp
@@ -26,7 +26,7 @@ class NSGInterfaceTest : public DataGen, public TestWithParam<::std::tuple<Confi
     void SetUp() override {
         //Init_with_default();
         FaissGpuResourceMgr::GetInstance().InitDevice(DEVICE_ID, 1024*1024*200, 1024*1024*600, 2);
-        Generate(256, 1000000, 1);
+        Generate(256, 10000, 1);
         index_ = std::make_shared<NSG>();
         std::tie(train_cfg, search_cfg) = GetParam();
     }
@@ -44,7 +44,7 @@ class NSGInterfaceTest : public DataGen, public TestWithParam<::std::tuple<Confi
 INSTANTIATE_TEST_CASE_P(NSGparameters, NSGInterfaceTest,
                         Values(std::make_tuple(
                             // search length > out_degree
-                            Config::object{{"nlist", 16384}, {"nprobe", 50}, {"knng", 100}, {"metric_type", "L2"},
+                            Config::object{{"nlist", 128}, {"nprobe", 50}, {"knng", 100}, {"metric_type", "L2"},
                                            {"search_length", 60}, {"out_degree", 70}, {"candidate_pool_size", 500}},
                             Config::object{{"k", 20}, {"search_length", 30}}))
 );
-- 
GitLab