tests/testthat/test_capply.R

In rje42/contingency: Discrete Multivariate Probability Distributions

set.seed(127)
x <- rprobMat(100, dim=c(2,3,4), alpha=2)
x2 <- capply(x, margin, c(1,3))

test_that("capply works on tables", {
  expect_equal(c(margin(x, c(1,3))), c(x2))
  expect_equal(capply(x, entropy), apply(as.array(x), 1, entropy))
  expect_equal(capply(x, mutualInf, m1=1, m2=3, condition=2), 
               apply(as.array(x), 1, mutualInf, m1=1,m2=3,condition=2))
})

set.seed(128)
p <- repTables(10, rprobdist, dim=3,d=3)

test_that("repTables works", {
  expect_equal(class(p), "tables")
  expect_equal(tdim(p), rep(3L,3))
  expect_equal(ntables(p), 10)
  expect_equal(capply(p, sum), rep(1,10))
})