Add runif(n()) translation (#526)

DESCRIPTION

@@ -40,8 +40,8 @@ Imports:
 Suggests: 
     blob,
     covr,
-    dbplyr (>= 2.2.1),
-    dplyr (>= 0.7.0),
+    dbplyr (>= 2.4.0),
+    dplyr (>= 1.1.0),
     hms,
     readr,
     sodium,

NEWS.md

@@ -1,5 +1,8 @@
 # bigrquery (development version)
 
+* Added a translation for `runif(n())`. This fixes the translation for
+  `slice_sample()` (@mgirlich, #448).
+
 * Add a `dbQuoteLiteral()` method for logicals to revert breaking change 
   introduced by DBI 1.1.2 (@meztez, #478).
 

R/dplyr.R

@@ -253,7 +253,11 @@ sql_translation.BigQueryConnection <- function(x) {
       pmin = sql_prefix("LEAST"),
 
       # Median
-      median = function(x) dbplyr::build_sql("APPROX_QUANTILES(", x, ", 2)[SAFE_ORDINAL(2)]")
+      median = function(x) dbplyr::build_sql("APPROX_QUANTILES(", x, ", 2)[SAFE_ORDINAL(2)]"),
+
+      runif = function(n = n(), min = 0, max = 1) {
+        dbplyr::sql_runif(RAND(), n = {{ n }}, min = min, max = max)
+      },
     ),
     dbplyr::sql_translator(.parent = dbplyr::base_agg,
       n = function() dplyr::sql("count(*)"),

tests/testthat/test-dplyr.R

@@ -130,6 +130,13 @@ test_that("suffixes use _", {
   expect_equal(dbplyr::sql_join_suffix(simulate_bigrquery()), c("_x", "_y"))
 })
 
+test_that("runif is correctly translated", {
+  expect_equal(
+    dbplyr::translate_sql(runif(n()), con = simulate_bigrquery()),
+    dbplyr::sql("RAND()")
+  )
+})
+
 test_that("can correctly print a lazy query", {
   con <- DBI::dbConnect(
     bigquery(),