From 94b4dcd37e662eb1e525dc241817c8dd5d4681fc Mon Sep 17 00:00:00 2001
From: Naeem Model <me@nmode.ca>
Date: Sat, 2 Nov 2024 18:13:28 +0000
Subject: Add input validation to estimators

---
 R/idea.R | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'R/idea.R')

diff --git a/R/idea.R b/R/idea.R
index 14ba838..dad4220 100644
--- a/R/idea.R
+++ b/R/idea.R
@@ -42,6 +42,13 @@
 #' # Obtain R0 when the serial distribution has a mean of three days.
 #' idea(cases, mu = 3 / 7)
 idea <- function(cases, mu) {
+  validate_cases(cases, min_length = 2, min_count = 1)
+  if (!is_real(mu) || mu <= 0) {
+    stop("The serial interval (`mu`) must be a number greater than 0.",
+      call. = FALSE
+    )
+  }
+
   s <- seq_along(cases) / mu
 
   x1 <- sum(s)
-- 
cgit v1.2.3