add tests

Author: hfrick

tests/testthat/test-proportional_hazards.R

@@ -0,0 +1,77 @@
+
+test_that("primary arguments", {
+  new_empty_quosure <- function(expr) {
+    rlang::new_quosure(expr, env = rlang::empty_env())
+  }
+
+  ph_penalty <- proportional_hazards(penalty = 0.05)
+  expect_equal(
+    ph_penalty$args,
+    list(penalty = new_empty_quosure(0.05),
+         mixture = new_empty_quosure(NULL))
+  )
+
+  ph_mixture <- proportional_hazards(mixture = 0.34)
+  expect_equal(
+    ph_mixture$args,
+    list(penalty = new_empty_quosure(NULL),
+         mixture = new_empty_quosure(0.34))
+  )
+
+  ph_mixture_v <- proportional_hazards(mixture = varying())
+  expect_equal(
+    ph_mixture_v$args,
+    list(penalty = new_empty_quosure(NULL),
+         mixture = new_empty_quosure(varying()))
+  )
+})
+
+test_that("printing", {
+  expect_output(
+    print(proportional_hazards()),
+    "Proportional Hazards Model Specification \\(censored regression\\)"
+  )
+})
+
+test_that("updating", {
+  new_empty_quosure <- function(expr) {
+    rlang::new_quosure(expr, env = rlang::empty_env())
+  }
+
+  basic <- proportional_hazards()
+
+  update_num <- update(basic, penalty = 0.05)
+  expect_equal(
+    update_num$args,
+    list(penalty = new_empty_quosure(0.05),
+         mixture = new_empty_quosure(NULL))
+  )
+
+  param_tibb <- tibble::tibble(penalty = 0.05)
+  update_tibb <- update(basic, param_tibb)
+  expect_equal(
+    update_tibb$args,
+    list(penalty = 0.05,
+         mixture = new_empty_quosure(NULL))
+  )
+
+  param_list <- as.list(param_tibb)
+  update_list <- update(basic, param_list)
+  expect_equal(
+    update_list$args,
+    list(penalty = 0.05,
+         mixture = new_empty_quosure(NULL))
+  )
+})
+
+
+test_that("bad input", {
+  expect_error(proportional_hazards(mode = ", classification"))
+})
+
+test_that("wrong fit interface", {
+  expect_error(
+    proportional_hazards() %>% fit_xy(),
+    "must use the formula interface"
+  )
+})