tests/testthat/utils.R

In tfio: Interface to 'TensorFlow IO'

library(tensorflow)

skip_if_no_tensorflow_io <- function() {
  if (!identical(Sys.getenv("NOT_CRAN"), "true")) {
    skip("Skip tests on CRAN due to lack of proper Python setup")
  }
  if (!reticulate::py_module_available("tensorflow_io"))
    skip("tensorflow_io Python module is not available for testing")
}

test_succeeds <- function(desc, expr) {
  test_that(desc, {
    skip_if_no_tensorflow_io()
    expect_error(force(expr), NA)
  })
}

iterate_all_batches <- function(dataset) {
  sess <- tf$Session()
  iterator <- make_iterator_one_shot(dataset)
  next_batch <- iterator_get_next(iterator)

  until_out_of_range({
    sess$run(next_batch)
  })
}