Add test

insightsengineering · May 16, 2023 · 87ec1ed · 87ec1ed
1 parent 90fc991
commit 87ec1ed
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/tests/testthat/_snaps/summarize_variables.md b/tests/testthat/_snaps/summarize_variables.md
@@ -1128,3 +1128,26 @@
       Median      NA      3.5         5.5   
       Min - Max   NA   3.0 - 4.0   5.0 - 6.0
 
+# `summarize_vars` na_level argument works with non-default input
+
+    Code
+      res
+    Output
+          all obs
+      ———————————
+      n      4   
+      a   2 (50%)
+      b   2 (50%)
+
+---
+
+    Code
+      res
+    Output
+           all obs 
+      —————————————
+      n       6    
+      a   2 (33.3%)
+      b   2 (33.3%)
+      c   2 (33.3%)
+
diff --git a/tests/testthat/test-summarize_variables.R b/tests/testthat/test-summarize_variables.R
@@ -345,3 +345,23 @@ testthat::test_that("`summarize_vars` works with empty named numeric variables",
   res <- testthat::expect_silent(result)
   testthat::expect_snapshot(res)
 })
+
+testthat::test_that("`summarize_vars` na_level argument works with non-default input", {
+  dta <- tibble::tibble(
+    foo = factor(c("a", "a", "b", "b", "c", "c"), levels = c("a", "b", "c"))
+  )
+
+  result <- basic_table() %>%
+    summarize_vars(vars = "foo", na_level = "c") %>%
+    build_table(dta)
+
+  res <- testthat::expect_silent(result)
+  testthat::expect_snapshot(res)
+
+  result <- basic_table() %>%
+    summarize_vars(vars = "foo", na_level = "c", na.rm = FALSE) %>%
+    build_table(dta)
+
+  res <- testthat::expect_silent(result)
+  testthat::expect_snapshot(res)
+})