Add test for mapping fn resolve

ElianHugh · Dec 22, 2023 · 1f848f9 · 1f848f9
1 parent dca452f
commit 1f848f9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/R/map.R b/R/map.R
@@ -1,4 +1,4 @@
-resolve_mapping_function <- function(.f, .type, .call) {
+resolve_mapping_function <- function(.f = NULL, .type = NULL, .call = NULL) {
     .f <- .f %||% switch(.type,
         block = qto_block,
         div = qto_div,
@@ -51,11 +51,7 @@ map_qto <- function(.x,
         .x,
         function(x) {
             x <- .f(x, ...)
-
-            if (inherits(x, "quarto_block")) {
-                return(x)
-            }
-
+            if (inherits(x, "quarto_block")) return(x)
             qto_block(x, sep = .sep, collapse = .collapse, .call = .call)
         }
     )

diff --git a/tests/testthat/test_map.R b/tests/testthat/test_map.R
@@ -5,6 +5,14 @@ check_types <- function(lst) {
     all(res)
 }
 
+test_that("mapping function is resolved correctly", {
+    expect_identical(
+        resolve_mapping_function(.type = "block"),
+        qto_block
+    )
+    expect_type(resolve_mapping_function(.f = ~.x + 1L), "closure")
+})
+
 test_that("map_qto works", {
     qto_list <- map_qto(
         list("This is a note.", "And this is a note.", "And this is a note"),