added tests for controls in probability_epidemic

tests/testthat/test-probability_epidemic.R

@@ -27,7 +27,7 @@ test_that("probability_epidemic works for k == Inf", {
   )
 })
 
-test_that("probability_epidemic works for different k & a", {
+test_that("probability_epidemic works for different k & num_init_infect", {
   res1 <- probability_epidemic(R = 1.5, k = 0.5, num_init_infect = 1)
   res2 <- probability_epidemic(R = 1.5, k = 0.2, num_init_infect = 1)
   res3 <- probability_epidemic(R = 1.5, k = 0.5, num_init_infect = 2)
@@ -36,6 +36,37 @@ test_that("probability_epidemic works for different k & a", {
   expect_true((res4 < res3))
 })
 
+test_that("probability_epidemic works with individual-level control", {
+  expect_equal(
+    probability_epidemic(
+      R = 1.5, k = 0.5, num_init_infect = 1, ind_control = 0.2
+    ),
+    0.091
+  )
+})
+
+test_that("probability_epidemic works with population-level control", {
+  expect_equal(
+    probability_epidemic(
+      R = 1.5, k = 0.5, num_init_infect = 1, pop_control = 0.2
+    ),
+    0.113
+  )
+})
+
+test_that("probability_epidemic works with both control measures", {
+  expect_equal(
+    probability_epidemic(
+      R = 1.5,
+      k = 0.5,
+      num_init_infect = 1,
+      ind_control = 0.1,
+      pop_control = 0.1
+    ),
+    0.108
+  )
+})
+
 test_that("probability_epidemic fails correctly", {
   expect_error(
     probability_epidemic(R = "1", k = 1, num_init_infect = 1),