tests/testthat/utils.R

skip_if_no_tfhub <- function(required_version = NULL) {
  if (!reticulate::py_module_available("tensorflow_hub"))
    skip("TensorFlow Hub not available for testing")
}

skip_if_no_tf_version <- function(required_version) {
  if (!reticulate::py_module_available("tensorflow"))
    skip("TensorFlow is not available.")

  if (tensorflow::tf_version() < required_version)
    skip(paste0("Needs TF version >= ", required_version))
}

test_succeeds <- function(desc, expr) {
  test_that(desc, {
    skip_if_no_tfhub()
    skip_if_no_tf_version("2.0")
    expect_error(force(expr), NA)
  })
}
rstudio/tfhub documentation built on Dec. 25, 2021, 7:51 p.m.