From c50cf6d7ff367ca40a4453f10d07edf569614f9f Mon Sep 17 00:00:00 2001
From: terrytangyuan <terrytangyuan@gmail.com>
Date: Mon, 7 Sep 2015 22:03:28 -0400
Subject: [PATCH] TST: Added test for poisson regression

---
 R-package/tests/testthat/test_poisson_regression.R | 13 +++++++++++++
 1 file changed, 13 insertions(+)
 create mode 100644 R-package/tests/testthat/test_poisson_regression.R

diff --git a/R-package/tests/testthat/test_poisson_regression.R b/R-package/tests/testthat/test_poisson_regression.R
new file mode 100644
index 000000000..5d3d78e27
--- /dev/null
+++ b/R-package/tests/testthat/test_poisson_regression.R
@@ -0,0 +1,13 @@
+context('Test poisson regression model')
+
+require(xgboost)
+
+test_that("poisson regression works", {
+  data(mtcars)
+  bst = xgboost(data=as.matrix(mtcars[,-11]),label=mtcars[,11],
+                objective='count:poisson',nrounds=5)
+  expect_equal(class(bst), "xgb.Booster")
+  pred = predict(bst,as.matrix(mtcars[,-11]))
+  expect_equal(length(pred), 32)
+  sqrt(mean((pred-mtcars[,11])^2))
+})
\ No newline at end of file