Skip to content

Commit c2c2c3d

Browse files
committed
updated unit test
1 parent 34f4ad5 commit c2c2c3d

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

tests/testthat/test_gen_additive_model.R

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,15 @@ context("generalized additive models")
1212

1313
# ------------------------------------------------------------------------------
1414

15-
reg_mod <- gen_additive_mod(select_features = TRUE) %>% set_engine("mgcv") %>% set_mode("regression")
1615

1716
test_that('regression', {
1817
skip_if_not_installed("mgcv")
1918

19+
reg_mod <-
20+
gen_additive_mod(select_features = TRUE) %>%
21+
set_engine("mgcv") %>%
22+
set_mode("regression")
23+
2024
expect_error(
2125
f_res <- fit(
2226
reg_mod,
@@ -51,14 +55,16 @@ test_that('regression', {
5155

5256
})
5357

54-
55-
5658
# ------------------------------------------------------------------------------
5759

58-
cls_mod <- gen_additive_mod(adjust_deg_free = 1.5) %>% set_engine("mgcv") %>% set_mode("classification")
59-
6060
test_that('classification', {
6161
skip_if_not_installed("mgcv")
62+
63+
cls_mod <-
64+
gen_additive_mod(adjust_deg_free = 1.5) %>%
65+
set_engine("mgcv") %>%
66+
set_mode("classification")
67+
6268
expect_error(
6369
f_res <- fit(
6470
cls_mod,

0 commit comments

Comments
 (0)