Merge pull request #51 from ncn-foreigners/main

pulling changes from main
ncn-foreigners · Jul 15, 2024 · f92547f · f92547f
2 parents b57f497 + b479e9c
commit f92547f
Showing 1 changed file with 38 additions and 3 deletions.
diff --git a/inst/tinytest/test_singleRcapture.R b/inst/tinytest/test_singleRcapture.R
@@ -16,7 +16,7 @@ expect_silent(
 )
 
 expect_silent(
-  estimatePopsize(
+  dd <- estimatePopsize(
     formula = TOTAL_SUB ~ .,
     model = "zotnegbin",
     method = "optim",
@@ -29,6 +29,13 @@ expect_silent(
   )
 )
 
+expect_silent(
+  predict(
+    dd,
+    type = "mean"
+  )
+)
+
 expect_silent(
   irls <- estimatePopsize(
     formula = TOTAL_SUB ~ .,
@@ -71,6 +78,13 @@ expect_silent(
   )
 )
 
+expect_silent(
+  predict(
+    ch,
+    type = "mean"
+  )
+)
+
 expect_equivalent(
   ch$populationSize$pointEstimate,
   21657,
@@ -97,6 +111,13 @@ expect_silent(
   )
 )
 
+expect_silent(
+  predict(
+    zl,
+    type = "mean"
+  )
+)
+
 expect_silent(
   zl2 <- estimatePopsize(
     formula = capture ~ 1,
@@ -260,7 +281,7 @@ expect_error(
 if (isTRUE(tolower(Sys.getenv("TEST_SINGLERCAPTURE_MULTICORE_DEVELOPER")) == "true")) {
   set.seed(123)
   expect_silent(
-    estimatePopsize(
+    xx <- estimatePopsize(
       formula = TOTAL_SUB ~ .,
       data = farmsubmission,
       model = "zotpoisson",
@@ -278,7 +299,14 @@ if (isTRUE(tolower(Sys.getenv("TEST_SINGLERCAPTURE_MULTICORE_DEVELOPER")) == "tr
   )
 
   expect_silent(
-    estimatePopsize(
+    predict(
+      xx,
+      type = "mean"
+    )
+  )
+
+  expect_silent(
+    xx <- estimatePopsize(
       formula = capture ~ gender,
       data = netherlandsimmigrant,
       model = "zotgeom",
@@ -293,6 +321,13 @@ if (isTRUE(tolower(Sys.getenv("TEST_SINGLERCAPTURE_MULTICORE_DEVELOPER")) == "tr
     )
   )
 
+  expect_silent(
+    predict(
+      xx,
+      type = "mean"
+    )
+  )
+
   expect_silent(
     estimatePopsize(
       formula = TOTAL_SUB ~ .,