tests/testthat/test_e2e_spec_removal.R

In kerasnip: A Bridge Between 'keras' and 'tidymodels'

test_that("E2E: Model spec removal works", {
  skip_if_no_keras()

  model_name <- "removable_model"

  input_block <- function(model, input_shape) {
    keras3::keras_model_sequential(input_shape = input_shape)
  }
  output_block <- function(model) {
    model |> keras3::layer_dense(units = 1)
  }

  create_keras_sequential_spec(
    model_name = model_name,
    layer_blocks = list(input = input_block, output = output_block),
    mode = "regression"
  )

  update_method_name <- paste0("update.", model_name)

  expect_true(exists(model_name, inherits = FALSE))
  expect_true(exists(update_method_name, inherits = FALSE))
  expect_error(parsnip:::check_model_doesnt_exist(model_name))

  remove_keras_spec(model_name)

  expect_false(exists(model_name, inherits = FALSE))
  expect_false(exists(update_method_name, inherits = FALSE))
  expect_no_error(parsnip:::check_model_doesnt_exist(model_name))
})

test_that("E2E: Model spec removal is not too aggressive", {
  skip_if_no_keras()

  model_name <- "my_mlp"
  model_name_2 <- "my_mlp_2"

  input_block <- function(model, input_shape) {
    keras3::keras_model_sequential(input_shape = input_shape)
  }
  output_block <- function(model) {
    model |> keras3::layer_dense(units = 1)
  }

  create_keras_sequential_spec(
    model_name = model_name,
    layer_blocks = list(input = input_block, output = output_block),
    mode = "regression"
  )

  create_keras_sequential_spec(
    model_name = model_name_2,
    layer_blocks = list(input = input_block, output = output_block),
    mode = "regression"
  )

  expect_true(exists(model_name, inherits = FALSE))
  expect_true(exists(model_name_2, inherits = FALSE))
  expect_error(parsnip:::check_model_doesnt_exist(model_name))
  expect_error(parsnip:::check_model_doesnt_exist(model_name_2))

  remove_keras_spec(model_name)

  expect_false(exists(model_name, inherits = FALSE))
  expect_true(exists(model_name_2, inherits = FALSE))
  expect_no_error(parsnip:::check_model_doesnt_exist(model_name))
  expect_error(parsnip:::check_model_doesnt_exist(model_name_2))
  
  # cleanup
  remove_keras_spec(model_name_2)
})