tests/testthat/test_tables.R

In rje42/contingency: Discrete Multivariate Probability Distributions

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

test_that("rprobMat creates tables object of distributions correctly", {
  expect_equal(dim(x), c(10, 24))
  expect_equal(tdim(x), c(2,3,4))
  expect_equal(apply(x,1,sum), rep(1,10))
})

test_that("[ operator for tables works correctly", {
  expect_equal(dim(x[1,]), c(2,3,4))
  expect_equal(dim(x[1,,keep=TRUE]), c(1,24))
  expect_equal(dim(x[,1,3,2:3]), c(10,2))  
  expect_equal(tdim(x[,1,3,2:3]), c(2))  
  expect_equal(dim(x[,1,3,2:3,drop=FALSE]), c(10,2))  
  expect_equal(tdim(x[,1,3,2:3,drop=FALSE]), c(1,1,2))  
})

x2 <- tbind(x,x)
x02 <- tbind(x[,,1,1],x[,,1,1])

test_that("tbind works correctly", {
  expect_equal(dim(x2), c(20,24))
  expect_equal(x2[1,], x[1,])
  expect_equal(x2[11,], x[1,])
  expect_equal(dim(x02), c(20,2))
  expect_error(tbind(x2,x02))
})

## now try with a reversed table
x <- rprobMat(10,c(2,3,4), rev=TRUE)

test_that("rprobMat creates tables object of distributions correctly", {
  expect_equal(dim(x), c(24, 10))
  expect_equal(tdim(x), c(2,3,4))
  expect_equal(apply(x,2,sum), rep(1,10))
})

test_that("[ operator for tables works correctly", {
  expect_equal(dim(x[1,]), c(2,3,4))
  expect_equal(dim(x[1,,keep=TRUE]), c(24,1))
  expect_equal(dim(x[,1,3,2:3]), c(2,10))  
  expect_equal(tdim(x[,1,3,2:3]), c(2))  
  expect_equal(dim(x[,1,3,2:3,drop=FALSE]), c(2,10))  
  expect_equal(tdim(x[,1,3,2:3,drop=FALSE]), c(1,1,2))  
})

x2 <- tbind(x,x)
x02 <- tbind(x[,,1,1],x[,,1,1])

test_that("tbind works correctly", {
  expect_equal(dim(x2), c(20,24))
  expect_equal(x2[1,], x[1,])
  expect_equal(x2[11,], x[1,])
  expect_equal(dim(x02), c(20,2))
  expect_error(tbind(x2,x02))
})