unit tests for tidymodels/parsnip#353

tidymodels · Jul 30, 2020 · 9b2cffc · 9b2cffc
1 parent bc5e0ca
commit 9b2cffc
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
 
 .DS_Store
 .Rproj.user
+.Rhistory
diff --git a/tests/testthat/test-encodings-glmnet.R b/tests/testthat/test-encodings-glmnet.R
@@ -22,6 +22,12 @@ test_that('parsnip models with formula interface', {
 
   expect_true(sum(grepl("(Intercept)", parsnip_form_names)) == 1)
   expect_true(sum(grepl("^Alley", parsnip_form_names)) == 2)
+
+  expect_error(
+    predict(parsnip_form_fit, ames %>% select(Year_Built, Alley)),
+    regex = NA
+  )
+
 })
 
 test_that('parsnip models with xy interface', {
@@ -41,6 +47,11 @@ test_that('parsnip models with xy interface', {
   parsnip_xy_names <- tidy(parsnip_xy_fit)$term
 
   expect_true(sum(grepl("(Intercept)", parsnip_xy_names)) == 1)
+
+  expect_error(
+    predict(parsnip_xy_fit, ames %>% select(Year_Built, Longitude)),
+    regex = NA
+  )
 })
 
 ## -----------------------------------------------------------------------------
@@ -65,6 +76,11 @@ test_that('workflows', {
 
   expect_true(sum(grepl("(Intercept)", parsnip_wflow_names)) == 1)
   expect_true(sum(grepl("^Alley", parsnip_wflow_names)) == 2)
+
+  expect_error(
+    predict(parsnip_wflow_fit, ames %>% select(Year_Built, Alley)),
+    regex = NA
+  )
 })
 
 

diff --git a/tests/testthat/test-encodings-randomForest.R b/tests/testthat/test-encodings-randomForest.R
@@ -22,6 +22,11 @@ test_that('parsnip models with formula interface', {
 
   parsnip_form_names <- rownames(parsnip_form_fit$fit$importance)
   expect_true(sum(grepl("Location", parsnip_form_names)) == 1)
+
+  expect_error(
+    predict(parsnip_form_fit, scat),
+    regex = NA
+  )
 })
 
 test_that('parsnip models with xy interface', {
@@ -34,6 +39,11 @@ test_that('parsnip models with xy interface', {
   parsnip_xy_names <- rownames(parsnip_xy_fit$fit$importance)
   expect_true(sum(grepl("Location", parsnip_xy_names)) == 1)
 
+  expect_error(
+    predict(parsnip_xy_fit, scat[, -1]),
+    regex = NA
+  )
+
 })
 
 ## -----------------------------------------------------------------------------
@@ -59,6 +69,10 @@ test_that('workflows', {
 
   expect_true(sum(grepl("Location", parsnip_wflow_names)) == 1)
 
+  expect_error(
+    predict(parsnip_wflow_fit, scat[, -1]),
+    regex = NA
+  )
 })
 
 

diff --git a/tests/testthat/test-encodings-ranger.R b/tests/testthat/test-encodings-ranger.R
@@ -22,6 +22,11 @@ test_that('parsnip models with formula interface', {
 
   parsnip_form_names <- names(parsnip_form_fit$fit$variable.importance)
   expect_true(sum(grepl("Location", parsnip_form_names)) == 1)
+
+  expect_error(
+    predict(parsnip_form_fit, scat),
+    regex = NA
+  )
 })
 
 test_that('parsnip models with xy interface', {
@@ -34,6 +39,10 @@ test_that('parsnip models with xy interface', {
   parsnip_xy_names <- names(parsnip_xy_fit$fit$variable.importance)
   expect_true(sum(grepl("Location", parsnip_xy_names)) == 1)
 
+  expect_error(
+    predict(parsnip_xy_fit, scat[, -1]),
+    regex = NA
+  )
 })
 
 ## -----------------------------------------------------------------------------
@@ -59,6 +68,10 @@ test_that('workflows', {
 
   expect_true(sum(grepl("Location", parsnip_wflow_names)) == 1)
 
+  expect_error(
+    predict(parsnip_wflow_fit, scat[, -1]),
+    regex = NA
+  )
 })