diff --git a/python/pyspark/tests.py b/python/pyspark/tests.py index 5b124d9308562..81ef80e713eee 100644 --- a/python/pyspark/tests.py +++ b/python/pyspark/tests.py @@ -26,7 +26,16 @@ import sys from tempfile import NamedTemporaryFile import time -import unittest + +if sys.version_info[:2] <= (2, 6): + try: + import unittest2 as unittest + except ImportError: + sys.stderr.write('Please install unittest2 to test with Python 2.6 or earlier') + sys.exit(1) +else: + import unittest + from pyspark.context import SparkContext from pyspark.files import SparkFiles diff --git a/python/run-tests b/python/run-tests index a986ac9380be4..6eeef4c13680e 100755 --- a/python/run-tests +++ b/python/run-tests @@ -33,6 +33,16 @@ function run_test() { FAILED=$((PIPESTATUS[0]||$FAILED)) } +echo "Running PySpark tests. Output is in python/unit-tests.log." + +# Try to test with Python 2.6, since that's the minimum version that we support: +if [ $(which python2.6) ]; then + export PYSPARK_PYTHON="python2.6" +fi + +echo "Testing with Python version:" +$PYSPARK_PYTHON --version + run_test "pyspark/rdd.py" run_test "pyspark/context.py" run_test "pyspark/conf.py"