Function TEST

tests/core/algorithm/flat/flat_searcher_test.cpp:93–209

}

TEST(FlatSearcher, NoBatch_FP32) {
  std::mt19937 gen((std::random_device())());
  auto dist = std::uniform_real_distribution<float>(0.0f, 1.0f);
  size_t dim = (std::uniform_int_distribution<size_t>(1, 512))(gen);

  IndexBuilder::Pointer builder = IndexFactory::CreateBuilder("FlatBuilder");
  ASSERT_NE(builder, nullptr);

  auto holder = std::make_shared<MultiPassIndexHolder<IndexMeta::DT_FP32>>(dim);
  uint32_t document_count =
      (std::uniform_int_distribution<size_t>(1, 10000))(gen);
  for (uint32_t i = 0; i < document_count; i++) {
    NumericalVector<float> vec(dim);
    for (size_t j = 0; j < vec.size(); ++j) {
      vec[j] = dist(gen) + static_cast<float>(i * 5);
    }
    ASSERT_TRUE(holder->emplace(i, vec));
  }

  IndexMeta meta1;
  meta1.set_meta(IndexMeta::DataType::DT_FP32, dim);
  meta1.set_metric("SquaredEuclidean", 0, Params());
  meta1.set_major_order(IndexMeta::MO_ROW);
  BuildIndex(meta1, holder, INDEX_PATH + ".1");

  IndexMeta meta2;
  meta2.set_meta(IndexMeta::DataType::DT_FP32, dim);
  meta2.set_metric("SquaredEuclidean", 0, Params());
  meta2.set_major_order(IndexMeta::MO_COLUMN);
  BuildIndex(meta2, holder, INDEX_PATH + ".2");

  IndexSearcher::Pointer searcher1, searcher2;
  LoadIndex(INDEX_PATH + ".1", searcher1);
  LoadIndex(INDEX_PATH + ".2", searcher2);

  auto context1 = searcher1->create_context();
  auto context2 = searcher2->create_context();
  auto context3 = searcher1->create_context();
  auto context4 = searcher2->create_context();
  auto context5 = searcher1->create_context();
  auto context6 = searcher2->create_context();
  uint32_t topk = std::min(10u, document_count);
  context1->set_topk(topk);
  context2->set_topk(topk);
  context3->set_topk(topk);
  context4->set_topk(topk);
  context3->set_filter([](uint64_t) { return false; });
  context4->set_filter([](uint64_t) { return false; });
  context5->set_filter([](uint64_t) { return true; });
  context6->set_filter([](uint64_t) { return true; });

  uint32_t query_count = (std::uniform_int_distribution<size_t>(1, 100))(gen);
  for (uint32_t i = 0; i < query_count; i++) {
    NumericalVector<float> vec(dim);
    for (uint32_t j = 0; j < vec.size(); ++j) {
      vec[j] = dist(gen);
    }
    ASSERT_EQ(