diff options
author | Gábor Lipták <gliptak@gmail.com> | 2015-10-22 15:27:11 -0700 |
---|---|---|
committer | Davies Liu <davies.liu@gmail.com> | 2015-10-22 15:27:11 -0700 |
commit | 163d53e829c166f061589cc379f61642d4c9a40f (patch) | |
tree | 72642ca829cecdaffc2c0bf8d45014b4696ef04c /python/pyspark/streaming | |
parent | 53e83a3a77cafc2ccd0764ecdb8b3ba735bc51fc (diff) | |
download | spark-163d53e829c166f061589cc379f61642d4c9a40f.tar.gz spark-163d53e829c166f061589cc379f61642d4c9a40f.tar.bz2 spark-163d53e829c166f061589cc379f61642d4c9a40f.zip |
[SPARK-7021] Add JUnit output for Python unit tests
WIP
Author: Gábor Lipták <gliptak@gmail.com>
Closes #8323 from gliptak/SPARK-7021.
Diffstat (limited to 'python/pyspark/streaming')
-rw-r--r-- | python/pyspark/streaming/tests.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/python/pyspark/streaming/tests.py b/python/pyspark/streaming/tests.py index 49634252fd..2c908daa8b 100644 --- a/python/pyspark/streaming/tests.py +++ b/python/pyspark/streaming/tests.py @@ -27,6 +27,11 @@ import struct import shutil from functools import reduce +try: + import xmlrunner +except ImportError: + xmlrunner = None + if sys.version_info[:2] <= (2, 6): try: import unittest2 as unittest @@ -1303,4 +1308,8 @@ if __name__ == "__main__": for testcase in testcases: sys.stderr.write("[Running %s]\n" % (testcase)) tests = unittest.TestLoader().loadTestsFromTestCase(testcase) - unittest.TextTestRunner(verbosity=3).run(tests) + if xmlrunner: + unittest.main(tests, verbosity=3, + testRunner=xmlrunner.XMLTestRunner(output='target/test-reports')) + else: + unittest.TextTestRunner(verbosity=3).run(tests) |