tests/testthat/test-collection-rf100-underwater.R

In torchvision: Models, Datasets and Transformations for Images

context("dataset-rf100-underwater")

t <- withr::local_tempdir()

test_that("rf100_underwater_collection handles missing files gracefully", {
  expect_error(
    rf100_underwater_collection(dataset = "pipes", split = "train", download = FALSE),
    class = "runtime_error"
  )
})

datasets <- c("pipes", "objects", "coral")

for (ds_name in datasets) {
  test_that(paste0("rf100_underwater_collection loads ", ds_name, " correctly"), {
    skip_if(Sys.getenv("TEST_LARGE_DATASETS", unset = 0) != 1,
            "Skipping test: set TEST_LARGE_DATASETS=1 to enable tests requiring large downloads.")
    ds <- rf100_underwater_collection(dataset = ds_name, split = "train", download = TRUE)

    expect_s3_class(ds, "rf100_underwater_collection")
    expect_gt(ds$.length(), 1)
    expect_type(ds$classes, "character")
    expect_gt(length(unique(ds$classes)), 1)

    item <- ds[1]

    expect_type(item$y, "list")
    expect_named(item$y, c("image_id","labels","boxes"))
    expect_type(item$y$labels, "integer")
    expect_tensor(item$y$boxes)
    expect_equal(item$y$boxes$ndim, 2)
    expect_equal(item$y$boxes$size(2), 4)
    expect_s3_class(item, "image_with_bounding_box")
  })
}

test_that(paste0("rf100_underwater_collection loads `aquarium` correctly"), {
  ds <- rf100_underwater_collection(dataset = "aquarium", split = "train", download = TRUE)

  expect_s3_class(ds, "rf100_underwater_collection")
  expect_gt(ds$.length(), 1)
  expect_type(ds$classes, "character")
  expect_gt(length(unique(ds$classes)), 1)

  item <- ds[1]

  expect_type(item$y, "list")
  expect_named(item$y, c("image_id","labels","boxes"))
  expect_type(item$y$labels, "integer")
  expect_tensor(item$y$boxes)
  expect_equal(item$y$boxes$ndim, 2)
  expect_equal(item$y$boxes$size(2), 4)
  expect_s3_class(item, "image_with_bounding_box")

  ds <- rf100_underwater_collection(dataset = "aquarium", split = "test", download = TRUE)

  expect_s3_class(ds, "rf100_underwater_collection")
  expect_gt(ds$.length(), 1)
  expect_type(ds$classes, "character")
  expect_gt(length(unique(ds$classes)), 1)

  ds <- rf100_underwater_collection(dataset = "aquarium", split = "val", download = TRUE)

  expect_s3_class(ds, "rf100_underwater_collection")
  expect_gt(ds$.length(), 1)
  expect_type(ds$classes, "character")
  expect_gt(length(unique(ds$classes)), 1)
})