Function TEST

src/colmap/controllers/image_reader_test.cc:183–216

                                      /*extension=*/".png")));

TEST(ImageReaderTest, SingleCamera) {
  auto database = Database::Open(kInMemorySqliteDatabasePath);
  const auto test_dir = CreateTestDir();

  ImageReaderOptions options;
  options.image_path = test_dir / "images";
  options.single_camera = true;
  CreateDirIfNotExists(options.image_path);

  // Create 3 test images with same dimensions
  Bitmap test_bitmap(10, 20, true);
  test_bitmap.Write(options.image_path / "0.png");
  test_bitmap.Write(options.image_path / "1.png");
  test_bitmap.Write(options.image_path / "2.png");

  ImageReader image_reader(options, database.get());
  EXPECT_EQ(image_reader.NumImages(), 3);

  Rig rig;
  Camera camera;
  Image image;
  PosePrior pose_prior;
  Bitmap bitmap;
  Bitmap mask;

  for (int i = 0; i < 3; ++i) {
    const auto status =
        image_reader.Next(&rig, &camera, &image, &pose_prior, &bitmap, &mask);
    ASSERT_EQ(status, ImageReader::Status::SUCCESS);
  }

  EXPECT_EQ(database->NumRigs(), 1);
  EXPECT_EQ(database->NumCameras(), 1);
}

TEST(ImageReaderTest, SingleCameraDimensionError) {
  auto database = Database::Open(kInMemorySqliteDatabasePath);