Added more test files

Author: damontoth

tests/testthat/test-pFinalSize.R

@@ -0,0 +1,11 @@
+test_that("pFinalSize() matches brute-force calculation", {
+  pfs <- pFinalSize(n=1, j=4, R=1.2, k=0.33)
+  
+  d <- dnbinom(0:3, mu=1.2, size=0.33)
+  dnb <- d[4] * d[1]^3 +
+    d[3] * d[2] * d[1]^2 * 2 +
+    d[2] * d[3] * d[1]^2 +
+    d[2]^3 * d[1]
+  
+  expect_equal(pfs, dnb, tolerance = sqrt(.Machine$double.eps))
+})

tests/testthat/test-pFinalSizeSwitch1.R

@@ -0,0 +1,12 @@
+test_that("pFinalSizeSwitch1() matches brute-force calculation", {
+  pfs <- pFinalSizeSwitch1(n=1, j=4, R0=1.2, k0=0.33, Rc=0.9, kc=1.1)
+  
+  d0 <- dnbinom(0:3, mu=1.2, size=0.33)
+  dc <- dnbinom(0:2, mu=0.9, size=1.1)
+  dnb <- d0[4] * dc[1]^3 +
+    d0[3] * dc[2] * dc[1]^2 * 2 +
+    d0[2] * dc[3] * dc[1]^2 +
+    d0[2] * dc[2]^2 * dc[1]
+  
+  expect_equal(pfs, dnb, tolerance = sqrt(.Machine$double.eps))
+})

tests/testthat/test-pNextGenSize.R

@@ -10,15 +10,3 @@ test_that("pNextGenSize() with multiple transmitters matches brute-force dnbinom
 
   expect_equal(pngs, dnb, tolerance = sqrt(.Machine$double.eps))
 })
-
-test_that("pFinalSize() matches brute-force calculation", {
-  pfs <- pFinalSize(n=1, j=4, R=1.2, k=0.33)
-  
-  d <- dnbinom(0:3, mu=1.2, size=0.33)
-  dnb <- d[4] * d[1]^3 +
-    d[3] * d[2] * d[1]^2 * 2 +
-    d[2] * d[3] * d[1]^2 +
-    d[2]^3 * d[1]
-  
-  expect_equal(pfs, dnb, tolerance = sqrt(.Machine$double.eps))
-})