From 5d5a20dce2b367c010477a1e8b734f10925eae95 Mon Sep 17 00:00:00 2001
From: FluorineDog <guilin.gou@zilliz.com>
Date: Fri, 25 Sep 2020 17:09:53 +0800
Subject: [PATCH] Enhance random generator of unittest

Signed-off-by: FluorineDog <guilin.gou@zilliz.com>
---
 sdk/examples/simple/insert.cpp |  8 +++++++-
 sdk/examples/simple/search.cpp | 20 ++++++++++++--------
 2 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/sdk/examples/simple/insert.cpp b/sdk/examples/simple/insert.cpp
index b19b70513..86c1967d2 100644
--- a/sdk/examples/simple/insert.cpp
+++ b/sdk/examples/simple/insert.cpp
@@ -19,6 +19,7 @@
 #include "grpc/ClientProxy.h"
 #include "interface/ConnectionImpl.h"
 #include "utils/TimeRecorder.h"
+#include <random>
 
 const int N = 200000;
 const int DIM = 16;
@@ -31,9 +32,14 @@ const milvus::FieldValue GetData() {
   for (int i = 0; i < N; i++) {
     int32_data.push_back(i);
   }
+  std::default_random_engine eng(rand() % 20);
+  std::normal_distribution<float> dis(0, 1);
   std::vector<milvus::VectorData> vector_data;
   for (int i = 0; i < N; i++) {
-    std::vector<float> float_data(DIM, 10.25);
+    std::vector<float> float_data(DIM);
+    for(auto &x: float_data) {
+      x = dis(eng);
+    }
     milvus::VectorData vectorData;
     vectorData.float_data = float_data;
     vector_data.push_back(vectorData);
diff --git a/sdk/examples/simple/search.cpp b/sdk/examples/simple/search.cpp
index f15781fa2..1da315232 100644
--- a/sdk/examples/simple/search.cpp
+++ b/sdk/examples/simple/search.cpp
@@ -17,6 +17,7 @@
 #include "interface/ConnectionImpl.h"
 #include "utils/TimeRecorder.h"
 #include "utils/Utils.h"
+#include <random>
 
 const int TOP_K = 10;
 
@@ -38,16 +39,19 @@ int main(int argc , char**argv) {
   partition_list.emplace_back("partition-3");
 
   milvus::VectorParam vectorParam;
-  milvus::VectorData vectorData;
-  std::vector<float> float_data;
-  for (int i = 0; i < 100; ++i) {
-    float_data.emplace_back(i);
-  }
-
-  vectorData.float_data = float_data;
-
   std::vector<milvus::VectorData> vector_records;
+
+  std::default_random_engine eng(rand() % 20);
+  std::normal_distribution<float> dis(0, 1);
+ 
   for (int j = 0; j < 10; ++j) {
+    milvus::VectorData vectorData;
+    std::vector<float> float_data;
+    for (int i = 0; i < 100; ++i) {
+      float_data.emplace_back(dis(eng));
+    }
+
+    vectorData.float_data = float_data;
     vector_records.emplace_back(vectorData);
   }
 
-- 
GitLab