From 4ec706e12570378e2a781189fac5386ea0127728 Mon Sep 17 00:00:00 2001
From: Daniel <mail@danielluedecke.de>
Date: Thu, 5 Oct 2023 09:31:33 +0200
Subject: [PATCH] test

---
 tests/testthat/test-check_predictions.R | 48 ++++++++++++++++++++++++-
 1 file changed, 47 insertions(+), 1 deletion(-)

diff --git a/tests/testthat/test-check_predictions.R b/tests/testthat/test-check_predictions.R
index caf0c66d4..61d97f9b9 100644
--- a/tests/testthat/test-check_predictions.R
+++ b/tests/testthat/test-check_predictions.R
@@ -23,7 +23,7 @@ test_that("check_predictions", {
   expect_equal(
     out$sim_1,
     c(
-      23.70112, 24.56502, 25.43419, 20.40954, 13.58266, 20.72532, 
+      23.70112, 24.56502, 25.43419, 20.40954, 13.58266, 20.72532,
       11.95366, 25.14559, 22.61286, 18.48403, 20.26737, 21.2291, 20.67149,
       10.07628, 0.25886, 10.64176, 10.18407, 20.68235, 28.10115, 27.55045,
       28.22301, 18.94021, 16.87727, 14.05421, 13.8378, 28.13797, 26.86451,
@@ -32,3 +32,49 @@ test_that("check_predictions", {
     tolerance = 1e-4
   )
 })
+
+
+test_that("check_predictions, glmmTMB", {
+  skip_if_not_installed("glmmTMB")
+  data(mtcars)
+  model <- glmmTMB::glmmTMB(vs ~ disp, data = mtcars, family = binomial())
+  set.seed(99)
+  out <- check_predictions(model)
+
+  expect_named(
+    out,
+    c(
+      "sim_1", "sim_2", "sim_3", "sim_4", "sim_5", "sim_6", "sim_7",
+      "sim_8", "sim_9", "sim_10", "sim_11", "sim_12", "sim_13", "sim_14",
+      "sim_15", "sim_16", "sim_17", "sim_18", "sim_19", "sim_20", "sim_21",
+      "sim_22", "sim_23", "sim_24", "sim_25", "sim_26", "sim_27", "sim_28",
+      "sim_29", "sim_30", "sim_31", "sim_32", "sim_33", "sim_34", "sim_35",
+      "sim_36", "sim_37", "sim_38", "sim_39", "sim_40", "sim_41", "sim_42",
+      "sim_43", "sim_44", "sim_45", "sim_46", "sim_47", "sim_48", "sim_49",
+      "sim_50", "y"
+    )
+  )
+  expect_equal(
+    out$sim_1,
+    c(
+      1, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 1, 1,
+      1, 0, 0, 0, 0, 1, 1, 1, 0, 1, 0, 1
+    ),
+    tolerance = 1e-4
+  )
+  expect_true(attributes(out)$model_info$is_bernoulli)
+
+  model <- glmmTMB::glmmTMB(vs ~ disp + (1 | cyl), data = mtcars, family = binomial())
+  set.seed(99)
+  out <- check_predictions(model)
+
+  expect_equal(
+    out$sim_1,
+    c(
+      0, 1, 1, 0, 0, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1, 1,
+      1, 0, 0, 0, 0, 1, 1, 1, 0, 1, 0, 0
+    ),
+    tolerance = 1e-4
+  )
+  expect_true(attributes(out)$model_info$is_bernoulli)
+})