tests/testthat/test-skeleton.R

In LSTbook: Data and Software for "Lessons in Statistical Thinking"

test_that("model_skeleton() generates a skeleton of inputs when requested", {
  mod <- mtcars |> model_train(mpg ~ wt + hp)
  Tmp <- model_skeleton(mod, ncont=5)
  expect_true(length(unique(Tmp$wt)) >= 5)
  expect_true(length(unique(Tmp$hp)) >= 5)
  expect_true(nrow(Tmp) > 25)
  expect_true(ncol(Tmp) == 2)
})

test_that("model_skeleton() gets untransformed explanatory variables", {
  mod <- mosaicData::Galton |> model_train(height ~ splines::ns(father, 4) * poly(mother,3) * sex)
  Tmp <- model_skeleton(mod)
  expect_true(all(names(Tmp) %in% c("father", "mother", "sex")))
})