tests/testthat/test_mask_counts.R

In countmaskr: Small Cell Masking Tool for One- & Two-Way Tabular Reports

library(testthat)

test_that("Primary masking works correctly with different inputs", {
  x <- c(5, 15, 0, NA)
  expect_equal(mask_counts(x, threshold = 11), c("<11", "<20", "0", NA_character_))
})

test_that("Secondary masking applies correctly with different conditions", {
  expect_equal(sum(grepl("<", mask_counts(c(5, 15, 20), threshold = 11))), 2)
  expect_equal(sum(grepl("<", mask_counts(c(1, 1, 15, 20), threshold = 11))), 3)
  expect_equal(sum(grepl("<", mask_counts(c(10, 10, 15, 20), threshold = 11))), 3)
})

test_that("Zero masking behaves as expected when zero_masking is TRUE", {
  x <- c(5, 15, 0)
  result <- mask_counts(x, zero_masking = TRUE, threshold = 11)
  expect_equal(result[c(1,3)], c("<11", "<11"))
})

test_that("Secondary cell selection methods work as expected", {
  x <- c(5, 14, 20)
  expect_equal(mask_counts(x, secondary_cell = "min"), c("<11", "<15", "20"))
  expect_equal(mask_counts(x, secondary_cell = "max"), c("<11", "14", "<25"))
  set.seed(123)
  expect_equal(sum(grepl("<", mask_counts(x, secondary_cell = "random"))), 2)
})

test_that("Function handles special inputs correctly", {
  expect_equal(mask_counts(c(5, NA, 15)), c("<11", NA_character_, "<20"))
  expect_true(is.character(mask_counts(c(5, 15, 20))))
})

test_that("Threshold parameter behaves correctly", {
  x <- c(4, 7, 9, 15)
  expect_equal(mask_counts(x, threshold = 5), c("<5", "<10", "9", "15"))
})

# Character input handling
test_that("Function handles various types of character inputs", {
  expect_equal(mask_counts(c("5", "15", "20"), threshold = 11), c("<11", "<20", "20"))
  expect_error(mask_counts(c("5 apples", "15 oranges", "20 bananas")), "disallowed characters")
  expect_error(mask_counts(c("five", "fifteen", "twenty")), "disallowed characters")
  expect_error(mask_counts(c("$5", "€15", "£20")), "disallowed characters")
})

test_that("Function handles negative numbers correctly", {
  expect_equal(mask_counts(c("-5", "4", "15", "20"), threshold = 11), c("-5", "<11", "<20", "20"))
})

# Additional edge cases
test_that("mask_counts handles different cases", {
  expect_equal(mask_counts(c(5, 11, 43, 55, 65, 121, 1213, 0, NA), zero_masking = FALSE),
               c("<11", "<15", "43", "55", "65", "121", "1,213", "0", NA))
  expect_equal(mask_counts(c(5, 11, 43, 55, 65, 121, 1213, 0, NA), zero_masking = TRUE),
               c("<11", "11", "43", "55", "65", "121", "1,213", "<11", NA))
  expect_equal(mask_counts(c(1, 1, 1, 5, 6, 10, 0, NA)), c("<11", "<11", "<11", "<11", "<11", "<11", "0", NA))
  expect_equal(mask_counts(c(5, "11", 43, "55")), c("<11", "<15", "43", "55"))
  expect_equal(mask_counts(c(5, 11, 43, 55), threshold = 20), c("<20", "<20", "43", "55"))
  expect_equal(mask_counts(c(5, 1000000, 2000000)), c("<11", "<1,000,005", "2,000,000"))
})

test_that("mask_counts handles pre-existing masked values", {
  x <- c("<11", "43", "55", "0")
  expect_equal(mask_counts(x, zero_masking = FALSE), c("<11", "<45", "55", "0"))
  expect_equal(mask_counts(x, zero_masking = TRUE), c("<11", "43", "55", "<11"))
})


test_that("mask_counts handles basic cases correctly", {
  x1 <- c(5, 11, 43, 55, 65, 121, 1213, 0, NA)
  expected1a <- c("<11", "<15", "43", "55", "65", "121", "1,213", "0", NA)
  expected1b <- c("<11", "11", "43", "55", "65", "121", "1,213", "<11", NA)

  expect_equal(mask_counts(x1,zero_masking = FALSE), expected1a)
  expect_equal(mask_counts(x1,zero_masking = TRUE), expected1b)
})

test_that("mask_counts handles all values below threshold", {
  x2 <- c(1, 1, 1, 5, 6, 10, 0, NA)
  expected2 <- c("<11", "<11", "<11", "<11", "<11", "<11", "0", NA)

  expect_equal(mask_counts(x2), expected2)
})

test_that("mask_counts handles secondary masking with one primary cell", {
  x3 <- c(5, 12, 43, 55,0)
  expected3a <- c("<11", "<15", "43", "55","0")
  expected3b <- c("<11", "12", "43", "55","<11")

  expect_equal(mask_counts(x3,zero_masking = FALSE), expected3a)
  expect_equal(mask_counts(x3,zero_masking = TRUE), expected3b)
})

test_that("mask_counts handles secondary masking with two primary cells equal to 1", {
  x4 <- c(1, 1, 43, 55)
  expected4 <- c("<11", "<11", "<45", "55")

  expect_equal(mask_counts(x4,zero_masking = FALSE), expected4)
  expect_equal(mask_counts(x4,zero_masking = TRUE), expected4)
})

test_that("mask_counts handles secondary masking with two primary cells equal to 10", {
  x5 <- c(10, 10, 43, 55,0)

  expected5a <- c("<11", "<11", "<45", "55","0")
  expected5b <- c("<11", "<11", "43", "55","<11")

  expect_equal(mask_counts(x5,zero_masking = FALSE), expected5a)
  expect_equal(mask_counts(x5,zero_masking = TRUE), expected5b)
})

test_that("mask_counts handles character input", {
  x6 <- c("5", "11", "43", "55")
  expected6 <- c("<11", "<15", "43", "55")

  expect_equal(mask_counts(x6), expected6)
})

test_that("mask_counts handles mixed numeric and character input", {
  x7 <- c(5, "11", 43, "55")
  expected7 <- c("<11", "<15", "43", "55")

  expect_equal(mask_counts(x7), expected7)
})

test_that("mask_counts handles custom threshold", {
  x8 <- c(5, 11, 43, 55)
  expected8 <- c("<20", "<20", "43", "55")

  expect_equal(mask_counts(x8, threshold = 20), expected8)
})

test_that("mask_counts handles large numbers correctly", {
  x9 <- c(5, 1000000, 2000000)
  expected9 <- c("<11", "<1,000,005", "2,000,000")

  expect_equal(mask_counts(x9), expected9)
})

test_that("mask_counts handles input with existing masked values", {
  x10 <- c("<11", "43", "55","0")
  expected10a <- c("<11", "<45", "55","0")
  expected10b <- c("<11", "43", "55","<11")

  expect_equal(mask_counts(x10,zero_masking = FALSE), expected10a)
  expect_equal(mask_counts(x10,zero_masking = TRUE), expected10b)
})