Test for case when time is specified but not present in the data

epiverse-trace · Sep 20, 2023 · a3cbf4a · a3cbf4a
1 parent a9f4b00
commit a3cbf4a
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/testthat/test-epichains.R b/tests/testthat/test-epichains.R
@@ -435,6 +435,22 @@ test_that("aggregate.epichains method returns correct objects", {
   )
 })
 
+test_that("aggregate.epichains method throws errors", {
+  expect_error(
+    aggregate(
+      simulate_tree(
+        nchains = 10,
+        statistic = "size",
+        offspring_dist = "pois",
+        stat_max = 10,
+        lambda = 2
+      ),
+      grouping_var = "time"
+    ),
+    "Object must have a time column"
+  )
+})
+
 test_that("aggregate.epichains method is numerically correct", {
   set.seed(12)
   #' Simulate a tree of infections without serials