[SPARK-3867] ./python/run-tests failed when it run with Python 2.6 an…

…d unittest2 is not installed
apache · Oct 11, 2014 · f068eb5 · f068eb5
1 parent 1d72a30
commit f068eb5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py
@@ -25,7 +25,11 @@
 from numpy import array, array_equal
 
 if sys.version_info[:2] <= (2, 6):
-    import unittest2 as unittest
+    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
 

diff --git a/python/pyspark/tests.py b/python/pyspark/tests.py
@@ -34,7 +34,11 @@
 from platform import python_implementation
 
 if sys.version_info[:2] <= (2, 6):
-    import unittest2 as unittest
+    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