From f04bde05fd6845d28c1bfcfba35294364992a096 Mon Sep 17 00:00:00 2001
From: Yun Ni <Euler57721@gmail.com>
Date: Tue, 5 Sep 2017 14:46:51 -0700
Subject: [PATCH] Add Coverage Report for Java and Python (#2667)

* Add coverage report for java

* Add coverage report for python

* Increase memory for JVM unit tests

* Increase memory for JVM unit tests
---
 .travis.yml              |  2 +-
 jvm-packages/pom.xml     | 19 +++++++++++++++++++
 tests/travis/run_test.sh |  8 ++++++--
 3 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index da91aeb19..ee43acf99 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -58,7 +58,7 @@ addons:
 before_install:
   - source dmlc-core/scripts/travis/travis_setup_env.sh
   - export PYTHONPATH=${PYTHONPATH}:${PWD}/python-package
-  - echo "MAVEN_OPTS='-Xmx2048m -XX:MaxPermSize=1024m -XX:ReservedCodeCacheSize=512m -Dorg.slf4j.simpleLogger.defaultLogLevel=error'" > ~/.mavenrc
+  - echo "MAVEN_OPTS='-Xmx4g -XX:MaxPermSize=1024m -XX:ReservedCodeCacheSize=512m -Dorg.slf4j.simpleLogger.defaultLogLevel=error'" > ~/.mavenrc
 
 install:
   - source tests/travis/setup.sh
diff --git a/jvm-packages/pom.xml b/jvm-packages/pom.xml
index 6cf06ee62..b1575096e 100644
--- a/jvm-packages/pom.xml
+++ b/jvm-packages/pom.xml
@@ -215,6 +215,25 @@
                     </execution>
                 </executions>
             </plugin>
+            <plugin>
+                <groupId>org.jacoco</groupId>
+                <artifactId>jacoco-maven-plugin</artifactId>
+                <version>0.7.9</version>
+                <executions>
+                    <execution>
+                        <goals>
+                            <goal>prepare-agent</goal>
+                        </goals>
+                    </execution>
+                    <execution>
+                        <id>report</id>
+                        <phase>test</phase>
+                        <goals>
+                            <goal>report</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
         </plugins>
     </build>
     <dependencies>
diff --git a/tests/travis/run_test.sh b/tests/travis/run_test.sh
index 37d27c56c..5fc3ddcea 100755
--- a/tests/travis/run_test.sh
+++ b/tests/travis/run_test.sh
@@ -32,8 +32,10 @@ if [ ${TASK} == "python_test" ]; then
     source activate python3
     python --version
     conda install numpy scipy pandas matplotlib nose scikit-learn
-    python -m pip install graphviz
+    python -m pip install graphviz pytest pytest-cov codecov
     python -m nose tests/python || exit -1
+    py.test tests/python --cov=python-package/xgboost
+    codecov
     source activate python2
     echo "-------------------------------"
     python --version
@@ -49,8 +51,10 @@ if [ ${TASK} == "python_lightweight_test" ]; then
     source activate python3
     python --version
     conda install numpy scipy nose
-    python -m pip install graphviz
+    python -m pip install graphviz pytest pytest-cov codecov
     python -m nose tests/python || exit -1
+    py.test tests/python --cov=python-package/xgboost
+    codecov
     source activate python2
     echo "-------------------------------"
     python --version