update test for tune. related to #22

The test is failing because of sample_size = tune() for lightgbm.
curso-r · Aug 2, 2020 · a568413 · a568413
1 parent e0b6c78
commit a568413
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/tests/testthat/test-catboost.R b/tests/testthat/test-catboost.R
@@ -8,7 +8,15 @@ test_that("catboost", {
 
 test_that("catboost with tune", {
 
-  model <- parsnip::boost_tree(mtry = 5, trees = tune())
+  model <- parsnip::boost_tree(
+    mtry = 5,
+    learn_rate = tune(),
+    loss_reduction = tune(),
+    sample_size = tune(),
+    trees = tune(),
+    min_n = tune(),
+    tree_depth = tune()
+  )
   model <- parsnip::set_engine(model, "catboost")
 
   expect_can_tune_boost_tree(model)

diff --git a/tests/testthat/test-lightgbm.R b/tests/testthat/test-lightgbm.R
@@ -8,7 +8,15 @@ test_that("lightgbm", {
 
 test_that("lightgbm with tune", {
 
-  model <- parsnip::boost_tree(mtry = 5, trees = tune())
+  model <- parsnip::boost_tree(
+    mtry = 5,
+    learn_rate = tune(),
+    loss_reduction = tune(),
+    sample_size = tune(),
+    trees = tune(),
+    min_n = tune(),
+    tree_depth = tune()
+  )
   model <- parsnip::set_engine(model, "lightgbm")
   model <- parsnip::set_mode(model, "regression")