diff --git a/internal/core/unittest/test_index_wrapper.cpp b/internal/core/unittest/test_index_wrapper.cpp
index c3f32bede8..b53593c426 100644
--- a/internal/core/unittest/test_index_wrapper.cpp
+++ b/internal/core/unittest/test_index_wrapper.cpp
@@ -19,6 +19,7 @@
#include "indexbuilder/utils.h"
#include "index/knowhere/knowhere/index/vector_index/helpers/IndexParameter.h"
#include "index/knowhere/knowhere/index/vector_index/adapter/VectorAdapter.h"
+#include "index/knowhere/knowhere/archive/KnowhereConfig.h"
#include "pb/index_cgo_msg.pb.h"
#include "test_utils/DataGen.h"
#include "test_utils/indexbuilder_test_utils.h"
@@ -32,6 +33,8 @@ class IndexWrapperTest : public ::testing::TestWithParam {
protected:
void
SetUp() override {
+ milvus::engine::KnowhereConfig::SetStatisticsLevel(3);
+
auto param = GetParam();
index_type = param.first;
metric_type = param.second;
@@ -124,6 +127,8 @@ TEST(IVFFLATNM, Build) {
}
TEST(IVFFLATNM, Query) {
+ milvus::engine::KnowhereConfig::SetStatisticsLevel(3);
+
auto index_type = milvus::knowhere::IndexEnum::INDEX_FAISS_IVFFLAT;
auto metric_type = milvus::knowhere::Metric::L2;
auto conf = generate_conf(index_type, metric_type);
@@ -142,6 +147,13 @@ TEST(IVFFLATNM, Query) {
auto xq_data = dataset.get_col(0);
auto xq_dataset = milvus::knowhere::GenDataset(NQ, DIM, xq_data.data());
auto result = index->Query(xq_dataset, conf, nullptr);
+
+ index->UpdateIndexSize();
+ ASSERT_GT(index->IndexSize(), 0);
+
+ auto stats = index->GetStatistics();
+ ASSERT_TRUE(stats != nullptr);
+ index->ClearStatistics();
}
TEST(NSG, Query) {
@@ -190,6 +202,8 @@ print_query_result(const std::unique_ptrdistances.data(), distances, sizeof(float) * nq * topk);
print_query_result(query_res);
+
+ index->UpdateIndexSize();
+ ASSERT_GT(index->IndexSize(), 0);
+
+ auto stats = index->GetStatistics();
+ ASSERT_TRUE(stats != nullptr);
+ index->ClearStatistics();
}
TEST(BINIDMAP, Build) {