fixed unit test in R

This commit is contained in:
Yuan Tang (Terry) 2015-09-06 20:00:25 -04:00
parent 92b996513e
commit 339a53d9d4

View File

@ -13,7 +13,7 @@ test_that("glm works", {
nthread = 2, alpha = 0.0001, lambda = 1) nthread = 2, alpha = 0.0001, lambda = 1)
watchlist <- list(eval = dtest, train = dtrain) watchlist <- list(eval = dtest, train = dtrain)
num_round <- 2 num_round <- 2
expect_that(bst <- xgb.train(param, dtrain, num_round, watchlist), not(throws_error())) bst <- xgb.train(param, dtrain, num_round, watchlist)
expect_that(ypred <- predict(bst, dtest), not(throws_error())) ypred <- predict(bst, dtest)
expect_equal(length(getinfo(dtest, 'label')), 1611) expect_equal(length(getinfo(dtest, 'label')), 1611)
}) })