tests/testthat/test-multiClassCM.R

In TalhoukLab/biostatUtil: Utility Functions Used in Biostatistics Projects

set.seed(23)
k <- 3
(x <- factor(sample(1:k, 100, replace = TRUE, prob = c(0.15, 0.25, 0.6))))
(y <- factor(sample(1:k, 100, replace = TRUE, prob = c(0.05, 0.4, 0.65))))
prop.table(table(y))

test_that("Results based on same confusion matrix as Caret", {
  (p <- factor(sample(1:2, 100, replace = TRUE, prob = c(0.15, 0.25))))
  (q <- factor(sample(1:2, 100, replace = TRUE, prob = c(0.05, 0.4))))
  a <- multiClassCM(p, q)
  b <- caret::confusionMatrix(p, q)
  expect_identical(unname(a$CM), unname(addmargins(b$table)))
})

test_that("Error generated if uniques of x and y are not equivalent", {
  z <- sample(letters[1:k], 100, replace = TRUE, prob = c(0.05, 0.4, 0.65))
  expect_error(multiClassCM(y, z))
})