funs: add support for replace_na() when input is a vector

NEWS.md

@@ -20,6 +20,9 @@
   - from package `lubridate`: `make_datetime()`.
 
   - from package `stringr`: `str_dup()`, `str_split()`, `str_split_i()`.
+
+  - from package `tidyr`: `replace_na()` (the data.frame method was already
+    translated but not the vector one that can be used in `mutate()` for example).
 
 ## Bug fixes
 

R/funs-default.R

@@ -483,6 +483,11 @@ pl_rank <- function(x, ...) {
   x$rank()
 }
 
+pl_replace_na <- function(data, replace, ...) {
+  check_empty_dots(...)
+  data$fill_null(replace)
+}
+
 pl_round <- function(x, digits = 0, ...) {
   check_empty_dots(...)
   x$round(decimals = digits)

tests/tinytest/test_replace_na.R

@@ -1,18 +1,22 @@
 source("helpers.R")
 using("tidypolars")
 
-pl_test <- pl$DataFrame(x = c(NA, 1), y = c(2, NA))
+pl_test <- polars::pl$DataFrame(x = c(NA, 1), y = c(2, NA))
 
 expect_is_tidypolars(replace_na(pl_test, 0))
 
 expect_equal(
-  replace_na(pl_test, 0) |>
-    as.data.frame(),
+  replace_na(pl_test, 0),
   data.frame(x = c(0, 1), y = c(2, 0))
 )
 
 expect_equal(
-  replace_na(pl_test, list(x = 0, y = 999)) |>
-    as.data.frame(),
+  replace_na(pl_test, list(x = 0, y = 999)),
   data.frame(x = c(0, 1), y = c(2, 999))
 )
+
+expect_equal(
+  pl_test |>
+    mutate(x = replace_na(x, 0)) ,
+  data.frame(x = c(0, 1), y = c(2, NA))
+)

tests/tinytest/test_replace_na_lazy.R

@@ -5,20 +5,24 @@ Sys.setenv('TIDYPOLARS_TEST' = TRUE)
 source("helpers.R")
 using("tidypolars")
 
-pl_test <- pl$LazyFrame(x = c(NA, 1), y = c(2, NA))
+pl_test <- polars::pl$LazyFrame(x = c(NA, 1), y = c(2, NA))
 
 expect_is_tidypolars(replace_na(pl_test, 0))
 
 expect_equal_lazy(
-  replace_na(pl_test, 0) |>
-    as.data.frame(),
+  replace_na(pl_test, 0),
   data.frame(x = c(0, 1), y = c(2, 0))
 )
 
 expect_equal_lazy(
-  replace_na(pl_test, list(x = 0, y = 999)) |>
-    as.data.frame(),
+  replace_na(pl_test, list(x = 0, y = 999)),
   data.frame(x = c(0, 1), y = c(2, 999))
 )
 
+expect_equal_lazy(
+  pl_test |>
+    mutate(x = replace_na(x, 0)) ,
+  data.frame(x = c(0, 1), y = c(2, NA))
+)
+
 Sys.setenv('TIDYPOLARS_TEST' = FALSE)

vignettes/r-and-polars-expressions.Rmd

@@ -287,6 +287,7 @@ out <- tribble(
   "`stringr`",          "`str_to_title`",
   "`stringr`",          "`str_to_upper`",
   "`stringr`",          "`word`",
+  "`tidyr`",            "`replace_na`",
   "`tools`",            "`toTitleCase`"
 ) |>
   mutate(Notes = case_when(