Function TEST

src/colmap/retrieval/visual_index_test.cc:332–357

}

TEST(VisualIndex, Print) {
  SetPRNGSeed(1);

  VisualIndex::BuildOptions build_options;
  build_options.num_iterations = 10;
  build_options.num_rounds = 1;
  build_options.num_visual_words = 10;

  auto descriptors =
      CreateRandomDescriptors(100, 128, FeatureExtractorType::SIFT);
  auto visual_index = VisualIndex::Create(128, 64);
  visual_index->Build(build_options, descriptors);

  VisualIndex::IndexOptions index_options;
  FeatureKeypoints keypoints(10);
  auto descriptors_small =
      CreateRandomDescriptors(10, 128, FeatureExtractorType::SIFT);
  visual_index->Add(index_options, 1, keypoints, descriptors_small);
  visual_index->Add(index_options, 2, keypoints, descriptors_small);

  std::ostringstream stream;
  stream << *visual_index;
  EXPECT_EQ(stream.str(),
            "VisualIndex(num_visual_words=10, num_images=2, desc_dim=128, "
            "embedding_dim=64, feature_type=SIFT)");
}

INSTANTIATE_TEST_SUITE_P(VisualIndexTests,
                         ParameterizedVisualIndexTests,