updated tests to use offspring_dist argument

tests/testthat/test-probability_contain.R

@@ -66,7 +66,10 @@ test_that("probability_contain works with <epidist>", {
   )
   expect_equal(
     probability_contain(
-      num_init_infect = 1, control = 0.1, stochastic = FALSE, epidist = edist
+      num_init_infect = 1,
+      control = 0.1,
+      stochastic = FALSE,
+      offspring_dist = edist
     ),
     0.904
   )

tests/testthat/test-probability_epidemic.R

@@ -82,7 +82,10 @@ test_that("probability_epidemic works with <epidist>", {
       single_epidist = TRUE
     )
   )
-  expect_equal(probability_epidemic(num_init_infect = 1, epidist = edist), 0.12)
+  expect_equal(
+    probability_epidemic(num_init_infect = 1, offspring_dist = edist),
+    0.12
+  )
 })
 
 test_that("probability_epidemic fails without R and k or <epidist>", {

tests/testthat/test-proportion_cluster_size.R

@@ -80,7 +80,7 @@ test_that("proportion_cluster_size works with <epidist>", {
       single_epidist = TRUE
     )
   )
-  res <- proportion_cluster_size(cluster_size = 20, epidist = edist)
+  res <- proportion_cluster_size(cluster_size = 20, offspring_dist = edist)
 
   expect_s3_class(res, "data.frame")
   expect_identical(dim(res), c(1L, 3L))

tests/testthat/test-proportion_transmission.R

@@ -137,7 +137,10 @@ test_that("proportion_transmission works with <epidist>", {
       single_epidist = TRUE
     )
   )
-  res <- proportion_transmission(percent_transmission = 0.8, epidist = edist)
+  res <- proportion_transmission(
+    percent_transmission = 0.8,
+    offspring_dist = edist
+  )
 
   expect_s3_class(res, "data.frame")
   expect_identical(dim(res), c(1L, 3L))