Fix Python environment for distributed unit tests (#3806)

tests/travis/run_test.sh

@@ -147,6 +147,11 @@ fi
 if [ ${TASK} == "distributed_test" ]; then
     set -e
     make all || exit -1
+    echo "-------------------------------"
+    source activate python3
+    python --version
+    conda install numpy scipy
+    python -m pip install kubernetes
     cd tests/distributed
     ./runtests.sh
 fi

tests/travis/setup.sh

@@ -10,8 +10,7 @@ if [ ${TASK} == "lint" ]; then
 fi
 
 
-if [ ${TASK} == "python_test" ] || [ ${TASK} == "python_lightweight_test" ]; then
-    # python2
+if [ ${TASK} == "python_test" ] || [ ${TASK} == "python_lightweight_test" ] || [ ${TASK} ==  "distributed_test" ]; then
     if [ ${TRAVIS_OS_NAME} == "osx" ]; then
         wget -O conda.sh https://repo.continuum.io/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
     else
@@ -27,7 +26,3 @@ if [ ${TASK} == "python_test" ] || [ ${TASK} == "python_lightweight_test" ]; the
     conda create -n python3 python=3.5
     conda create -n python2 python=2.7
 fi
-
-if [ ${TASK} == "distributed_test" ]; then
-    pip install --user kubernetes numpy scipy
-fi