From cd3f06f7c32448b601308236f1a70856bae198cf Mon Sep 17 00:00:00 2001
From: Andrew Pulsipher <a.pulsipher@utah.edu>
Date: Thu, 14 Nov 2024 08:07:49 -0700
Subject: [PATCH] Create test-sis.R and populate with tests

---
 inst/tinytest/test-multiple.R |  2 +-
 inst/tinytest/test-sis.R      | 34 ++++++++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+), 1 deletion(-)
 create mode 100644 inst/tinytest/test-sis.R

diff --git a/inst/tinytest/test-multiple.R b/inst/tinytest/test-multiple.R
index 64f9eb56..af7e2ecf 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 00000000..1b57afa6
--- /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