tests/testthat/test-split-chains.R

In universals: S3 Generics for Bayesian Analyses

test_that("split_chains nlist::nlists", {
  nlists <- nlist::nlists(nlist::nlist(x = matrix(1:6, 2)), nlist::nlist(x = matrix(3:8, 2)))
  nlists2 <- nlists
  attr(nlists2, "nchains") <- 2L

  expect_identical(split_chains(nlists), nlists2)
})

test_that("split_chains nlist::nlists error", {
  nlists <- nlist::nlists(nlist::nlist(x = matrix(1:6, 2)))
  expect_error(split_chains(nlists), "^`x` must have at least two iterations[.]$")
})

test_that("split_chains nlist::nlists extra iters", {
  nlists <- nlist::nlists(
    nlist::nlist(x = matrix(1:6, 2)), nlist::nlist(x = matrix(3:8, 2)),
    nlist::nlist(x = matrix(2:7, 2))
  )

  nlists2 <- nlist::nlists(
    nlist::nlist(x = matrix(1:6, 2)), nlist::nlist(x = matrix(3:8, 2))
  )
  attr(nlists2, "nchains") <- 2L

  expect_identical(
    split_chains(nlists),
    nlists2)
})