Update test_inspect.R

integrated-inferences · Oct 21, 2024 · 1e53497 · 1e53497
1 parent f2d1ba3
commit 1e53497
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tests/testthat/test_inspect.R b/tests/testthat/test_inspect.R
@@ -114,9 +114,20 @@ test_that("inspect outputs correct stanfit", {
 
 
 test_that("inspect handles dots", {
-  expect_true(make_model("X->Y") |>
+  model <- make_model("X->Y")
+
+  expect_true(model |>
                 inspect("prior_hyperparameters", nodes = "Y") |>
                 length() == 4)
+
+  expect_true(
+    grab(model, what = "data_types", complete_data = TRUE) |>
+      nrow() == 4)
+
+  expect_true(
+    grab(model, what = "data_types", complete_data = FALSE) |>
+      nrow() == 9)
+
 })