提交 5d5a20dc 编写于 作者: F FluorineDog 提交者: yefu.chen

Enhance random generator of unittest

Signed-off-by: NFluorineDog <guilin.gou@zilliz.com>
上级 6554b2cf
......@@ -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);
......
......@@ -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);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册