tests/testthat/test-snapshot.R

In rstudio/tfdatasets: Interface to 'TensorFlow' Datasets

test_that("dataset_snapshot works", {

  skip_if(tf_version() < "2.3")
  fi <- tempfile()

  res <- tensor_slices_dataset(1:10) %>%
    dataset_snapshot(fi) %>%
    reticulate::as_iterator() %>%
    reticulate::iterate()

  expect_length(res, 10)
  expect(length(dir(fi)) > 0, "snapshots not found")
})