R version 4.4.0 RC (2024-04-16 r86468 ucrt) -- "Puppy Cup"
Copyright (C) 2024 The R Foundation for Statistical Computing
Platform: x86_64-w64-mingw32/x64

R is free software and comes with ABSOLUTELY NO WARRANTY.
You are welcome to redistribute it under certain conditions.
Type 'license()' or 'licence()' for distribution details.

R is a collaborative project with many contributors.
Type 'contributors()' for more information and
'citation()' on how to cite R or R packages in publications.

Type 'demo()' for some demos, 'help()' for on-line help, or
'help.start()' for an HTML browser interface to help.
Type 'q()' to quit R.

> library(epigrowthfit)
> library(tools)
> options(warn = 2L, error = if (interactive()) recover)
> 
> 
> ## finalsize ###########################################################
> 
> n <- 10L
> R0 <- rlnorm(n, 0, 2)
> S0 <- runif(n, 0, 1)
> I0 <- runif(n, 0, 1 - S0)
> Z <- S0 + emdbook::lambertW(-R0 * S0 * exp(-R0 * (S0 + I0))) / R0
> 
> stopifnot(exprs = {
+ 	all.equal(finalsize(R0 = R0, S0 = S0, I0 = I0), Z)
+ 	all.equal(finalsize(R0 = 0, S0 = S0, I0 = I0), double(n))
+ 	all.equal(finalsize(R0 = Inf, S0 = S0, I0 = I0), S0)
+ })
> assertWarning(finalsize(R0 = -1, S0 = 1, I0 = 0))
> assertWarning(finalsize(R0 = 1, S0 = 1, I0 = 0.1))
> 
> 
> ## R0 ##################################################################
> 
> r <- c(rlnorm(10L, -3, 1), 0, NaN, Inf)
> breaks <- seq(0, 20, 1)
> probs <- diff(pgamma(breaks, shape = 1, scale = 2.5))
> probs <- probs / sum(probs)
> 
> n <- length(breaks)
> f <- function(r) r / sum(probs * (exp(-r * breaks[-n]) - exp(-r * breaks[-1L])) / (breaks[-1L] - breaks[-n]))
> 
> R0.1 <- function(x) R0(r = x, breaks = breaks, probs = probs)
> R0.3 <- function(x) R0(r = r, breaks = breaks, probs = x)
> 
> stopifnot(exprs = {
+ 	all.equal(R0.1(r), c(vapply(r[1:10], f, 0), 1, NaN, Inf))
+ 	all.equal(R0.3(probs), R0.3(100 * probs))
+ })
> assertWarning(R0.1(-1))
> 
> 
> ## timescale ###########################################################
> 
> r <- c(rlnorm(10L, -3, 1), 0, NaN, Inf)
> units <- "days"
> stopifnot(exprs = {
+ 	all.equal(timescale(r), 1 / r)
+ 	all.equal(timescale(r, units), .difftime(1 / r, units))
+ })
> assertWarning(timescale(-1))
> 
> proc.time()
   user  system elapsed 
   1.48    0.17    1.64