tests/testthat/test-guess-the-correlation.R

In mlverse/torchdatasets: Ready to Use Extra Datasets for Torch

test_that("guess_the_correlation_dataset works", {

  tmp <- tempfile()
  dataset <- guess_the_correlation_dataset(
    root = tmp,
    download = TRUE,
    transform = function(x) torch::torch_zeros(3,3)
  )

  expect_length(dataset$.getitem(1), 3)
  expect_equal(dim(dataset$.getitem(1)$x), c(3,3))
  expect_true(dataset$.getitem(1)$y$dtype == torch::torch_float())

})