add vector ids check in unittest

Former-commit-id: 26c60e303c5ede2200c77efee11b17a782d0dc46
pull/191/head
Yu Kun 2019-08-20 16:49:04 +08:00
parent da6e37c8b8
commit c1e82cb749
1 changed files with 13 additions and 0 deletions

View File

@ -432,4 +432,17 @@ TEST_F(DBTest, VECTOR_IDS_TEST)
vector_ids.clear();
stat = db_->InsertVectors(TABLE_NAME, nb, xb.data(), vector_ids);
ASSERT_STATS(stat);
nb = 100;
xb.clear();
BuildVectors(nb, xb);
vector_ids.clear();
vector_ids.resize(nb);
for (auto i = 0; i < nb; i++) {
vector_ids[i] = i + nb;
}
stat = db_->InsertVectors(TABLE_NAME, nb, xb.data(), vector_ids);
for (auto i = 0; i < nb; i++) {
ASSERT_EQ(vector_ids[i], i + nb);
}
}