tests/testthat/test-merge_chains.R

In stan-dev/posterior: Tools for Working with Posterior Distributions

test_that("merge_chains works correctly", {
   x <- example_draws()

   xm <- merge_chains(as_draws_matrix(x))
   expect_equal(nchains(xm), 1)
   expect_equal(ndraws(xm), ndraws(x))
   expect_equal(variables(xm), variables(x))

   xm <- merge_chains(as_draws_array(x))
   expect_equal(nchains(xm), 1)
   expect_equal(ndraws(xm), ndraws(x))
   expect_equal(variables(xm), variables(x))

   xm <- merge_chains(as_draws_df(x))
   expect_equal(nchains(xm), 1)
   expect_equal(ndraws(xm), ndraws(x))
   expect_equal(variables(xm), variables(x))

   xm <- merge_chains(as_draws_list(x))
   expect_equal(nchains(xm), 1)
   expect_equal(ndraws(xm), ndraws(x))
   expect_equal(variables(xm), variables(x))

   # need to keep x_rvar around for this test because draws_rvars counts
   # variables differently from the other formats
   x_rvar <- as_draws_rvars(x)
   xm <- merge_chains(x_rvar)
   expect_equal(nchains(xm), 1)
   expect_equal(ndraws(xm), ndraws(x))
   expect_equal(variables(xm), variables(x_rvar))
})