From fe999bf968e173e79b89b786f59a6fe496d99151 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <ybbs.daans@hotmail.com>
Date: Thu, 15 Nov 2018 22:17:35 +1300
Subject: [PATCH] Add back python2 tests for Travis light weight tests. (#3901)

---
 tests/travis/run_test.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/travis/run_test.sh b/tests/travis/run_test.sh
index b2fd2dcf4..b0780c4cb 100755
--- a/tests/travis/run_test.sh
+++ b/tests/travis/run_test.sh
@@ -76,6 +76,7 @@ fi
 
 if [ ${TASK} == "python_lightweight_test" ]; then
     make all || exit -1
+
     echo "-------------------------------"
     source activate python3
     python --version
@@ -83,12 +84,15 @@ if [ ${TASK} == "python_lightweight_test" ]; then
     python -m pip install graphviz pytest pytest-cov codecov
     py.test -v --fulltrace -s tests/python --cov=python-package/xgboost || exit -1
     codecov
+
     source activate python2
     echo "-------------------------------"
     python --version
-    conda install numpy scipy
+    conda install numpy scipy pytest
     python -m pip install graphviz
     python -m pip install flake8==3.4.1
+    py.test -v --fulltrace -s tests/python || exit -1
+
     flake8 --ignore E501 python-package || exit -1
     flake8 --ignore E501 tests/python || exit -1
     exit 0