tests/testthat/test-tornado_survreg.R

In bertcarnell/tornado: Plots for Model Sensitivity and Variable Importance

context("test-tornado_survreg")

test_that("survreg tornado works", {
  torn <- tornado(survreg_model, modeldata = survival::ovarian, type = "PercentChange", alpha = 0.10)
  expect_equal(class(torn), "tornado_plot")
  g <- plot(torn, plot = FALSE, xlabel = "Survival Time")
  g <- g + ggtitle("Test: survival regression")
  plot(g)

  # change to a factor
  mydat <- survival::ovarian
  mydat$resid.ds <- factor(mydat$resid.ds)
  gtest <- survival::survreg(survival::Surv(futime, fustat) ~ ecog.ps + rx + resid.ds, mydat, dist = 'weibull', scale = 1)
  torn <- tornado(gtest, modeldata = mydat, type = "PercentChange", alpha = 0.10)
  expect_equal(class(torn), "tornado_plot")
  g <- plot(torn, plot = FALSE, xlabel = "Survival Time")
  g <- g + ggtitle("Test: survival regression with factor")
  plot(g)
})

test_that("survreg tornado works with weights", {
  torn <- tornado(survreg_model_weighted, modeldata = survival::ovarian, type = "PercentChange", alpha = 0.10)
  expect_equal(class(torn), "tornado_plot")
  g <- plot(torn, plot = FALSE, xlabel = "Survival Time")
  g <- g + ggtitle("Test: survival regression with weights")
  plot(g)

  # weighted is different from unweighted
  torn_base <- tornado(survreg_model, modeldata = survival::ovarian, type = "PercentChange", alpha = 0.10)
  expect_true(abs(torn_base$data$pmeans - torn$data$pmeans) > 1E-6)
})