add test

easystats · Oct 10, 2024 · b45a517 · b45a517
1 parent 30f6ba5
commit b45a517
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/R/row_count.R b/R/row_count.R
@@ -75,6 +75,11 @@ row_count <- function(data,
 
   data <- .coerce_to_dataframe(data[select])
 
+  # check if we have a data framme with at least two columns
+  if (nrow(data) < 1) {
+    insight::format_error("`data` must be a data frame with at least one row.")
+  }
+
   # check if we have a data framme with at least two columns
   if (ncol(data) < 2) {
     insight::format_error("`data` must be a data frame with at least two numeric columns.")

diff --git a/tests/testthat/test-row_count.R b/tests/testthat/test-row_count.R
@@ -23,6 +23,7 @@ test_that("row_count, errors or messages", {
   expect_error(expect_warning(row_count(iris, select = "abc")), regex = "must be a valid")
   expect_error(expect_warning(row_count(iris, select = "abc", count = 3)), regex = "No columns")
   expect_error(row_count(iris[1], count = 3), regex = "with at least")
+  expect_error(row_count(d_mn[-c(1:4), , drop = FALSE], count = 2), regex = "one row")
 })
 
 test_that("row_count, exact match", {