tests/testthat/utils.R

In henry090/tfaddons: Interface to 'TensorFlow SIG Addons'

skip_if_no_tfaddons <- function(required_version = NULL) {
  if (!reticulate::py_module_available("tensorflow_addons"))
    skip("TensorFlow Addons 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_tfaddons()
    skip_if_no_tf_version("2.0")
    expect_error(force(expr), NA)
  })
}