tests/testthat/test_margin.R

In rje42/contingency: Discrete Multivariate Probability Distributions

set.seed(123)
x <- rprobMat(10,c(2,3,4,2))

test_that("conditioning works for object of class 'tables'", {
  expect_equal(c(conditional2(x,3:4,1:2)), c(x/c(margin(x,1:2))))
  expect_equal(c(conditional(x,3:4,1:2,condition.value=list(1:2,1))), 
               c(aperm((x/c(margin(x,1:2)))[,1:2,1,,,drop=FALSE],c(3,4,1,2))))
  #expect_equal(c(intervention(x,3,1:2)), c(t((apply(as.array(x), 1, intervention, 3, 1:2)))))
})

test_that("intervention works for object of class 'tables'", {
  expect_equal(c(intervention(x,3,1:2)), c(c(margin(x,1:2))*conditional2(x,4,1:3)))
  expect_equal(c(intervention(x,3,1:2)), c(t((apply(as.array(x), 1, intervention, 3, 1:2)))))
})