added test, fixed typo

tidyverse · Aug 15, 2024 · b10faab · b10faab
1 parent 8d2ad6e
commit b10faab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/R/imap.R b/R/imap.R
@@ -60,7 +60,7 @@ imap_dbl <- function(.x, .f, ...) {
 #' @export
 imap_vec <- function(.x, .f, ...) {
   .f <- as_mapper(.f, ...)
-  map_vec(.x, vec_index(.x), .f, ...)
+  map2_vec(.x, vec_index(.x), .f, ...)
 }
 
 

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", {