tests/testthat/test-simulation-e2e.R

In mrc-ide/malariasimulation: An individual based model for malaria

test_that('Simulation runs for a few timesteps', {
  sim <- run_simulation(100)
  expect_equal(nrow(sim), 100)
})

test_that('run_metapop_simulation fails with incorrect mixing matrix', {
  population <- 4
  timestep <- 5
  parameters <- get_parameters(list(human_population = population))
  paramlist <- list(parameters, parameters)
  # incorrect params
  mixing <- matrix(c(1, 1), nrow = 1, ncol = 2)
  p_captured <- diag(nrow=2)
  expect_error(
    run_metapop_simulation(
      timesteps,
      parameters,
      NULL,
      mixing_tt = 1,
      export_mixing = list(mixing),
      import_mixing = list(mixing),
      p_captured_tt = 1,
      p_captured = list(diag(nrow=2)),
      p_success = 1
    )
  )
})

test_that('run_metapop_simulation integrates two models correctly', {
  population <- 4
  timesteps <- 5
  parameters <- get_parameters(list(human_population = population))
  parametersets <- list(parameters, parameters)
  mixing <- diag(nrow = 2)
  p_captured <- 1 - diag(nrow = 2)

  outputs <- run_metapop_simulation(
    timesteps,
    parametersets,
    NULL,
    mixing_tt = 1,
    export_mixing = list(mixing),
    import_mixing = list(mixing),
    p_captured_tt = 1,
    p_captured = list(p_captured),
    p_success = 1
  )
  expect_equal(length(outputs), 2)
  expect_equal(nrow(outputs[[1]]), 5)
  expect_equal(nrow(outputs[[2]]), 5)
})