tests

Author: B0ydT

tests/testthat/test-dplyr_methods.R

@@ -377,10 +377,32 @@ test_that("rowwise()", {
     expect_identical(fd[[1]], sapply(df$lys, sum))
 })
 
-test_that("group_split()", {
+test_that("group_split() works for one variable", {
+  fd <- df |> 
+    group_split(groups)
+  expect_equal(length(fd), length(unique(df$groups)))
+})
+
+test_that("group_split() works for combination of variables", {
     fd <- df |> 
       group_split(groups, ident)
     expect_equal(length(fd), length(unique(df$groups)) *
                    length(unique(df$ident)))
 })
 
+test_that("group_split() works for one logical statement", {
+  fd_log <- df |> 
+    group_split(groups=="g1")
+  fd_var <- df |> 
+    group_split(groups=="g1")
+  expect_equal(lapply(fd_var, count), lapply(fd_log, count))
+})
+
+test_that("group_split() works for two logical statements", {
+  fd <- df |>
+    group_split(PC_1>0 & groups=="g1")
+  fd_counts <- lapply(fd, count)
+  expect_equal(c(fd_counts[[1]], fd_counts[[2]], use.names = FALSE), 
+               list(75, 5))
+})
+