tests/testthat/test_survival_reg.R

In parsnip: A Common API to Modeling and Analysis Functions

test_that("primary argument", {

  normal <- survival_reg(dist = "lnorm")

  expect_snapshot(normal)

  expect_equal(
    normal$args,
    list(dist = rlang::quo("lnorm"))
  )

  dist_v <- survival_reg(dist = tune())

  expect_snapshot(dist_v)

  expect_equal(
    dist_v$args,
    list(dist = rlang::quo(tune()))
  )
})

test_that("updating", {
  expect_snapshot(
    survival_reg() %>%
      update(dist = "lnorm")
  )
})

test_that("bad input", {
  expect_error(survival_reg(mode = ", classification"))
})