test output of evaluate_smooths

greta-dev · Dec 10, 2024 · 627b407 · 627b407
1 parent 8d9a1df
commit 627b407
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 2 deletions.
diff --git a/tests/testthat/_snaps/evaluate-smooths.md b/tests/testthat/_snaps/evaluate-smooths.md
@@ -6,3 +6,27 @@
       Error in `evaluate_smooths()`:
       ! Can only evaluate smooths from greta arrays created with `greta.gam::smooths()`
 
+---
+
+    Code
+      (eval_z <- evaluate_smooths(z, newdata = data.frame(x = x_plot)))
+    Message
+      greta array <operation>
+      
+    Output
+            [,1]
+       [1,]  ?  
+       [2,]  ?  
+       [3,]  ?  
+       [4,]  ?  
+       [5,]  ?  
+       [6,]  ?  
+       [7,]  ?  
+       [8,]  ?  
+       [9,]  ?  
+      [10,]  ?  
+    Message
+      
+      i 190 more values
+      Use `print(n = ...)` to see more values
+
diff --git a/tests/testthat/test-evaluate-smooths.R b/tests/testthat/test-evaluate-smooths.R
@@ -15,7 +15,12 @@ test_that("evaluate_smooths errors appropriately", {
     error = TRUE,
     evaluate_smooths("thing")
   )
-  expect_no_error(
-    z_pred <- evaluate_smooths(z, newdata = data.frame(x = x_plot))
+  expect_snapshot(
+    (eval_z <- evaluate_smooths(z, newdata = data.frame(x = x_plot)))
+  )
+
+  expect_s3_class(
+    object = eval_z,
+    "greta_array"
   )
 })