Merge pull request #14 from Yue-Jiang/master

add plotting tests

Author: Yue-Jiang

R/min_depth_distribution.R

@@ -143,7 +143,7 @@ plot_min_depth_distribution <- function(min_depth_frame, k = 10, min_no_of_trees
                                         mean_sample = "top_trees", mean_scale = FALSE, mean_round = 2,
                                         main = "Distribution of minimal depth and its mean"){
   minimal_depth <- NULL; mean_minimal_depth_label <- NULL; mean_minimal_depth <- NULL
-  if("randomForest" %in% class(min_depth_frame)){
+  if(any(c("randomForest", "ranger") %in% class(min_depth_frame))){
     min_depth_frame <- min_depth_distribution(min_depth_frame)
   }
   min_depth_count_list <- min_depth_count(min_depth_frame)

inst/testdata/test_ranger.rda

@@ -39,6 +39,35 @@ test_that("min_depth_interactions works", {
                     0)
 })
 
+test_that("plot_multi_way_importance works", {
+  p <- plot_multi_way_importance(rf_c)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_distribution works", {
+  p <- plot_min_depth_distribution(rf_c)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_interactions works", {
+  p <- plot_min_depth_interactions(rf_c)
+  expect_silent(print(p))
+})
+
+test_that("plot_predict_interaction works", {
+  p <- plot_predict_interaction(rf_c, data = iris, variable1 = "Petal.Width", variable2 = "Petal.Length")
+  expect_silent(print(p))
+})
+
+test_that("plot_importance_ggpairs works", {
+  p <- plot_importance_ggpairs(rf_c)
+  expect_silent(suppressMessages(print(p)))
+})
+
+test_that("plot_importance_rankings works", {
+  p <- plot_importance_rankings(rf_c)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
 
 context("Test randomForest regression forests")
 
@@ -69,6 +98,35 @@ test_that("min_depth_interactions works", {
                     1)
 })
 
+test_that("plot_multi_way_importance works", {
+  p <- plot_multi_way_importance(rf_r)
+  expect_silent(suppressWarnings(print(p)))
+})
+
+test_that("plot_min_depth_distribution works", {
+  p <- plot_min_depth_distribution(rf_r)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_interactions works", {
+  p <- plot_min_depth_interactions(rf_r)
+  expect_silent(print(p))
+})
+
+test_that("plot_predict_interaction works", {
+  p <- plot_predict_interaction(rf_r, data = mtcars, variable1 = "cyl", variable2 = "wt")
+  expect_silent(print(p))
+})
+
+test_that("plot_importance_ggpairs works", {
+  p <- plot_importance_ggpairs(rf_r)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
+
+test_that("plot_importance_rankings works", {
+  p <- plot_importance_rankings(rf_r)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
 
 context("Test randomForest unsupervised forests")
 
@@ -98,3 +156,33 @@ test_that("min_depth_interactions works", {
   expect_equivalent(min_depth_int[min_depth_int$interaction == "Petal.Width:Sepal.Length", ]$mean_min_depth,
                     1)
 })
+
+test_that("plot_multi_way_importance works", {
+  p <- plot_multi_way_importance(rf_u)
+  expect_silent(suppressWarnings(print(p)))
+})
+
+test_that("plot_min_depth_distribution works", {
+  p <- plot_min_depth_distribution(rf_u)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_interactions works", {
+  p <- plot_min_depth_interactions(rf_u)
+  expect_silent(print(p))
+})
+
+test_that("plot_predict_interaction does not work for unsupervised forests", {
+  expect_warning(p <- plot_predict_interaction(rf_u, data = iris, variable1 = "Petal.Width", variable2 = "Sepal.Length"))
+  expect_null(p)
+})
+
+test_that("plot_importance_ggpairs works", {
+  p <- plot_importance_ggpairs(rf_u)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
+
+test_that("plot_importance_rankings works", {
+  p <- plot_importance_rankings(rf_u)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})

tests/testthat/test_randomForest.R

@@ -5,11 +5,10 @@ load(system.file("testdata/test_ranger.rda", package="randomForestExplainer", mu
 # library(ranger)
 # library(survival)
 # set.seed(12345)
-# ranger_c <- ranger(Species ~ ., data = iris, importance = "impurity", num.trees = 2)
+# ranger_c <- ranger(Species ~ ., data = iris, importance = "impurity", num.trees = 2, probability = TRUE)
 # ranger_r <- ranger(mpg ~ ., data = mtcars, importance = "impurity", num.trees = 2)
 # ranger_s <- ranger(Surv(futime, fustat) ~ ., data = ovarian, importance = "impurity", num.trees = 2)
-# save(ranger_c, ranger_r, ranger_s, file = "inst/testdata/test_ranger.rda")
-
+# save(ranger_c, ranger_r, ranger_s, ovarian, file = "inst/testdata/test_ranger.rda")
 
 context("Test ranger classification forests")
 
@@ -24,7 +23,7 @@ test_that("important_variables works", {
   imp_vars <- important_variables(ranger_c, k = 3,
                                   measures = c("mean_min_depth", "impurity",
                                                "no_of_nodes", "times_a_root", "p_value"))
-  expect_equal(imp_vars, c("Petal.Width", "Petal.Length", "Sepal.Length"))
+  expect_equal(imp_vars, c("Petal.Width", "Sepal.Length", "Petal.Length"))
 })
 
 test_that("min_depth_distribution works", {
@@ -39,6 +38,35 @@ test_that("min_depth_interactions works", {
                     1.5)
 })
 
+test_that("plot_multi_way_importance works", {
+  p <- plot_multi_way_importance(ranger_c)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_distribution works", {
+  p <- plot_min_depth_distribution(ranger_c)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_interactions works", {
+  p <- plot_min_depth_interactions(ranger_c)
+  expect_silent(print(p))
+})
+
+test_that("plot_predict_interaction works", {
+  p <- plot_predict_interaction(ranger_c, data = iris, variable1 = "Petal.Width", variable2 = "Petal.Length")
+  expect_silent(print(p))
+})
+
+test_that("plot_importance_ggpairs works", {
+  p <- plot_importance_ggpairs(ranger_c)
+  expect_silent(suppressMessages(print(p)))
+})
+
+test_that("plot_importance_rankings works", {
+  p <- plot_importance_rankings(ranger_c)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
 
 context("Test ranger regression forests")
 
@@ -69,6 +97,35 @@ test_that("min_depth_interactions works", {
                     0.5)
 })
 
+test_that("plot_multi_way_importance works", {
+  p <- plot_multi_way_importance(ranger_r)
+  expect_silent(suppressWarnings(print(p)))
+})
+
+test_that("plot_min_depth_distribution works", {
+  p <- plot_min_depth_distribution(ranger_r)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_interactions works", {
+  p <- plot_min_depth_interactions(ranger_r)
+  expect_silent(print(p))
+})
+
+test_that("plot_predict_interaction works", {
+  p <- plot_predict_interaction(ranger_r, data = mtcars, variable1 = "cyl", variable2 = "wt")
+  expect_silent(print(p))
+})
+
+test_that("plot_importance_ggpairs works", {
+  p <- plot_importance_ggpairs(ranger_r)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
+
+test_that("plot_importance_rankings works", {
+  p <- plot_importance_rankings(ranger_c)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
 
 context("Test ranger survival forests")
 
@@ -98,3 +155,33 @@ test_that("min_depth_interactions works", {
   expect_equivalent(min_depth_int[min_depth_int$interaction == "age:ecog.ps", ]$mean_min_depth,
                     0.5)
 })
+
+test_that("plot_multi_way_importance works", {
+  p <- plot_multi_way_importance(ranger_s)
+  expect_silent(suppressWarnings(print(p)))
+})
+
+test_that("plot_min_depth_distribution works", {
+  p <- plot_min_depth_distribution(ranger_s)
+  expect_silent(print(p))
+})
+
+test_that("plot_min_depth_interactions works", {
+  p <- plot_min_depth_interactions(ranger_s)
+  expect_silent(print(p))
+})
+
+test_that("plot_predict_interaction works", {
+  p <- plot_predict_interaction(ranger_s, data = ovarian, variable1 = "age", variable2 = "ecog.ps")
+  expect_silent(print(p))
+})
+
+test_that("plot_importance_ggpairs works", {
+  p <- plot_importance_ggpairs(ranger_s)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})
+
+test_that("plot_importance_rankings works", {
+  p <- plot_importance_rankings(ranger_s)
+  expect_silent(suppressWarnings(suppressMessages(print(p))))
+})