Function TEST_F

tests/core/algorithm/flat/flat_streamer_test.cc:65–98

}

TEST_F(FlatStreamerTest, TestAddVector) {
  IndexStreamer::Pointer streamer =
      IndexFactory::CreateStreamer("FlatStreamer");
  ASSERT_TRUE(streamer != nullptr);

  Params params;
  ASSERT_EQ(0, streamer->init(*index_meta_ptr_, params));
  auto storage = IndexFactory::CreateStorage("MMapFileStorage");
  ASSERT_NE(nullptr, storage);
  Params stg_params;
  ASSERT_EQ(0, storage->init(stg_params));
  ASSERT_EQ(0, storage->open(dir_ + "Test/AddVector", true));
  ASSERT_EQ(0, streamer->open(storage));

  auto ctx = streamer->create_context();
  auto provider = streamer->create_provider();
  ASSERT_TRUE(!!ctx);

  IndexQueryMeta qmeta(IndexMeta::DT_FP32, dim);
  for (size_t i = 0; i < 1000UL; i++) {
    NumericalVector<float> vec(dim);
    for (size_t j = 0; j < dim; ++j) {
      vec[j] = i;
    }
    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_FLOAT_EQ(data[j], i);
    }
  }

  streamer->flush(0UL);
  streamer.reset();
}

TEST_F(FlatStreamerTest, TestLinearSearch) {
  IndexStreamer::Pointer streamer =