remove some warnings in tests

Author: topepo

tests/testthat/test_boost_tree.R

@@ -120,12 +120,12 @@ test_that('updating', {
 test_that('bad input', {
   expect_error(boost_tree(mode = "bogus"))
   expect_error({
-    bt <- boost_tree(trees = -1)
-    fit(bt, Species ~ ., iris) %>% set_engine("xgboost")
+    bt <- boost_tree(trees = -1) %>% set_engine("xgboost")
+    fit(bt, Species ~ ., iris)
   })
   expect_error({
-    bt <- boost_tree(min_n = -10)
-    fit(bt, Species ~ ., iris)  %>% set_engine("xgboost")
+    bt <- boost_tree(min_n = -10)  %>% set_engine("xgboost")
+    fit(bt, Species ~ ., iris)
   })
   expect_message(translate(boost_tree(mode = "classification"), engine = NULL))
   expect_error(translate(boost_tree(formula = y ~ x)))

tests/testthat/test_decision_tree.R

@@ -110,12 +110,12 @@ test_that('updating', {
 test_that('bad input', {
   expect_error(decision_tree(mode = "bogus"))
   expect_error({
-    bt <- decision_tree(cost_complexity = -1)
-    fit(bt, Species ~ ., iris) %>% set_engine("rpart")
+    bt <- decision_tree(cost_complexity = -1) %>% set_engine("rpart")
+    fit(bt, Species ~ ., iris)
   })
   expect_error({
-    bt <- decision_tree(min_n = 0)
-    fit(bt, Species ~ ., iris)  %>% set_engine("rpart")
+    bt <- decision_tree(min_n = 0)  %>% set_engine("rpart")
+    fit(bt, Species ~ ., iris)
   })
   expect_error(translate(decision_tree(), engine = NULL))
   expect_error(translate(decision_tree(formula = y ~ x)))

tests/testthat/test_linear_reg_keras.R

@@ -101,9 +101,12 @@ test_that('regression prediction', {
     )
 
   keras_pred <-
-    predict(lm_fit$fit, as.matrix(iris[1:3,2:4])) %>%
-    as_tibble() %>%
-    setNames(".pred")
+    predict(lm_fit$fit, as.matrix(iris[1:3,2:4]))
+  colnames(keras_pred) <- ".pred"
+
+  keras_pred <-
+    keras_pred %>%
+    as_tibble()
   parsnip_pred <- predict(lm_fit, iris[1:3,2:4])
   expect_equal(as.data.frame(keras_pred), as.data.frame(parsnip_pred))