ó
    ýZh>t  ã                   ó  • S SK r S SKrS SKrS SKrS SKrS SKrS SKrS SKrS SKrS SK	r	S SK
r
S SKrS SKrS SKJr  S SKJr  S SK
JrJrJrJrJrJr  SSKJr  S SKJr   S SKJr  S
 r\" SS9S 5       rS r S r! " S S\"5      r# " S S\
RH                  5      r% " S S\RL                  5      r' " S S\
RP                  5      r)Sr*S r+S r,SS.S jr-S5S jr.S r/S  r0 " S! S"\15      r2 " S# S$\ Rf                  5      r4 " S% S&\Rj                  5      r6 " S' S(\RL                  5      r7 " S) S*\Rj                  5      r8 " S+ S,\"5      r9 " S- S.\"5      r: " S/ S0\"5      r;S1 r<S2r= " S3 S4\RL                  5      r>g! \ a
    S S	KJr   GN#f = f)6é    N)Ú	lru_cache)ÚStringIO)ÚresultÚrunnerÚsignalsÚsuiteÚloaderÚcaseé   )Ú
TestLoader)Úconfig)ÚTimeoutError)ÚEmptyc                 ó   ^ • U 4S jnU$ )zG
Create a decorator allowing tests to be tagged with the *known_tags*.
c                  óR   >^ • T  H  nUT;  d  M  [        SU< 35      e   U 4S jnU$ )zz
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: c                 ó$  >• [        U 5      (       a5  [        U [        5      (       d   U R                  R	                  S5      (       d  [        S5      e U R                  nUR                  T5        U $ ! [         a    [        5       =ol         N/f = f)NÚtest_z(@tag(...) should be used on test methods)
ÚcallableÚ
isinstanceÚtypeÚ__name__Ú
startswithÚ	TypeErrorÚtagsÚAttributeErrorÚsetÚupdate)ÚfuncÚsr   s     €Úa/home/gothic/public_html/Fooocus/fooocus_env/lib64/python3.13/site-packages/numba/testing/main.pyÚdecorateÚ1make_tag_decorator.<locals>.tag.<locals>.decorate+   sv   ø€ Ü˜T—N‘N¤j°´t×&<Ñ&<Ø—}‘}×/Ñ/°×8Ñ8ÜÐ JÓKÐKð&Ø—I‘I�ð �H‰H�TŒNØˆKøô "ó &Ü #£Ð%�–Ið&ús   ÁA2 Á2BÂB©Ú
ValueError)r   Útr!   Ú
known_tagss   `  €r    ÚtagÚmake_tag_decorator.<locals>.tag!   s1   ù€ ó ˆAØ˜
Õ"Ý ²aÐ!9Ó:Ð:ñ õ		ð ˆó    © )r&   r'   s   ` r    Úmake_tag_decoratorr+      s   ø€ õ
ð, €Jr)   é€   )Úmaxsizec                 óz   • [        [        R                  R                  [        R
                  " U 5      5      5      $ )z>
Gets the mtime of the file in which a test class is defined.
)ÚstrÚosÚpathÚgetmtimeÚinspectÚgetfile)Úclss    r    Ú
_get_mtimer6   ?   s'   € ô
 Œr�w‰w×Ñ¤§¢°Ó 4Ó5Ó6Ð6r)   c                 óä   • U R                   n[        U5      [        U 5      -   nSSKJn  X1R                  5       ;   a3  [        UR                  5      < S[        UR                  5      < SU< 3nU$ )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   )ÚCUDATestCaseÚ.Ú )Ú	__class__r6   r/   Únumba.cuda.testingr8   ÚmroÚ
__module__r   )Úxr5   Úkeyr8   s       r    Úcuda_sensitive_mtimerA   G   sQ   € ð �+‰+€CÜ
�S‹/œC ›FÑ
"€Cå/Ø—w‘w“yÓ Ü §¡Ö/´°S·\±\Ö1BÂCÐHˆà€Jr)   c                 ó.  ^^• [        U 5      (       a  U $ U (       d  S $  U R                  S5      u  mm[        T5      m[        T5      mTS:X  a  S $ TS:  d  TS:  d  TT:¼  a  [	        S5      eUU4S jnU$ ! [         a    SU -  n[	        U5      ef = f)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                 ó   • g©NTr*   ©r?   s    r    Ú<lambda>Úparse_slice.<locals>.<lambda>b   ó   € ˜r)   Ú: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                 ó   • grD   r*   rE   s    r    rF   rG   n   rH   r)   zSharding out of rangec                 óÖ   >• [        X R                  5      nS[        US0 5      ;   a  g[        [        R                  " U R                  5       R                  S5      5      5      T-  T:H  $ )NÚalways_testr   Tzutf-8)ÚgetattrÚ_testMethodNameÚabsÚzlibÚcrc32ÚidÚencode)Útestr   ÚcountÚindexs     €€r    ÚdecideÚparse_slice.<locals>.decider   sV   ø€ Ü˜4×!5Ñ!5Ó6ˆDØ¤¨¨f°bÓ 9Ó9ØÜ”t—z’z $§'¡'£)×"2Ñ"2°7Ó";Ó<Ó=ÀÑEÈÑNÐNr)   )r   ÚsplitÚintÚ	Exceptionr$   )ÚusesliceÚmsgrW   rU   rV   s      @@r    Úparse_slicer^   Y   s³   ù€ ô �×ÑØˆÞÙÐð	Ø!Ÿ™¨Ó,‰ˆ�Ü�E“
ˆÜ�E“
ˆð �ƒzÙÐØ	�‹�e˜a“i 5¨E£>ÜÐ0Ó1Ð1ö	Oð
 ˆøô! ó ð@àBJñKð 	ô ˜‹oÐðús    *A9 Á9Bc                   ó$   • \ rS rSrSrS rS rSrg)Ú
TestListeréz   z5Simply list available tests rather than running them.c                 ó$   • [        U5      U l        g ©N)r^   r\   )Úselfr\   s     r    Ú__init__ÚTestLister.__init__|   s   € Ü# HÓ-ˆ�r)   c                 ó–  • [         R                  " [        R                  SSS9n[	        U5      U l        [        [        U R                  U R
                  5      5      nU R
                  R                  [        S9  U H  n[        UR                  5       5        M     [        S[        U R
                  5      [        U5      4-  5        U$ )NTr   )ÚdescriptionsÚ	verbosity©r@   z%d tests found. %s selected)r   ÚTextTestResultÚsysÚstderrÚ_flatten_suiteÚ
_test_listÚlistÚfilterr\   ÚsortrA   ÚprintrR   Úlen)rd   rT   r   Úmasked_listr%   s        r    ÚrunÚTestLister.run   sŸ   € Ü×&Ò&¤s§z¡zÀØ12ñ4ˆä(¨Ó.ˆŒÜœ6 $§-¡-°·±ÓAÓBˆØ�‰×ÑÔ!5ÐÑ6ÛˆAÜ�!—$‘$“&ŽMñ äÐ+¬s°4·?±?Ó/CÜ/2°;Ó/?ð/Añ Aô 	Bàˆr)   )ro   r\   N)r   r>   Ú__qualname__Ú__firstlineno__Ú__doc__re   rv   Ú__static_attributes__r*   r)   r    r`   r`   z   s   † Ù?ò.õ
r)   r`   c                   ó,   ^ • \ rS rSrSrU 4S jrSrU =r$ )ÚSerialSuiteéŒ   a-  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                 ó®   >• [        U[        R                  5      (       d  U H  nU R                  U5        M     g SUl        [
        [        U ]  U5        g )NF)r   ÚunittestÚTestCaseÚaddTestÚ_numba_parallel_test_Úsuperr}   )rd   rT   r%   r;   s      €r    r‚   ÚSerialSuite.addTest•   sD   ø€ Ü˜$¤× 1Ñ 1×2Ñ2ã�Ø—‘˜Q–ò ð */ˆDÔ&Ü”+˜tÑ,¨TÕ2r)   r*   )r   r>   rx   ry   rz   r‚   r{   Ú__classcell__©r;   s   @r    r}   r}   Œ   s   ø† ñ÷3ó 3r)   r}   c                   ó.   ^ • \ rS rSrS rU 4S jrSrU =r$ )ÚBasicTestRunneré    c                 óf   • [         R                  R                  " U 40 UD6  [        U5      U l        g rc   )r   ÚTextTestRunnerre   r^   r\   )rd   r\   Úkwargss      r    re   ÚBasicTestRunner.__init__¡   s'   € Ü×Ñ×&Ò& tÑ6¨vÒ6Ü# HÓ-ˆ�r)   c                 óÌ   >• [        [        U R                  [        U5      5      5      nUR	                  [
        S9  [        R                  " U5      n[        [        U ]+  U5      $ )Nrj   )rp   rq   r\   rn   rr   rA   r€   Ú	TestSuiter„   r‰   rv   )rd   rT   rv   Úwrappedr;   s       €r    rv   ÚBasicTestRunner.run¥   sN   ø€ Ü”6˜$Ÿ-™-¬¸Ó)=Ó>Ó?ˆØ�‰Ô)ˆÑ*Ü×$Ò$ SÓ)ˆÜ”_ dÑ/°Ó8Ð8r)   )r\   )r   r>   rx   ry   re   rv   r{   r†   r‡   s   @r    r‰   r‰       s   ø† ò.÷9ó 9r)   r‰   c                   ó„   ^ • \ rS rSrSrSrSrSrSrSr	Sr
SrSrSrU 4S jrU 4S jrS rU 4S	 jrSS
 jrU 4S jrSrU =r$ )ÚNumbaTestProgramé¯   a  
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                 óÚ   >• UR                  SS 5      n[        U5      US'   [        R                  R	                  S5        UR                  SS5      U l        [        [        U ]"  " U0 UD6  g )NÚtopleveldirÚ
testLoaderz:xÚnomultiprocF)	Úpopr   rl   ÚwarnoptionsÚappendrš   r„   r”   re   )rd   Úargsr�   r˜   r;   s       €r    re   ÚNumbaTestProgram.__init__Ä   s^   ø€ Ø—j‘j °Ó5ˆÜ)¨+Ó6ˆˆ|Ñô 	�‰×Ñ˜tÔ$Ø!Ÿ:™: m°UÓ;ˆÔÜÔ Ò.°Ð?¸Ó?r)   c           
      óô  >• [         [        U ]  5       nU R                  c  UR	                  SSSSSS9  UR	                  SSS	S
[
        [        R                  " 5       SS9  UR	                  SSSSSS9  UR	                  SS[        SS9  UR	                  SS[        SS9  UR	                  SS[        SS9  UR	                  SSSSS9  UR	                  SSS S
[        S!S"S9  S# nUR	                  S$S%S&US'S
S(S)9  U$ )*Nz-Rz	--refleakÚrefleakÚ
store_truezDetect reference / memory leaks)ÚdestÚactionÚhelpz-mz--multiprocessÚmultiprocessÚ?zParallelize tests)r£   Únargsr   Úconstr¥   ú-lz--listrp   zList tests without running themz--tagsr   zAComma-separated list of tags to select a subset of the test suite)r£   r   r¥   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--slicer\   ÚNonezShard the test sequencec                 ó(   • U S:w  a  [        S5      eU $ )NÚancestorzinvalid option for --gitdiffr#   rE   s    r    Úgit_diff_strÚ:NumbaTestProgram._getParentArgParser.<locals>.git_diff_strî   s   € Ø�J‹Ü Ð!?Ó@Ð@ØˆHr)   z-gz	--gitdiffÚgitdiffFz˜Run tests from changes made against origin/release0.61 as identified by `git diff`. If set to "ancestor", the diff compares against the common ancestor.)r£   r   Údefaultr¨   r¥   )
r„   r”   Ú_getParentArgParserÚ
testRunnerÚadd_argumentrZ   ÚmultiprocessingÚ	cpu_countr/   Úfloat)rd   Úparserr±   r;   s      €r    rµ   Ú$NumbaTestProgram._getParentArgParserÎ   si  ø€ ô Ô'¨ÑBÓDˆØ�?‰?Ñ"Ø×Ñ  k¸	Ø'3Ø%Fð  ñ Hð 	×Ñ˜DÐ"2¸Ø"%Ü!$Ü"1×";Ò";Ó"=Ø!4ð	 	ñ 	6ð
 	×Ñ˜D (°Ø#/Ø!Bð 	ñ 	Dð 	×Ñ˜H¨6¼ð">ð 	ñ 	?ð 	×ÑÐ,°>Ìð">ð 	ñ 	?ð 	×Ñ˜J¨_Ä5Ø!Gð 	ñ 	Ià×Ñ˜K¨iØ#/Ø!7ð 	ñ 	9ð 	×Ñ˜D )°*ÀCÜ!$¨FØ!:ð 	ñ 	<ò	ð
 	×Ñ˜D +°IÀLØ$)°ð#Að 	ñ 	Cð ˆr)   c                 óD  • S nU H  nX$;   d  M
  Uc  UnM  [        SU-  5      e   UbÛ  UR                  U5      n X2:X  a'  XS-      R                  5       nUR                  U5        O4SU;   a#  UR	                  S5      S   R                  5       nO[        S5      eUR                  S5      (       a  [        S5      eSU;   a  Sn[        Xr-  5      eUS	S  R                  SS
5      n[        XU5        UR                  U5        g g ! [         a    Sn[        Xr-  5      ef = f)Nzargument %s supplied repeatedlyr   Ú=Ú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$   rV   ÚstripÚremoverY   ÚAssertionErrorÚ
IndexErrorr   ÚreplaceÚsetattr)	rd   ÚargvÚtagstrÚfoundr?   ÚposnÚtag_argsr]   Úattrs	            r    Ú_handle_tagsÚNumbaTestProgram._handle_tagsû   s/  € ØˆÛˆAØ�{Ø‘=Ø’Eä$Ð%FÈÑ%OÓPÐPñ ð ÑØ—:‘:˜eÓ$ˆDð/Ø“?Ø#¨1¡H™~×3Ñ3Ó5�HØ—K‘K Õ)à˜e“|Ø$)§K¡K°Ó$4°QÑ$7×$=Ñ$=Ó$?™ä,¨]Ó;Ð;ð ×"Ñ" 3×'Ñ'Ü Ð!OÓPÐPð �h‹ØI�Ü  ¡Ó.Ð.Ø˜!˜"�:×%Ñ% c¨3Ó/ˆDÜ�D Ô)Ø�K‰K˜Õð3 øô ó /àD�Ü  ¡Ó.Ð.ð/ús   »+D Á'(D ÂD ÄDc                 óÈ  >• SU;   a  UR                  S5        SU l        [        [        U ]  U5        [        U S5      (       a  U R                  R                  5       (       d"  U R                  4U l	        U R                  5         U R                  (       aR  U R                  R                  S5       Vs/ s H  o"R                  5       PM     nn[        U R                  USS9U l        U R                  (       aR  U R                  R                  S5       Vs/ s H  o"R                  5       PM     nn[        U R                  USS9U l        U R                   (       a0  [#        U R                  U R                   U R$                  5      U l        U R&                  SLa&  [)        U R                  U R&                  S	:H  S
9U l        U R*                  S::  a  SU l        g g s  snf s  snf )Nrª   TrT   Ú,Úinclude)ÚmodeÚexcludeFr°   ©Úuse_common_ancestorr   )rÄ   rp   r„   r”   Ú	parseArgsÚhasattrrT   ÚcountTestCasesÚdefaultTestÚ	testNamesÚcreateTestsr   rY   rÃ   Ú_choose_tagged_testsr«   r¬   Ú_choose_random_testsÚrandom_seedr³   Ú_choose_gitdiff_testsri   Úbuffer)rd   rÉ   r   r   r;   s       €r    rØ   ÚNumbaTestProgram.parseArgs   st  ø€ Ø�4‹<Ø�K‰K˜ÔØˆDŒIäÔ Ñ/°Ô5ô �t˜V×$Ñ$¨D¯I©I×,DÑ,D×,FÑ,FØ"×.Ñ.Ð0ˆDŒNØ×ÑÔà�9�9Ø'+§y¡y§¡°sÔ';Ó<Ò'; !—G‘G–IÑ';ˆDÐ<Ü,¨T¯Y©Y¸À9ÑMˆDŒIà××Ø'+×'8Ñ'8×'>Ñ'>¸sÔ'CÓDÒ'C !—G‘G–IÑ'CˆDÐDÜ,¨T¯Y©Y¸À9ÑMˆDŒIà××Ü,¨T¯Y©Y¸×8JÑ8JØ-1×-=Ñ-=ó?ˆDŒIð �<‰<˜uÒ$Ü-Ø—	‘	Ø%)§\¡\°ZÑ%?ñˆDŒIð
 �>‰>˜QÓð ˆD�Kð ùò# =ùò Es   Â3GÄGc                 ó   • g rc   r*   )rd   rÉ   ÚLoaders      r    Ú_do_discoveryÚNumbaTestProgram._do_discoveryD  s   € ð
 	r)   c           	      óì  >^ • T R                   (       a7  [        T l        [        [        S5      (       d  [
        R                  " S5        OrT R                  (       a  [        T R                  5      T l        OFT R                  c9  [        T R                  T R                  T R                  T R                  S9T l        T R                  (       a�  T R                  (       dp  T R                  S:  a  Sn[!        U5      e[#        [$        R&                  T R                  T R                  T R                  T R                  T R                  S9T l        UU 4S jnT R(                  (       aÒ  [*        R,                  R/                  [*        R,                  R1                  [        R2                  S   R4                  5      5      S   S	-   n[6        R8                  " [:        R<                  S
9nUR?                  5          URA                  U5        URC                  5         [E        SU< 35        URG                  U5        g U" 5         g ! URC                  5         [E        SU< 35        URG                  U5        f = f)NÚgettotalrefcountz^detecting reference leaks requires a debug build of Python, only memory leaks will be detected)ri   Úfailfastrâ   r   zSValue specified for the number of processes to use in running the suite must be > 0c                  ó*   >• [         [        T]  5         g rc   )r„   r”   ÚrunTests)r;   rd   s   €€r    Úrun_tests_realÚ1NumbaTestProgram.runTests.<locals>.run_tests_realh  s   ø€ ÜÔ" DÑ2Õ4r)   Ú__main__r   z.prof)ÚtimerzWriting test profile data into )$r¡   ÚRefleakTestRunnerr¶   rÙ   rl   ÚwarningsÚwarnrp   r`   r\   r‰   ri   rê   râ   r¦   rš   r$   ÚParallelTestRunnerr   rŒ   r­   r0   r1   ÚsplitextÚbasenameÚmodulesÚ__file__ÚcProfileÚProfileÚtimeÚperf_counterÚenableÚruncallÚdisablers   Ú
dump_stats)rd   r]   rí   ÚfilenameÚpr;   s   `    €r    rì   ÚNumbaTestProgram.runTestsK  s¹  ù€ Ø�<�<Ü/ˆDŒOäœ3Ð 2×3Ñ3Ü—’ð Nô Oøð �Y�YÜ(¨¯©Ó7ˆD�Oà�_‰_Ñ$Ü-¨d¯m©mØ8<¿¹Ø7;·}±}Ø59·[±[ñBˆDŒOð
 ×× T×%5×%5Ø× Ñ  1Ó$ð4�ä  “oÐ%Ü0´×1FÑ1FØ15×1BÑ1BØ15·±Ø;?¿>¹>Ø:>¿-¹-Ø8<¿¹ñEˆDŒOö	5ð �<�<Ü—w‘w×'Ñ'Ü—‘× Ñ ¤§¡¨ZÑ!8×!AÑ!AÓBóàñàñˆHô × Ò ¤t×'8Ñ'8Ñ9ˆAØ�H‰HŒJð'Ø—	‘	˜.Ô)à—	‘	”ÝºXÐGÔHØ—‘˜XÕ&áÕøð	 —	‘	”ÝºXÐGÔHØ—‘˜XÕ&ús   Ç7I É2I3)râ   rp   rš   rT   rÜ   r¶   rc   )r   r>   rx   ry   rz   r¡   r­   r¦   r\   rp   r   r«   r¬   rà   re   rµ   rÏ   rØ   ræ   rì   r{   r†   r‡   s   @r    r”   r”   ¯   s^   ø† ñð €GØ€GØ€LØ€HØ€DØ€DØ€LØ€MØ€Kõ@õ+òZ"õJ"ôH÷-ó -r)   r”   )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                 óÊ   • / n[        U [        R                  [        [        45      (       a%  U  H  nUR                  [        U5      5        M     U$ UR                  U 5        U$ )z
Workhorse for _flatten_suite
)r   r€   r�   rp   ÚtupleÚextendÚ_flatten_suite_innerr�   )rT   Útestsr?   s      r    r  r  ‰  sY   € ð €EÜ�$œ×+Ñ+¬T´5Ð9×:Ñ:ÛˆAØ�L‰LÔ-¨aÓ0Ö1ñ ð €Lð 	�‰�TÔØ€Lr)   c                 ó  • [        U 5      n[        5       nU H2  n[         H%  nU[        U5      ;   d  M  UR	                  U5        M'     M4     [        U5      U-
  nS n[        XVS9nUR                  [        [        U5      US95        U$ )z.
Expand nested suite into list of test cases.
c                 óZ   • U R                   [        U 5      R                  U R                  4$ rc   )r>   r   r   rN   rE   s    r    r@   Ú_flatten_suite.<locals>.key¦  s#   € Ø�|‰|œT !›W×-Ñ-¨q×/@Ñ/@Ð@Ð@r)   rj   )r  r   Ú
_GENERATEDr/   ÚaddÚsortedr  rp   )rT   r  Ú	generatedr%   ÚgÚnormalr@   s          r    rn   rn   –  s~   € ô ! Ó&€Eô “€IÛˆßˆAØ”C˜“F�{Ø—‘˜aÖ ó ñ ô �‹Z˜)Ñ#€FòAä�6Ñ#€EØ	‡L�L”œ˜Y›¨SÑ1Ô2Ø€Lr)   FrÖ   c                ó*  •  SSK Jn  U" S5      n[        R
                  R                  SS5      nU(       a  [        S5        SnOS	nUR                   R                  XTS
S9R                  5       nU Vs/ s H"  n[        R
                  R                  U5      PM$     nn/ nU Vs/ s H-  n[        R
                  R                  UR                  U5      PM/     nn[        U 5       H\  n	[        U	[        R                  5      (       d   e[         R"                  " U	R$                  5      n
X¦;   d  MK  UR'                  U	5        M^     [        S[)        U5      -  5        [        R*                  " U5      $ ! [         a    [        S5      ef = fs  snf s  snf )Nr   )ÚRepoz&gitpython needed for git functionalityr9   Únumbar  zGit diff by common ancestorzorigin/release0.61...HEADzorigin/release0.61..HEADT)Ú	name_onlyzGit diff identified %s tests)Úgitr  ÚImportErrorr$   r0   r1   Újoinrs   ÚdiffrY   ÚnormpathÚworking_dirrn   r   r€   r�   r3   Úgetsourcefiler;   r�   rt   r�   )r  r×   r  Úrepor1   ÚtargetÚgdiff_pathsr?   ÚselectedrT   Úfnames              r    rá   rá   ­  sI  € ðCÝñ �‹9€DÜ�7‰7�<‰<˜ Ó)€DÞÜÐ+Ô-Ø,‰à+ˆØ—(‘(—-‘- ¸�-Ð=×CÑCÓE€Ká0;Ó<²¨1”2—7‘7×#Ñ# AÖ&±€KÐ<Ø€HÙ>IÓJºk¸”2—7‘7—<‘< × 0Ñ 0°!Ö4¹k€KÐJÜ˜uÖ%ˆÜ˜$¤× 1Ñ 1×2Ñ2Ð2Ð2Ü×%Ò% d§n¡nÓ5ˆØÕØ�O‰O˜DÖ!ñ	 &ô
 
Ð
(¬3¨x«=Ñ
8Ô9Ü×Ò˜hÓ'Ð'øô) ó CÜÐAÓBÐBðCüò =ùâJs   ‚E2 Á3)FÂ$4FÅ2Fc                 óä  • / n[        U5      n[        U 5       H¯  n[        U[        R                  5      (       d   e[        XDR                  5      n UR                  n[        USS5      nUS:X  a#  Ub  Xa-  (       a  UR                  U5        Mx  Mz  M|  US:X  a!  Ub	  Xa-  (       d  UR                  U5        M¡  M£  [        SU-  5      e   [        R                  " U5      $ ! [         a     N�f = f)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   rn   r   r€   r�   rM   rN   Úim_funcr   r�   r$   r�   )r  r   rÔ   r   rT   r   Ú
found_tagss          r    rÞ   rÞ   Æ  sí   € ð €HÜˆt‹9€DÜ˜uÖ%ˆÜ˜$¤× 1Ñ 1×2Ñ2Ð2Ð2Ü�t×1Ñ1Ó2ˆð	à—<‘<ˆDô ˜T 6¨4Ó0ˆ
à�9ÓØÑ%¨*×*;Ø—‘ Ö%ñ +<Ñ%à�YÓàÑ!¨*×*;Ø—‘ Ö%ñ +<ô Ð;¸dÑBÓCÐCñ' &ô( ×Ò˜hÓ'Ð'øô ó 	Ùð	ús   ÁC"Ã"
C/Ã.C/c                 ó2  • [         R                  " 5       nUR                  U5        [        U [        R
                  5      (       a  [        U 5      n UR                  U [        [        U 5      U-  5      5      n [        U S S9n [        R
                  " U 5      $ )z/
Choose a given proportion of tests at random.
c                 ó$   • U R                   " 5       $ rc   ©rR   )r
   s    r    rF   Ú&_choose_random_tests.<locals>.<lambda>î  s
   € ¨4¯7ª7¬9r)   rj   )ÚrandomÚRandomÚseedr   r€   r�   rn   ÚsamplerZ   rt   r  )r  Úratior+  Úrnds       r    rß   rß   å  ss   € ô �-Š-‹/€CØ‡H�HˆT„NÜ�%œ×+Ñ+×,Ñ,Ü˜uÓ%ˆØ�J‰J�uœc¤# e£*¨uÑ"4Ó5Ó6€EÜ�5Ñ4Ñ5€EÜ×Ò˜eÓ$Ð$r)   c                  óŽ  • [         R                  n  [         R                  n[         R                  [         R
                  [         R                  [         R                  4 H  nUc  M  UR                  5         M     [         R                  " 5         [        R                  " 5         U " 5       U" 5       4$ ! [         a    S n N¡f = f)Nc                  ó   • g)Nr–   r*   r*   r)   r    rF   Ú"_refleak_cleanup.<locals>.<lambda>û  s   € ˜r)   )rl   Úgetallocatedblocksré   r   Ústdoutrm   Ú
__stdout__Ú
__stderr__ÚflushÚ_clear_type_cacheÚgcÚcollect)Úfunc1Úfunc2Ústreams      r    Ú_refleak_cleanupr=  õ  sŽ   € ä×"Ñ"€EðÜ×$Ñ$ˆô —:‘:œsŸz™z¬3¯>©>¼3¿>¹>ÓJˆØÓØ�L‰LŽNñ Kô ×ÒÔä‡J‚J„LÙ‹7‘E“GÐÐøô ó ÙŠðús   ’B4 Â4CÃCc                   ó   • \ rS rSrSrg)ÚReferenceLeakErrori	  r*   N)r   r>   rx   ry   r{   r*   r)   r    r?  r?  	  s   † Úr)   r?  c                   ó   • \ rS rSrS rSrg)ÚIntPooli  c                 ó   • U$ rc   r*   )rd   r@   s     r    Ú__missing__ÚIntPool.__missing__  s   € Øˆ
r)   r*   N)r   r>   rx   ry   rC  r{   r*   r)   r    rA  rA    s   † õr)   rA  c                   ó6   ^ • \ rS rSrSrSrS rU 4S jrSrU =r	$ )ÚRefleakTestResulti  é   é   c                 ó`  • U R                   R                  5         U R                  nU R                  nS/X#-
  -  nS/X#-
  -  n[	        5       n[        SS5       H  nXg     M	     [        U5       Hº  n[        R                  " 5       nUR                  U5        UR                  5       (       dP  U R                  R                  UR                  5        U R                  R                  UR                  5        [        eA[        5       u  pšXs:¼  a  XjW-
     XGU-
  '   XiW-
     XWU-
  '   Xšp¼M¼     XE4$ )Nr   i8ÿÿÿéÈ   )r<  r6  ÚrepetitionsÚwarmuprA  Úranger   Ú
TestResultrv   ÚwasSuccessfulÚfailuresr  ÚerrorsrÅ   r=  )rd   rT   ÚrepcountÚnwarmupÚ	rc_deltasÚalloc_deltasÚ	_int_poolÚiÚresÚalloc_afterÚrc_afterÚ	rc_beforeÚalloc_befores                r    Ú
_huntLeaksÚRefleakTestResult._huntLeaks  s  € Ø�‰×ÑÔà×#Ñ#ˆØ—+‘+ˆØ�C˜8Ñ-Ñ.ˆ	Ø�s˜hÑ0Ñ1ˆô “Iˆ	Ü�t˜SÖ!ˆAØŒLñ "ô �x–ˆAä×#Ò#Ó%ˆCØ�H‰H�SŒMð ×$Ñ$×&Ñ&Ø—‘×$Ñ$ S§\¡\Ô2Ø—‘×"Ñ" 3§:¡:Ô.Ü$Ð$ØÜ$4Ó$6Ñ!ˆKØ‹|Ø)2¸iÑ3GÑ)H�	˜g™+Ñ&Ø,5ÀLÑ6PÑ,Q� ™[Ñ)Ø&1š)ñ !ð  Ð&Ð&r)   c           
      ó,  >•  U R                  U5      u  p#S nS nSnUSU4USU44 H;  u  pxn	U	" U5      (       d  M  U< SU< SU< S[        U5      < 3n
S	n [	        U
5      e   U(       d  [        [        U ];  U5        g g ! [         a    U R                  5       (       a   e g f = f! [
         a    [        R                  " 5       n Of = fU R                  (       a#  U R                  R                  U< S
U< S35        U R                  X5        Mî  )Nc                 ó   • [        U 5      $ rc   )Úany©Údeltass    r    Úcheck_rc_deltasÚ5RefleakTestResult.addSuccess.<locals>.check_rc_deltas@  s   € Ü�v“;Ðr)   c                 óz   • SU R                  S5      -  [        U 5      :  a  g[        U 5      [        S5      ::  d  gg)NrG  r   T)r   r   éÿÿÿÿF)rU   rt   r   rb  s    r    Úcheck_alloc_deltasÚ8RefleakTestResult.addSuccess.<locals>.check_alloc_deltasC  s4   € à�6—<‘< “?Ñ"¤S¨£[Ó0Øä�v“;¤# j£/Ó1ØØr)   FÚ
referenceszmemory blocksz leaked r:   z, sum=Tz = )r]  rÅ   rO  Úsumr?  r[   rl   Úexc_infoÚshowAllr<  ÚwriteÚ
addFailurer„   rF  Ú
addSuccess)rd   rT   rT  rU  rd  rh  Úfailedrc  Ú	item_nameÚcheckerr]   rl  r;   s               €r    rp  ÚRefleakTestResult.addSuccess7  s  ø€ ð	Ø&*§o¡o°dÓ&;Ñ#ˆIò	ò	ð ˆð ˜ oÐ6Ø˜?Ð,>Ð?ó+AÑ&ˆF˜wñ �v�‹ã›&£)¬S°­[ð:�à�ð.Ü,¨SÓ1Ð1ñ+Aö ÜÔ# TÑ5°dÕ;ð øôG ó 	à×)Ñ)×+Ñ+Ð+Ð+Ùð	ûô: !ó .Ü"Ÿ|š|›~’Hð.úà—<—<Ø—K‘K×%Ñ%³I»vÐ&FÔGØ—‘ Ö/s#   ƒB ÁB*Â!B'Â&B'Â*CÃCr*   )
r   r>   rx   ry   rL  rK  r]  rp  r{   r†   r‡   s   @r    rF  rF    s   ø† à€FØ€Kò'÷>'<ó '<r)   rF  c                   ó   • \ rS rSr\rSrg)rñ   ia  r*   N)r   r>   rx   ry   rF  Úresultclassr{   r*   r)   r    rñ   rñ   a  s   † Ø#ƒKr)   rñ   c                   ó   • \ rS rSrSrS rSrg)ÚParallelTestResultie  z9
A TestResult able to inject results from other results.
c                 óN  • U R                   R                  UR                   R                  5       5        U R                   R                  5         U =R                  UR                  -  sl        U R
                  R                  UR
                  5        U R                  R                  UR                  5        U R                  R                  UR                  5        U R                  R                  UR                  5        U R                  R                  UR                  5        g)z9
Add the results from the other *result* to this result.
N)r<  rn  Úgetvaluer6  ÚtestsRunrP  r  rQ  ÚskippedÚexpectedFailuresÚunexpectedSuccesses)rd   r   s     r    Úadd_resultsÚParallelTestResult.add_resultsj  s»   € ð 	�‰×Ñ˜&Ÿ-™-×0Ñ0Ó2Ô3Ø�‰×ÑÔØ�Š˜Ÿ™Ñ(�Ø�‰×Ñ˜VŸ_™_Ô-Ø�‰×Ñ˜6Ÿ=™=Ô)Ø�‰×Ñ˜FŸN™NÔ+Ø×Ñ×$Ñ$ V×%<Ñ%<Ô=Ø× Ñ ×'Ñ'¨×(BÑ(BÕCr)   r*   N)r   r>   rx   ry   rz   r  r{   r*   r)   r    rx  rx  e  s   † ñõDr)   rx  c                   ó,   • \ rS rSrSrSrS rSS jrSrg)	Ú_MinimalResultix  z/
A minimal, picklable TestResult-alike object.
)	rP  rQ  r|  r}  r~  r<  Ú
shouldStopr{  Útest_idc                 ó   • SUl         g)zB
Remove any unpicklable attributes from TestCase instance *case*.
N)Ú_outcomeForDoCleanups)rd   r
   s     r    Ú
fixup_caseÚ_MinimalResult.fixup_case‚  s   € ð
 &*ˆÕ"r)   Nc           
      óF  • U R                    H  n[        X[        XS 5      5        M     U R                   H  u  pEU R	                  U5        M     U R
                   H  u  pEU R	                  U5        M     U R                   H  u  pEU R	                  U5        M     X l        g rc   )Ú	__slots__rÈ   rM   r}  r‡  rQ  rP  r„  )rd   Úoriginal_resultr„  rÎ   r
   rÂ   s         r    re   Ú_MinimalResult.__init__‰  s|   € Ø—N”NˆDÜ�D¤¨¸tÓ DÖEñ #à×,Ô,‰GˆDØ�O‰O˜DÖ!ñ -à—{”{‰GˆDØ�O‰O˜DÖ!ñ #à—}”}‰GˆDØ�O‰O˜DÖ!ñ %à�r)   )r„  rc   )	r   r>   rx   ry   rz   rŠ  r‡  re   r{   r*   r)   r    r‚  r‚  x  s   † ñð€Iò
*÷	r)   r‚  c                   ó$   • \ rS rSrSrS rS rSrg)Ú_FakeStringIOi•  z2
A trivial picklable StringIO-alike for Python 2.
c                 ó   • Xl         g rc   ©Ú_value)rd   Úvalues     r    re   Ú_FakeStringIO.__init__š  s   € Ø�r)   c                 ó   • U R                   $ rc   r�  )rd   s    r    rz  Ú_FakeStringIO.getvalue�  s   € Ø�{‰{Ðr)   r�  N)r   r>   rx   ry   rz   re   rz  r{   r*   r)   r    rŽ  rŽ  •  s   † ñòõr)   rŽ  c                   óH   • \ rS rSrSrS rS r\R                  S 5       r	Sr
g)Ú_MinimalRunneri¡  zi
A minimal picklable object able to instantiate a runner in a
child process and run a test case with it.
c                 ó   • Xl         X l        g rc   )Ú
runner_clsÚrunner_args)rd   r™  rš  s      r    re   Ú_MinimalRunner.__init__§  s   € Ø$ŒØ&Õr)   c                 óü  • U R                   n[        5       US'   U R                  " S0 UD6nUR                  " 5       n[        R
                  " 5         [        R                  " U5        UR                  Ul        UR                  Ul        U R                  U5         U" U5        S S S 5        [        UR                  R                  5       5      Ul        [        XAR                  5       5      $ ! , (       d  f       NO= f)Nr<  r*   )rš  r   r™  Ú_makeResultr   ÚinstallHandlerÚregisterResultrê   râ   Úcleanup_objectrŽ  r<  rz  r‚  rR   )rd   rT   r�   r   r   s        r    Ú__call__Ú_MinimalRunner.__call__®  s¹   € à×!Ñ!ˆô $›:ˆˆxÑØ—’Ñ* 6Ñ*ˆØ×#Ò#Ó%ˆä×ÒÔ Ü×Ò˜vÔ&Ø Ÿ/™/ˆŒØŸ™ˆŒØ× Ñ  Õ&Ù�ŒL÷ 'ô & f§m¡m×&<Ñ&<Ó&>Ó?ˆŒÜ˜f§g¡g£iÓ0Ð0÷	 'Õ&ús   Â	C-Ã-
C;c              #   ó  #   • [        UR                  5      n Uv •  [        UR                  5      U-
  nU H  nUR                  U	 M     g! [        UR                  5      U-
  nU H  nUR                  U	 M     f = f7f)z]
A context manager which cleans up unwanted attributes on a test case
(or any other object).
N)r   Ú__dict__)rd   rT   Úvanilla_attrsÚspurious_attrsÚnames        r    r   Ú_MinimalRunner.cleanup_objectÁ  st   é € ô ˜DŸM™MÓ*ˆð	(ØŠJä  §¡Ó/°-Ñ?ˆNÛ&�Ø—M‘M $Ò'ò 'øô ! §¡Ó/°-Ñ?ˆNÛ&�Ø—M‘M $Ò'ò 'üs   ‚A?™A �/A?Á0A<Á<A?)rš  r™  N)r   r>   rx   ry   rz   re   r¡  Ú
contextlibÚcontextmanagerr   r{   r*   r)   r    r—  r—  ¡  s+   † ñò
'ò1ð& ×Ññ(ó ó(r)   r—  c                 ó¸   • / n/ n/ [        U[        U 5      5      QnS nU H4  nU" U5      (       a  UR                  U5        M#  UR                  U5        M6     X#4$ )z2
Split test suite into parallel and serial tests.
c                 óŒ   • U R                   n[        X5      nUR                  U:w  a  UR                  S:X  a  g[        U SS5      $ )NÚtestFailureFrƒ   T)rN   rM   r   )rT   Úmethod_nameÚmethods      r    Úis_parallelizable_test_caseÚ=_split_nonparallel_tests.<locals>.is_parallelizable_test_caseÙ  sC   € ð ×*Ñ*ˆÜ˜Ó+ˆØ�?‰?˜kÓ)¨f¯o©oÀÓ.NØä�tÐ4°dÓ;Ð;r)   )rq   rn   r�   )rT   ÚslicedÚptestsÚstestsÚflatr°  r%   s          r    Ú_split_nonparallel_testsr¶  Ð  sb   € ð €FØ€Fà2ŒV�FœN¨4Ó0Ó1Ð2€Dò<ó ˆÙ& q×)Ñ)Ø�M‰M˜!Öà�M‰M˜!Öñ	 ð ˆ>Ðr)   i°  c                   óF   ^ • \ rS rSrSr\r\rS r	S r
S rU 4S jrSrU =r$ )rô   iî  zP
A test runner which delegates the actual running to a pool of child
processes.
c                 óŠ   • [         R                  R                  " U 40 UD6  Xl        X l        [        U5      U l        X@l        g rc   )r   rŒ   re   r™  Únprocsr^   r\   rš  )rd   r™  r¹  r\   r�   s        r    re   ÚParallelTestRunner.__init__÷  s7   € Ü×Ñ×&Ò& tÑ6¨vÒ6Ø$ŒØŒÜ# HÓ-ˆŒØ!Õr)   c                 óº  • [        U R                  U R                  5      nSn[        S[	        U R
                  5      U5       Vs/ s H  nU R
                  XDU-    PM     nnU H‹  n[        R                  " U R                  5      n U R                  XX&5        UR                  (       a#  UR                  5          UR                  5           O%UR                  5          UR                  5         M�     UR                  (       d(  [        U R                  5      nUR!                  U5        U$ g s  snf !   UR                  5         e = f! UR                  5         f = f)Néd   r   )r—  r™  rš  rM  rt   Ú_ptestsr¸   ÚPoolr¹  Ú_run_parallel_testsrƒ  Ú	terminater  Úcloser}   Ú_stestsrv   )	rd   r   Úchild_runnerÚ
chunk_sizerW  Úsplitted_testsr  Úpoolr´  s	            r    Ú
_run_innerÚParallelTestRunner._run_innerþ  s%  € ô & d§o¡o°t×7GÑ7GÓHˆð ˆ
ä#(¨¬C°·±Ó,=¸zÔ#JóLÚ#J˜að Ÿ,™, q¨Z©Ó8Ù#Jð 	ð Ló $ˆEÜ"×'Ò'¨¯©Ó4ˆDðØ×(Ñ(¨°|ÔKð ×$×$Ø—N‘NÔ$Øð
 —	‘	–ð —J‘J•Lð —	‘	–ñ% $ð& × × Ü  §¡Ó.ˆFØ�J‰J�vÔØˆMð !ùò-Løðà—‘Ô àûð —	‘	•ús*   ÁD-ÂD2Â!EÃEÄ2EÅEÅEc                 óð  • [        S U 5       5      nUR                  [        S9  UR                  X45      n  UR	                  U R
                  5      nUR                  " U5        UR                  UR                  5        UR                  (       a  SUl	        g  Me  ! [         a     g [         aC  nSSR                  S [        U5       5       5      -  n	U	4UR                  SS  -   Ul        UeS nAff = f)Nc              3   ó@   #   • U  H  oR                  5       v •  M     g 7frc   r'  )Ú.0r%   s     r    Ú	<genexpr>Ú9ParallelTestRunner._run_parallel_tests.<locals>.<genexpr>!  s   é € Ð2ªE qŸD™DŸF˜FªEùs   ‚rj   Tz3Tests didn't finish before timeout (or crashed):
%sÚ c              3   ó,   #   • U  H
  nS U-  v •  M     g7f)z- %r
Nr*   )rË  Útids     r    rÌ  rÍ  ,  s   é € Ð QÒ;P°C ¨C¦Ò;Pùs   ‚r   )r   rr   rA   Úimap_unorderedÚ__next__Útimeoutr  Údiscardr„  rƒ  ÚStopIterationr   r  r  rž   )
rd   r   rÆ  rÃ  r  Úremaining_idsÚitÚchild_resultÚer]   s
             r    r¿  Ú&ParallelTestRunner._run_parallel_tests   së   € ÜÑ2©EÓ2Ó2ˆØ�
‰
Ô+ˆ
Ñ,Ø× Ñ  Ó5ˆØðØ!Ÿ{™{¨4¯<©<Ó8�ð ×"Ò" <Ô0Ø×%Ñ% l×&:Ñ&:Ô;Ø×*×*Ø(,�FÔ%Øð +ñ øô !ó ÙÜó àMØŸ™Ñ Q¼6À-Ô;PÓ QÓQñR�ð ˜ !§&¡&¨¨ *Ñ,�”Ø�ûðús   ¹B Â
C5Â)	C5Â2>C0Ã0C5c                 óö   >• [        UU R                  5      u  U l        U l        [	        S[        U R                  5      < S[        U R                  5      < 35        [        [        U ]#  U R                  5      $ )Nz
Parallel: z
. Serial: )
r¶  r\   r½  rÂ  rs   rt   r„   rô   rv   rÇ  )rd   rT   r;   s     €r    rv   ÚParallelTestRunner.run7  s^   ø€ Ü%=¸dØ>B¿m¹mó&MÑ"ˆŒ�d”lå¬C°·±Ö,=Ü,/°·±Õ,=ð?ô 	@ô Ô'¨Ñ2°4·?±?ÓCÐCr)   )r½  rÂ  r¹  rš  r™  r\   )r   r>   rx   ry   rz   rx  rv  Ú_TIMEOUTrÓ  re   rÇ  r¿  rv   r{   r†   r‡   s   @r    rô   rô   î  s0   ø† ñð
 %€KØ€Gò"ò òD÷.Dó Dr)   rô   )rÓ   )?Úcollectionsr©  rù   r3   r8  r¸   r0   r)  rl   rû   r€   rò   rP   Ú	functoolsr   Úior   r   r   r   r   r	   r
   r   Ú
numba.corer   r   r  ÚQueuer   r+   r6   rA   r^   Úobjectr`   r�   r}   rŒ   r‰   Úmainr”   r  r  rn   rá   rÞ   rß   r=  ÚRuntimeErrorr?  ÚdefaultdictrA  rk   rF  rñ   rx  r‚  rŽ  r—  r¶  rÝ  rô   r*   r)   r    Ú<module>rç     s•  ðÛ Û Û Û Û 	Û Û 	Û Û 
Û Û Û Û å Ý ß A× Aå Ý ð,Ý,ò
ñF �3Ññ7ó ð7òò$ôB�ô ô$3�(×$Ñ$ô 3ô(	9�f×+Ñ+ô 	9ôI�x—}‘}ô Iðb€
ò
òð. 9>õ (ô2(ò>
%ò ô(	˜ô 	ôˆk×%Ñ%ô ôK<˜×-Ñ-ô K<ô\$˜×-Ñ-ô $ôD˜×.Ñ.ô Dô&�Vô ô:	�Fô 	ô,(�Vô ,(ò^ð8 €ôPD˜×.Ñ.õ PDøðm ó ,ß+Ð+ð,ús   ÁE8 Å8FÆF