
import sys

import unittest
from unittest import TestCase

try:
    import faulthandler
except ImportError:
    pass
else:
    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 . 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)

