added test for maximum date

epiverse-trace · Oct 3, 2023 · 5993222 · 5993222
1 parent 9908f54
commit 5993222
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/testthat/test-immunization_date.R b/tests/testthat/test-immunization_date.R
@@ -113,3 +113,22 @@ test_that("`get_immunization_date`: Take first vaccination", {
     all(vax_difference == immunization_delay)
   )
 })
+
+test_that("`get_immunization_date`: end_cohort > max_date", {
+  data("cohortdata")
+  cohortdata <- as.data.frame(cohortdata)
+
+  # get immunization dates
+  expect_warning(
+    get_immunization_date(
+      data = cohortdata,
+      outcome_date_col = "death_date",
+      outcome_delay = 0,
+      immunization_delay = 14,
+      vacc_date_col = c("vaccine_date_1", "vaccine_date_2"),
+      end_cohort = as.Date("2101-12-31"),
+      take_first = FALSE
+    )
+  )
+
+})