Add back python2 tests for Travis light weight tests. (#3901)

dmlc · Nov 15, 2018 · fe999bf · fe999bf
1 parent 2ea0f88
commit fe999bf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/travis/run_test.sh b/tests/travis/run_test.sh
@@ -76,19 +76,23 @@ fi
 
 if [ ${TASK} == "python_lightweight_test" ]; then
     make all || exit -1
+
     echo "-------------------------------"
     source activate python3
     python --version
     conda install numpy scipy
     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