tests/testthat/test-dataset.R

In Athospd/guessthecorrelation: Guess The Correlation Torch Dataset and Models

test_that("dataset works", {

  dataset <- guess_the_correlation_dataset(
    root = tempfile(),
    token = "kaggle.json",
    download = TRUE
  )

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

})