diff --git a/R-package/tests/testthat/test_helpers.R b/R-package/tests/testthat/test_helpers.R index c51fef1bd..11368216b 100644 --- a/R-package/tests/testthat/test_helpers.R +++ b/R-package/tests/testthat/test_helpers.R @@ -23,7 +23,7 @@ test_that("xgb.dump works", { test_that("xgb.importance works", { expect_true(xgb.dump(bst, 'xgb.model.dump', with.stats = T)) - importance <- xgb.importance(sparse_matrix@Dimnames[[2]], 'xgb.model.dump') + importance <- xgb.importance(sparse_matrix@Dimnames[[2]], model = bst) expect_equal(dim(importance), c(7, 4)) expect_equal(colnames(importance), c("Feature", "Gain", "Cover", "Frequency")) })