Expose base::rank().

Author: Sun Rui

R/pkg/R/functions.R

@@ -2179,6 +2179,13 @@ setMethod("rank",
             column(jc)
           })
 
+# Expose rank() in the R base package
+setMethod("rank",
+          signature(x = "ANY"),
+          function(x, ...) {
+            base::rank(x, ...)
+          })
+
 #' rowNumber
 #'
 #' Window function: returns a sequential number starting at 1 within a window partition.

R/pkg/R/generics.R

@@ -904,7 +904,7 @@ setGeneric("randn", function(seed) { standardGeneric("randn") })
 
 #' @rdname rank
 #' @export
-setGeneric("rank", function(x) { standardGeneric("rank") })
+setGeneric("rank", function(x, ...) { standardGeneric("rank") })
 
 #' @rdname regexp_extract
 #' @export

R/pkg/inst/tests/test_sparkSQL.R

@@ -833,6 +833,10 @@ test_that("column functions", {
   c13 <- cumeDist() + ntile(1)
   c14 <- denseRank() + percentRank() + rank() + rowNumber()
 
+  # Test if base::rank() is exposed
+  expect_equal(class(rank())[[1]], "Column")
+  expect_equal(rank(1:3), as.numeric(c(1:3)))
+
   df <- jsonFile(sqlContext, jsonPath)
   df2 <- select(df, between(df$age, c(20, 30)), between(df$age, c(10, 20)))
   expect_equal(collect(df2)[[2, 1]], TRUE)