xiaoxue_admin/zvec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test: fix ut random float (#182)

Browse Source
This commit is contained in:
ZeFeng Yin
2026-02-28 16:11:13 +08:00
committed by GitHub
parent 120d9e3e68
commit 596d5795e9
11 changed files with 112 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/core/algorithm/cluster/opt_kmeans_cluster_test.cc
+1 -1
View File
@@ -514,7 +514,7 @@ TEST(OptKmeansCluster, IN4Correctness) {
EXPECT_EQ(centroids1.size(), centroids2.size());
for (size_t i = 0; i < centroids1.size(); ++i) {
EXPECT_EQ(centroids1[i].follows(), centroids2[i].follows());
EXPECT_EQ(centroids1[i].score(), centroids2[i].score());
EXPECT_DOUBLE_EQ(centroids1[i].score(), centroids2[i].score());
}
}
tests/core/algorithm/flat/flat_searcher_test.cpp
+1 -1
View File
@@ -1346,7 +1346,7 @@ TEST(FlatProvider, Provider_FP32) {
const float *features1 = (const float *)provider1->get_vector(it1->key());
const float *features2 = (const float *)provider2->get_vector(it2->key());
for (size_t idx = 0; idx < dim; idx++) {
ASSERT_EQ(*features1, *features2);
ASSERT_FLOAT_EQ(*features1, *features2);
features1++;
features2++;
}
tests/core/algorithm/flat/flat_streamer_buffer_test.cpp
+4 -4
View File
@@ -104,7 +104,7 @@ TEST_F(FlatStreamerTest, TestLinearSearch) {
ASSERT_EQ(0, provider->get_vector(result1[0].key(), block));
const float *data = (float *)block.data();
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(data[j], i);
ASSERT_FLOAT_EQ(data[j], i);
}
ASSERT_EQ(i, result1[0].key());
@@ -150,7 +150,7 @@ TEST_F(FlatStreamerTest, TestLinearSearch) {
ASSERT_EQ(0, provider->get_vector(result1[0].key(), block));
const float *data = (float *)block.data();
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(data[j], i);
ASSERT_FLOAT_EQ(data[j], i);
}
ASSERT_EQ(i, result1[0].key());
@@ -226,7 +226,7 @@ TEST_F(FlatStreamerTest, TestLinearSearchMMap) {
ASSERT_EQ(0, provider->get_vector(result1[0].key(), block));
const float *data = (float *)block.data();
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(data[j], i);
ASSERT_FLOAT_EQ(data[j], i);
}
ASSERT_EQ(i, result1[0].key());
@@ -320,7 +320,7 @@ TEST_F(FlatStreamerTest, TestBufferStorage) {
EXPECT_EQ(topk, result1.size());
for (size_t j = 0; j < dim; ++j) {
const float *data = (float *)provider->get_vector(result1[0].key());
EXPECT_EQ(data[j], i);
EXPECT_FLOAT_EQ(data[j], i);
}
EXPECT_EQ(i, result1[0].key());
tests/core/algorithm/flat/flat_streamer_test.cc
+7 -7
View File
@@ -93,7 +93,7 @@ TEST_F(FlatStreamerTest, TestAddVector) {
streamer->add_impl(i, vec.data(), qmeta, ctx);
const float *data = (float *)provider->get_vector(i);
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(data[j], i);
ASSERT_FLOAT_EQ(data[j], i);
}
}
@@ -141,7 +141,7 @@ TEST_F(FlatStreamerTest, TestLinearSearch) {
ASSERT_EQ(topk, result1.size());
for (size_t j = 0; j < dim; ++j) {
const float *data = (float *)provider->get_vector(result1[0].key());
ASSERT_EQ(data[j], i);
ASSERT_FLOAT_EQ(data[j], i);
}
ASSERT_EQ(i, result1[0].key());
@@ -376,7 +376,7 @@ TEST_F(FlatStreamerTest, TestOpenClose) {
while (iter->is_valid()) {
float *data = (float *)provider->get_vector(cur);
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)cur, data[d]);
ASSERT_FLOAT_EQ((float)cur, data[d]);
}
iter->next();
cur += 2;
@@ -463,7 +463,7 @@ TEST_F(FlatStreamerTest, TestForceFlush) {
while (iter->is_valid()) {
float *data = (float *)provider->get_vector(cur);
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)cur, data[d]);
ASSERT_FLOAT_EQ((float)cur, data[d]);
}
iter->next();
cur++;
@@ -501,7 +501,7 @@ TEST_F(FlatStreamerTest, TestForceFlush) {
const float *data = (const float *)provider->get_vector(i);
ASSERT_NE(data, nullptr);
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(i, data[j]);
ASSERT_FLOAT_EQ(i, data[j]);
}
}
}
@@ -556,7 +556,7 @@ TEST_F(FlatStreamerTest, TestMultiThread) {
while (iter->is_valid()) {
float *data = (float *)iter->data();
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)iter->key(), data[d]);
ASSERT_FLOAT_EQ((float)iter->key(), data[d]);
}
total++;
min = std::min(min, iter->key());
@@ -716,7 +716,7 @@ TEST_F(FlatStreamerTest, TestConcurrentAddAndSearch) {
while (iter->is_valid()) {
float *data = (float *)iter->data();
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)iter->key(), data[d]);
ASSERT_FLOAT_EQ((float)iter->key(), data[d]);
}
total++;
min = std::min(min, iter->key());
tests/core/algorithm/flat_sparse/flat_sparse_streamer_test.cc
+5 -5
View File
@@ -387,7 +387,7 @@ TEST_F(FlatSparseStreamerTest, TestCreateIterator) {
float *sparse_data = (float *)iter->sparse_data();
ASSERT_EQ(cur, iter->key());
for (size_t d = 0; d < sparse_dim_count; ++d) {
ASSERT_EQ((float)cur, sparse_data[d]);
ASSERT_FLOAT_EQ((float)cur, sparse_data[d]);
}
iter->next();
cur++;
@@ -487,7 +487,7 @@ TEST_F(FlatSparseStreamerTest, TestOpenAndClose) {
float *sparse_data = (float *)iter->sparse_data();
ASSERT_EQ(cur, iter->key());
for (size_t d = 0; d < sparse_dim_count; ++d) {
ASSERT_EQ((float)cur, sparse_data[d]);
ASSERT_FLOAT_EQ((float)cur, sparse_data[d]);
}
iter->next();
cur += 2;
@@ -589,7 +589,7 @@ TEST_F(FlatSparseStreamerTest, TestForceFlush) {
const float *data = reinterpret_cast<const float *>(iter->sparse_data());
for (size_t j = 0; j < sparse_dim_count; ++j) {
ASSERT_EQ((float)cur, data[j]);
ASSERT_FLOAT_EQ((float)cur, data[j]);
}
iter->next();
@@ -710,7 +710,7 @@ TEST_F(FlatSparseStreamerTest, TestMultiThread) {
const float *data = reinterpret_cast<const float *>(iter->sparse_data());
for (size_t j = 0; j < sparse_dim_count; ++j) {
ASSERT_EQ((float)iter->key(), data[j]);
ASSERT_FLOAT_EQ((float)iter->key(), data[j]);
}
total++;
min = std::min(min, iter->key());
@@ -915,7 +915,7 @@ TEST_F(FlatSparseStreamerTest, TestConcurrentAddAndSearch) {
const float *data = reinterpret_cast<const float *>(iter->sparse_data());
for (size_t j = 0; j < sparse_dim_count; ++j) {
ASSERT_EQ((float)iter->key(), data[j]);
ASSERT_FLOAT_EQ((float)iter->key(), data[j]);
}
total++;
min = std::min(min, iter->key());
tests/core/algorithm/hnsw/hnsw_streamer_buffer_test.cpp
+1 -1
View File
@@ -229,7 +229,7 @@ TEST_F(HnswStreamerTest, TestHnswSearchMMap) {
ASSERT_EQ(0, provider->get_vector(result1[0].key(), block));
const float *data = (float *)block.data();
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(data[j], i);
ASSERT_FLOAT_EQ(data[j], i);
}
ASSERT_EQ(i, result1[0].key());
tests/core/algorithm/hnsw/hnsw_streamer_test.cc
+10 -10
View File
@@ -576,7 +576,7 @@ TEST_F(HnswStreamerTest, TestOpenClose) {
float *data = (float *)iter->data();
ASSERT_EQ(cur, iter->key());
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)cur, data[d]);
ASSERT_FLOAT_EQ((float)cur, data[d]);
}
iter->next();
cur += 2;
@@ -657,7 +657,7 @@ TEST_F(HnswStreamerTest, TestCreateIterator) {
float *data = (float *)iter->data();
ASSERT_EQ(cur, iter->key());
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)cur, data[d]);
ASSERT_FLOAT_EQ((float)cur, data[d]);
}
iter->next();
cur++;
@@ -689,7 +689,7 @@ TEST_F(HnswStreamerTest, TestCreateIterator) {
const float *data = (const float *)provider->get_vector(i);
ASSERT_NE(data, nullptr);
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(i, data[j]);
ASSERT_FLOAT_EQ(i, data[j]);
}
}
}
@@ -730,7 +730,7 @@ TEST_F(HnswStreamerTest, TestForceFlush) {
float *data = (float *)iter->data();
ASSERT_EQ(cur, iter->key());
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)cur, data[d]);
ASSERT_FLOAT_EQ((float)cur, data[d]);
}
iter->next();
cur++;
@@ -768,7 +768,7 @@ TEST_F(HnswStreamerTest, TestForceFlush) {
const float *data = (const float *)provider->get_vector(i);
ASSERT_NE(data, nullptr);
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(i, data[j]);
ASSERT_FLOAT_EQ(i, data[j]);
}
}
}
@@ -830,7 +830,7 @@ TEST_F(HnswStreamerTest, TestKnnMultiThread) {
while (iter->is_valid()) {
float *data = (float *)iter->data();
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)iter->key(), data[d]);
ASSERT_FLOAT_EQ((float)iter->key(), data[d]);
}
total++;
min = std::min(min, iter->key());
@@ -1008,7 +1008,7 @@ TEST_F(HnswStreamerTest, TestKnnConcurrentAddAndSearch) {
while (iter->is_valid()) {
float *data = (float *)iter->data();
for (size_t d = 0; d < dim; ++d) {
ASSERT_EQ((float)iter->key(), data[d]);
ASSERT_FLOAT_EQ((float)iter->key(), data[d]);
}
total++;
min = std::min(min, iter->key());
@@ -1584,7 +1584,7 @@ TEST_F(HnswStreamerTest, TestCheckDuplicateAndGetVector) {
const float *data = (const float *)provider->get_vector(i);
ASSERT_NE(data, nullptr);
for (size_t j = 0; j < dim; ++j) {
ASSERT_EQ(i, data[j]);
ASSERT_FLOAT_EQ(i, data[j]);
}
}
@@ -2275,7 +2275,7 @@ TEST_F(HnswStreamerTest, TestFetchVector) {
ASSERT_NE(vector, nullptr);
float vector_value = *(float *)(vector);
ASSERT_EQ(vector_value, i);
ASSERT_FLOAT_EQ(vector_value, i);
}
auto linearCtx = streamer->create_context();
@@ -2310,7 +2310,7 @@ TEST_F(HnswStreamerTest, TestFetchVector) {
ASSERT_NE(knnResult[0].vector(), nullptr);
float vector_value = *((float *)(knnResult[0].vector()));
ASSERT_EQ(vector_value, i);
ASSERT_FLOAT_EQ(vector_value, i);
}
std::cout << "knnTotalTime: " << knnTotalTime << std::endl;
std::cout << "linearTotalTime: " << linearTotalTime << std::endl;
tests/core/algorithm/hnsw_sparse/hnsw_sparse_searcher_test.cc → tests/core/algorithm/hnsw_sparse/hnsw_sparse_searcher_test.cpp
View File
tests/core/algorithm/hnsw_sparse/hnsw_sparse_streamer_test.cc
+4 -4
View File
@@ -494,7 +494,7 @@ TEST_F(HnswSparseStreamerTest, TestOpenClose) {
float *sparse_data = (float *)iter->sparse_data();
ASSERT_EQ(cur, iter->key());
for (size_t d = 0; d < sparse_dim_count; ++d) {
ASSERT_EQ((float)cur, sparse_data[d]);
ASSERT_FLOAT_EQ((float)cur, sparse_data[d]);
}
iter->next();
cur += 2;
@@ -587,7 +587,7 @@ TEST_F(HnswSparseStreamerTest, TestCreateIterator) {
float *sparse_data = (float *)iter->sparse_data();
ASSERT_EQ(cur, iter->key());
for (size_t d = 0; d < sparse_dim_count; ++d) {
ASSERT_EQ((float)cur, sparse_data[d]);
ASSERT_FLOAT_EQ((float)cur, sparse_data[d]);
}
iter->next();
cur++;
@@ -678,7 +678,7 @@ TEST_F(HnswSparseStreamerTest, TestForceFlush) {
const float *data = reinterpret_cast<const float *>(iter->sparse_data());
for (size_t j = 0; j < sparse_dim_count; ++j) {
ASSERT_EQ((float)cur, data[j]);
ASSERT_FLOAT_EQ((float)cur, data[j]);
}
iter->next();
@@ -1017,7 +1017,7 @@ TEST_F(HnswSparseStreamerTest, TestKnnConcurrentAddAndSearch) {
const float *data = reinterpret_cast<const float *>(iter->sparse_data());
for (size_t j = 0; j < sparse_dim_count; ++j) {
ASSERT_EQ((float)iter->key(), data[j]);
ASSERT_FLOAT_EQ((float)iter->key(), data[j]);
}
total++;
min = std::min(min, iter->key());
tests/core/algorithm/ivf/ivf_searcher_test.cc
+69 -69
View File
@@ -282,7 +282,7 @@ TEST_F(IVFSearcherTest, TestSimple) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -297,7 +297,7 @@ TEST_F(IVFSearcherTest, TestSimple) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)32 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -312,7 +312,7 @@ TEST_F(IVFSearcherTest, TestSimple) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -480,7 +480,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
ASSERT_EQ((uint64_t)(total - 1) - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -495,7 +495,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -510,7 +510,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -525,7 +525,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -604,7 +604,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithFilter) {
EXPECT_EQ((size_t)1, result.size());
for (size_t i = 0; i < 1; ++i) {
EXPECT_EQ((uint64_t)0, result[i].key());
EXPECT_EQ((float)999 * 999 * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)999 * 999 * dimension_, result[i].score());
}
}
@@ -619,7 +619,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithFilter) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)1, result.size());
EXPECT_EQ((uint64_t)0, result[0].key());
EXPECT_EQ((float)q * q * dimension_, result[0].score());
EXPECT_FLOAT_EQ((float)q * q * dimension_, result[0].score());
}
}
@@ -634,7 +634,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithFilter) {
EXPECT_EQ((size_t)1, result.size());
for (size_t i = 0; i < 1; ++i) {
EXPECT_EQ((uint64_t)0, result[i].key());
EXPECT_EQ((float)999 * 999 * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)999 * 999 * dimension_, result[i].score());
}
}
@@ -649,7 +649,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithFilter) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)1, result.size());
EXPECT_EQ((uint64_t)0, result[0].key());
EXPECT_EQ((float)q * q * dimension_, result[0].score());
EXPECT_FLOAT_EQ((float)q * q * dimension_, result[0].score());
}
}
@@ -725,7 +725,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -740,7 +740,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -755,7 +755,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -770,7 +770,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -850,7 +850,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithFilter) {
EXPECT_EQ((size_t)1, result.size());
for (size_t i = 0; i < 1; ++i) {
EXPECT_EQ((uint64_t)0, result[i].key());
EXPECT_EQ((float)999 * 999 * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)999 * 999 * dimension_, result[i].score());
}
}
@@ -865,7 +865,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithFilter) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)1, result.size());
EXPECT_EQ((uint64_t)0, result[0].key());
EXPECT_EQ((float)q * q * dimension_, result[0].score());
EXPECT_FLOAT_EQ((float)q * q * dimension_, result[0].score());
}
}
@@ -880,7 +880,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithFilter) {
EXPECT_EQ((size_t)1, result.size());
for (size_t i = 0; i < 1; ++i) {
EXPECT_EQ((uint64_t)0, result[i].key());
EXPECT_EQ((float)999 * 999 * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)999 * 999 * dimension_, result[i].score());
}
}
@@ -895,7 +895,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWithFilter) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)1, result.size());
EXPECT_EQ((uint64_t)0, result[0].key());
EXPECT_EQ((float)q * q * dimension_, result[0].score());
EXPECT_FLOAT_EQ((float)q * q * dimension_, result[0].score());
}
}
@@ -977,7 +977,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWith1LevelAndBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -992,7 +992,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWith1LevelAndBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1007,7 +1007,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWith1LevelAndBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -1022,7 +1022,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFloatWith1LevelAndBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1104,7 +1104,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWith1LevelAndBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -1119,7 +1119,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWith1LevelAndBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1134,7 +1134,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWith1LevelAndBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -1149,7 +1149,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWith1LevelAndBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1228,7 +1228,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorInt8WithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)127 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -1243,7 +1243,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorInt8WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1258,7 +1258,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorInt8WithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)127 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -1273,7 +1273,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorInt8WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1352,7 +1352,7 @@ TEST_F(IVFSearcherTest, TestRowMajorInt8WithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)127 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -1367,7 +1367,7 @@ TEST_F(IVFSearcherTest, TestRowMajorInt8WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1382,7 +1382,7 @@ TEST_F(IVFSearcherTest, TestRowMajorInt8WithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)127 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -1397,7 +1397,7 @@ TEST_F(IVFSearcherTest, TestRowMajorInt8WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1478,7 +1478,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorBinaryWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)256 - i, result[i].key());
EXPECT_EQ((float)i, result[i].score());
EXPECT_FLOAT_EQ((float)i, result[i].score());
}
}
@@ -1493,7 +1493,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorBinaryWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1508,7 +1508,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorBinaryWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)256 - i, result[i].key());
EXPECT_EQ((float)i, result[i].score());
EXPECT_FLOAT_EQ((float)i, result[i].score());
}
}
@@ -1523,7 +1523,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorBinaryWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1604,7 +1604,7 @@ TEST_F(IVFSearcherTest, TestRowMajorBinaryWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)256 - i, result[i].key());
EXPECT_EQ((float)i, result[i].score());
EXPECT_FLOAT_EQ((float)i, result[i].score());
}
}
@@ -1619,7 +1619,7 @@ TEST_F(IVFSearcherTest, TestRowMajorBinaryWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1634,7 +1634,7 @@ TEST_F(IVFSearcherTest, TestRowMajorBinaryWithBuildMemory) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)256 - i, result[i].key());
EXPECT_EQ((float)i, result[i].score());
EXPECT_FLOAT_EQ((float)i, result[i].score());
}
}
@@ -1649,7 +1649,7 @@ TEST_F(IVFSearcherTest, TestRowMajorBinaryWithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1861,7 +1861,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFp16WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -1893,7 +1893,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFp16WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2000,7 +2000,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFp16WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2032,7 +2032,7 @@ TEST_F(IVFSearcherTest, TestRowMajorFp16WithBuildMemory) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2110,7 +2110,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithHnswGraphType) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)(total - 1) - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -2125,7 +2125,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithHnswGraphType) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2140,7 +2140,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithHnswGraphType) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -2155,7 +2155,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithHnswGraphType) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2234,7 +2234,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithSsgGraphType) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)(total - 1) - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -2249,7 +2249,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithSsgGraphType) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2264,7 +2264,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithSsgGraphType) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -2279,7 +2279,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithSsgGraphType) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2356,7 +2356,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithInt8Converter) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)(total - 1) - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -2371,7 +2371,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithInt8Converter) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2386,7 +2386,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithInt8Converter) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((uint64_t)999 - i, result[i].key());
EXPECT_EQ((float)i * i * dimension_, result[i].score());
EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -2401,7 +2401,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithInt8Converter) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2497,7 +2497,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithFloat16Quantizer) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2529,7 +2529,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithFloat16Quantizer) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2626,7 +2626,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithConverterAndQuantizer) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
ASSERT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2658,7 +2658,7 @@ TEST_F(IVFSearcherTest, TestColumnMajorFloatWithConverterAndQuantizer) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
EXPECT_EQ((uint64_t)q, result[0].key());
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2755,7 +2755,7 @@ TEST_F(IVFSearcherTest, TestQuantizedPerCentroid) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
ASSERT_NEAR((uint64_t)(total - 1) - q, result[0].key(), 100);
// EXPECT_EQ((float)0, result[0].score());
// EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -2770,7 +2770,7 @@ TEST_F(IVFSearcherTest, TestQuantizedPerCentroid) {
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_NEAR((uint64_t)total - i - 1, result[i].key(), 100);
// EXPECT_EQ((float)i * i * dimension_, result[i].score());
// EXPECT_FLOAT_EQ((float)i * i * dimension_, result[i].score());
}
}
@@ -2785,7 +2785,7 @@ TEST_F(IVFSearcherTest, TestQuantizedPerCentroid) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((size_t)topk, result.size());
ASSERT_NEAR((uint64_t)(total - 1) - q, result[0].key(), 100);
// EXPECT_EQ((float)0, result[0].score());
// EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -3474,7 +3474,7 @@ TEST_F(IVFSearcherTest, TestSameValue) {
for (size_t q = 0; q < qnum; ++q) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}
@@ -3488,7 +3488,7 @@ TEST_F(IVFSearcherTest, TestSameValue) {
const IndexDocumentList &result = context->result(0);
EXPECT_EQ((size_t)topk, result.size());
for (size_t i = 0; i < topk; ++i) {
EXPECT_EQ((float)0, result[i].score());
EXPECT_FLOAT_EQ((float)0, result[i].score());
}
}
@@ -3501,7 +3501,7 @@ TEST_F(IVFSearcherTest, TestSameValue) {
for (size_t q = 0; q < qnum; ++q) {
const IndexDocumentList &result = context->result(q);
EXPECT_EQ((float)0, result[0].score());
EXPECT_FLOAT_EQ((float)0, result[0].score());
}
}