tests/testthat/test-cut-with-nas.R

In OuhscBbmc/OuhscMunge: Data Manipulation Operations

library(testthat)

w       <- c(0L, NA_integer_, 22:25, NA_integer_, 40)
breaks  <- c(0, 25, 50)
labels  <- c("lower", "upper")

test_that("default", {

  expected_1  <- structure(c(1L, 3L, 2L, 2L, 2L, 2L, 3L, 2L), .Label = c("(-0.04,20]", "(20,40]", "Unknown"), class = "factor")
  expected_2  <- structure(c(3L, 3L, 1L, 1L, 1L, 1L, 3L, 2L), .Label = c("lower", "upper", "Unknown"), class = "factor")
  expected_3  <- structure(c(3L, 3L, 1L, 1L, 1L, 1L, 3L, 2L), .Label = c("(0,25]", "(25,50]", "Unknown"), class = "factor")
  expected_4  <- structure(c(1L, 3L, 1L, 1L, 1L, 1L, 3L, 2L), .Label = c("[0,25]", "(25,50]", "Unknown"), class = "factor")
  expected_5  <- structure(c(1L, 3L, 1L, 1L, 1L, 2L, 3L, 2L), .Label = c("[0,25)", "[25,50]", "Unknown"), class = "factor")
  expected_6  <- structure(c(1L, 3L, 1L, 1L, 1L, 2L, 3L, 2L), .Label = c("[0,25)", "[25,50)", "Unknown"), class = "factor")

  actual_1  <- cut_with_nas(w, breaks = 2)
  actual_2  <- cut_with_nas(w, breaks = breaks, labels = labels)
  actual_3  <- cut_with_nas(w, breaks = breaks               )
  actual_4  <- cut_with_nas(w, breaks = breaks               , include.lowest = TRUE)
  actual_5  <- cut_with_nas(w, breaks = breaks               , include.lowest = TRUE, right = FALSE)
  actual_6  <- cut_with_nas(w, breaks = breaks                                      , right = FALSE)

  testthat::expect_equal(actual_1, expected_1)
  testthat::expect_equal(actual_2, expected_2)
  testthat::expect_equal(actual_3, expected_3)
  testthat::expect_equal(actual_4, expected_4)
  testthat::expect_equal(actual_5, expected_5)
  testthat::expect_equal(actual_6, expected_6)
})

test_that("error - bad x", {
  expect_error(
    cut_with_nas(letters, breaks = 2),
    "Assertion on 'x' failed: Must be of type 'numeric', not 'character'\\."
  )
})
test_that("error - bad missing label", {
  expect_error(
    cut_with_nas(w, breaks = 2, .missing = 3L),
    "Assertion on '.missing' failed: Must be of type 'character', not 'integer'\\."
  )
})
test_that("error - missing label too many elements", {
  expect_error(
    cut_with_nas(w, breaks = 2, .missing = letters),
    "Assertion on '\\.missing' failed: Must have length 1, but has length 26\\."
  )
})