TST: Added one minor check for xgb.importance

2015-09-07 22:51:14 -04:00 · 2015-09-07 22:51:14 -04:00 · 33f1ab3ae1
commit 33f1ab3ae1
parent fbf2a5feed
1 changed files with 1 additions and 0 deletions
--- a/R-package/tests/testthat/test_helpers.R
+++ b/R-package/tests/testthat/test_helpers.R
@ -24,6 +24,7 @@ test_that("xgb.dump works", {
 test_that("xgb.importance works", {
  xgb.dump(bst, 'xgb.model.dump', with.stats = T)
  importance <- xgb.importance(sparse_matrix@Dimnames[[2]], 'xgb.model.dump')
+  expect_equal(dim(importance), c(7, 4))
 })

 test_that("xgb.plot.tree works", {