diff options
Diffstat (limited to 'R/pkg/inst/tests/testthat/test_mllib.R')
| -rw-r--r-- | R/pkg/inst/tests/testthat/test_mllib.R | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/R/pkg/inst/tests/testthat/test_mllib.R b/R/pkg/inst/tests/testthat/test_mllib.R index 1597306bb6..63ec84e497 100644 --- a/R/pkg/inst/tests/testthat/test_mllib.R +++ b/R/pkg/inst/tests/testthat/test_mllib.R @@ -204,6 +204,18 @@ test_that("naiveBayes", { "Yes", "Yes", "Yes", "Yes", "Yes", "Yes", "Yes", "Yes", "No", "No", "Yes", "Yes", "No", "No")) + # Test model save/load + modelPath <- tempfile(pattern = "naiveBayes", fileext = ".tmp") + ml.save(m, modelPath) + expect_error(ml.save(m, modelPath)) + ml.save(m, modelPath, overwrite = TRUE) + m2 <- ml.load(modelPath) + s2 <- summary(m2) + expect_equal(s$apriori, s2$apriori) + expect_equal(s$tables, s2$tables) + + unlink(modelPath) + # Test e1071::naiveBayes if (requireNamespace("e1071", quietly = TRUE)) { expect_that(m <- e1071::naiveBayes(Survived ~ ., data = t1), not(throws_error())) |