tests/testthat/test-group-trim.R

In duckdblabs/duckplyr: A 'DuckDB'-Backed Version of 'dplyr'

test_that("duckplyr_group_trim() is identity on non grouped data", {
  expect_identical(duckplyr_group_trim(iris), iris)
})

test_that("duckplyr_group_trim() always regroups even if no factors", {
  res <- mtcars %>%
    duckplyr_group_by(cyl) %>%
    duckplyr_filter(cyl == 6, .preserve = TRUE) %>%
    duckplyr_group_trim()
  expect_equal(duckplyr_n_groups(res), 1L)
})

test_that("duckplyr_group_trim() drops factor levels in data and grouping structure", {
  res <- iris %>%
    duckplyr_group_by(Species) %>%
    duckplyr_filter(Species == "setosa") %>%
    duckplyr_group_trim()

  expect_equal(duckplyr_n_groups(res), 1L)
  expect_equal(levels(res$Species), "setosa")
  expect_equal(levels(attr(res, "groups")$Species), "setosa")
})