TST: Added test for xgb.plot.tree

2015-09-07 21:49:27 -04:00 · 2015-09-07 21:49:27 -04:00 · 408c3a62a8
commit 408c3a62a8
parent d833038ba1
1 changed files with 4 additions and 0 deletions
--- a/R-package/tests/testthat/test_helpers.R
+++ b/R-package/tests/testthat/test_helpers.R
@ -25,3 +25,7 @@ test_that("xgb.importance works", {
  xgb.dump(bst, 'xgb.model.dump', with.stats = T)
  importance <- xgb.importance(sparse_matrix@Dimnames[[2]], 'xgb.model.dump')
 })
+
+test_that("xgb.plot.tree works", {
+  xgb.plot.tree(agaricus.train$data@Dimnames[[2]], model = bst)
+})