
    shl                        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rS SKJr  S SKJr  S SKJrJ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$J%r%J&r&J'r'   S SK(r(S	 r*\*" 5       r+S
r,\	RZ                  " \+\,5      r.S r/S r0S r1S r2S r3S r4 " S S\\ 5      r5S r6 " S S\ 5      r7 " S S\75      r8 " S S\85      r9 " S S\85      r: " S S\85      r;\& " S S\85      5       r< " S  S!\85      r= " S" S#\75      r>\% " S$ S%\	R@                  5      5       r?S& r@ " S' S(\	R@                  5      rA " S) S*\75      rB\CS+:X  a  \	R                  " 5         gg! \) a    Sr( Nf = f),    N)Path)njit)codegen)_UserWideCacheLocator_ZipCacheLocator)NumbaWarning)parfor)	SerialMixinTestCasecapture_cache_logimport_dynamicoverride_configrun_in_new_process_cachingskip_if_typeguardskip_parfors_unsupportedtemp_directoryc                  D   [        S5      n [        R                  R                  U S5      n[        R                  " U5        [        [        R                  R                  US5      S5       nUR                  S5        S S S 5        [        R                  " US5         [        [        R                  R                  US5      S5       nUR                  S5        S S S 5         [        R                  " US	5        [        R                  " U5        g
! , (       d  f       N= f! , (       d  f       NN= f! [         a0     [        R                  " US	5        [        R                  " U5        gf = f! [        R                  " US	5        [        R                  " U5        f = f)N
test_cachewritable_testwrite_okwtcheck1@  write_forbiddencheck2i  FT)r   ospathjoinmkdiropenwritechmodshutilrmtreePermissionError)tempdirtest_dirfs      l/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/tests/test_caching.pycheck_access_is_preventabler*   )   s    \*GWo6HHHX	bggll8Z0$	71	 
8 HHXu "'',,x):;TBaGGH C  	5!h% 
8	7
 CB   
5!h 	5!hsH   ,D*D3 D"D3 
D"
D0,D3 3
E-=E0 ,E--E0 0/Fz9Cannot create a directory to which writes are preventablec                       Sn [        U 5      U 4$ )Nabcd)hash)cs    r)   constant_unicode_cacher/   M   s    A7A:    c                  n    [         n [        SS9" U 5      nU " 5       u  p#U" 5       u  pEX$:X  d   eX5:X  d   eg NTcache)r/   r   )pyfunccfuncexp_hvexp_strgot_hvgot_strs         r)   check_constant_unicode_cacher;   R   s@    #FtV$EhOFgOFr0   c                      SSS.$ )N      )ab rA   r0   r)   
dict_cacherB   [   s    r0   c                  X    [         n [        SS9" U 5      nU " 5       nU" 5       nX#:X  d   eg r2   )rB   r   r5   r6   expgots       r)   check_dict_cacherG   _   s.    FtV$E
(C
'C::r0   c               #   &   #    S H  n U v   M	     g 7f)N)r=   r>      rA   )vs    r)   generator_cacherK   g   s      s   c                  |    [         n [        SS9" U 5      n[        U " 5       5      n[        U" 5       5      nX#:X  d   eg r2   )rK   r   listrD   s       r)   check_generator_cacherN   l   s6    FtV$E
vx.C
uw-C::r0   c                   2    \ rS rSrS rS rS rS rS rSr	g)	TestCachingt   c                 R    U" 5         [        U5      nU R                  US   S5        g )Nexitcoder   )r   assertEqual)selffuncress      r)   run_testTestCaching.run_testu   s%    (.Z!,r0   c                 .    U R                  [        5        g N)rX   r;   rU   s    r)   test_constant_unicode_cache'TestCaching.test_constant_unicode_cachez   s    23r0   c                 .    U R                  [        5        g r[   )rX   rG   r\   s    r)   test_dict_cacheTestCaching.test_dict_cache}   s    &'r0   c                 .    U R                  [        5        g r[   )rX   rN   r\   s    r)   test_generator_cache TestCaching.test_generator_cache   s    +,r0   c                 X   [        U R                  R                  5      n[        R                  " 5       nUR                  5       nUR                  [        X1S4S9nUR                  5         UR                  5         UR                  5       u  pVU(       d  U R                  U5        U R                  USS5        UR                  [        X1S4S9nUR                  5         UR                  5         UR                  5       u  pVU(       d  U R                  U5        U R                  USS5        g )NFtargetargs  z-Omitted function returned an incorrect outputT)r   	__class____name__multiprocessingget_contextQueueProcessomitted_child_test_wrapperstartr   getfailrT   )rU   	cache_dirctxresult_queueprocsuccessoutputs          r)   test_omittedTestCaching.test_omitted   s    #4>>#:#:;	))+yy{{{-51  
 	

		&**, IIf;	
 {{-40  
 	

		&**, IIf;	
r0   rA   N)
rk   
__module____qualname____firstlineno__rX   r]   r`   rc   rz   __static_attributes__rA   r0   r)   rP   rP   t   s    -
4(-(
r0   rP   c                 .   [        SU5         [        SS9SS j5       n U" 5       nU(       aO  UR                  UR                  S      S:X  d   S5       eUR                  UR                  S      S:X  d   S5       eONUR                  UR                  S      S:X  d   S	5       eUR                  UR                  S      S:X  d   S
5       eSnU R                  XT45        S S S 5        g !   [
        R                  " 5       nSn N6= f! , (       d  f       g = f)N	CACHE_DIRTr3   c                     U $ r[   rA   )nums    r)   test(omitted_child_test_wrapper.<locals>.test   s    Jr0   r   r=   zCache did not hit as expectedzCache has an unexpected misszCache did not miss as expectedzCache has an unexpected hitF)ri   )r   r   _cache_hits
signatures_cache_misses	traceback
format_excput)rv   rt   second_callr   ry   rx   s         r)   rp   rp      s)   	i	0	D		 
		VF ''(:;q@ 434@))$//!*<=B 323B ))$//!*<=B 545B''(:;q@ 212@G
 	'*+1 
1	0*	))+FG/ 
1	0s#   DB-C(D(DD
Dc                   F    \ 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g)BaseCacheTest   Nc                 2   [        S5      U l        [        R                  R	                  SU R                  5        [
        R                  R                  U R                  U R                  S-   5      U l        [
        R                  R                  U R                  S5      U l	        [        R                  " U R                  U R                  5        [
        R                  " U R                  [        R                  [        R                   -  5        S U l        g )Nr   r   z.py__pycache__)r   r&   sysr   insertr   r   modnamemodfilert   r#   copyusecases_filer"   statS_IREADS_IWRITEmaxDiffr\   s    r)   setUpBaseCacheTest.setUp   s    %l34<<(ww||DLL$,,2FGdllMBD&&5
t||dmm;<r0   c                     [         R                  R                  U R                  S 5        [         R                  R                  U R                  5        g r[   )r   modulespopr   r   remover&   r\   s    r)   tearDownBaseCacheTest.tearDown   s,    d+%r0   c                 p   [         R                  R                  U R                  S 5      nUb-  UR                  /nU H  n [
        R                  " U5        M     [        U R                  5      nU R                  UR                  R                  S5      U R                  5        U$ ! [         a     Mw  f = f)Nco)r   r   r   r   
__cached__r   unlinkFileNotFoundErrorr   rT   __file__rstripr   )rU   oldcachedfnmods        r)   import_moduleBaseCacheTest.import_module   s     kkoodllD1?nn%FIIbM 
 T\\*,,T2DLLA
	 ) s   B''
B54B5c                      [         R                  " U R                  5       Vs/ s H  nUR                  S5      (       a  M  UPM     sn$ s  snf ! [         a    / s $ f = f)N)z.pycz.pyo)r   listdirrt   endswithr   )rU   r   s     r)   cache_contentsBaseCacheTest.cache_contents   sZ    	!#DNN!; :!;2;;'78 !; : : :  	I	s(   "A AAA A AAc                 \   ^  [        U 4S j[        T R                  5       5       5       5      $ )Nc              3      >#    U  HN  nU[         R                  R                  [         R                  R                  TR                  U5      5      4v   MP     g 7fr[   )r   r   getmtimer   rt   ).0r   rU   s     r)   	<genexpr>1BaseCacheTest.get_cache_mtimes.<locals>.<genexpr>   s?      =; ))"'',,t~~r*JKL;s   AA)dictsortedr   r\   s   `r)   get_cache_mtimesBaseCacheTest.get_cache_mtimes   s+     =$T%8%8%:;= = 	=r0   c                 Z    U R                  5       nU R                  [        U5      X5        g r[   )r   rT   len)rU   nr.   s      r)   check_pycacheBaseCacheTest.check_pycache   s$    !Q&r0   c                     g r[   rA   r\   s    r)   
dummy_testBaseCacheTest.dummy_test   s    r0   )rt   r   r   r&   )rk   r|   r}   r~   r   r   r   r   r   r   r   r   r   r   rA   r0   r)   r   r      s/    MG&"='r0   r   c                       \ rS rSr\R
                  R                  \5      r\R
                  R                  \S5      r
Sr0 S.S jrS	S jrSrg)
DispatcherCacheUsecasesTesti  cache_usecases.pydispatcher_caching_test_fodderenvvarsc          	         S[        U R                  U R                  S9-  n[        R                  R                  5       nUR                  U5        [        R                  " [        R                  SU/[        R                  [        R                  US9nUR                  5       u  pVUR                  S:w  a>  [        SUR                  < SUR                  5       < SUR                  5       < S	35      eg )
Nzif 1:
            import sys

            sys.path.insert(0, %(tempdir)r)
            mod = __import__(%(modname)r)
            mod.self_test()
            r&   r   -c)stdoutstderrenvr   process failed with code z: 
stdout follows
z
stderr follows

)r   r&   r   r   environr   update
subprocessPopenr   
executablePIPEcommunicate
returncodeAssertionErrordecode)rU   r   codesubp_envpopenouterrs          r)   run_in_separate_process3DispatcherCacheUsecasesTest.run_in_separate_process  s     t||T\\BC ::??$   #..$!=(2
%-/ $$&q   ##SZZ\3::<A  !r0   Nc                    UR                   nU R                  [        UR                  R	                  5       5      X$R                  5        Ub?  U R                  [        UR
                  R	                  5       5      UUR
                  5        g g r[   )statsrT   sum
cache_hitsvaluescache_misses)rU   rV   hitsmissessts        r)   
check_hits&DispatcherCacheUsecasesTest.check_hits   sg    ZZR]]1134dMMJS!7!7!9:F__. r0   rA   r[   )rk   r|   r}   r~   r   r   dirnamer   herer   r   r   r   r   r   rA   r0   r)   r   r     s;    77??8$DGGLL':;M.G13 4.r0   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S rS rS rS rS rS r\\R,                  " \R0                  S:H  S5      S 5       5       r\\R,                  " \R0                  S:H  S5      S 5       5       rS r\R,                  " \SL =(       d    \R:                  S   S:  S5      S 5       rSrg)	TestCachei(  c                 l   U R                  S5        U R                  5       nU R                  S5        UR                  nU R                  U" SS5      S5        U R                  S5        U R                  U" SS5      S5        U R                  S5        U R	                  USS5        UR
                  nU R                  U" SS5      S5        U R                  S5        U R                  U" SS5      S5        U R                  S5        U R	                  USS5        UR                  nU" UR                  S5      nU R                  [        U5      S	5        U" UR                  S5      nU R                  [        U5      S	5        U R                  S
5        U R	                  USS5        U R                  5         g )Nr   r>   rI            @      @   r=   )r>   g     E@	   )r   r   add_usecaseassertPreciseEqualr   add_objmode_usecaserecord_returnaligned_arrtuple
packed_arrr   )rU   r   r(   recs       r)   test_cachingTestCache.test_caching*  so   1  "1OO!Q+1#q	3/11a ##!Q+1#q	3/11a #c
I6"c
I611a  	$$&r0   c                    U R                  S5        U R                  5       nU R                  S5        UR                  nU R                  U" SS5      S5        U R                  S5        U R                  U" S[        R
                  " S5      5      S[        R
                  " S5      -   S-   5        U R                  S5        U R                  USS5        g )Nr   r>   rI   r   r=   )r   r   r   r   nparanger   rU   r   r(   s      r)   test_caching_nrt_pruned!TestCache.test_caching_nrt_prunedH  s    1  "1OO!Q+1!RYYq\ 2A		!4Dq4HI11a r0   c                 B   U R                  5       nU R                  UR                  SS5      S5        U R                  S5        UR                  nU R                  U" SS5      S5        U R                  S5        U R                  5       nUR                  nU R                  U" SS5      S5        U R                  S5        UR
                  nU R                  U" SS5      S5        U R                  S5        U R                  U" SS5      S5        U R                  S5        g )NrI   r>   r            @r   )r   r   innerr   outer_uncachedouterr	  s      r)   test_inner_then_outerTestCache.test_inner_then_outerU  s      "		!Q31!Q+1  "!Q+1II!Q+1#q	3/1r0   c                    U R                  5       nU R                  UR                  SS5      S5        U R                  S5        U R                  UR	                  SS5      S5        U R                  S5        U R                  5       nUR
                  nU R                  U" SS5      S5        U R                  S5        U R                  U" SS5      S5        U R                  S5        g )NrI   r>   r  r   r  r   r   )r   r   r  r   r  r  r	  s      r)   test_outer_then_innerTestCache.test_outer_then_inneri  s      "		!Q31 2 21a 8!<1  "II!Q+1#q	3/1r0   c                     U R                  5       nUR                  nU R                  U" SS5      S5        U R                  S5        g )Nr>   rI   r   r   )r   add_nocache_usecaser   r   r	  s      r)   test_no_cachingTestCache.test_no_cachingw  s?      "##!Q+1r0   c                    U R                  5       n[        R                  " SS9 n[        R                  " S[        5        UR
                  nU R                  U" S5      S5        U R                  S5        S S S 5        U R                  [        W5      S5        U R                  S[        US   R                  5      5        g ! , (       d  f       NR= f)	NTrecordalwaysr  r   r   r=   zBCannot cache compiled function "looplifted" as it uses lifted code)r   warningscatch_warningssimplefilterr   
loopliftedr   r   rT   r   assertInstrmessagerU   r   wr(   s       r)   test_loopliftedTestCache.test_looplifted~  s      "$$D1Q!!(L9A##AaD!,q! 2 	Q# /03AaDLL0A	C 21s   AC
Cc                    U R                  5       n[        R                  " SS9 n[        R                  " S[        5        UR
                  n[        R                  R                  U" 5       UR                  5        U R                  S5        S S S 5        U R                  [        W5      S5        U R                  S[        US   R                  5      5        g ! , (       d  f       NR= f)NTr  r  r   r=   zICannot cache compiled function "use_big_array" as it uses dynamic globals)r   r  r   r!  r   use_big_arrayr  testingassert_equalbiggier   rT   r   r#  r$  r%  r&  s       r)   test_big_arrayTestCache.test_big_array  s      "$$D1Q!!(L9!!AJJ##AC4q! 2 	Q# 347!4E	G 21s   A(C
C'c                    U R                  5       nUR                  UR                  UR                  4 H  n[        R
                  " SS9 n[        R                  " S[        5        U R                  U" S5      S5        U R                  S5        S S S 5        U R                  [        W5      S5        U R                  SR                  UR                  5      [        US   R                   5      5        M     g ! , (       d  f       No= f)NTr  r  g        r   r=   z#Cannot cache compiled function "{}")r   	use_c_sinuse_c_sin_nest1use_c_sin_nest2r  r   r!  r   r   r   rT   r   r#  formatrk   r$  r%  )rU   r   r(   r'  s       r)   test_ctypesTestCache.test_ctypes  s       "--!4!4c6I6IJA((5%%h=''#4""1%	 6 SVQ'MM5<<QZZHAaDLL! K55s   AC99
D	c                    U R                  5       n[        R                  " 5          [        R                  " S[        5        UR
                  nU R                  U" S5      S5        UR                  nU R                  U" S5      S5        UR                  nU R                  U" S5      S5        UR                  nU R                  U" S5      S5        U R                  S5        S S S 5        g ! , (       d  f       g = f)NerrorrI   r      
      r   )r   r  r   r!  r   closure1r   closure2closure3closure4r   r	  s      r)   test_closureTestCache.test_closure  s      "$$&!!'<8A##AaD!,A##AaD!,A##AaD"-A##AaD"-q! '&&s   B=C,,
C:c                 v   U R                  5       nUR                  nU R                  U" UR                  S5      S5        U R                  U" UR                  S5      S5        U R                  U" UR                  S5      S5        U R                  U" UR                  S5      S5        U R                  S5        g )Nr=   r;  d   r>         )r   first_class_function_usecaserT   first_class_function_mulfirst_class_function_addr   r	  s      r)   test_first_class_function#TestCache.test_first_class_function  s      ",,377;Q?377<cB377;Q?377<bA 	1r0   c                 \   U R                  5       nUR                  SS5        UR                  SS5        UR                  SS5        UR                  SS5        UR	                  SS5        UR                  UR                  S5        UR                  UR                  S5        U R                  5       nU R                  UR                  SS5        U R                  5       nU R                  X5        UR                  nU" SS5        U R                  USS5        U" SS5        U R                  USS5        UR                  nU" SS5        U R                  USS5        U R                  U R                  5       U5        U R                  5         U R                  U R                  5       U5        g )Nr>   rI   r   r  r   r=   )r   r   r   r  r  r   r  r   r   r   assertIsNotrT   r   )rU   r   mtimesmod2r(   s        r)   test_cache_reuseTestCache.test_cache_reuse  s`     "1S!1%1a 		!Q#..!,#//1-&&(A.!!##	!Q1a 	#s1a $$	!Q1a  	..0&9$$&..0&9r0   c                    U R                  5       nUR                  nU R                  U" SS5      S5        [        U R                  S5       nUR                  S5        S S S 5        U R                  5       nUR                  nU R                  U" SS5      S5        UR                  nU R                  U" SS5      S5        g ! , (       d  f       Ni= f)Nr>   rI   r   r?   z
Z = 10
   )r   r   r   r    r   r!   r   r	  s      r)   test_cache_invalidateTestCache.test_cache_invalidate  s      "OO!Q+ $,,$GGL! %   "OO!Q,##!Q, %$s   C
Cc                    U R                  5       nUR                  nU R                  U" SS5      S5        U R                  5       nUR                  nSUl        U R                  U" SS5      S5        UR	                  5         U R                  U" SS5      S5        U R                  5       nUR                  nU R                  U" SS5      S5        g )Nr>   rI   r   r;  rS  )r   r   r   Z	recompiler	  s      r)   test_recompileTestCache.test_recompile  s      "OO!Q+  "OO!Q+	!Q,   "OO!Q,r0   c                     U R                  5       nUR                  nU R                  U" S5      S5        UR                  nU R                  U" S5      S5        g )Nr>   r  r:  )r   renamed_function1r   renamed_function2r	  s      r)   test_same_namesTestCache.test_same_names  sN      "!!!a(!!!a(r0   c                 r   SSK Jn  UR                  n[        SSS5      n X1l        [        R
                  " U5      n[        R                  " X5      nU R                  U5        S[        l
        [        R                  " X5      nU R                  U[        5        X!l        [        ?
g ! X!l        [        ?
f = f)Nr=   )functionpassztests/dummy_module.pyexecT)dummy_modulera  __code__compileinspectgetfiler   from_functionassertIsNoner   frozenassertIsInstance)rU   ra  old_codecode_objsourcelocators         r)   test_frozenTestCache.test_frozen  s    *$$6#:FC	 (__X.F ,99(KGg&CJ ,99(KG!!'+@A !)
 !)
s   A:B( (B6c                    U R                  5       nUR                  nU R                  [        R                  UR
                  R                  SS9  U R                  U" SS5      S5        U R                  USS5        U R                  5       nUR                  nU R                  U" SS5      S5        U R                  USS5        U R                  S5        g)	za
With a disabled __pycache__, test there is a working fallback
(e.g. on the user-wide cache dir)
T)ignore_errorsr>   rI   r   r   r=   N)
r   r   
addCleanupr#   r$   r   
cache_pathr   r   r   )rU   r   r(   rO  s       r)   _test_pycache_fallback TestCache._test_pycache_fallback&  s    
   "OO 	qww'9'9N!Q+1a  !!#!Q+1a  	1r0   ntz3cannot easily make a directory read-only on Windowsc                    [         R                  " U R                  5      R                  n[         R                  " U R                  S5        U R                  [         R                  U R                  U5        U R                  5         g )Nr   )r   r   r&   st_moder"   ru  rw  )rU   	old_permss     r)   test_non_creatable_pycache$TestCache.test_non_creatable_pycache?  sR    
 GGDLL)11	
u%$,,	:##%r0   c                 R   [         R                  R                  U R                  S5      n[         R                  " U5        [         R
                  " U5      R                  n[         R                  " US5        U R                  [         R                  X5        U R                  5         g )Nr   r   )
r   r   r   r&   r   r   r{  r"   ru  rw  )rU   pycacher|  s      r)   test_non_writable_pycache#TestCache.test_non_writable_pycacheJ  sh    
 '',,t||];
GGG$,,	
% '5##%r0   c                 \  ^ ^^ [         R                  SS/mT/ SQ-  m [        R                  " TS/-   5      nWR                  5       R                  5       n[        R                  R                  T R                  S5      m[        TS5       nUR                  S5        S S S 5        UUU 4S	 jnU" 5         U" 5       nT R                  S
UR                  5       5        g ! [        R                   a(  nT R                  SUR                  -  5         S nANS nAff = f! , (       d  f       N= f)N-mIPythonz--quietz--quickz--no-bannerz--colors=NoColor	--version%ipython not available: return code %dzipython_cache_usecase.txtr'  }  
                import os
                import sys

                from numba import jit

                # IPython 5 does not support multiline input if stdin isn't
                # a tty (https://github.com/ipython/ipython/issues/9752)
                f = jit(cache=True)(lambda: 42)

                res = f()
                # IPython writes on stdout, so use stderr instead
                sys.stderr.write(u"cache hits = %d\n" % f.stats.cache_hits[()])

                # IPython hijacks sys.exit(), bypass it
                sys.stdout.flush()
                sys.stderr.flush()
                os._exit(res)
                c            	      B  > [        TS5       n [        R                  " TU [        R                  [        R                  SS9nUR	                  5       u  p#UR
                  S:w  a   TR                  SUR
                  X#4-  5        UsS S S 5        $ ! , (       d  f       g = fNrbT)stdinr   r   universal_newlines*   z6unexpected return code %d
-- stdout:
%s
-- stderr:
%s
r    r   r   r   r   r   rs   r  pr   r   base_cmdinputfnrU   s       r)   execute_with_input2TestCache.test_ipython.<locals>.execute_with_inputx      gt$$$XU,6OO,6OO8<> ==?<<2%II 1 "#s 89 :  %$$   A8B
Bcache hits = 1)r   r   r   check_outputCalledProcessErrorskipTestr   stripr   r   r   r   r&   r    r!   r#  )rU   verer(   r  r   r  r  s   `     @@r)   test_ipythonTestCache.test_ipythonW  s    NND)4MM	*))(k]*BCC iik  "'',,t||-HI'31GG    *	 	 "&		4[ ,, 	*MMALL) * *	*  s#   C DD2DD
D+Nr   r   zrequires ipykernel >= 6c                 4  ^ ^^ [         R                  SS/mT/ SQ-  m [        R                  " TS/-   5      nWR                  5       R                  5       nSSK	J
n  UR                  5       m[        TS5       nUR                  S	5        S S S 5        UUU 4S
 jnU" 5         U" 5       nT R                  SUR                  5       5        g ! [        R                   a(  nT R                  SUR                  -  5         S nANS nAff = f! , (       d  f       N= f)Nr  r  r  r  r  r   )compilerr'  r  c            	      B  > [        TS5       n [        R                  " TU [        R                  [        R                  SS9nUR	                  5       u  p#UR
                  S:w  a   TR                  SUR
                  X#4-  5        UsS S S 5        $ ! , (       d  f       g = fr  r  r  s       r)   r  4TestCache.test_ipykernel.<locals>.execute_with_input  r  r  r  )r   r   r   r  r  r  r   r  r   	ipykernelr  get_tmp_directoryr    r!   r#  )	rU   r  r  r  r(   r  r   r  r  s	   `      @@r)   test_ipykernelTestCache.test_ipykernel  s    
 NND)4MM	*))(k]*BCC iik  "&,,.'31GG    *	 	 "&		4] ,, 	*MMALL) * *	*  s#   C
 9D	
DDD	
DrA   ) rk   r|   r}   r~   r  r
  r  r  r  r(  r/  r6  rA  rJ  rP  rT  rY  r^  rq  rw  skip_bad_accessunittestskipIfr   namer}  r  r  r  version_infor  r   rA   r0   r)   r   r   (  s    '<!(CG$" 	:<--$).2 __RWW_JL&L & __RWW_JL&L &35j __i4'KY-C-CA-F-J.055055r0   r   c                   4   ^  \ rS rSrU 4S jrS rS rSrU =r$ )TestCacheZipi  c                 T  > [         TU ]  5         SnSnSn[        R                  R	                  U R
                  U5      n[        R                  " US5       nUR                  X!5        S S S 5        [        R                  R                  SU5        SU l        g ! , (       d  f       N6= f)NzI
from numba import jit

@jit(cache=True)
def add(x, y):
    return x + y
ztest_module.pyztest_archive.zipr'  r   test_module)superr   r   r   r   r&   zipfileZipFilewritestrr   r   r   )rU   mod_contentmod_filenamezip_filenamezip_pathzfrj   s         r)   r   TestCacheZip.setUp  s~     () 77<<l;__Xs+rKK2 , 	8$$ ,+s   B
B'c                     [         R                  R                  S5        [         R                  R                  SS 5        g )Nr   r  )r   r   r   r   r\   s    r)   r   TestCacheZip.tearDown  s#    Qt,r0   c                    SS K nUR                  SS5      nU R                  US5        U R                  UR                  SS5        U R                  UR                  S5        [        R
                  S	 [        R                  " 5         SS K nUR                  SS5      nU R                  US5        U R                  UR                  S5        g )Nr   r>   rI   r   r=   r  )r  addrT   r   r   r   	importlibinvalidate_caches)rU   r  result1result2s       r)   test_zip_cachingTestCacheZip.test_zip_caching  s     	//!Q'!$A. 	+ KK&##% //!Q'!$ 	+r0   )r   )	rk   r|   r}   r~   r   r   r  r   __classcell__rj   s   @r)   r  r    s    %.-, ,r0   r  c                   $    \ rS rSrSrS rS rSrg)TestCacheZipLibi  zJ
ZipCache tests that don't require the setup/teardown from `TestCacheZip`
c                     S nSn[         R                  " X5      nU R                  U5        U R                  UR                  [        [        S5      5      5        U R                  UR                  S5        g )Nc                      g r[   rA   rA   r0   r)   	mock_func<TestCacheZipLib.test_zip_locator_creation.<locals>.mock_func      r0   z/path/to/archive.zip/module.pyz/path/to/archive.zipz	module.py)r   ri  assertIsNotNonerT   	_zip_pathr$  r   _internal_path)rU   r  r  rp  s       r)   test_zip_locator_creation)TestCacheZipLib.test_zip_locator_creation  sc    	 4"00EW%**C5K0L,MN//=r0   c                 \    S nSn[         R                  " X5      nU R                  U5        g )Nc                      g r[   rA   rA   r0   r)   r  @TestCacheZipLib.test_zip_locator_non_zip_path.<locals>.mock_func  r  r0   z/path/to/module.py)r   ri  rj  )rU   r  non_zip_pathrp  s       r)   test_zip_locator_non_zip_path-TestCacheZipLib.test_zip_locator_non_zip_path  s,    	 ,"00I'"r0   rA   N)rk   r|   r}   r~   __doc__r  r  r   rA   r0   r)   r  r    s    
>#r0   r  c                   :   ^  \ rS rSrU 4S jrU 4S jrS rSrU =r$ )TestSequentialParForsCachei  c                 @   > [         [        U ]  5         S[        l        g )NT)r  r  r   r	   sequential_parfor_loweringrU   rj   s    r)   r    TestSequentialParForsCache.setUp  s    ($57,0)r0   c                 @   > [         [        U ]  5         S[        l        g )NF)r  r  r   r	   r  r  s    r)   r   #TestSequentialParForsCache.tearDown#  s    ($8:,1)r0   c                    U R                  5       nU R                  S5        UR                  n[        R                  " S5      nU R                  U" U5      X3-  U-   5        UR                  R                  5        Vs/ s H  nUR                  R                  PM     nnU R                  US/5        U R                  S5        g s  snf )Nr   r;  Fr>   )r   r   parfor_usecaser  onesr   	overloadsr   libraryhas_dynamic_globalsrT   )rU   r   r(   arycresdynamic_globalss         r)   r  'TestSequentialParForsCache.test_caching(  s      "1ggbk#	C8'({{'9'9';=';t  <<;;'; 	 =5'21=s   = CrA   )	rk   r|   r}   r~   r   r   r  r   r  r  s   @r)   r  r    s    1
2
	 	r0   r  c                   *    \ rS rSrSrS rS rS rSrg)TestCacheWithCpuSettingi4  Fc                     SnU R                  5       R                  5        H%  u  p4X1;   d  M  U R                  XAU   5        US-  nM'     U R                  USSS9  g )Nr   r=   znothing to comparemsg)r   itemsassertGreaterEqualassertGreater)rU   
mtimes_oldmatch_countkrJ   s        r)   check_later_mtimes*TestCacheWithCpuSetting.check_later_mtimes8  sb    ))+113DA''a=9q  4 	;3 	 	5r0   c                 `   U R                  S5        U R                  5       nUR                  5         [        U R	                  5       5      nU R                  5       nU R                  SS0S9  U R                  U5        U R                  [        U R	                  5       5      U5        UR                  R                  nUR                  nUR                  5       nU R                  [        U5      S5        UR                  5       u  pxUS   S   [        R                   " 5       :X  a  XxpOXpU R                  U	S   S   [        R                   " 5       5        U R                  U	S   S   ["        R$                  " 5       5        U R                  U
S   S   S5        U R                  U
S   S   S5        g )Nr   NUMBA_CPU_NAMEgenericr   r>   r=    )r   r   	self_testr   r   r   r   r  r  r   _cache_cache_file_load_indexrT   keysllget_host_cpu_namer   get_host_cpu_features)rU   r   
cache_sizerN  r4   
cache_filecache_indexkey_akey_bkey_hostkey_generics              r)   test_user_set_cpu_name.TestCacheWithCpuSetting.test_user_set_cpu_nameA  so   1  ",,./
&&($$.>	-J$K'3t2245zB&&&&
 ,,.[)1-$))+8A;"..00$)k$)k!Q)=)=)?@!Q)F)F)HIQ*I6Q*B/r0   c                    U R                  S5        U R                  5       nUR                  5         [        U R	                  5       5      nU R                  5       nSn[        R                  " 5       nU R                  XT5        U R                  SU0S9  U R                  U5        U R                  [        U R	                  5       5      U5        UR                  R                  nUR                  nUR                  5       nU R!                  [        U5      S5        UR#                  5       u  pU	S   S   U:X  a  XpOXpU R!                  US   S   [$        R&                  " 5       5        U R!                  US   S   U5        U R!                  US   S   [$        R&                  " 5       5        U R!                  US   S   U5        g )Nr   z	-sse;-avxNUMBA_CPU_FEATURESr   r>   r=   )r   r   r  r   r   r   r   r  assertNotEqualr   r  r  r   r   r  r  rT   r  r  r  )rU   r   r  rN  my_cpu_featuressystem_featuresr4   r  r	  r
  r  r  r  s                r)   test_user_set_cpu_features2TestCacheWithCpuSetting.test_user_set_cpu_features\  s   1  ",,./
&&(%!779O=$$)?; 	% 	
 	'3t2245zB&&&&
 ,,.[)1-$))+8A;/)$)k$)k!Q)=)=)?@!Q9Q*B,@,@,BCQ*O<r0   rA   N)	rk   r|   r}   r~   _numba_parallel_test_r  r  r  r   rA   r0   r)   r  r  4  s    !506!=r0   r  c                       \ rS rSrSr\R                  R                  \5      r	\R                  R                  \	S5      rSrS rSrg)TestMultiprocessCachei  Fr   r   c                    U R                  5       nUR                  nSn [        R                  " S5      nUR                  U5      n [        UR                  U[        U5      5      5      nUR                  5         U R                  XcUS-
  -  S-  5        g ! [         a	    [        n Nsf = f! UR                  5         f = f)NrI   spawnr=   r>   )r   simple_usecase_callerrl   rm   AttributeErrorPoolr   imaprangecloserT   )rU   r   r(   r   ru   poolrW   s          r)   test_multiprocessing*TestMultiprocessCache.test_multiprocessing  s      " %%	"!--g6C xx{	dii58,-CJJL1q5kQ./  	"!C	" JJLs   B $B- B*)B*-B?rA   N)rk   r|   r}   r~   r  r   r   r   r   r   r   r   r   r#  r   rA   r0   r)   r  r    s;     "77??8$DGGLL':;M.G0r0   r  c                       \ rS rSrSr\R                  R                  \5      r	\R                  R                  \	S5      rSrSrSrS rS rS	 rS
 rS r\R*                  " \" \S5      S5      S 5       rSrg)TestCacheFileCollisioni  Fr   caching_file_loc_fodderzD
from numba import njit
@njit(cache=True)
def bar():
    return 123
zD
from numba import njit
@njit(cache=True)
def bar():
    return 321
c                 4   [        S5      U l        [        R                  R	                  SU R                  5        SU l        U R                  U R
                  [        R                  5        U R
                  U l        SR                  U R
                  S/5      U l
        [        R                  R                  U R                  U R
                  5      n[        R                  " U5        [        [        R                  R                  US5      S5       n[        U R                  US9  S S S 5        [        [        R                  R                  US	5      S5       n[        U R                   US9  S S S 5        g ! , (       d  f       NV= f! , (       d  f       g = f)
Ntest_cache_file_locr   module_name_that_is_unlikely.fooz__init__.pyr'  filezfoo.py)r   r&   r   r   r   r   assertNotInr   modname_bar1r   modname_bar2r   r   r    printsource_text_1source_text_2)rU   foomodfouts      r)   r   TestCacheFileCollision.setUp  s   %&;<4<<(5s{{3 LLHHdllE%:;dllDLL9
"'',,v}5s;t$$$40 <"'',,vx0#6$$$$40 76 <;66s   E8F	8
F	
Fc                     [         R                  R                  U R                  S 5        [         R                  R                  U R                  S 5        [         R
                  R                  U R                  5        g r[   )r   r   r   r0  r1  r   r   r&   r\   s    r)   r   TestCacheFileCollision.tearDown  sF    ))40))40%r0   c                 @    [        U R                  5      R                  $ r[   )r   r0  barr\   s    r)   import_bar1"TestCacheFileCollision.import_bar1      d//0444r0   c                 @    [        U R                  5      R                  $ r[   )r   r1  r;  r\   s    r)   import_bar2"TestCacheFileCollision.import_bar2  r>  r0   c                 f   U R                  5       nU R                  5       nUR                  R                  R                  nUR                  R                  R                  nU R                  X45        U R                  UR                  S5      5        U R                  UR                  S5      5        g )Nz__init__.bar-3.pyzfoo.bar-3.py)r<  r@  r   r  _index_namer  
assertTrue
startswith)rU   bar1bar2idxname1idxname2s        r)   test_file_location)TestCacheFileCollision.test_file_location  s    !!;;**66;;**66H/++,?@A++N;<r0   rm   z)Test requires multiprocessing.get_contextc                 l   U R                  5       nU R                  5       n[        5        nU" 5       nS S S 5        WR                  5       nU R	                  UR                  S5      S5        U R	                  UR                  S5      S5        U R	                  UR                  S5      S5        U R	                  UR                  S5      S5        [        5        nU" 5       nS S S 5        UR                  5       nU R	                  UR                  S5      S5        U R	                  UR                  S5      S5        U R	                  UR                  S5      S5        U R	                  UR                  S5      S5        U R                  WW5         [        R                  " S5      nWR                  5       nUR                  [        XR                  U R                  U R                   4S	9n	U	R#                  5         UR%                  5       n
UR%                  5       nUR%                  5       nUR%                  5       nU	R'                  5         U R	                  X5        U R	                  X5        U R	                  U
R                  S5      S5        U R	                  U
R                  S5      S5        U R	                  U
R                  S5      S5        U R	                  U
R                  S5      S5        U R	                  UR                  S5      S5        U R	                  UR                  S5      S5        U R	                  UR                  S5      S5        U R	                  UR                  S5      S5        g ! , (       d  f       GNJ= f! , (       d  f       GN= f! [         a    [        S5         GNf = f)
Nzindex savedr=   z
data savedzindex loadedr   zdata loadedr  zmissing spawn contextrf   )r<  r@  r   getvaluerT   countr  rl   rm   
ValueErrorr2  rn   ro   cache_file_collision_testerr&   r0  r1  rq   rr   r   )rU   rF  rG  bufres1cachelogres2mpqrw   log1got1log2got2s                 r)   test_no_collision(TestCacheFileCollision.test_no_collision  s    !! C6D !<<>6:5q97;6: C6D !<<>6:5q97;6:D$'	+ !,,W5B HHJzz!< !<< $ 1 1 $ 1 1 3  4 	

uuwuuwuuwuuw		 	$$ 	M2A6L115N3Q7M2A6 	M2A6L115N3Q7M2A6m !  !   	+)*	+s)   M6NN 6
N
NN32N3)r   r0  r1  r&   N)rk   r|   r}   r~   r  r   r   r   r   r   r   r   r   r3  r4  r   r   r<  r@  rJ  r  
skipUnlesshasattrrl   r[  r   rA   r0   r)   r&  r&    s    !77??8$DGGLL':;M'GMM1&
55= -@DF97F97r0   r&  c                    [         R                  R                  SU5        [        U5      R                  n[        U5      R                  n[        5        nU" 5       nS S S 5        U R                  WR                  5       5        U R                  W5        [        5        nU" 5       nS S S 5        U R                  UR                  5       5        U R                  W5        g ! , (       d  f       N= f! , (       d  f       NP= f)Nr   )r   r   r   r   r;  r   r   rM  )	rV  r&   r0  r1  rF  rG  rQ  r1r2s	            r)   rP  rP    s    HHOOAw,'++D,'++D		V 
EE#,,.EE"I		V 
EE#,,.EE"I 
	 
	s   C C1 
C.1
C?c                   2    \ rS rSrSrSrSrS rS rS r	Sr
g	)
#TestCacheMultipleFilesWithSignaturei   Fz
from file2 import function2
z
from numba import njit

@njit('float64(float64)', cache=True)
def function1(x):
    return x

@njit('float64(float64)', cache=True)
def function2(x):
    return x
c                    [        S5      U l        [        R                  R	                  U R                  S5      U l        [        U R
                  S5       n[        U R                  US9  S S S 5        [        R                  R	                  U R                  S5      U l	        [        U R                  S5       n[        U R                  US9  S S S 5        g ! , (       d  f       Nq= f! , (       d  f       g = f)Nr)  zfile1.pyr'  r-  zfile2.py)r   r&   r   r   r   file1r    r2  source_text_file1file2source_text_file2)rU   r6  s     r)   r   )TestCacheMultipleFilesWithSignature.setUp4  s    %&;<WW\\$,,
;
$**c"d$((t4 # WW\\$,,
;
$**c"d$((t4 #"	 #" #"s   C8C'
C$'
C5c                 D    [         R                  " U R                  5        g r[   )r#   r$   r&   r\   s    r)   r   ,TestCacheMultipleFilesWithSignature.tearDown?  s    dll#r0   c                 |   [         R                  " [        R                  U R                  /[         R
                  [         R
                  S9nUR                  5       u  p#SUR                  5        SUR                  5        3nU R                  UR                  SUS9  [         R                  " [        R                  U R                  /[         R
                  [         R
                  S9nUR                  5       u  p#SUR                  5        SUR                  5        3nU R                  UR                  SU5        g )Nr   r   zstdout:
z


stderr:
r   r  )r   r   r   r   re  r   r   r   rT   r   rg  )rU   r   r   r   r  s        r)   *test_caching_mutliple_files_with_signatureNTestCacheMultipleFilesWithSignature.test_caching_mutliple_files_with_signatureB  s      #..$**!=(2(29 $$&#**,}SZZ\NC))1#6   #..$**!=(2(29 $$&#**,}SZZ\NC))1c2r0   )re  rg  r&   N)rk   r|   r}   r~   r  rf  rh  r   r   rn  r   rA   r0   r)   rc  rc     s)     "
	5$3r0   rc  c                       \ rS rSr\R
                  R                  \5      r\R
                  R                  \S5      r
SrS rS rS rSrg)	TestCFuncCacheiT  zcfunc_cache_usecases.pycfunc_caching_test_fodderc                 d   S[        U R                  U R                  S9-  n[        R                  " [
        R                  SU/[        R                  [        R                  S9nUR                  5       u  p4UR                  S:w  a*  [        SUR                   SUR                  5        S35      eg )	Naf  if 1:
            import sys

            sys.path.insert(0, %(tempdir)r)
            mod = __import__(%(modname)r)
            mod.self_test()

            f = mod.add_usecase
            assert f.cache_hits == 1
            f = mod.outer
            assert f.cache_hits == 1
            f = mod.div_usecase
            assert f.cache_hits == 1
            r   r   rm  r   r   z:stderr follows
r   )r   r&   r   r   r   r   r   r   r   r   r   r   )rU   r   r   r   r   s        r)   r   &TestCFuncCache.run_in_separate_processZ  s     t||T\\BC   #..$!=(2
Q$$&q  #<U=M=M<N O447JJL>"E F F !r0   c                 $    UR                  5         g r[   )r  rU   r   s     r)   check_moduleTestCFuncCache.check_moduler  s    r0   c                 n   U R                  S5        U R                  5       nU R                  S5        U R                  UR                  R                  S5        U R                  UR
                  R                  S5        U R                  UR                  R                  S5        U R                  UR                  R                  S5        U R                  U5        U R                  5       nU R                  S5        U R                  UR                  R                  S5        U R                  UR
                  R                  S5        U R                  UR                  R                  S5        U R                  UR                  R                  S5        U R                  U5        U R                  5         g )Nr   r   r=   )
r   r   rT   r   r   r  r  div_usecaserw  r   rv  s     r)   r  TestCFuncCache.test_cachingu  sF   1  "133Q7--q100;;Q?33Q7#   "133Q7--q100;;Q?33Q7#$$&r0   rA   N)rk   r|   r}   r~   r   r   r   r   r   r   r   r   r   rw  r  r   rA   r0   r)   rq  rq  T  s>    77??8$DGGLL'@AM)GF0'r0   rq  __main__)Er  rg  rl   r   r#   r   r   r   r   r  r  r  pathlibr   llvmlite.bindingbindingr  numpyr  numbar   
numba.corer   numba.core.cachingr   r   numba.core.errorsr   numba.parforsr	   numba.tests.supportr
   r   r   r   r   r   r   r   r   r  ImportErrorr*   _access_preventable_access_msgr]  r  r/   r;   rB   rG   rK   rN   rP   rp   r   r   r   r  r  r  r  r  r&  rP  rc  rq  rk   mainrA   r0   r)   <module>r     s      	    
          F *  
 
 

 > 23 I%%&9;G

7
+x 7
t,84H 4n$.- $.N[5+ [5|8,. 8,v#1 #6 !<  .I=9 I=X0M 0: q7X.. q7 q7h13(*;*; 13h6'] 6'r zMMO Q#  Is   8E/ /E:9E: