From 33f1ab3ae1237f788f09e81a0106aff46869b2bd Mon Sep 17 00:00:00 2001 From: terrytangyuan Date: Mon, 7 Sep 2015 22:51:14 -0400 Subject: [PATCH] TST: Added one minor check for xgb.importance --- R-package/tests/testthat/test_helpers.R | 1 + 1 file changed, 1 insertion(+) diff --git a/R-package/tests/testthat/test_helpers.R b/R-package/tests/testthat/test_helpers.R index e26ec155b..4d80146e3 100644 --- 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", {