Add input checks to offspring_ll()

epiverse-trace · Nov 10, 2023 · 5ddef94 · 5ddef94
1 parent 8a1d936
commit 5ddef94
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/R/stat_likelihoods.R b/R/stat_likelihoods.R
@@ -168,6 +168,17 @@ geom_length_ll <- function(x, prob) {
 #' )
 offspring_ll <- function(x, offspring_dist, statistic,
                          nsim_offspring = 100, ...) {
+  # Input checking
+  checkmate::assert_numeric(
+    x, lower = 0, any.missing = FALSE
+  )
+  # check that offspring is properly specified
+  check_offspring_valid(offspring_dist)
+  checkmate::assert_character(statistic)
+  checkmate::assert_numeric(
+    nsim_offspring, lower = 1
+  )
+
   # Simulate the chains
   dist <- simulate_summary(
     nchains = nsim_offspring,