tests/testthat/test-condense.R

In kleu046/wr.data.table: What the Package Does (One Line, Title Case)

mkdt()
test_that("condense without grouping is working", {
  expect_equal(condense(dt, mu_mpg = mean(mpg)), copy(dt)[,.(mu_mpg = mean(mpg))])
  expect_equal(condense(dt, mu_mpg = mean("mpg")), copy(dt)[,.(mu_mpg = mean(mpg))])
  expect_equal(condense(dt, mu_mpg = mean(mpg), sd_mpg = sd(mpg)), copy(dt)[,.(mu_mpg = mean(mpg), sd_mpg = sd(mpg))])
})

test_that("condense with grouping is working", {
  expect_equal(condense(dt |> set_group(vs), mu_mpg = mean(mpg)), copy(dt)[,.(mu_mpg = mean(mpg)), by=c("vs")])
  expect_equal(condense(dt |> set_group(vs), mu_mpg = mean("mpg")), copy(dt)[,.(mu_mpg = mean(mpg)), by=c("vs")])
  expect_equal(condense(dt |> set_group(vs), mu_mpg = mean("mpg"), sd_mpg = sd(mpg)), copy(dt)[,.(mu_mpg = mean(mpg), sd_mpg = sd(mpg)), by=c("vs")])
})

rm(dt)