added imap_vec() (#1137)

Fixes #1084
tidyverse · Aug 15, 2024 · d022c4e · d022c4e
1 parent 770bbfe
commit d022c4e
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 0 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -60,6 +60,7 @@ export(imap_dfr)
 export(imap_int)
 export(imap_lgl)
 export(imap_raw)
+export(imap_vec)
 export(imodify)
 export(insistently)
 export(invoke)

diff --git a/NEWS.md b/NEWS.md
@@ -1,5 +1,6 @@
 # purrr (development version)
 
+* Added `imap_vec()` (#1084)
 * `list_transpose()` inspects all elements to determine the correct
   template if it's not provided by the user  (#1128, @krlmlr).
 

diff --git a/R/imap.R b/R/imap.R
@@ -56,6 +56,13 @@ imap_dbl <- function(.x, .f, ...) {
   map2_dbl(.x, vec_index(.x), .f, ...)
 }
 
+#' @rdname imap
+#' @export
+imap_vec <- function(.x, .f, ...) {
+  .f <- as_mapper(.f, ...)
+  map2_vec(.x, vec_index(.x), .f, ...)
+}
+
 
 #' @export
 #' @rdname imap

diff --git a/man/imap.Rd b/man/imap.Rd
diff --git a/tests/testthat/test-imap.R b/tests/testthat/test-imap.R
@@ -13,6 +13,7 @@ test_that("atomic vector imap works", {
   expect_length(imap_chr(x, paste), 3)
   expect_equal(imap_int(x, ~ .x + as.integer(.y)), x * 2)
   expect_equal(imap_dbl(x, ~ .x + as.numeric(.y)), x * 2)
+  expect_equal(imap_vec(x, ~ .x + as.numeric(.y)), x * 2)
 })
 
 test_that("iwalk returns invisibly", {