58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
import sys
|
|
|
|
import unittest
|
|
from unittest import TestCase
|
|
|
|
import faulthandler
|
|
|
|
|
|
try:
|
|
# May fail in IPython Notebook with UnsupportedOperation
|
|
faulthandler.enable()
|
|
except BaseException as e:
|
|
msg = "Failed to enable faulthandler due to:\n{err}"
|
|
warnings.warn(msg.format(err=e))
|
|
|
|
|
|
# Try to inject Numba's unittest customizations.
|
|
from llvmlite.tests import customize
|
|
|
|
|
|
def discover_tests(startdir):
|
|
"""Discover test under a directory
|
|
"""
|
|
# Avoid importing unittest
|
|
loader = unittest.TestLoader()
|
|
suite = loader.discover(startdir)
|
|
return suite
|
|
|
|
|
|
def run_tests(suite=None, xmloutput=None, verbosity=1):
|
|
"""
|
|
args
|
|
----
|
|
- suite [TestSuite]
|
|
A suite of all tests to run
|
|
- xmloutput [str or None]
|
|
Path of XML output directory (optional)
|
|
- verbosity [int]
|
|
Verbosity level of tests output
|
|
|
|
Returns the TestResult object after running the test *suite*.
|
|
"""
|
|
if suite is None:
|
|
suite = discover_tests("llvmlite.tests")
|
|
if xmloutput is not None:
|
|
import xmlrunner
|
|
runner = xmlrunner.XMLTestRunner(output=xmloutput)
|
|
else:
|
|
runner = None
|
|
prog = unittest.main(suite=suite, testRunner=runner, exit=False,
|
|
verbosity=verbosity)
|
|
return prog.result
|
|
|
|
|
|
def main():
|
|
res = run_tests()
|
|
sys.exit(0 if res.wasSuccessful() else 1)
|