]> nmode's Git Repositories - Rnaught/blobdiff - R/id.R
Update about.html
[Rnaught] / R / id.R
diff --git a/R/id.R b/R/id.R
index 5277591fc2cb92591c8d702cdec554b15910fbfd..c7c28d3f87f0eced67e1df8bfb61476f7b688821 100644 (file)
--- a/R/id.R
+++ b/R/id.R
 #' # Obtain R0 when the serial distribution has a mean of three days.
 #' id(cases, mu = 3 / 7)
 id <- function(cases, mu) {
 #' # Obtain R0 when the serial distribution has a mean of three days.
 #' id(cases, mu = 3 / 7)
 id <- function(cases, mu) {
+  validate_cases(cases, min_length = 1, min_count = 1)
+  if (!is_real(mu) || mu <= 0) {
+    stop("The serial interval (`mu`) must be a number greater than 0.",
+      call. = FALSE
+    )
+  }
+
   exp(sum((log(cases) * mu) / seq_along(cases)) / length(cases))
 }
   exp(sum((log(cases) * mu) / seq_along(cases)) / length(cases))
 }