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

fix: fix Indexing.Iterator ut: build index with all data at once (#32844) 

issue: #32843

Signed-off-by: Buqian Zheng <zhengbuqian@gmail.com>
pull/32927/head
Buqian Zheng 2024-05-10 11:31:30 +08:00 committed by GitHub
parent 1e1fba0588
commit bb7765cbd6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
internal/core/unittest/test_indexing.cpp
View File

@ -410,11 +410,10 @@ INSTANTIATE_TEST_SUITE_P(
#endif
std::pair(knowhere::IndexEnum::INDEX_HNSW, knowhere::metric::L2)));
/*TEST(Indexing, Iterator) {
TEST(Indexing, Iterator) {
constexpr int N = 10240;
constexpr int TOPK = 100;
constexpr int dim = 128;
constexpr int chunk_size = 5120;
auto [raw_data, timestamps, uids] = generate_data<dim>(N);
milvus::index::CreateIndexInfo create_index_info;
@ -437,21 +436,8 @@ INSTANTIATE_TEST_SUITE_P(
{knowhere::indexparam::NPROBE, 4},
};
std::vector<knowhere::DataSetPtr> datasets;
auto raw = raw_data.data();
for (int beg = 0; beg < N; beg += chunk_size) {
auto end = beg + chunk_size;
if (end > N) {
end = N;
}
std::vector<float> ft(raw + dim * beg, raw + dim * end);
auto ds = knowhere::GenDataSet(end - beg, dim, ft.data());
datasets.push_back(ds);
}
for (auto& ds : datasets) {
index->BuildWithDataset(ds, build_conf);
}
index->BuildWithDataset(knowhere::GenDataSet(N, dim, raw_data.data()),
build_conf);
auto bitmap = BitsetType(N, false);
@ -477,7 +463,7 @@ INSTANTIATE_TEST_SUITE_P(
ASSERT_TRUE(off >= 0);
ASSERT_TRUE(dis >= 0);
}
}*/
}
TEST_P(IndexTest, BuildAndQuery) {
milvus::index::CreateIndexInfo create_index_info;