tests/testthat/test-estimates.R

In mcmcr: Manipulate MCMC Samples

test_that("estimates.mcarray", {
  expect_equal(estimates(as.mcarray(mcmcr_example[[1]])), c(3.718025, 4.718025))

  expect_snapshot_data(estimates(as.mcarray(mcmcr_example[[1]]), as_df = TRUE),
                       "estimates")
})

test_that("estimates.mcmcarray", {
  expect_equal(estimates(mcmcr_example[[1]]), c(3.718025, 4.718025))
  expect_equal(estimates(mcmcr_example[[3]]), c(0.7911975))
  expect_identical(estimates(mcmcr_example)$alpha, estimates(mcmcr_example[[1]]))
})

test_that("estimates.mcmc", {
  expect_equal(estimates(as.mcmc(collapse_chains(mcmcr_example)))$alpha, c(3.718025, 4.718025))
  expect_equal(
    estimates(as.mcmc(collapse_chains(mcmcr_example)), as_df = TRUE)$estimate,
    c(
      3.718025, 4.718025, 0.9716535, 1.9716535, 1.9716535, 2.9716535,
      0.7911975
    )
  )
})

test_that("estimates.mcmc.list", {
  expect_equal(estimates(coda::as.mcmc.list(mcmcr_example))$alpha, c(3.718025, 4.718025))
  expect_equal(
    estimates(coda::as.mcmc.list(mcmcr_example), as_df = TRUE)$estimate,
    c(
      3.718025, 4.718025, 0.9716535, 1.9716535, 1.9716535, 2.9716535,
      0.7911975
    )
  )
})

test_that("estimates not as list", {
  expect_equal(estimates(mcmcr_example[[1]], as_df = TRUE), tibble(term = as_term(c("parameter[1]", "parameter[2]")), estimate = estimates(mcmcr_example[[1]])))
  expect_equal(estimates(mcmcr_example[[2]], as_df = TRUE), tibble(term = as_term(c("parameter[1,1]", "parameter[2,1]", "parameter[1,2]", "parameter[2,2]")), estimate = as.vector(estimates(mcmcr_example[[2]]))))
  expect_equal(estimates(mcmcr_example[[3]], as_df = TRUE), tibble(term = as_term("parameter"), estimate = estimates(mcmcr_example[[3]])))
  expect_identical(sort(estimates(mcmcr_example, as_df = TRUE)$term), sort(as_term(c("alpha[1]", "alpha[2]", "beta[1,1]", "beta[2,1]", "beta[1,2]", "beta[2,2]", "sigma"))))
})

test_that("estimates not as list", {
  expect_equal(estimates(mcmcr_example[[3]], fun = sum), 774.4415, tolerance = 0.0000001)
  expect_equal(estimates(mcmcr_example[[3]], fun = sum, as_df = TRUE)$estimate, 774.4415, tolerance = 0.0000001)
  expect_identical(estimates(mcmcr_example[[3]], fun = length), 800L)
  expect_identical(estimates(mcmcr_example[[3]], fun = length, as_df = TRUE)$estimate, 800L)
})

test_that("estimates not scalar", {
  expect_error(estimates(mcmcr::mcmcr_example, fun = function(x) NULL),
    "^`fun` must return a scalar[.]$",
    class = "chk_error"
  )
  expect_error(estimates(mcmcr::mcmcr_example, fun = identity),
    "^`fun` must return a scalar[.]$",
    class = "chk_error"
  )
})

test_that("estimates not numeric", {
  expect_error(estimates(mcmcr::mcmcr_example, fun = function(x) as.character(sum(x))),
    "^`fun` must return a numeric[.]$",
    class = "chk_error"
  )
})

test_that("estimates ...", {

  mean1 <- function(x, plus_one = FALSE) {
    x <- mean(x)
    if(isFALSE(plus_one)) return(x)
    x + 1
  }

  expect_equal(estimates(mcmcr::mcmcr_example, fun = mean1, plus_one = TRUE)$alpha,
               estimates(mcmcr::mcmcr_example, fun = mean1)$alpha + 1)
})