TST move test to test_multinom_reg_glmnet

Author: patr1ckm

tests/testthat/test_multinom_reg.R

@@ -105,13 +105,4 @@ test_that('bad input', {
   expect_warning(translate(multinom_reg() %>% set_engine("glmnet", x = iris[,1:3], y = iris$Species)))
 })
 
-test_that("predictions are factors with all levels", {
-  basic <- multinom_reg() %>% set_engine("glmnet") %>% fit(Species ~ ., data = iris)
-  nd <- iris[iris$Species == "setosa", ]
-  yhat <- predict(basic, new_data = nd, penalty = .1)
-  expect_is(yhat$.pred_class, "factor")
-  expect_equal(levels(yhat$.pred_class), levels(iris$Species))
-  yhat_multi <- multi_predict(basic, new_data =  nd, penalty = .1)$.pred
-  expect_is(yhat_multi[[1]]$.pred_class, "factor")
-  expect_equal(levels(yhat_multi[[1]]$.pred_class), levels(iris$Species))
-})
+

tests/testthat/test_multinom_reg_glmnet.R

@@ -138,6 +138,10 @@ test_that('glmnet probabilities, mulitiple lambda', {
     multi_predict(xy_fit, iris[rows, 1:4], penalty = lams)$.pred
   )
 
+  expect_is(yhat$.pred_class, "factor")
+  expect_equal(levels(yhat$.pred_class), levels(iris$Species))
+
+
   expect_error(
     multi_predict(xy_fit, newdata = iris[rows, 1:4], penalty = lams),
     "Did you mean"
@@ -150,3 +154,12 @@ test_that('glmnet probabilities, mulitiple lambda', {
   )
 
 })
+
+test_that("predictions are factors with all levels", {
+  basic <- multinom_reg() %>% set_engine("glmnet") %>% fit(Species ~ ., data = iris[rows, ])
+  nd <- iris[iris$Species == "setosa", ]
+  yhat <- predict(basic, new_data = nd, penalty = .1)
+  yhat_multi <- multi_predict(basic, new_data =  nd, penalty = .1)$.pred
+  expect_is(yhat_multi[[1]]$.pred_class, "factor")
+  expect_equal(levels(yhat_multi[[1]]$.pred_class), levels(iris$Species))
+})