diff --git a/inst/tinytest/test-multiple.R b/inst/tinytest/test-multiple.R index 64f9eb5..af7e2ec 100644 --- a/inst/tinytest/test-multiple.R +++ b/inst/tinytest/test-multiple.R @@ -27,7 +27,7 @@ saver <- make_saver( run_multiple( model_seircon, ndays=days, - nsim=nsims, + nsims=nsims, seed=1972, saver=saver, nthreads = 2L diff --git a/inst/tinytest/test-sis.R b/inst/tinytest/test-sis.R new file mode 100644 index 0000000..1b57afa --- /dev/null +++ b/inst/tinytest/test-sis.R @@ -0,0 +1,34 @@ +# Test just this file: tinytest::run_test_file("inst/tinytest/test-sis.R") + +# Create small world population SIS Model -------------------------------------- +expect_silent(sis_0 <- ModelSIS( + name = "SIS", + prevalence = .1, + transmission_rate = .3, + recovery_rate = .3 +)) + +# Check model initialization +expect_inherits(sis_0, "epiworld_sis") +expect_inherits(sis_0, "epiworld_model") +expect_silent(agents_smallworld( + sis_0, + n = 2000, + k = 5, + d = FALSE, + p = .01 +)) + +# Check model run -------------------------------------------------------------- +expect_silent(verbose_off(sis_0)) +expect_warning(expect_error(plot(sis_0))) # Plot fails before model is run +expect_silent(run_multiple( + sis_0, + ndays=100, + nsims=10, + seed=1231, + reset=TRUE, + verbose = TRUE, + nthreads = 1 +)) +expect_silent(plot(sis_0)) # Plot succeeds after model is run