tests/testthat/test-if-else2.R

In poissonconsulting/tidyplus: Additional 'tidyverse' Functions

test_that("scalar true and false are vectorised", {
  x <- c(TRUE, TRUE, FALSE, FALSE)
  expect_identical(if_else2(x, 1, 2), c(1, 1, 2, 2))
})

test_that("vector true and false are ok", {
  x <- c(-1, 0, 1)

  expect_identical(if_else2(x < 0, x, 0), c(-1, 0, 0))
  expect_identical(if_else2(x > 0, x, 0), c(0, 0, 1))
})

test_that("missing values are preserved if x", {
  x <- c(TRUE, NA, FALSE)
  expect_identical(if_else2(x, FALSE, x), c(FALSE, NA, FALSE))
})

test_that("missing values are replaced", {
  expect_identical(if_else2(c(TRUE, NA, FALSE), -1, 1), c(-1, 1, 1))
})

test_that("works with lists", {
  x <- list(1, 2, 3)

  expect_equal(
    if_else2(c(TRUE, TRUE, FALSE), x, list(NULL)),
    list(1, 2, NULL)
  )
})

test_that("throws an informative error when error = TRUE and no matches are found", {
  x <- c(1, 2, 3, 4, 5)
  expect_error(
    if_else2(x < 0, 0, x, error = TRUE),
    "No matches found. Did not make any replacements."
  )
})

test_that("doesn't throws an informative error when error = TRUE and matches are found", {
  x <- c(-1, 2, 3, 4, 5)
  expect_equal(
    if_else2(x < 0, 0, x, error = TRUE),
    c(0, 2, 3, 4, 5)
  )
})

test_that("doesn't throw an informative error when error = FALSE (the default) and no matches are found", {
  x <- c(1, 2, 3, 4, 5)
  expect_equal(
    if_else2(x < 0, 0, x),
    x
  )
})