tests/testthat/test_sampling.R

In mashr: Multivariate Adaptive Shrinkage

test_that("Samples from the posterior look right", {
  set.seed(100)
  test = simple_sims()
  data = mash_set_data(test$Bhat, test$Shat)
  U = cov_canonical(data)
  # res = mash(data,U, algorithm.version = 'R', posterior_samples = 100,
  #             usepointmass = FALSE)
  res = mash(data,U, algorithm.version = "R", posterior_samples = 100,
             verbose = FALSE)$result
  expect_equal(dim(res$PosteriorSamples), c(400,5,100))
  expect_equal(t(apply(res$PosteriorSamples, 1, rowMeans)),
               res$PosteriorMean,tolerance = max(res$PosteriorSD))
})

test_that("Samples from the posterior with linear transformation look right", {
  set.seed(100)
  test = simple_sims()
  data = mash_set_data(test$Bhat, test$Shat)
  U = cov_canonical(data)
  res = mash(data,U, algorithm.version = "R", posterior_samples = 100,
             verbose = FALSE, A = matrix(c(1,1,0,0,0), 1, 5))$result
  expect_equal(dim(res$PosteriorSamples), c(400,1,100))
  expect_equal(as.matrix(apply(res$PosteriorSamples, 1, rowMeans)),
               res$PosteriorMean,tolerance = max(res$PosteriorSD))
})