
    %YHhl                        d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlmZ d dlmZ d dlZd dlmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d dlmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z' 	 d dl(Z(d	 Z* e*       Z+d
Z, e	jZ                  e+e,      Z.d Z/d Z0d Z1d Z2d Z3d Z4 G d dee       Z5d Z6 G d de       Z7 G d de7      Z8 G d de8      Z9 G d de8      Z: G d de8      Z;e& G d de8             Z< G d  d!e8      Z= G d" d#e7      Z>e% G d$ d%e	j@                               Z?d& Z@ G d' d(e	j@                        ZA G d) d*e7      ZBeCd+k(  r e	j                          yy# e)$ r dZ(Y w xY w),    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                     t        d      } t        j                  j                  | d      }t        j                  |       t        t        j                  j                  |d      d      5 }|j                  d       d d d        t        j                  |d       	 t        t        j                  j                  |d      d      5 }|j                  d       d d d        	 t        j                  |d	       t        j                  |       y
# 1 sw Y   xY w# 1 sw Y   BxY w# t        $ r. Y t        j                  |d	       t        j                  |       yw xY w# t        j                  |d	       t        j                  |       w xY w)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      Q/var/www/html/planif/env/lib/python3.12/site-packages/numba/tests/test_caching.pycheck_access_is_preventabler*   )   s)    \*GWo6HHHX	bggll8Z0$	7 1	 HHXu "'',,x):;TB 	aGGH	  	5!h% 
	 	   
5!h 	5!hsH   +D*D% DD% DD"D% %	E.E EE -Fz9Cannot create a directory to which writes are preventablec                       d} t        |       | fS )Nabcd)hash)cs    r)   constant_unicode_cacher/   M   s    A7A:    c                  x    t         }  t        d      |       } |        \  }} |       \  }}||k(  sJ ||k(  sJ y NTcache)r/   r   )pyfunccfuncexp_hvexp_strgot_hvgot_strs         r)   check_constant_unicode_cacher;   R   sJ    #FDtV$EhOFGgOFGVgr0   c                      dddS )N      )ab rA   r0   r)   
dict_cacherB   [   s    r0   c                  ^    t         }  t        d      |       } |        } |       }||k(  sJ y r2   )rB   r   r5   r6   expgots       r)   check_dict_cacherG   _   s2    FDtV$E
(C
'C#::r0   c               #   "   K   dD ]  } |   y w)N)r=   r>      rA   )vs    r)   generator_cacherK   g   s      s   c                      t         }  t        d      |       }t         |              }t         |             }||k(  sJ y r2   )rK   r   listrD   s       r)   check_generator_cacherN   l   s:    FDtV$E
vx.C
uw-C#::r0   c                   *    e Zd Zd Zd Zd Zd Zd Zy)TestCachingc                 R     |        t        |      }| j                  |d   d       y )Nexitcoder   )r   assertEqual)selffuncress      r)   run_testzTestCaching.run_testu   s%    (.Z!,r0   c                 .    | j                  t               y N)rW   r;   rT   s    r)   test_constant_unicode_cachez'TestCaching.test_constant_unicode_cachez   s    23r0   c                 .    | j                  t               y rY   )rW   rG   rZ   s    r)   test_dict_cachezTestCaching.test_dict_cache}   s    &'r0   c                 .    | j                  t               y rY   )rW   rN   rZ   s    r)   test_generator_cachez TestCaching.test_generator_cache   s    +,r0   c                 V   t        | j                  j                        }t        j                         }|j                         }|j                  t        ||df      }|j                          |j                          |j                         \  }}|s| j                  |       | j                  |dd       |j                  t        ||df      }|j                          |j                          |j                         \  }}|s| j                  |       | j                  |dd       y )NFtargetargs  z-Omitted function returned an incorrect outputT)r   	__class____name__multiprocessingget_contextQueueProcessomitted_child_test_wrapperstartr   getfailrS   )rT   	cache_dirctxresult_queueprocsuccessoutputs          r)   test_omittedzTestCaching.test_omitted   s    #4>>#:#:;	))+yy{{{-	51  
 	

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

		&**, IIf;	
r0   N)rf   
__module____qualname__rW   r[   r]   r_   ru   rA   r0   r)   rP   rP   t   s    -
4(-(
r0   rP   c                    t        d|      5  t        d      dd       }	  |       }|rL|j                  |j                  d      dk(  sJ d       |j                  |j                  d      dk(  sSJ d       |j                  |j                  d      dk(  sJ d	       |j                  |j                  d      dk(  sJ d
       d}| j                  ||f       d d d        y #  t        j                         }d}Y 6xY w# 1 sw Y   y xY w)N	CACHE_DIRTr3   c                     | S rY   rA   )nums    r)   testz(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)rd   )r   r   _cache_hits
signatures_cache_misses	traceback
format_excput)rq   ro   second_callr|   rt   rs   s         r)   rk   rk      s4   	i	0 ,	D		 
		VF ''(:;q@ 434@))$//!*<=B 323B ))$//!*<=B 545B''(:;q@ 212@G
 	'6*+1, ,*	))+FG/, ,s#   C>B#C!C>!C;9C>>Dc                   >    e Zd ZdZdZd Zd Zd Zd Zd Z	d Z
d Zy)	BaseCacheTestNc                 .   t        d      | _        t        j                  j	                  d| j                         t
        j                  j                  | j                  | j                  dz         | _        t
        j                  j                  | j                  d      | _	        t        j                  | j                  | j                         t        j                  | j                  t        j                  t        j                   z         d | _        y )Nr   r   z.py__pycache__)r   r&   sysr   insertr   r   modnamemodfilero   r#   copyusecases_filer"   statS_IREADS_IWRITEmaxDiffrZ   s    r)   setUpzBaseCacheTest.setUp   s    %l34<<(ww||DLL$,,2FGdllMBD&&5
t||dmm;<r0   c                     t         j                  j                  | j                  d        t         j                  j                  | j                         y rY   )r   modulespopr   r   remover&   rZ   s    r)   tearDownzBaseCacheTest.tearDown   s,    d+%r0   c                 d   t         j                  j                  | j                  d       }|*|j                  g}|D ]  }	 t        j                  |        t        | j                        }| j                  |j                  j                  d      | j                         |S # t        $ r Y rw xY w)Nco)r   r   r   r   
__cached__r   unlinkFileNotFoundErrorr   rS   __file__rstripr   )rT   oldcachedfnmods        r)   import_modulezBaseCacheTest.import_module   s     kkoodllD1?nn%F IIbM
 T\\*,,T2DLLA
	 ) s    B##	B/.B/c                     	 t        j                  | j                        D cg c]  }|j                  d      s| c}S c c}w # t        $ r g cY S w xY w)N)z.pycz.pyo)r   listdirro   endswithr   )rT   r   s     r)   cache_contentszBaseCacheTest.cache_contents   sT    	!#DNN!; :2;;'78  : : :  	I	s!   !A A A  A AAc                 Z     t         fdt         j                               D              S )Nc              3      K   | ]M  }|t         j                  j                  t         j                  j                  j                  |            f O y wrY   )r   r   getmtimer   ro   ).0r   rT   s     r)   	<genexpr>z1BaseCacheTest.get_cache_mtimes.<locals>.<genexpr>   s@      = ))"'',,t~~r*JKL =s   AA)dictsortedr   rZ   s   `r)   get_cache_mtimeszBaseCacheTest.get_cache_mtimes   s+     =$T%8%8%:;= = 	=r0   c                 \    | j                         }| j                  t        |      ||       y rY   )r   rS   len)rT   nr.   s      r)   check_pycachezBaseCacheTest.check_pycache   s&    !QA&r0   c                      y rY   rA   rZ   s    r)   
dummy_testzBaseCacheTest.dummy_test   s    r0   )rf   rv   rw   r   r   r   r   r   r   r   r   r   rA   r0   r)   r   r      s/    MG&"='r0   r   c                       e Zd Zej                  j                  e      Zej                  j                  ed      Z	dZ
i ddZddZy)DispatcherCacheUsecasesTestcache_usecases.pydispatcher_caching_test_fodderenvvarsc          	         dt        | j                  | j                        z  }t        j                  j                         }|j                  |       t        j                  t        j                  d|gt        j                  t        j                  |      }|j                         \  }}|j                  dk7  r;t        d|j                  d|j                         d|j                         d	      y )
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)rT   r   codesubp_envpopenouterrs          r)   run_in_separate_processz3DispatcherCacheUsecasesTest.run_in_separate_process  s     t||T\\BC ::??$   #..$!=(2
%-/ $$&Sq   ##SZZ\3::<A  !r0   Nc                    |j                   }| j                  t        |j                  j	                               ||j                         |?| j                  t        |j
                  j	                               ||j
                         y y rY   )statsrS   sum
cache_hitsvaluescache_misses)rT   rU   hitsmissessts        r)   
check_hitsz&DispatcherCacheUsecasesTest.check_hits   si    ZZR]]1134dBMMJS!7!7!9:F__. r0   rY   )rf   rv   rw   r   r   dirnamer   herer   r   r   r   r   rA   r0   r)   r   r     s;    77??8$DGGLL':;M.G13 4.r0   r   c                   |   e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zd Zd Ze ej*                  ej.                  dk(  d      d               Ze ej*                  ej.                  dk(  d      d               Zd Z ej*                  edu xs ej8                  d   dk  d      d        Zy)	TestCachec                 l   | j                  d       | j                         }| j                  d       |j                  }| j                   |dd      d       | j                  d       | j                   |dd      d       | j                  d       | j	                  |dd       |j
                  }| j                   |dd      d       | j                  d       | j                   |dd      d       | j                  d       | j	                  |dd       |j                  } ||j                  d      }| j                  t        |      d	        ||j                  d      }| j                  t        |      d	       | j                  d
       | j	                  |dd       | j                          y )Nr   r>   rI            @      @   r=   )r>   g     E@	   )r   r   add_usecaseassertPreciseEqualr   add_objmode_usecaserecord_returnaligned_arrtuple
packed_arrr   )rT   r   r(   recs       r)   test_cachingz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                    | j                  d       | j                         }| j                  d       |j                  }| j                   |dd      d       | j                  d       | j                   |dt	        j
                  d            dt	        j
                  d      z   dz          | j                  d       | j                  |dd       y )Nr   r>   rI   r   r=   )r   r   r   r   nparanger   rT   r   r(   s      r)   test_caching_nrt_prunedz!TestCache.test_caching_nrt_prunedH  s    1  "1OO!Q+1!RYYq\ 2A		!4Dq4HI11a r0   c                 B   | j                         }| j                  |j                  dd      d       | j                  d       |j                  }| j                   |dd      d       | j                  d       | j                         }|j                  }| j                   |dd      d       | j                  d       |j
                  }| j                   |dd      d       | j                  d       | j                   |dd      d       | j                  d       y )NrI   r>   r            @r   )r   r   innerr   outer_uncachedouterr   s      r)   test_inner_then_outerzTestCache.test_inner_then_outerU  s      "		!Q31!Q+1  "!Q+1II!Q+1#q	3/1r0   c                    | j                         }| j                  |j                  dd      d       | j                  d       | j                  |j	                  dd      d       | j                  d       | j                         }|j
                  }| j                   |dd      d       | j                  d       | j                   |dd      d       | j                  d       y )NrI   r>   r   r   r   r   r   )r   r   r   r   r   r   r   s      r)   test_outer_then_innerzTestCache.test_outer_then_inneri  s      "		!Q31 2 21a 8!<1  "II!Q+1#q	3/1r0   c                     | j                         }|j                  }| j                   |dd      d       | j                  d       y )Nr>   rI   r   r   )r   add_nocache_usecaser   r   r   s      r)   test_no_cachingzTestCache.test_no_cachingw  s?      "##!Q+1r0   c                    | j                         }t        j                  d      5 }t        j                  dt               |j
                  }| j                   |d      d       | j                  d       d d d        | j                  t              d       | j                  dt        |d   j                               y # 1 sw Y   MxY w)	NTrecordalwaysr   r   r   r=   zBCannot cache compiled function "looplifted" as it uses lifted code)r   warningscatch_warningssimplefilterr   
loopliftedr   r   rS   r   assertInstrmessagerT   r   wr(   s       r)   test_loopliftedzTestCache.test_looplifted~  s      "$$D1 	"Q!!(L9A##AaD!,q!	" 	Q# /03AaDLL0A	C	" 	"s   ACCc                    | j                         }t        j                  d      5 }t        j                  dt               |j
                  }t        j                  j                   |       |j                         | j                  d       d d d        | j                  t              d       | j                  dt        |d   j                               y # 1 sw Y   MxY w)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   rS   r   r
  r  r  r  s       r)   test_big_arrayzTestCache.test_big_array  s      "$$D1 	"Q!!(L9!!AJJ##AC4q!	" 	Q# 347!4E	G	" 	"s   A'CC#c                    | j                         }|j                  |j                  |j                  fD ]  }t	        j
                  d      5 }t	        j                  dt               | j                   |d      d       | j                  d       d d d        | j                  t              d       | j                  dj                  |j                        t        |d   j                                 y # 1 sw Y   hxY w)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   rS   r   r
  formatrf   r  r  )rT   r   r(   r  s       r)   test_ctypeszTestCache.test_ctypes  s       "--!4!4c6I6IJ 	A((5 &%%h=''#4""1%	& SVQ'MM5<<QZZHAaDLL!	& &s   AC88D	c                    | j                         }t        j                         5  t        j                  dt               |j
                  }| j                   |d      d       |j                  }| j                   |d      d       |j                  }| j                   |d      d       |j                  }| j                   |d      d       | j                  d       d d d        y # 1 sw Y   y xY w)NerrorrI   r      
      r   )r   r  r  r  r   closure1r   closure2closure3closure4r   r   s      r)   test_closurezTestCache.test_closure  s      "$$& 	"!!'<8A##AaD!,A##AaD!,A##AaD"-A##AaD"-q!	" 	" 	"s   B<C**C3c                 v   | j                         }|j                  }| j                   ||j                  d      d       | j                   ||j                  d      d       | j                   ||j                  d      d       | j                   ||j                  d      d       | j                  d       y )Nr=   r  d   r>         )r   first_class_function_usecaserS   first_class_function_mulfirst_class_function_addr   r   s      r)   test_first_class_functionz#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                 ^   | j                         }|j                  dd       |j                  dd       |j                  dd       |j                  dd       |j	                  dd       |j                  |j                  d       |j                  |j                  d       | j                         }| j                  |j                  dd       | j                         }| j                  ||       |j                  } |dd       | j                  |dd        |dd       | j                  |dd       |j                  } |dd       | j                  |dd       | j                  | j                         |       | j                          | j                  | j                         |       y )Nr>   rI   r   r   r   r=   )r   r   r   r   r   r   r   r   r   r   assertIsNotrS   r   )rT   r   mtimesmod2r(   s        r)   test_cache_reusezTestCache.test_cache_reuse  sb     "1S!1%1a 		!Q#..!,#//1-&&(A.!!#d#	!Q1a 	#s1a $$	!Q1a  	..0&9$$&..0&9r0   c                    | j                         }|j                  }| j                   |dd      d       t        | j                  d      5 }|j                  d       d d d        | j                         }|j                  }| j                   |dd      d       |j                  }| j                   |dd      d       y # 1 sw Y   dxY w)Nr>   rI   r   r?   z
Z = 10
   )r   r   r   r    r   r!   r   r   s      r)   test_cache_invalidatezTestCache.test_cache_invalidate  s      "OO!Q+ $,,$ 	"GGL!	"   "OO!Q,##!Q,	" 	"s   CC
c                    | j                         }|j                  }| j                   |dd      d       | j                         }|j                  }d|_        | j                   |dd      d       |j	                          | j                   |dd      d       | j                         }|j                  }| j                   |dd      d       y )Nr>   rI   r   r  r4  )r   r   r   Z	recompiler   s      r)   test_recompilezTestCache.test_recompile  s      "OO!Q+  "OO!Q+	!Q,   "OO!Q,r0   c                     | j                         }|j                  }| j                   |d      d       |j                  }| j                   |d      d       y )Nr>   r   r  )r   renamed_function1r   renamed_function2r   s      r)   test_same_nameszTestCache.test_same_names  sN      "!!!a(!!!a(r0   c                 v   ddl m} |j                  }t        ddd      }	 ||_        t	        j
                  |      }t        j                  ||      }| j                  |       dt        _
        t        j                  ||      }| j                  |t               ||_        t        `
y # ||_        t        `
w xY w)Nr=   )functionpassztests/dummy_module.pyexecT)dummy_moduler?  __code__compileinspectgetfiler   from_functionassertIsNoner   frozenassertIsInstance)rT   r?  old_codecode_objsourcelocators         r)   test_frozenzTestCache.test_frozen  s    *$$6#:FC	 (H__X.F ,99(FKGg&CJ ,99(FKG!!'+@A !)H
 !)H
s   A:B) )B8c                    | j                         }|j                  }| j                  t        j                  |j
                  j                  d       | j                   |dd      d       | j                  |dd       | j                         }|j                  }| j                   |dd      d       | j                  |dd       | j                  d       y)	zy
        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   )rT   r   r(   r1  s       r)   _test_pycache_fallbackz 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                    t        j                  | j                        j                  }t        j                  | j                  d       | j                  t         j                  | j                  |       | j                          y )Nr   )r   r   r&   st_moder"   rR  rT  )rT   	old_permss     r)   test_non_creatable_pycachez$TestCache.test_non_creatable_pycache?  sR    
 GGDLL)11	
u%$,,	:##%r0   c                 N   t         j                  j                  | j                  d      }t        j                  |       t        j
                  |      j                  }t        j                  |d       | j                  t         j                  ||       | j                          y )Nr   r   )
r   r   r   r&   r   r   rW  r"   rR  rT  )rT   pycacherX  s      r)   test_non_writable_pycachez#TestCache.test_non_writable_pycacheJ  sj    
 '',,t||];
GGG$,,	
% '95##%r0   c                 L    t         j                  ddgg dz  	 t        j                  dgz         }j                         j                         }t        j                  j                   j                  d      t        d      5 }|j                  d       d d d         fd	} |         |       } j                  d
|j                                y # t        j                  $ r(} j                  d|j                  z         Y d }~d }~ww xY w# 1 sw Y   }xY w)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                  >   t        d      5 } t        j                  | t        j                  t        j                  d      }|j	                         \  }}|j
                  dk7  r!j                  d|j
                  ||fz         |cd d d        S # 1 sw Y   y xY wNrbT)stdinr   r   universal_newlines*   z6unexpected return code %d
-- stdout:
%s
-- stderr:
%s
r    r   r   r   r   r   rn   rg  pr   r   base_cmdinputfnrT   s       r)   execute_with_inputz2TestCache.test_ipython.<locals>.execute_with_inputx      gt$ $$XU,6OO,6OO8<> ==?S<<2%II 1 "#sC 89 :      A;BBcache hits = 1)r   r   r   check_outputCalledProcessErrorskipTestr   stripr   r   r   r   r&   r    r!   r
  )rT   verer(   ro  r   rm  rn  s   `     @@r)   test_ipythonz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/DDD#Nr   r   zrequires ipykernel >= 6c                 $    t         j                  ddgg dz  	 t        j                  dgz         }j                         j                         }ddl	m
} |j                         t        d      5 }|j                  d	       d d d         fd
} |         |       } j                  d|j                                y # t        j                  $ r(} j                  d|j                  z         Y d }~d }~ww xY w# 1 sw Y   }xY w)Nr^  r_  r`  ra  rb  r   )compilerr  rc  c                  >   t        d      5 } t        j                  | t        j                  t        j                  d      }|j	                         \  }}|j
                  dk7  r!j                  d|j
                  ||fz         |cd d d        S # 1 sw Y   y xY wre  rj  rk  s       r)   ro  z4TestCache.test_ipykernel.<locals>.execute_with_input  rp  rq  rr  )r   r   r   rs  rt  ru  r   rv  r   	ipykernelr{  get_tmp_directoryr    r!   r
  )	rT   rw  rx  r{  r(   ro  r   rm  rn  s	   `      @@r)   test_ipykernelzTestCache.test_ipykernel  s    
 NND)4MM	*))(k]*BCC iik  "&,,.'3 	1GG  	*	 	 "&		4] ,, 	*MMALL) * *	*	 	s#   C 8DDC>>DD)rf   rv   rw   r   r   r   r   r  r  r  r  r%  r-  r2  r5  r9  r=  rO  rT  skip_bad_accessunittestskipIfr   namerY  r\  ry  r}  version_infor  rA   r0   r)   r   r   (  s   '<!(CG$" 	:<--$).2 X__RWW_JL&L & X__RWW_JL&L &35j X__i4'KY-C-CA-F-J.055055r0   r   c                   *     e Zd Z fdZd Zd Z xZS )TestCacheZipc                 J   t         |           d}d}d}t        j                  j	                  | j
                  |      }t        j                  |d      5 }|j                  ||       d d d        t        j                  j                  d|       d| _        y # 1 sw Y   1xY w)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   )rT   mod_contentmod_filenamezip_filenamezip_pathzfre   s         r)   r   zTestCacheZip.setUp  s     () 77<<l;__Xs+ 	3rKKk2	3 	8$$	3 	3s   BB"c                     t         j                  j                  d       t         j                  j                  dd        y )Nr   r  )r   r   r   r   rZ   s    r)   r   zTestCacheZip.tearDown  s#    Qt,r0   c                    dd l }|j                  dd      }| j                  |d       | j                  |j                  dd       | j                  |j                  d       t        j
                  d= t        j                          dd l }|j                  dd      }| j                  |d       | j                  |j                  d       y )Nr   r>   rI   r   r=   r  )r  addrS   r   r   r   	importlibinvalidate_caches)rT   r  result1result2s       r)   test_zip_cachingzTestCacheZip.test_zip_caching  s     	//!Q'!$A. 	+ KK&##% //!Q'!$ 	+r0   )rf   rv   rw   r   r   r  __classcell__re   s   @r)   r  r    s    %.-,r0   r  c                       e Zd ZdZd Zd Zy)TestCacheZipLibzR
    ZipCache tests that don't require the setup/teardown from `TestCacheZip`
    c                     d }d}t        j                  ||      }| j                  |       | j                  |j                  t        t        d                   | j                  |j                  d       y )Nc                       y rY   rA   rA   r0   r)   	mock_funcz<TestCacheZipLib.test_zip_locator_creation.<locals>.mock_func      r0   z/path/to/archive.zip/module.pyz/path/to/archive.zipz	module.py)r   rG  assertIsNotNonerS   	_zip_pathr  r   _internal_path)rT   r  r  rN  s       r)   test_zip_locator_creationz)TestCacheZipLib.test_zip_locator_creation  se    	 4"00HEW%**C5K0L,MN//=r0   c                 \    d }d}t        j                  ||      }| j                  |       y )Nc                       y rY   rA   rA   r0   r)   r  z@TestCacheZipLib.test_zip_locator_non_zip_path.<locals>.mock_func  r  r0   z/path/to/module.py)r   rG  rH  )rT   r  non_zip_pathrN  s       r)   test_zip_locator_non_zip_pathz-TestCacheZipLib.test_zip_locator_non_zip_path  s.    	 ,"00LI'"r0   N)rf   rv   rw   __doc__r  r  rA   r0   r)   r  r    s    
>#r0   r  c                   .     e Zd Z fdZ fdZd Z xZS )TestSequentialParForsCachec                 @    t         t        |           dt        _        y )NT)r  r  r   r	   sequential_parfor_loweringrT   re   s    r)   r   z TestSequentialParForsCache.setUp  s    ($57,0)r0   c                 @    t         t        |           dt        _        y )NF)r  r  r   r	   r  r  s    r)   r   z#TestSequentialParForsCache.tearDown#  s    ($8:,1)r0   c                    | j                         }| j                  d       |j                  }t        j                  d      }| j                   ||      ||z  |z          |j                  j                         D cg c]  }|j                  j                   }}| j                  |dg       | j                  d       y c c}w )Nr   r  Fr>   )r   r   parfor_usecaser   onesr   	overloadsr   libraryhas_dynamic_globalsrS   )rT   r   r(   arycresdynamic_globalss         r)   r   z'TestSequentialParForsCache.test_caching(  s      "1ggbk#c	C8'({{'9'9';=#  <<;; = =5'21=s   =C)rf   rv   rw   r   r   r   r  r  s   @r)   r  r    s    1
2
	r0   r  c                   "    e Zd ZdZd Zd Zd Zy)TestCacheWithCpuSettingFc                     d}| j                         j                         D ]$  \  }}||v s| j                  |||          |dz  }& | j                  |dd       y )Nr   r=   znothing to comparemsg)r   itemsassertGreaterEqualassertGreater)rT   
mtimes_oldmatch_countkrJ   s        r)   check_later_mtimesz*TestCacheWithCpuSetting.check_later_mtimes8  sm    ))+113 	!DAqJ'':a=9q 	! 	;3 	 	5r0   c                 h   | j                  d       | j                         }|j                          t        | j	                               }| j                         }| j                  ddi       | j                  |       | j                  t        | j	                               |       |j                  j                  }|j                  }|j                         }| j                  t        |      d       |j                         \  }}|d   d   t        j                          k(  r||}
}	n||}
}	| j                  |	d   d   t        j                                 | j                  |	d   d   t#        j$                                | j                  |
d   d   d       | j                  |
d   d   d       y )Nr   NUMBA_CPU_NAMEgenericr   r>   r=    )r   r   	self_testr   r   r   r   r  r  r   _cache_cache_file_load_indexrS   keysllget_host_cpu_namer   get_host_cpu_features)rT   r   
cache_sizer0  r4   
cache_filecache_indexkey_akey_bkey_hostkey_generics              r)   test_user_set_cpu_namez.TestCacheWithCpuSetting.test_user_set_cpu_nameA  sy   1  ",,./
&&($$.>	-J$K'3t2245zB&&&&
 ,,.[)1-$))+8A;"..00$)5kH$)5kH!Q)=)=)?@!Q)F)F)HIQ*I6Q*B/r0   c                    | j                  d       | j                         }|j                          t        | j	                               }| j                         }d}t        j                         }| j                  ||       | j                  d|i       | j                  |       | j                  t        | j	                               |       |j                  j                  }|j                  }|j                         }| j!                  t        |      d       |j#                         \  }	}
|	d   d   |k(  r|	|
}}n|
|	}}| j!                  |d   d   t%        j&                                | j!                  |d   d   |       | j!                  |d   d   t%        j&                                | j!                  |d   d   |       y )Nr   z	-sse;-avxNUMBA_CPU_FEATURESr   r>   r=   )r   r   r  r   r   r   r   r  assertNotEqualr   r  r  r   r  r  r  rS   r  r  r  )rT   r   r  r0  my_cpu_featuressystem_featuresr4   r  r  r  r  r  r  s                r)   test_user_set_cpu_featuresz2TestCacheWithCpuSetting.test_user_set_cpu_features\  s   1  ",,./
&&(%!779O_=$$)?; 	% 	
 	'3t2245zB&&&&
 ,,.[)1-$))+8A;/)$)5kH$)5kH!Q)=)=)?@!Q9Q*B,@,@,BCQ*O<r0   N)rf   rv   rw   _numba_parallel_test_r  r  r  rA   r0   r)   r  r  4  s    !506!=r0   r  c                       e Zd ZdZej
                  j                  e      Zej
                  j                  ed      Z
dZd Zy)TestMultiprocessCacheFr   r   c                    | j                         }|j                  }d}	 t        j                  d      }|j                  |      }	 t        |j                  |t        |                  }|j                          | j                  |||dz
  z  dz         y # t        $ r	 t        }Y sw xY w# |j                          w xY w)NrI   spawnr=   r>   )r   simple_usecase_callerrg   rh   AttributeErrorPoolr   imaprangecloserS   )rT   r   r(   r   rp   poolrV   s          r)   test_multiprocessingz*TestMultiprocessCache.test_multiprocessing  s      " %%	"!--g6C xx{	dii58,-CJJLa1q5kQ./  	"!C	" JJLs   B $B, B)(B),B>N)rf   rv   rw   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                       e Zd ZdZej
                  j                  e      Zej
                  j                  ed      Z
dZdZdZd Zd Zd Zd	 Zd
 Z ej(                   eed      d      d        Zy)TestCacheFileCollisionFr   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                 *   t        d      | _        t        j                  j	                  d| j                         d| _        | j                  | j
                  t        j                         | j
                  | _        dj                  | j
                  dg      | _
        t        j                  j                  | j                  | j
                        }t        j                  |       t        t        j                  j                  |d      d      5 }t        | j                  |       d d d        t        t        j                  j                  |d	      d      5 }t        | j                   |       d d d        y # 1 sw Y   TxY w# 1 sw Y   y xY w)
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)rT   foomodfouts      r)   r   zTestCacheFileCollision.setUp  s   %&;<4<<(5s{{3 LLHHdllE%:;dllDLL9
"'',,v}5s; 	1t$$$40	1"'',,vx0#6 	1$$$$40	1 	1	1 	1	1 	1s   E=F	=F	Fc                     t         j                  j                  | j                  d        t         j                  j                  | j                  d        t         j
                  j                  | j                         y rY   )r   r   r   r  r  r   r   r&   rZ   s    r)   r   zTestCacheFileCollision.tearDown  sF    ))40))40%r0   c                 @    t        | j                        j                  S rY   )r   r  barrZ   s    r)   import_bar1z"TestCacheFileCollision.import_bar1      d//0444r0   c                 @    t        | j                        j                  S rY   )r   r  r  rZ   s    r)   import_bar2z"TestCacheFileCollision.import_bar2  r  r0   c                 h   | j                         }| j                         }|j                  j                  j                  }|j                  j                  j                  }| j                  ||       | j                  |j                  d             | j                  |j                  d             y )Nz__init__.bar-3.pyzfoo.bar-3.py)r  r  r  r  _index_namer  
assertTrue
startswith)rT   bar1bar2idxname1idxname2s        r)   test_file_locationz)TestCacheFileCollision.test_file_location  s    !!;;**66;;**66Hh/++,?@A++N;<r0   rh   z)Test requires multiprocessing.get_contextc                 `   | j                         }| j                         }t               5 } |       }d d d        j                         }| j	                  |j                  d      d       | j	                  |j                  d      d       | j	                  |j                  d      d       | j	                  |j                  d      d       t               5 } |       }d d d        |j                         }| j	                  |j                  d      d       | j	                  |j                  d      d       | j	                  |j                  d      d       | j	                  |j                  d      d       | j                         	 t        j                  d      }j                         }|j                  t        || j                  | j                  | j                   f	      }	|	j#                          |j%                         }
|j%                         }|j%                         }|j%                         }|	j'                          | j	                  ||       | j	                  ||       | j	                  |
j                  d      d       | j	                  |
j                  d      d       | j	                  |
j                  d      d       | j	                  |
j                  d      d       | j	                  |j                  d      d       | j	                  |j                  d      d       | j	                  |j                  d      d       | j	                  |j                  d      d       y # 1 sw Y   JxY w# 1 sw Y   xY w# t        $ r t        d       Y w xY w)
Nzindex savedr=   z
data savedzindex loadedr   zdata loadedr  zmissing spawn contextra   )r  r  r   getvaluerS   countr  rg   rh   
ValueErrorr  ri   rj   cache_file_collision_testerr&   r  r  rl   rm   r   )rT   r
  r  bufres1cachelogres2mpqrr   log1got1log2got2s                 r)   test_no_collisionz(TestCacheFileCollision.test_no_collision  s    !!  	C6D	<<>6:5q97;6:  	C6D	<<>6:5q97;6:D$'	+ !,,W5B HHJzz!< !4<< $ 1 1 $ 1 1 3  4 	

uuwuuwuuwuuw		 	t$t$ 	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)   M;NN ;NNN-,N-N)rf   rv   rw   r  r   r   r   r   r   r   r   r   r  r  r   r   r  r  r  r  
skipUnlesshasattrrg   r  rA   r0   r)   r  r    s    !77??8$DGGLL':;M'GMM1&
55= X-@DF97F97r0   r  c                    t         j                  j                  d|       t        |      j                  }t        |      j                  }t               5 } |       }d d d        | j                  j                                | j                         t               5 } |       }d d d        | j                  |j                                | j                         y # 1 sw Y   xY w# 1 sw Y   FxY w)Nr   )r   r   r   r   r  r   r   r  )	r  r&   r  r  r
  r  r  r1r2s	            r)   r  r    s    HHOOAw,'++D,'++D		 VEE#,,.EE"I		 VEE#,,.EE"I  s   C C, C),C5c                   *    e Zd ZdZdZdZd Zd Zd Zy)#TestCacheMultipleFilesWithSignatureFz
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                    t        d      | _        t        j                  j	                  | j                  d      | _        t        | j
                  d      5 }t        | j                  |       d d d        t        j                  j	                  | j                  d      | _	        t        | j                  d      5 }t        | j                  |       d d d        y # 1 sw Y   oxY w# 1 sw Y   y xY w)Nr  zfile1.pyr  r  zfile2.py)r   r&   r   r   r   file1r    r  source_text_file1file2source_text_file2)rT   r  s     r)   r   z)TestCacheMultipleFilesWithSignature.setUp4  s    %&;<WW\\$,,
;
$**c" 	5d$((t4	5 WW\\$,,
;
$**c" 	5d$((t4	5 	5		5 	5	5 	5s   C;C(C%(C1c                 B    t        j                  | j                         y rY   )r#   r$   r&   rZ   s    r)   r   z,TestCacheMultipleFilesWithSignature.tearDown?  s    dll#r0   c                    t        j                  t        j                  | j                  gt         j
                  t         j
                        }|j                         \  }}d|j                          d|j                          }| j                  |j                  d|       t        j                  t        j                  | j                  gt         j
                  t         j
                        }|j                         \  }}d|j                          d|j                          }| j                  |j                  d|       y )Nr   r   zstdout:
z


stderr:
r   r  )r   r   r   r   r'  r   r   r   rS   r   r)  )rT   r   r   r   r  s        r)   *test_caching_mutliple_files_with_signaturezNTestCacheMultipleFilesWithSignature.test_caching_mutliple_files_with_signatureB  s      #..$**!=(2(29 $$&S#**,}SZZ\NC))1#6   #..$**!=(2(29 $$&S#**,}SZZ\NC))1c2r0   N)	rf   rv   rw   r  r(  r*  r   r   r.  rA   r0   r)   r%  r%     s)     "
	5$3r0   r%  c                       e Zd Zej                  j                  e      Zej                  j                  ed      Z	dZ
d Zd Zd Zy)TestCFuncCachezcfunc_cache_usecases.pycfunc_caching_test_fodderc                 n   dt        | j                  | j                        z  }t        j                  t
        j                  d|gt        j                  t        j                        }|j                         \  }}|j                  dk7  r*t        d|j                   d|j                          d      y )	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   r-  r   r   z:stderr follows
r   )r   r&   r   r   r   r   r   r   r   r   r   r   )rT   r   r   r   r   s        r)   r   z&TestCFuncCache.run_in_separate_processZ  s     t||T\\BC   #..$!=(2
Q$$&Sq  #<U=M=M<N O447JJL>"E F F !r0   c                 $    |j                          y rY   )r  rT   r   s     r)   check_modulezTestCFuncCache.check_moduler  s    r0   c                 n   | j                  d       | j                         }| j                  d       | j                  |j                  j                  d       | j                  |j
                  j                  d       | j                  |j                  j                  d       | j                  |j                  j                  d       | j                  |       | j                         }| j                  d       | j                  |j                  j                  d       | j                  |j
                  j                  d       | j                  |j                  j                  d       | j                  |j                  j                  d       | j                  |       | j                          y )Nr   r   r=   )
r   r   rS   r   r   r   r   div_usecaser5  r   r4  s     r)   r   z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   N)rf   rv   rw   r   r   r   r   r   r   r   r   r   r5  r   rA   r0   r)   r0  r0  T  s>    77??8$DGGLL'@AM)GF0'r0   r0  __main__)Er  rE  rg   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   rk   r   r   r   r  r  r  r  r  r  r  r%  r0  rf   mainrA   r0   r)   <module>rG     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HMMO Q#  Is   8E. .E87E8