tests/testthat/test_Summarize.R

In KO112/KO: Kevin O'Brien's Personal Package

context("Summarize")


# Test count_distinct
test_that("count_distinct", {
  expect_equal(count_distinct(mtcars),
               setNames(c(25, 3, 27, 22, 22, 29, 30, 2, 2, 3, 6), names(mtcars)))
  expect_equal(count_distinct(mtcars),
               count_distinct(mtcars, use_base = FALSE))
  expect_equal(count_distinct(as.matrix(mtcars)),
               setNames(c(25, 3, 27, 22, 22, 29, 30, 2, 2, 3, 6), names(mtcars)))
  expect_equal(count_distinct(as.matrix(mtcars)),
               count_distinct(as.matrix(mtcars), use_base = FALSE))
  expect_equal(count_distinct(mtcars$mpg),
               count_distinct(mtcars$mpg, use_base = FALSE))
  expect_equal(count_distinct(mtcars$mpg),
               25)
  expect_equal(count_distinct(c(1, 1, 1, NULL, NA)),
               2)
  expect_equal(count_distinct(c(1, 1, 1, NULL, NA)),
               count_distinct(c(1, 1, 1, NULL, NA), use_base = FALSE))
  expect_equal(count_distinct(c("a", "b", "c", "a", "A", NA)),
               5)
  expect_equal(count_distinct(c("a", "b", "c", "a", "A", NA)),
               count_distinct(c("a", "b", "c", "a", "A", NA), use_base = FALSE))
})


# Test table_df
test_that("table_df", {
  expect_identical(table_df(mtcars),
                   sapply(mtcars, table, useNA = "ifany", dnn = NULL))
  expect_identical(table_df(mtcars$am),
                   table(mtcars$am, dnn = NULL))

})


# Test mode_stat
test_that("mode_stat", {
  expect_equal(mode_stat(c(1, 1, 1, 2, 2)),
               1)
  expect_equal(mode_stat(c("a", "b", "b", "b", "c")),
               "b")
  expect_equal(mode_stat(c("a", "b")),
               "b")
  expect_equal(mode_stat(c("a", "b"), return_all_modes = TRUE),
               c("a", "b"))
})


# Test summarize_vec
# test_that("summarize_vec", {
#   expect_equal(summarize_vec(),
#                )
# })


# Test summarize_df
# test_that("summarize_df", {
#   expect_equal(summarize_df(),
#                )
# })