diff options
author | Josh Rosen <joshrosen@eecs.berkeley.edu> | 2013-02-03 06:44:11 +0000 |
---|---|---|
committer | Josh Rosen <joshrosen@eecs.berkeley.edu> | 2013-02-03 06:44:11 +0000 |
commit | 2415c18f48fc28d88f29b88c312f98054f530f20 (patch) | |
tree | 182f6a82babf093b902407e2cdb8717bf97413cc | |
parent | ae26911ec0d768dcdae8b7d706ca4544e36535e6 (diff) | |
download | spark-2415c18f48fc28d88f29b88c312f98054f530f20.tar.gz spark-2415c18f48fc28d88f29b88c312f98054f530f20.tar.bz2 spark-2415c18f48fc28d88f29b88c312f98054f530f20.zip |
Fix reporting of PySpark doctest failures.
-rw-r--r-- | python/pyspark/context.py | 4 | ||||
-rw-r--r-- | python/pyspark/rdd.py | 4 |
2 files changed, 6 insertions, 2 deletions
diff --git a/python/pyspark/context.py b/python/pyspark/context.py index 6831f9b7f8..657fe6f989 100644 --- a/python/pyspark/context.py +++ b/python/pyspark/context.py @@ -256,8 +256,10 @@ def _test(): globs['sc'] = SparkContext('local[4]', 'PythonTest', batchSize=2) globs['tempdir'] = tempfile.mkdtemp() atexit.register(lambda: shutil.rmtree(globs['tempdir'])) - doctest.testmod(globs=globs) + (failure_count, test_count) = doctest.testmod(globs=globs) globs['sc'].stop() + if failure_count: + exit(-1) if __name__ == "__main__": diff --git a/python/pyspark/rdd.py b/python/pyspark/rdd.py index 41ea6e6e14..fb144bc45d 100644 --- a/python/pyspark/rdd.py +++ b/python/pyspark/rdd.py @@ -748,8 +748,10 @@ def _test(): # The small batch size here ensures that we see multiple batches, # even in these small test examples: globs['sc'] = SparkContext('local[4]', 'PythonTest', batchSize=2) - doctest.testmod(globs=globs) + (failure_count, test_count) = doctest.testmod(globs=globs) globs['sc'].stop() + if failure_count: + exit(-1) if __name__ == "__main__": |