tests/testthat/test-matrix.R

In mrc-ide/squire: SEIR transmission model of COVID-19

test_that("matrix manipulation", {
  im <- matrix(rep(1, 16 * 16), ncol = 16)
  m1 <- process_contact_matrix(im, rep(1, 16))
  expect_identical(m1, im)

  for(i in 1:20){
    i_rand <- matrix(sample(1:100, 16 * 16, replace  = TRUE), ncol = 16)
    diag(i_rand) <- 1
    p_rand <- sample(1:100, 16, replace  = TRUE)
    m1 <- process_contact_matrix(i_rand, p_rand)
  }

})