U
    hâËdbt  ã                   @   s@  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlZd dlmZ d dlmZ d dl
mZmZmZmZmZmZ ddlmZ d dlmZ zd dlmZ W n  ek
rè   d d	lmZ Y nX d
d„ Zedd�dd„ ƒZdd„ Z dd„ Z!G dd„ de"ƒZ#G dd„ de
j$ƒZ%G dd„ dej&ƒZ'G dd„ de
j(ƒZ)dZ*dd„ Z+dd „ Z,d!d"œd#d$„Z-dAd&d'„Z.d(d)„ Z/d*d+„ Z0G d,d-„ d-e1ƒZ2G d.d/„ d/e j3ƒZ4G d0d1„ d1ej5ƒZ6G d2d3„ d3ej&ƒZ7G d4d5„ d5ej5ƒZ8G d6d7„ d7e"ƒZ9G d8d9„ d9e"ƒZ:G d:d;„ d;e"ƒZ;d<d=„ Z<d>Z=G d?d@„ d@ej&ƒZ>dS )Bé    N)Ú	lru_cache)ÚStringIO)ÚresultÚrunnerÚsignalsÚsuiteÚloaderÚcaseé   )Ú
TestLoader)Úconfig)ÚTimeoutError)ÚEmptyc                    s   ‡ fdd„}|S )zO
    Create a decorator allowing tests to be tagged with the *known_tags*.
    c                     s0   ˆ D ]}|ˆkrt d|f ƒ‚q‡ fdd„}|S )zš
        Tag a test method with the given tags.
        Can be used in conjunction with the --tags command-line argument
        for runtests.py.
        zunknown tag: %rc                    s`   t | ƒrt| tƒs| j d¡s&tdƒ‚z
| j}W n  tk
rP   tƒ  }| _Y nX | 	ˆ ¡ | S )NZtest_z(@tag(...) should be used on test methods)
ÚcallableÚ
isinstanceÚtypeÚ__name__Ú
startswithÚ	TypeErrorÚtagsÚAttributeErrorÚsetÚupdate)ÚfuncÚs©r   © úK/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/testing/main.pyÚdecorate+   s    
ÿ

z1make_tag_decorator.<locals>.tag.<locals>.decorate©Ú
ValueError)r   Útr   ©Ú
known_tagsr   r   Útag!   s
    
zmake_tag_decorator.<locals>.tagr   )r#   r$   r   r"   r   Úmake_tag_decorator   s    r%   é€   )Úmaxsizec                 C   s   t tj t | ¡¡ƒS )zF
    Gets the mtime of the file in which a test class is defined.
    )ÚstrÚosÚpathÚgetmtimeÚinspectÚgetfile)Úclsr   r   r   Ú
_get_mtime?   s    r/   c                 C   sL   | j }t|ƒt| ƒ }ddlm} || ¡ krHdt|jƒt|jƒ|f }|S )a>  
    Return a key for sorting tests bases on mtime and test name. For CUDA
    tests, interleaving tests from different classes is dangerous as the CUDA
    context might get reset unexpectedly between methods of a class, so for
    CUDA tests the key prioritises the test module and class ahead of the
    mtime.
    r   )ÚCUDATestCasez%s.%s %s)Ú	__class__r/   r(   Znumba.cuda.testingr0   ÚmroÚ
__module__r   )Úxr.   Úkeyr0   r   r   r   Úcuda_sensitive_mtimeG   s    r6   c                    s¨   t | ƒr| S | sdd„ S z"|  d¡\‰‰ tˆƒ‰tˆ ƒ‰ W n$ tk
r^   d|  }t|ƒ‚Y nX ˆ dkrpdd„ S ˆ dk sˆˆdk sˆˆˆ kr’tdƒ‚n‡ ‡fdd	„}|S d
S )zçParses the argument string "useslice" as a shard index and number and
    returns a function that filters on those arguments. i.e. input
    useslice="1:3" leads to output something like `lambda x: zlib.crc32(x) % 3
    == 1`.
    c                 S   s   dS ©NTr   ©r4   r   r   r   Ú<lambda>b   ó    zparse_slice.<locals>.<lambda>ú:z�Expected arguments shard index and count to follow option `-j i:t`, where i is the shard number and t is the total number of shards, found '%s'r   c                 S   s   dS r7   r   r8   r   r   r   r9   n   r:   zSharding out of rangec                    s@   t | | jƒ}dt |di ƒkr dS tt |  ¡  d¡¡ƒˆ  ˆkS )NZalways_testr   Tzutf-8)ÚgetattrÚ_testMethodNameÚabsÚzlibÚcrc32ÚidÚencode)Útestr   ©ÚcountÚindexr   r   Údecider   s    zparse_slice.<locals>.decideN)r   ÚsplitÚintÚ	Exceptionr    )ÚusesliceÚmsgrG   r   rD   r   Úparse_sliceY   s(    þÿ
rM   c                   @   s    e Zd ZdZdd„ Zdd„ ZdS )Ú
TestListerz5Simply list available tests rather than running them.c                 C   s   t |ƒ| _d S ©N)rM   rK   )ÚselfrK   r   r   r   Ú__init__|   s    zTestLister.__init__c                 C   sp   t jtjddd�}t|ƒ| _tt| j| jƒƒ}| jj	t
d� |D ]}t| ¡ ƒ q@tdt| jƒt|ƒf ƒ |S )NTr
   )ZdescriptionsÚ	verbosity©r5   z%d tests found. %s selected)r   ÚTextTestResultÚsysÚstderrÚ_flatten_suiteZ
_test_listÚlistÚfilterrK   Úsortr6   ÚprintrA   Úlen)rP   rC   r   Zmasked_listr!   r   r   r   Úrun   s    
ÿ
ÿzTestLister.runN)r   r3   Ú__qualname__Ú__doc__rQ   r]   r   r   r   r   rN   z   s   rN   c                       s    e Zd ZdZ‡ fdd„Z‡  ZS )ÚSerialSuiteaA  A simple marker to make sure tests in this suite are run serially.

    Note: As the suite is going through internals of unittest,
          it may get unpacked and stuffed into a plain TestSuite.
          We need to set an attribute on the TestCase objects to
          remember they should not be run in parallel.
    c                    s<   t |tjƒs"|D ]}|  |¡ qnd|_tt| ƒ |¡ d S )NF)r   ÚunittestÚTestCaseÚaddTestÚ_numba_parallel_test_Úsuperr`   )rP   rC   r!   ©r1   r   r   rc   •   s
    zSerialSuite.addTest)r   r3   r^   r_   rc   Ú__classcell__r   r   rf   r   r`   Œ   s   r`   c                       s$   e Zd Zdd„ Z‡ fdd„Z‡  ZS )ÚBasicTestRunnerc                 K   s   t jj| f|Ž t|ƒ| _d S rO   )r   ÚTextTestRunnerrQ   rM   rK   )rP   rK   Úkwargsr   r   r   rQ   ¡   s    zBasicTestRunner.__init__c                    s:   t t| jt|ƒƒƒ}|jtd� t |¡}tt	| ƒ 
|¡S )NrS   )rX   rY   rK   rW   rZ   r6   ra   Ú	TestSuitere   rh   r]   )rP   rC   r]   Úwrappedrf   r   r   r]   ¥   s    
zBasicTestRunner.run)r   r3   r^   rQ   r]   rg   r   r   rf   r   rh       s   rh   c                       sz   e Zd ZdZdZdZdZdZdZdZ	dZ
dZdZ‡ fdd„Z‡ fdd„Zd	d
„ Z‡ fdd„Zddd„Z‡ fdd„Z‡  ZS )ÚNumbaTestPrograma3  
    A TestProgram subclass adding the following options:
    * a -R option to enable reference leak detection
    * a --profile option to enable profiling of the test run
    * a -m option for parallel execution
    * a -l option to (only) list tests

    Currently the options are only added in 3.4+.
    FNé*   c                    sH   |  dd ¡}t|ƒ|d< tj d¡ |  dd¡| _tt| ƒj||Ž d S )NÚtopleveldirZ
testLoaderz:xÚnomultiprocF)	Úpopr   rU   ÚwarnoptionsÚappendrp   re   rm   rQ   )rP   Úargsrj   ro   rf   r   r   rQ   Ä   s
    zNumbaTestProgram.__init__c              	      sà   t t| ƒ ¡ }| jd kr,|jdddddd� |jddd	d
tt ¡ dd� |jdddddd� |jddtdd� |jddtdd� |jddt	dd� |jddddd� |jddd d
td!d"d� d#d$„ }|jd%d&d'|d(d
d)d*� |S )+Nz-Rz	--refleakÚrefleakÚ
store_truezDetect reference / memory leaks)ÚdestÚactionÚhelpz-mz--multiprocessÚmultiprocessú?zParallelize tests)rw   Únargsr   Úconstry   ú-lz--listrX   zList tests without running themz--tagsr   zAComma-separated list of tags to select a subset of the test suite)rw   r   ry   z--exclude-tagsÚexclude_tagszDComma-separated list of tags to de-select a subset of the test suitez--randomÚrandom_selectz$Random proportion of tests to selectz	--profileÚprofilezProfile the test runz-jz--slicerK   ÚNonezShard the test sequencec                 S   s   | dkrt dƒ‚| S )NÚancestorzinvalid option for --gitdiffr   r8   r   r   r   Úgit_diff_strî   s    z:NumbaTestProgram._getParentArgParser.<locals>.git_diff_strz-gz	--gitdiffÚgitdiffFz˜Run tests from changes made against origin/release0.57 as identified by `git diff`. If set to "ancestor", the diff compares against the common ancestor.)rw   r   Údefaultr|   ry   )
re   rm   Ú_getParentArgParserÚ
testRunnerÚadd_argumentrI   ÚmultiprocessingÚ	cpu_countr(   Úfloat)rP   Úparserr„   rf   r   r   r‡   Î   sR    

þ
ü
þ
ÿ
ÿ
ÿþ þ þz$NumbaTestProgram._getParentArgParserc           	      C   s  d }|D ]&}||kr|d kr"|}qt d| ƒ‚q|d k	�r
| |¡}zL||krj||d   ¡ }| |¡ n$d|kr†| d¡d  ¡ }ntdƒ‚W n$ tk
r´   d}t || ƒ‚Y nX | d¡rÈt dƒ‚d|kràd}t || ƒ‚|d	d …  dd
¡}t	| ||ƒ | |¡ d S )Nzargument %s supplied repeatedlyr
   ú=Zunreachablez,%s requires at least one tag to be specifiedú-z,tag starts with '-', probably a syntax errorz1%s argument contains '=', probably a syntax erroré   Ú_)
r    rF   ÚstripÚremoverH   ÚAssertionErrorÚ
IndexErrorr   ÚreplaceÚsetattr)	rP   ÚargvZtagstrÚfoundr4   ZposnZtag_argsrL   Úattrr   r   r   Ú_handle_tagsü   s4    


zNumbaTestProgram._handle_tagsc                    sø   d|kr|  d¡ d| _tt| ƒ |¡ t| dƒr<| j ¡ sN| jf| _	|  
¡  | jr|dd„ | j d¡D ƒ}t| j|dd�| _| jrªd	d„ | j d¡D ƒ}t| j|d
d�| _| jrÄt| j| j| jƒ| _| jdk	rät| j| jdkd�| _| jdkrôd| _d S )Nr~   TrC   c                 S   s   g | ]}|  ¡ ‘qS r   ©r’   ©Ú.0r   r   r   r   Ú
<listcomp>/  s     z.NumbaTestProgram.parseArgs.<locals>.<listcomp>ú,Úinclude)Úmodec                 S   s   g | ]}|  ¡ ‘qS r   rœ   r�   r   r   r   rŸ   3  s     ÚexcludeFrƒ   ©Úuse_common_ancestorr   )r“   rX   re   rm   Ú	parseArgsÚhasattrrC   ZcountTestCasesZdefaultTestZ	testNamesZcreateTestsr   rH   Ú_choose_tagged_testsr   r€   Ú_choose_random_testsÚrandom_seedr…   Ú_choose_gitdiff_testsrR   Úbuffer)rP   r˜   r   rf   r   r   r¦   !  s0    


ÿ
þ
zNumbaTestProgram.parseArgsc                 C   s   d S rO   r   )rP   r˜   ÚLoaderr   r   r   Ú_do_discoveryE  s    zNumbaTestProgram._do_discoveryc              	      s,  ˆ j r"tˆ _ttdƒsZt d¡ n8ˆ jr6tˆ j	ƒˆ _n$ˆ jd krZt
ˆ j	ˆ jˆ jˆ jd�ˆ _ˆ jržˆ jsžˆ jdk r|d}t|ƒ‚ttjˆ jˆ j	ˆ jˆ jˆ jd�ˆ _‡‡ fdd„}ˆ j�r"tj tj tjd j¡¡d	 d
 }tjtjd�}| ¡  z| #|¡ W 5 |  ¡  t!d|f ƒ | "|¡ X n|ƒ  d S )NÚgettotalrefcountz^detecting reference leaks requires a debug build of Python, only memory leaks will be detected)rR   Úfailfastr¬   r
   zSValue specified for the number of processes to use in running the suite must be > 0c                      s   t tˆƒ ¡  d S rO   )re   rm   ÚrunTestsr   )r1   rP   r   r   Úrun_tests_reali  s    z1NumbaTestProgram.runTests.<locals>.run_tests_realÚ__main__r   z.prof)Útimerz!Writing test profile data into %r)$ru   ÚRefleakTestRunnerrˆ   r§   rU   ÚwarningsÚwarnrX   rN   rK   rh   rR   r°   r¬   rz   rp   r    ÚParallelTestRunnerr   ri   r�   r)   r*   ÚsplitextÚbasenameÚmodulesÚ__file__ÚcProfileZProfileÚtimeÚperf_counterÚenableÚdisabler[   Z
dump_statsZruncall)rP   rL   r²   ÚfilenameÚprf   ©rP   r   r±   L  sP    

ý
ûÿþþzNumbaTestProgram.runTests)N)r   r3   r^   r_   ru   r�   rz   rK   rX   r   r   r€   rª   rQ   r‡   r›   r¦   r®   r±   rg   r   r   rf   r   rm   ¯   s    

.%$
rm   )z?numba.cuda.tests.cudapy.test_libdevice.TestLibdeviceCompilationznumba.tests.test_num_threadsz!numba.tests.test_parallel_backendznumba.tests.test_svmlznumba.tests.test_ufuncsc                 C   s>   g }t | tjttfƒr0| D ]}| t|ƒ¡ qn
| | ¡ |S )z&
    Workhorse for _flatten_suite
    )r   ra   rk   rX   ÚtupleÚextendÚ_flatten_suite_innerrs   )rC   Útestsr4   r   r   r   rÇ   Š  s    
rÇ   c                 C   sr   t | ƒ}tƒ }|D ]$}tD ]}|t|ƒkr| |¡ qqt|ƒ| }dd„ }t||d�}| tt|ƒ|d�¡ |S )z6
    Expand nested suite into list of test cases.
    c                 S   s   | j t| ƒj| jfS rO   )r3   r   r   r=   r8   r   r   r   r5   §  s    z_flatten_suite.<locals>.keyrS   )rÇ   r   Ú
_GENERATEDr(   ÚaddÚsortedrÆ   rX   )rC   rÈ   Ú	generatedr!   ÚgÚnormalr5   r   r   r   rW   —  s    rW   Fr¤   c          	         sê   zddl m} W n tk
r,   tdƒ‚Y nX |dƒ‰ tj dd¡}|rVtdƒ d}nd	}ˆ j j||d
d� 	¡ }dd„ |D ƒ}g }‡ fdd„|D ƒ}t
| ƒD ]2}t|tjƒs°t‚t |j¡}||krœ| |¡ qœtdt|ƒ ƒ t |¡S )Nr   )ÚRepoz&gitpython needed for git functionalityÚ.ZnumbarÈ   zGit diff by common ancestorzorigin/release0.57...HEADzorigin/release0.57..HEADT)Z	name_onlyc                 S   s   g | ]}t j |¡‘qS r   )r)   r*   Únormpath©rž   r4   r   r   r   rŸ   ¼  s     z)_choose_gitdiff_tests.<locals>.<listcomp>c                    s   g | ]}t j ˆ j|¡‘qS r   )r)   r*   ÚjoinZworking_dirrÒ   ©Úrepor   r   rŸ   ¾  s     zGit diff identified %s tests)ÚgitrÏ   ÚImportErrorr    r)   r*   rÓ   r[   ÚdiffrH   rW   r   ra   rb   r”   r,   Úgetsourcefiler1   rs   r\   rk   )	rÈ   r¥   rÏ   r*   ÚtargetZgdiff_pathsÚselectedrC   Úfnamer   rÔ   r   r«   ®  s*    r«   r¡   c              	   C   sÀ   g }t |ƒ}t| ƒD ] }t|tjƒs(t‚t||jƒ}z
|j}W n t	k
rR   Y nX t|ddƒ}|dkr„|dk	r´||@ r´| 
|¡ q|dkr¨|dksœ||@ s´| 
|¡ qtd| ƒ‚qt |¡S )zÅ
    Select tests that are tagged/not tagged with at least one of the given tags.
    Set mode to 'include' to include the tests with tags, or 'exclude' to
    exclude the tests with the tags.
    r   Nr¡   r£   zInvalid 'mode' supplied: %s.)r   rW   r   ra   rb   r”   r<   r=   Zim_funcr   rs   r    rk   )rÈ   r   r¢   rÛ   rC   r   Z
found_tagsr   r   r   r¨   Ç  s$    
r¨   c                 C   sX   t  ¡ }| |¡ t| tjƒr&t| ƒ} | | tt	| ƒ| ƒ¡} t
| dd„ d�} t | ¡S )z7
    Choose a given proportion of tests at random.
    c                 S   s   |   ¡ S rO   ©rA   )r	   r   r   r   r9   ï  r:   z&_choose_random_tests.<locals>.<lambda>rS   )ÚrandomÚRandomÚseedr   ra   rk   rW   ÚsamplerI   r\   rË   )rÈ   Úratiorà   Zrndr   r   r   r©   æ  s    
r©   c                  C   st   t j} z
t j}W n tk
r,   dd„ }Y nX t jt jt jt jfD ]}|d k	rB| ¡  qBt  	¡  t
 ¡  | ƒ |ƒ fS )Nc                   S   s   dS )Nrn   r   r   r   r   r   r9   ü  r:   z"_refleak_cleanup.<locals>.<lambda>)rU   Úgetallocatedblocksr¯   r   ÚstdoutrV   Ú
__stdout__Ú
__stderr__ÚflushÚ_clear_type_cacheÚgcZcollect)Zfunc1Zfunc2Ústreamr   r   r   Ú_refleak_cleanupö  s    

rë   c                   @   s   e Zd ZdS )ÚReferenceLeakErrorN)r   r3   r^   r   r   r   r   rì   
  s   rì   c                   @   s   e Zd Zdd„ ZdS )ÚIntPoolc                 C   s   |S rO   r   )rP   r5   r   r   r   Ú__missing__  s    zIntPool.__missing__N)r   r3   r^   rî   r   r   r   r   rí     s   rí   c                       s,   e Zd ZdZdZdd„ Z‡ fdd„Z‡  ZS )ÚRefleakTestResulté   é   c                 C   sæ   | j  ¡  | j}| j}dg||  }dg||  }tƒ }tddƒD ]}||  qBt|ƒD ]„}t ¡ }| |¡ | 	¡ s–| j
 |j
¡ | j |j¡ t‚~tƒ \}	}
||krÒ||
|  ||| < ||	|  ||| < |	|
 }}qX||fS )Nr   i8ÿÿÿéÈ   )rê   rç   ÚrepetitionsÚwarmuprí   Úranger   Z
TestResultr]   ÚwasSuccessfulÚfailuresrÆ   Úerrorsr”   rë   )rP   rC   ZrepcountZnwarmupÚ	rc_deltasÚalloc_deltasZ	_int_poolÚiÚresZalloc_afterZrc_afterZ	rc_beforeZalloc_beforer   r   r   Ú
_huntLeaks  s,    



zRefleakTestResult._huntLeaksc              	      sò   z|   |¡\}}W n" tk
r4   |  ¡ r.t‚Y d S X dd„ }dd„ }d}|d|f|d|ffD ]z\}}}	|	|ƒr^d|||t|ƒf }
d	}zt|
ƒ‚W n tk
r°   t ¡ }Y nX | jrÌ| j	 
d
||f ¡ |  ||¡ q^|sîtt| ƒ |¡ d S )Nc                 S   s   t | ƒS rO   )Úany©Údeltasr   r   r   Úcheck_rc_deltasA  s    z5RefleakTestResult.addSuccess.<locals>.check_rc_deltasc                 S   s2   d|   d¡ t| ƒk rdS t| ƒtdƒks.dS dS )Nrð   r   T)r
   r   éÿÿÿÿF)rE   r\   r   rÿ   r   r   r   Úcheck_alloc_deltasD  s
    z8RefleakTestResult.addSuccess.<locals>.check_alloc_deltasFZ
referenceszmemory blocksz%s leaked %s %s, sum=%sTz%s = %r )rý   r”   rö   Úsumrì   rJ   rU   Úexc_infoZshowAllrê   ÚwriteZ
addFailurere   rï   Ú
addSuccess)rP   rC   rù   rú   r  r  Úfailedr   Z	item_nameÚcheckerrL   r  rf   r   r   r  8  s8    	þ   ÿzRefleakTestResult.addSuccess)r   r3   r^   rô   ró   rý   r  rg   r   r   rf   r   rï     s   rï   c                   @   s   e Zd ZeZdS )rµ   N)r   r3   r^   rï   Úresultclassr   r   r   r   rµ   b  s   rµ   c                   @   s   e Zd ZdZdd„ ZdS )ÚParallelTestResultzA
    A TestResult able to inject results from other results.
    c                 C   sv   | j  |j  ¡ ¡ | j  ¡  |  j|j7  _| j |j¡ | j |j¡ | j |j¡ | j	 |j	¡ | j
 |j
¡ dS )zI
        Add the results from the other *result* to this result.
        N)rê   r  Úgetvaluerç   ÚtestsRunr÷   rÆ   rø   ÚskippedÚexpectedFailuresÚunexpectedSuccesses)rP   r   r   r   r   Úadd_resultsk  s    
zParallelTestResult.add_resultsN)r   r3   r^   r_   r  r   r   r   r   r  f  s   r  c                   @   s&   e Zd ZdZdZdd„ Zddd„ZdS )	Ú_MinimalResultz7
    A minimal, picklable TestResult-alike object.
    )	r÷   rø   r  r  r  rê   Ú
shouldStopr  Útest_idc                 C   s
   d|_ dS )zR
        Remove any unpicklable attributes from TestCase instance *case*.
        N)Z_outcomeForDoCleanups)rP   r	   r   r   r   Ú
fixup_caseƒ  s    z_MinimalResult.fixup_caseNc                 C   sx   | j D ]}t| |t||d ƒƒ q| jD ]\}}|  |¡ q&| jD ]\}}|  |¡ q@| jD ]\}}|  |¡ qZ|| _d S rO   )Ú	__slots__r—   r<   r  r  rø   r÷   r  )rP   Zoriginal_resultr  rš   r	   r‘   r   r   r   rQ   Š  s    
z_MinimalResult.__init__)N)r   r3   r^   r_   r  r  rQ   r   r   r   r   r  y  s   r  c                   @   s    e Zd ZdZdd„ Zdd„ ZdS )Ú_FakeStringIOz:
    A trivial picklable StringIO-alike for Python 2.
    c                 C   s
   || _ d S rO   ©Ú_value)rP   Úvaluer   r   r   rQ   ›  s    z_FakeStringIO.__init__c                 C   s   | j S rO   r  rÄ   r   r   r   r  ž  s    z_FakeStringIO.getvalueN)r   r3   r^   r_   rQ   r  r   r   r   r   r  –  s   r  c                   @   s.   e Zd ZdZdd„ Zdd„ Zejdd„ ƒZdS )	Ú_MinimalRunnerzu
    A minimal picklable object able to instantiate a runner in a
    child process and run a test case with it.
    c                 C   s   || _ || _d S rO   )Ú
runner_clsÚrunner_args)rP   r  r  r   r   r   rQ   ¨  s    z_MinimalRunner.__init__c              	   C   s‚   | j }tƒ |d< | jf |Ž}| ¡ }t ¡  t |¡ |j|_|j|_|  	|¡� ||ƒ W 5 Q R X t
|j ¡ ƒ|_t|| ¡ ƒS )Nrê   )r  r   r  Z_makeResultr   ZinstallHandlerZregisterResultr°   r¬   Úcleanup_objectr  rê   r  r  rA   )rP   rC   rj   r   r   r   r   r   Ú__call__¯  s    

z_MinimalRunner.__call__c              	   c   s<   t |jƒ}z
|V  W 5 t |jƒ| }|D ]}|j|= q(X dS )zu
        A context manager which cleans up unwanted attributes on a test case
        (or any other object).
        N)r   Ú__dict__)rP   rC   Zvanilla_attrsZspurious_attrsÚnamer   r   r   r  Â  s    

z_MinimalRunner.cleanup_objectN)	r   r3   r^   r_   rQ   r  Ú
contextlibÚcontextmanagerr  r   r   r   r   r  ¢  s
   r  c                 C   sP   g }g }t |t| ƒƒ•}dd„ }|D ]"}||ƒr<| |¡ q$| |¡ q$||fS )z:
    Split test suite into parallel and serial tests.
    c                 S   s4   | j }t| |ƒ}|j|kr(|jdkr(dS t| ddƒS )NZtestFailureFrd   T)r=   r<   r   )rC   Úmethod_nameÚmethodr   r   r   Úis_parallelizable_test_caseÚ  s
    
z=_split_nonparallel_tests.<locals>.is_parallelizable_test_case)rY   rW   rs   )rC   ZslicedZptestsÚstestsZflatr&  r!   r   r   r   Ú_split_nonparallel_testsÑ  s    
r(  iX  c                       s@   e Zd ZdZeZeZdd„ Zdd„ Z	dd„ Z
‡ fdd	„Z‡  ZS )
r¸   z\
    A test runner which delegates the actual running to a pool of child
    processes.
    c                 K   s0   t jj| f|Ž || _|| _t|ƒ| _|| _d S rO   )r   ri   rQ   r  ÚnprocsrM   rK   r  )rP   r  r)  rK   rj   r   r   r   rQ   ø  s
    
zParallelTestRunner.__init__c              	      sÈ   t ˆjˆjƒ}d‰ ‡ ‡fdd„tdtˆjƒˆ ƒD ƒ}|D ]l}t ˆj¡}zPzˆ 
||||¡ W n   | ¡  ‚ Y n"X |jrŽ| ¡  W ¢ q¦n| ¡  W 5 | 	¡  X q8|jsÄtˆjƒ}| |¡ |S d S )Néd   c                    s   g | ]}ˆj ||ˆ  … ‘qS r   )Ú_ptests)rž   rû   ©Ú
chunk_sizerP   r   r   rŸ     s   ÿz1ParallelTestRunner._run_inner.<locals>.<listcomp>r   )r  r  r  rõ   r\   r+  rŠ   ZPoolr)  rÓ   Ú_run_parallel_testsÚ	terminater  Úcloser`   Ú_stestsr]   )rP   r   Úchild_runnerZsplitted_testsrÈ   Úpoolr'  r   r,  r   Ú
_run_innerÿ  s*    ÿ


zParallelTestRunner._run_innerc           
   
   C   sÔ   t dd„ |D ƒƒ}|jtd� | ||¡}z| | j¡}W nl tk
rP   Y d S  tk
r¦ } z:dd dd„ t	|ƒD ƒ¡ }	|	f|j
dd …  |_
|‚W 5 d }~X Y q*X | |¡ | |j¡ |jr*d|_d S q*d S )	Nc                 s   s   | ]}|  ¡ V  qd S rO   rÝ   )rž   r!   r   r   r   Ú	<genexpr>"  s     z9ParallelTestRunner._run_parallel_tests.<locals>.<genexpr>rS   z3Tests didn't finish before timeout (or crashed):
%sÚ c                 s   s   | ]}d | V  qdS )z- %r
Nr   )rž   Útidr   r   r   r5  -  s     r
   T)r   rZ   r6   Zimap_unorderedÚ__next__ÚtimeoutÚStopIterationr   rÓ   rË   rt   r  Údiscardr  r  )
rP   r   r3  r2  rÈ   Zremaining_idsÚitZchild_resultÚerL   r   r   r   r.  !  s$    ÿ
z&ParallelTestRunner._run_parallel_testsc                    sB   t || jƒ\| _| _tdt| jƒt| jƒf ƒ tt| ƒ | j	¡S )NzParallel: %s. Serial: %s)
r(  rK   r+  r1  r[   r\   re   r¸   r]   r4  )rP   rC   rf   r   r   r]   8  s    ÿÿzParallelTestRunner.run)r   r3   r^   r_   r  r
  Ú_TIMEOUTr9  rQ   r4  r.  r]   rg   r   r   rf   r   r¸   ï  s   "r¸   )r¡   )?Úcollectionsr"  r½   r,   ré   rŠ   r)   rÞ   rU   r¾   ra   r¶   r?   Ú	functoolsr   Úior   r   r   r   r   r   r	   r   Z
numba.corer   r   r×   ÚQueuer   r%   r/   r6   rM   ÚobjectrN   rk   r`   ri   rh   Úmainrm   rÉ   rÇ   rW   r«   r¨   r©   rë   ÚRuntimeErrorrì   Údefaultdictrí   rT   rï   rµ   r  r  r  r  r(  r>  r¸   r   r   r   r   Ú<module>   sb    #
! S	
N/