also test status_y error messages

Author: jgabry

R/ppc-censoring.R

@@ -90,8 +90,9 @@ ppc_km_overlay <- function(
   suggested_package("survival")
   suggested_package("ggfortify")
 
-  stopifnot(is.numeric(status_y))
-  stopifnot(all(status_y %in% c(0, 1)))
+  if (!is.numeric(status_y) || length(status_y) != length(y) || !all(status_y %in% c(0, 1))) {
+    stop("`status_y` must be a numeric vector of 0s and 1s the same length as `y`.")
+  }
 
   if (!is.null(left_truncation_y)) {
     if (!is.numeric(left_truncation_y) || length(left_truncation_y) != length(y)) {

tests/testthat/test-ppc-censoring.R

@@ -32,6 +32,22 @@ test_that("ppc_km_overlay_grouped returns a ggplot object", {
                                    status_y = status_y2))
 })
 
+test_that("ppc_km_overlay errors if bad status_y value", {
+  skip_if_not_installed("ggfortify")
+  expect_error(
+    ppc_km_overlay(y, yrep, status_y = FALSE),
+    "`status_y` must be a numeric vector of 0s and 1s the same length as `y`."
+  )
+  expect_error(
+    ppc_km_overlay(y, yrep, status_y = 1:10),
+    "`status_y` must be a numeric vector of 0s and 1s the same length as `y`."
+  )
+  expect_error(
+    ppc_km_overlay(y, yrep, status_y = rep(10, length(y))),
+    "`status_y` must be a numeric vector of 0s and 1s the same length as `y`."
+  )
+})
+
 test_that("ppc_km_overlay errors if bad left_truncation_y value", {
   skip_if_not_installed("ggfortify")
   expect_error(