tests/testthat/test-contact_matrices.R

In mrc-ide/hypatia: Individual-Based Model of COVID Transmission

test_that('get_contact_matrices returns the first matrix', {
  data <- c(rep(0, 9), rep(1, 9)) # Matrix 1 and Matrix 2
  m <- aperm(array(data, dim=c(3, 3, 2)), c(3, 1, 2)) # Organise it how R does
  expect_equal(
    get_contact_matrix(m),
    array(rep(0, 9), dim=c(3, 3))
  )
})