tests/testthat/test_surv_ceteris_paribus.R

In survxai: Visualization of the Local and Global Survival Model Explanations

context("surv_ceteris_paribus")

source("objects_for_tests.R")

test_that("Creating surv_ceteris_paribus", {
  expect_is(cp_cph, "surv_ceteris_paribus_explainer")
  expect_is(cp_cph, "data.frame")
  expect_is(ceteris_paribus(surve_cph, pbcTest[1,-c(1,5)], selected_variables = "sex"), "data.frame")
})


test_that("Wrong input",{
  expect_error(ceteris_paribus(surve_cph))
  expect_error(ceteris_paribus(pbcTest[1,-c(1,5)]))
  expect_error(ceteris_paribus(surve_cph_null_data, pbcTest[1,-c(1,5)]))
})