Add tests for collect()

R/collect.R

@@ -113,7 +113,7 @@ collect.RPolarsLazyFrame <- function(
 ) {
   if (isTRUE(verbose)) {
     rlang::warn(
-      "As of `tidypolars` 0.8.0, `collect()` will automatically convert the Polars DataFrame to a standard R data.frame. Use `compute()` instead (with the same arguments) to have a Polars DataFrame as output.\nUse `verbose = FALSE` to remove this warning."
+      "As of `tidypolars` 0.8.0, `collect()` will automatically convert the Polars DataFrame to a standard R data.frame. \nUse `compute()` instead (with the same arguments) to have a Polars DataFrame as output.\nUse `verbose = FALSE` to remove this warning."
     )
   }
   x |>

tests/tinytest/setup.R

@@ -1,7 +1,7 @@
 lf <- list.files("tests/tinytest", pattern = "^test")
 eager <- lf[grep("lazy", lf, invert = TRUE)]
 
-exceptions <- c("test_benchmark.R", "test_compute.R",
+exceptions <- c("test_benchmark.R", "test_compute.R", "test_collect.R",
                 "test_describe.R", "test_fetch.R",
                 "test_group_split.R",
                 "test_pivot_wider.R", "test_sink_csv.R", "test_sink_parquet.R",

tests/tinytest/test_collect.R

@@ -0,0 +1,40 @@
+source("helpers.R")
+using("tidypolars")
+
+pl_iris <- pl$DataFrame(iris)
+pl_iris_lazy <- pl$LazyFrame(iris)
+
+expect_warning(
+  collect(pl_iris_lazy),
+  "automatically convert the"
+)
+expect_is_tidypolars(collect(pl_iris_lazy, verbose = FALSE))
+
+expect_equal(
+  collect(pl_iris_lazy, verbose = FALSE),
+  pl_iris
+)
+
+expect_equal(
+  pl_iris_lazy |>
+    filter(Species == "setosa") |>
+    collect(verbose = FALSE),
+  pl_iris |>
+    filter(Species == "setosa")
+)
+
+expect_error(collect(pl_iris, verbose = FALSE))
+
+out <- pl_iris_lazy |>
+  group_by(Species, maintain_order = TRUE) |>
+  collect(verbose = FALSE)
+
+expect_equal(
+  attr(out, "pl_grps"),
+  "Species"
+)
+
+expect_equal(
+  attr(out, "maintain_grp_order"),
+  TRUE
+)