test cases for printing

Author: topepo

tests/testthat/test_boost_tree_C50.R

@@ -32,6 +32,8 @@ test_that('C5.0 execution', {
     ),
     regexp = NA
   )
+  expect_output(print(res), "parsnip model object")
+
   expect_error(
     res <- fit_xy(
       lc_basic,

tests/testthat/test_linear_reg.R

@@ -225,6 +225,8 @@ test_that('lm execution', {
     ),
     regexp = NA
   )
+  expect_output(print(res), "parsnip model object")
+
   expect_error(
     res <- fit_xy(
       iris_basic,

tests/testthat/test_logistic_reg.R

@@ -245,6 +245,8 @@ test_that('glm execution', {
     ),
     regexp = NA
   )
+  expect_output(print(res), "parsnip model object")
+
   expect_error(
     res <- fit(
       lc_basic,

tests/testthat/test_mars.R

@@ -139,6 +139,8 @@ test_that('mars execution', {
     ),
     regexp = NA
   )
+  expect_output(print(res), "parsnip model object")
+
   expect_error(
     res <- fit_xy(
       iris_basic,

tests/testthat/test_misc.R

@@ -56,3 +56,34 @@ test_that('getting y names from terms', {
 
 })
 
+# ------------------------------------------------------------------------------
+
+test_that('S3 method dispatch/registration', {
+
+  expect_error(
+    res <-
+      null_model() %>%
+      set_engine("parsnip") %>%
+      set_mode("regression") %>%
+      fit(mpg ~ ., data = mtcars) %>%
+      tidy(),
+    regex = NA
+  )
+  expect_true(tibble::is_tibble(res))
+
+  expect_error(
+    res <-
+      null_model() %>%
+      set_engine("parsnip") %>%
+      set_mode("classification") %>%
+      fit(Species ~ ., data = iris) %>%
+      tidy(),
+    regex = NA
+  )
+  expect_true(tibble::is_tibble(res))
+
+
+
+})
+
+

tests/testthat/test_mlp_nnet.R

@@ -27,6 +27,8 @@ test_that('nnet execution, classification', {
     ),
     regexp = NA
   )
+  expect_output(print(res), "parsnip model object")
+
   expect_error(
     res <- parsnip::fit_xy(
       iris_nnet,

tests/testthat/test_multinom_reg_nnet.R

@@ -41,6 +41,7 @@ test_that('model fitting', {
       ),
     regexp = NA
   )
+  expect_output(print(fit1), "parsnip model object")
 
   set.seed(257)
   expect_error(

tests/testthat/test_rand_forest_ranger.R

@@ -34,6 +34,7 @@ test_that('ranger classification execution', {
     ),
     regexp = NA
   )
+  expect_output(print(res), "parsnip model object")
 
   expect_error(
     res <- fit_xy(

tests/testthat/test_surv_reg_survreg.R

@@ -28,6 +28,7 @@ test_that('survival execution', {
     ),
     regexp = NA
   )
+  expect_output(print(res), "parsnip model object")
   expect_error(
     res <- fit(
       surv_lnorm,

tests/testthat/test_svm_poly.R

@@ -139,6 +139,7 @@ test_that('svm poly regression', {
 
   expect_false(has_multi_predict(res))
   expect_equal(multi_predict_args(res), NA_character_)
+  expect_output(print(res), "parsnip model object")
 
   expect_error(
     fit(