tests/testthat/test-survreg.R

In butcher: Model Butcher

test_that("survreg + predict() works", {
  skip_on_cran()
  skip_if_not_installed("survival")
  suppressPackageStartupMessages(library(survival))
  survreg_fit <- survreg(Surv(futime, fustat) ~ 1, data = ovarian)
  x <- axe_call(survreg_fit)
  expect_equal(x$call, rlang::expr(dummy_call()))
  expect_error(model.matrix(x))
  x <- axe_data(survreg_fit)
  expect_error(residuals(x))
  expect_equal(x$y, numeric(0))
  x <- axe_env(survreg_fit)
  expect_error(model.matrix(x))
  expect_identical(attr(x$terms, ".Environment"), rlang::base_env())
  x <- butcher(survreg_fit)
  expect_equal(predict(x, ovarian[1, 1:2])[1], c(`1` = 1225.4189589287))
  expect_equal(predict(survreg_fit, newdata = ovarian[1, 1:2], type = c("quantile")), c(160.794978494583, 2601.21611951026))
  expect_equal(predict(survreg_fit, newdata = ovarian[1, 1:2], type = c("linear")), c(`1` = 7.1110380717964))
  expect_equal(predict(survreg_fit, newdata = ovarian[1, 1:2], se.fit = TRUE)$se.fit, c(`1` = 358.714386983321))
})