tests/testthat/test-solve_ode.R

In bristolmathmodellers/biddmodellingcourse: Practicals and Code for the BIDD Modelling Short Course

context("solve_ode.R")

## Intialise
N = 1000
I_0 = 1
S_0 = N - I_0
R_0 = 3
beta = R_0

##Time for model to run over
tbegin = 0
tend = 50
times <- seq(tbegin, tend, 1)

##Vectorise input
parameters <- as.matrix(c(beta = beta))
inits <- as.matrix(c(S = S_0, I = I_0))

test <- solve_ode(model = SI_ode, inits, parameters, times, as.data.frame = TRUE)
test <- head(test)
test <- round(test, digits = 0)


result <- tibble::tibble(time = 0:5, 
                         S = c(999, 980, 712, 110, 6, 0),
                         I = c(1, 20, 288, 890, 994, 1000)
                         )
result <- round(result, digits = 0)


test_that("solve_ode can run a simple model simulation", {
  expect_equal(result, test)
})