Merge pull request #17 from mayer79/missing_tests

Add some tests
ModelOriented · Jul 1, 2023 · 51d0f29 · 51d0f29
2 parents 0c4501b + 5e25261
commit 51d0f29
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/testthat/test_interact.R b/tests/testthat/test_interact.R
@@ -9,6 +9,7 @@ test_that("Additive models show 0 interactions (univariate)", {
   )
   expect_equal(H2(inter), 0)
   expect_s3_class(H2_j(inter), "ggplot")
+  expect_s3_class(plot(inter), "ggplot")
 })
 
 test_that("Additive models show 0 interactions (multivariate)", {
@@ -24,6 +25,7 @@ test_that("Additive models show 0 interactions (multivariate)", {
   )
   expect_equal(H2(inter), c(Sepal.Length = 0, Sepal.Width = 0))
   expect_s3_class(H2_j(inter), "ggplot")
+  expect_s3_class(plot(inter), "ggplot")
 })
 
 test_that("Non-additive models show interactions > 0 (one interaction)", {

diff --git a/tests/testthat/test_pdp.R b/tests/testthat/test_pdp.R
@@ -5,6 +5,7 @@ g <- 2:1
 test_that("pd_raw() works for simple example", {
   pd <- pd_raw(1, v = "a", X = X, grid = g, pred_fun = pfun)
   expect_equal(pd, cbind(pred = g / 2))
+  capture_output(expect_no_error(print(pd)))
 })
 
 test_that("pd_raw() works on simple example with and without grid compression", {