add tests

R/data_tabulate.R

@@ -10,7 +10,7 @@
 #' @param by Optional vector or factor. If supplied, a crosstable is created.
 #' If `x` is a data frame, `by` can also be a character string indicating the
 #' name of a variable in `x`.
-#' @param drop_levels Logical, if `TRUE`, factor levels that do not occur in
+#' @param drop_levels Logical, if `FALSE`, factor levels that do not occur in
 #' the data are included in the table (with frequency of zero), else unused
 #' factor levels are dropped from the frequency table.
 #' @param name Optional character string, which includes the name that is used

tests/testthat/test-data_tabulate.R

@@ -243,6 +243,16 @@ test_that("data_tabulate print, collapse groups, drop levels", {
   )
 })
 
+test_that("data_tabulate drop levels", {
+  x <- factor(rep(letters[1:3], 3), levels = letters[1:5])
+  out1 <- data_tabulate(x, drop_levels = FALSE)
+  out2 <- data_tabulate(x, drop_levels = TRUE)
+  expect_identical(out1$N, c(3L, 3L, 3L, 0L, 0L, 0L))
+  expect_identical(as.character(out1$Value), c("a", "b", "c", "d", "e", NA))
+  expect_identical(out2$N, c(3L, 3L, 3L, 0L))
+  expect_identical(as.character(out2$Value), c("a", "b", "c", NA))
+})
+
 
 # select helpers ------------------------------
 test_that("data_tabulate regex", {
@@ -254,6 +264,8 @@ test_that("data_tabulate regex", {
 })
 
 
+# missing values ------------------------------
+
 test_that("data_tabulate exclude/include missing values", {
   data(efc, package = "datawizard")
   set.seed(123)