correct argument checking

simonpcouch · Jul 29, 2024 · 109ab48 · 109ab48
1 parent 056f116
commit 109ab48
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/R/utils.R b/R/utils.R
@@ -51,8 +51,14 @@ check_numeric <- function(x, arg = caller_arg(x), call = caller_env()) {
   }
 }
 
+check_integer <- function(x, arg = caller_arg(x), call = caller_env()) {
+  if (!inherits(x, "integer")) {
+    cli::cli_abort("{.arg {arg}} must be an integer.", call = call)
+  }
+}
+
 check_outcome <- function(x, arg = caller_arg(x), call = caller_env()) {
   if (!inherits_any(x, c("integer", "factor"))) {
-    cli::cli_abort("{.arg {arg}} must be an integer.", call = call)
+    cli::cli_abort("{.arg {arg}} must be an integer or factor.", call = call)
   }
 }
diff --git a/tests/testthat/_snaps/utils.md b/tests/testthat/_snaps/utils.md
@@ -36,7 +36,7 @@
       check_y(rnorm(10), "classification")
     Condition
       Error:
-      ! `y` must be an integer.
+      ! `y` must be an integer or factor.
 
 ---