expect_message -> expect_snapshot

tidymodels · Sep 9, 2024 · 4562458 · 4562458
1 parent ebc0e73
commit 4562458
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/tests/testthat/_snaps/boost_tree.md b/tests/testthat/_snaps/boost_tree.md
@@ -23,6 +23,21 @@
       Error in `boost_tree()`:
       ! "bogus" is not a known mode for model `boost_tree()`.
 
+---
+
+    Code
+      translate(boost_tree(mode = "classification"), engine = NULL)
+    Message
+      Used `engine = 'xgboost'` for translation.
+    Output
+      Boosted Tree Model Specification (classification)
+      
+      Computational engine: xgboost 
+      
+      Model fit template:
+      parsnip::xgb_train(x = missing_arg(), y = missing_arg(), weights = missing_arg(), 
+          nthread = 1, verbose = 0)
+
 ---
 
     Code

diff --git a/tests/testthat/test-boost_tree.R b/tests/testthat/test-boost_tree.R
@@ -12,7 +12,7 @@ test_that('updating', {
 
 test_that('bad input', {
   expect_snapshot(error = TRUE, boost_tree(mode = "bogus"))
-  expect_message(translate(boost_tree(mode = "classification"), engine = NULL))
+  expect_snapshot(translate(boost_tree(mode = "classification"), engine = NULL))
   expect_snapshot(error = TRUE, translate(boost_tree(formula = y ~ x)))
 })