diff options
Diffstat (limited to 'python/pyspark/heapq3.py')
-rw-r--r-- | python/pyspark/heapq3.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/python/pyspark/heapq3.py b/python/pyspark/heapq3.py index 4ef2afe035..b27e91a4cc 100644 --- a/python/pyspark/heapq3.py +++ b/python/pyspark/heapq3.py @@ -883,6 +883,7 @@ except ImportError: if __name__ == "__main__": - import doctest - print(doctest.testmod()) + (failure_count, test_count) = doctest.testmod() + if failure_count: + exit(-1) |