ARROW-17674: [R] Implement dplyr::across() inside arrange() (apache#1…

…4092) Authored-by: SHIMA Tatsuya <[email protected]> Signed-off-by: Neal Richardson <[email protected]>
fatemehp · Oct 17, 2022 · 5fa8ef1 · 5fa8ef1
1 parent 93de2b5
commit 5fa8ef1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/r/R/dplyr-arrange.R b/r/R/dplyr-arrange.R
@@ -20,7 +20,8 @@
 
 arrange.arrow_dplyr_query <- function(.data, ..., .by_group = FALSE) {
   call <- match.call()
-  exprs <- quos(...)
+  exprs <- expand_across(.data, quos(...))
+
   if (.by_group) {
     # when the data is is grouped and .by_group is TRUE, order the result by
     # the grouping columns first

diff --git a/r/tests/testthat/test-dplyr-arrange.R b/r/tests/testthat/test-dplyr-arrange.R
@@ -201,3 +201,18 @@ test_that("arrange() with bad inputs", {
     fixed = TRUE
   )
 })
+
+test_that("Can use across() within arrange()", {
+  compare_dplyr_binding(
+    .input %>%
+      arrange(across(starts_with("d"))) %>%
+      collect(),
+    example_data
+  )
+  compare_dplyr_binding(
+    .input %>%
+      arrange(across(starts_with("d"), desc)) %>%
+      collect(),
+    example_data
+  )
+})