Merge pull request #804 from markfairbanks/pmap-data_frame

`pmap()` works on data frame inputs

NEWS.md

@@ -2,6 +2,7 @@
 
 #### Bug fixes
 * Attempting to rename columns using `group_by()` now leads to an error (#799)
+* `pmap()` family works with data frame inputs (#803)
 
 # tidytable 0.11.0
 

R/purrr-pmap.R

@@ -62,12 +62,11 @@ pmap_vec <- function(.l, .f, ..., .ptype = NULL) {
 }
 
 .args_recycle <- function(args) {
-  lengths <- map_int(args, length)
-  n <- max(lengths)
+  args <- as.list(args)
+  sizes <- list_sizes(args)
+  n <- max(sizes)
 
-  stopifnot(all(lengths == 1L | lengths == n))
-  to_recycle <- lengths == 1L
-  args[to_recycle] <- map(args[to_recycle], function(x) rep.int(x, n))
+  args <- map(args, vec_recycle, n)
 
   args
 }

tests/testthat/test-purrr.R

@@ -61,6 +61,10 @@ test_that("pmap works", {
   expect_equal(.df, tidytable(x = 1:3, y = 1:3))
   .vec <- pmap_vec(list(1:3, 1:3), ~ .x + .y)
   expect_equal(.vec, c(2, 4, 6))
+  # Works on data.frame inputs, #803
+  df <- tidytable(a = 1:3, b = 4:6)
+  out <- pmap(df, ~ .x + .y)
+  expect_equal(out, list(5, 7, 9))
 })