o ۾i@sddlZddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z ddl m Z mZmZddlmZddZ  d d d ZdS) N)fnmatch)joinisfilerelpathnormpathsplitext)NumbaTestProgram SerialSuitemake_tag_decorator)configcCszXt}g}t|D]%}t||}t|r"t|dr"||q tt|dr1|| |q |D]!}t t|||j }t t |tjjdd}|||q4|WStyntjtjdtdYdSw)zFind tests in 'dir'.z test_*.pyz __init__.py.r)fileN)unittest TestSuiteoslistdirrrrappendaddTestsdiscoverr_top_level_dirrrreplacepathseploadTestsFromName Exception traceback print_excsysstderrexit)loaderdirsuitefilesfrr'J/home/ubuntu/.local/lib/python3.10/site-packages/numba/testing/__init__.pyload_testsuite s&   r)Fc Cs@|durddl}|j|d}nd}t|d|||d||d}|jS)z args ---- - 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*. Nr)outputF)argvmodule defaultTest topleveldir testRunnerr! verbosity nomultiproc) xmlrunner XMLTestRunnerr result) r+r-r. xmloutputr0r1r2runnerprogr'r'r( run_tests$s r8)NNNNrF)rr functoolsrrros.pathrrrrrmainr r r numba.corer r)r8r'r'r'r(s