tests/testthat/test-cavi_generic_functions.R

In jpmeagher/vbar: Variational Bayes Ancestral Reconstruction

test_that("Gaussian outer product", {
  L <- 5
  z <- rnorm(L)
  S <- stats::rWishart(1, df = L, Sigma = diag(L)/ L^2)[,,1]
  inv_S <- chol2inv(chol(S))

  expect_equal(
    gaussian_outer_product_expectation(expected_value = z, covariance_matrix = S),
    S + z %*% t(z)
  )

  expect_equal(
    gaussian_outer_product_expectation(expected_value = z, precision_matrix = inv_S),
    S + z %*% t(z)
  )

})