tests/testthat/test-sm_predict.R

In ddsjoberg/sjosmooth: Kernel smoothing for time to event data

context("test-sm_predict")

test_that("sm_predict creates output without error/warning: glm", {
  glm_ex <-
    sm_predict(
      data = mtcars,
      method = "glm",
      formula = am ~ mpg,
      method.args = list(family = binomial(link = "logit")),
      type = "response"
    )

  expect_error(glm_ex, NA)
  expect_warning(glm_ex, NA)
})

test_that("sm_predict creates output without error/warning: coxph", {
  library(survival)
  coxph_ex <-
    sm_predict(
      data = lung,
      method = "coxph",
      formula = Surv(time, status) ~ age,
      type = "survival"
    )

  expect_error(coxph_ex, NA)
  expect_warning(coxph_ex, NA)
})

test_that("sm_predict with each kernel", {
  coxph_kernel <-
    purrr::map(
      c("epanechnikov", "tricube", "gaussian", "flat"),
      ~sm_predict(
        data = lung,
        method = "coxph",
        formula = Surv(time, status) ~ age,
        type = "survival",
        kernel = .x
      )
    )

  expect_error(coxph_kernel, NA)
  expect_warning(coxph_kernel, NA)
})