tests/testthat/test-utils.R

In XiaoqiLu/ziclust: What the Package Does (One Line, Title Case)

test_that("standard logistic function is correct", {
  expect_equal(logistic(0), 0.5)
  expect_equal(logistic(-Inf), 0)
  expect_equal(logistic(Inf), 1)
})

test_that("rowwise Kronecker function is correect", {
  x <- matrix(
    c(
      1, 1,
      1, 2,
      1, 3
    ),
    3, 2,
    byrow = TRUE
  )
  y <- matrix(
    c(
      1, 0,
      0, 1,
      1, 1
    ),
    3, 2,
    byrow = TRUE
  )

  got <- rowwise_kronecker(x, y)
  want <- matrix(
    c(
      1, 1, 0, 0,
      0, 0, 1, 2,
      1, 3, 1, 3
    ),
    3, 4,
    byrow = TRUE
  )

  expect_equal(got, want)
})

test_that("rng for Dirichlet distribution is correct (by checking moments)", {
  alphas <- c(2, 4, 6)

  alpha0 <- sum(alphas)
  z_expectation <- alphas / alpha0
  z_covariance <- (alpha0 * diag(alphas) - alphas %o% alphas) / (alpha0 ^ 2 * (alpha0 + 1))

  n <- 10000
  Z <- rdirichlet(n, alphas)
  z_mean <- colMeans(Z)
  z_sample_covariance <- var(Z)

  expect_equal(colMeans(Z), z_expectation, tolerance = 10 / sqrt(n))
  expect_equal(var(Z), z_covariance, tolerance = 10 / sqrt(n))
})

test_that("log-sum-exp works as expected", {
  expect_equal(log_sum_exp(c(1, 2, 3)), log(sum(exp(c(1, 2, 3)))))
  expect_equal(log_sum_exp(c(1, 2, 3000)), 3000)
  expect_equal(log_sum_exp(c(-3000, -4000, -5000)), -3000)
})

test_that("soft_max works as expected", {
  expect_equal(soft_max(c(1, 2, 3)), exp(c(1, 2, 3)) / sum(exp(c(1, 2, 3))))
  expect_equal(soft_max(c(1, 2, 3000)),c(0, 0, 1))
  expect_equal(soft_max(c(-3000, -4000, -5000)), c(1, 0, 0))
})