tests/testthat/test-dogs-vs-cats.R

test_that("dogs-vs-cats dataset", {

  dataset <- dogs_vs_cats_dataset(
    "./dogs-vs-cats",
    download = TRUE
  )

  expect_length(dataset$.getitem(1), 2)

})
mlverse/torchdatasets documentation built on April 5, 2025, 3:12 a.m.