tests/testthat/test-ssvs.R

In sabainter/SSVS: Functions for Stochastic Search Variable Selection (SSVS)

test_that("ssvs argument error checking works", {
  predictors <- c("cyl", "disp", "hp", "drat", "wt", "vs", "am", "gear", "carb", "mpg")
  outcome <- "qsec"

  expect_error(ssvs())
  expect_error(ssvs("nodata", x = predictors, y = outcome))
  expect_error(ssvs(mtcars, x = "novar", y = outcome))
  expect_error(ssvs(mtcars, x = predictors, y = "novar"))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, inprob = 2))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, burn = 0))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, burn = 1000, runs = 1000))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, burn = 0))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, burn = 1.5))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, runs = 0))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, a1 = 0))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, b1 = 0))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, prec.beta = 0))
  expect_error(ssvs(mtcars, x = predictors, y = outcome, progress = "nobool"))
})

test_that("ssvs works", {
  predictors <- c("cyl", "disp", "hp", "drat", "wt", "vs", "am", "gear", "carb", "mpg")
  outcome <- "qsec"

  set.seed(1000)
  results_simple <- ssvs(data = mtcars, y = outcome, x = predictors, progress = FALSE)
  expect_equal(results_simple, readRDS(system.file("testdata/results_simple.rds", package = "SSVS")))
})