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

test(db): small change 

Former-commit-id: 8cf0fe13bfd7715a491f00cc24e2efa37fd8decd
pull/191/head
Xu Peng 2019-04-30 14:17:19 +08:00
parent 3a05719e5a
commit ce54690910
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cpp/unittest/db/db_tests.cpp
View File

@ -54,19 +54,21 @@ TEST_F(DBTest, DB_TEST) {
INIT_TIMER; INIT_TIMER;
std::stringstream ss; std::stringstream ss;
long count = 0; long count = 0;
long prev_count = -1;
for (auto j=0; j<15; ++j) { for (auto j=0; j<10; ++j) {
ss.str(""); ss.str("");
db_->count(group_name, count); db_->count(group_name, count);
prev_count = count;
ss << "Search " << j << " With Size " << count;
START_TIMER; START_TIMER;
stat = db_->search(group_name, k, qb, qxb, results); stat = db_->search(group_name, k, qb, qxb, results);
ss << "Search " << j << " With Size " << (float)(count*group_dim*sizeof(float))/(1024*1024) << " M";
STOP_TIMER(ss.str()); STOP_TIMER(ss.str());
ASSERT_STATS(stat); ASSERT_STATS(stat);
ASSERT_EQ(results[0][0], target_ids[0]); ASSERT_EQ(results[0][0], target_ids[0]);
ASSERT_TRUE(count >= prev_count);
std::this_thread::sleep_for(std::chrono::seconds(1)); std::this_thread::sleep_for(std::chrono::seconds(1));
} }
}); });
@ -79,7 +81,7 @@ TEST_F(DBTest, DB_TEST) {
} else { } else {
db_->add_vectors(group_name, nb, xb, vector_ids); db_->add_vectors(group_name, nb, xb, vector_ids);
} }
std::this_thread::sleep_for(std::chrono::microseconds(5)); std::this_thread::sleep_for(std::chrono::microseconds(1));
} }
search.join(); search.join();