
    xKgd                        d dl Z d dlmZ d dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlmZ 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jP                  e&e'      Z)d
 Z*d Z+d Z,d Z-d Z.d Z/ G d dee      Z0d Z1 G d de      Z2 G d de2      Z3 G d de3      Z4e! G d de3             Z5 G d de3      Z6 G d de2      Z7e  G d d ej4                               Z8d! Z9 G d" d#ej4                        Z: G d$ d%e2      Z;e<d&k(  r ejz                          yy# e$$ r dZ#Y w xY w)'    N)njit)codegen)_UserWideCacheLocator)NumbaWarning)parfor)	TestCaseSerialMixin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      \/home/alanp/www/video.onchill/myenv/lib/python3.12/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*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-   H   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_cacher9   M   sJ    #FDtV$EhOFGgOFGVgr.   c                      dddS )N      )ab r?   r.   r'   
dict_cacher@   V   s    r.   c                  ^    t         }  t        d      |       } |        } |       }||k(  sJ y r0   )r@   r   r3   r4   expgots       r'   check_dict_cacherE   Z   s2    FDtV$E
(C
'C#::r.   c               #   "   K   dD ]  } |   y w)N)r;   r<      r?   )vs    r'   generator_cacherI   b   s      s   c                      t         }  t        d      |       }t         |              }t         |             }||k(  sJ y r0   )rI   r   listrB   s       r'   check_generator_cacherL   g   s:    FDtV$E
vx.C
uw-C#::r.   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_testp   s%    (.Z!,r.   c                 .    | j                  t               y N)rU   r9   rR   s    r'   test_constant_unicode_cachez'TestCaching.test_constant_unicode_cacheu   s    23r.   c                 .    | j                  t               y rW   )rU   rE   rX   s    r'   test_dict_cachezTestCaching.test_dict_cachex   s    &'r.   c                 .    | j                  t               y rW   )rU   rL   rX   s    r'   test_generator_cachez TestCaching.test_generator_cache{   s    +,r.   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failrQ   )rR   	cache_dirctxresult_queueprocsuccessoutputs          r'   test_omittedzTestCaching.test_omitted~   s    #4>>#:#:;	))+yy{{{-	51  
 	

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

		&**, IIf;	
r.   N)rd   
__module____qualname__rU   rY   r[   r]   rs   r?   r.   r'   rN   rN   o   s    -
4(-(
r.   rN   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_DIRTr1   c                     | S rW   r?   )nums    r'   testz(omitted_child_test_wrapper.<locals>.test   s    Jr.   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)rb   )r   r   _cache_hits
signatures_cache_misses	traceback
format_excput)ro   rm   second_callrz   rr   rq   s         r'   ri   ri      s+   	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 
1	0*	))+FG/ 
1	0s#   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modfilerm   r!   copyusecases_filer    statS_IREADS_IWRITEmaxDiffrX   s    r'   setUpzBaseCacheTest.setUp   s    %l34<<(ww||DLL$,,2FGdllMBD&&5
t||dmm;<r.   c                     t         j                  j                  | j                  d        t         j                  j                  | j                         y rW   )r   modulespopr   r   remover$   rX   s    r'   tearDownzBaseCacheTest.tearDown   s,    d+%r.   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   rQ   __file__rstripr   )rR   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   listdirrm   endswithr   )rR   r   s     r'   cache_contentszBaseCacheTest.cache_contents   sZ    	!#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rW   )r   r   getmtimer   rm   ).0r   rR   s     r'   	<genexpr>z1BaseCacheTest.get_cache_mtimes.<locals>.<genexpr>   s@      =; ))"'',,t~~r*JKL;s   AA)dictsortedr   rX   s   `r'   get_cache_mtimeszBaseCacheTest.get_cache_mtimes   s+     =$T%8%8%:;= = 	=r.   c                 \    | j                         }| j                  t        |      ||       y rW   )r   rQ   len)rR   nr,   s      r'   check_pycachezBaseCacheTest.check_pycache   s&    !QA&r.   c                      y rW   r?   rX   s    r'   
dummy_testzBaseCacheTest.dummy_test   s    r.   )rd   rt   ru   r   r   r   r   r   r   r   r   r   r?   r.   r'   r   r      s/    MG&"='r.   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)rR   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  !r.   Nc                    |j                   }| j                  t        |j                  j	                               ||j                         |?| j                  t        |j
                  j	                               ||j
                         y y rW   )statsrQ   sum
cache_hitsvaluescache_misses)rR   rS   hitsmissessts        r'   
check_hitsz&DispatcherCacheUsecasesTest.check_hits  si    ZZR]]1134dBMMJS!7!7!9:F__. r.   rW   )rd   rt   ru   r   r   dirnamer   herer   r   r   r   r   r?   r.   r'   r   r      s;    77??8$DGGLL':;M.G13 4.r.   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<   rG            @      @   r;   )r<   g     E@	   )r   r   add_usecaseassertPreciseEqualr   add_objmode_usecaserecord_returnaligned_arrtuple
packed_arrr   )rR   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  	$$&r.   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<   rG   r   r;   )r   r   r   r   nparanger   rR   r   r&   s      r'   test_caching_nrt_prunedz!TestCache.test_caching_nrt_prunedC  s    1  "1OO!Q+1!RYYq\ 2A		!4Dq4HI11a r.   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 )NrG   r<   r            @r   )r   r   innerr   outer_uncachedouterr   s      r'   test_inner_then_outerzTestCache.test_inner_then_outerP  s      "		!Q31!Q+1  "!Q+1II!Q+1#q	3/1r.   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 )NrG   r<   r   r   r   r   r   )r   r   r   r   r   r   r   s      r'   test_outer_then_innerzTestCache.test_outer_then_innerd  s      "		!Q31 2 21a 8!<1  "II!Q+1#q	3/1r.   c                     | j                         }|j                  }| j                   |dd      d       | j                  d       y )Nr<   rG   r   r   )r   add_nocache_usecaser   r   r   s      r'   test_no_cachingzTestCache.test_no_cachingr  s?      "##!Q+1r.   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   rQ   r   assertInstrmessagerR   r   wr&   s       r'   test_loopliftedzTestCache.test_loopliftedy  s      "$$D1Q!!(L9A##AaD!,q! 2 	Q# /03AaDLL0A	C 21s   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   rQ   r   r  r	  r
  r  s       r'   test_big_arrayz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                    | 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   rQ   r   r  formatrd   r	  r
  )rR   r   r&   r  s       r'   test_ctypeszTestCache.test_ctypes  s       "--!4!4c6I6IJA((5%%h=''#4""1%	 6 SVQ'MM5<<QZZHAaDLL! K55s   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errorrG   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_usecaserQ   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r.   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<   rG   r   r   r   r;   )r   r   r   r   r   r   r   r   r   r   assertIsNotrQ   r   )rR   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&9r.   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<   rG   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<   rG   r   r  r2  )r   r   r   Z	recompiler   s      r'   test_recompilezTestCache.test_recompile  s      "OO!Q+  "OO!Q+	!Q,   "OO!Q,r.   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(r.   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)rR   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<   rG   r   r   r;   N)
r   r   
addCleanupr!   r"   r   
cache_pathr   r   r   )rR   r   r&   r/  s       r'   _test_pycache_fallbackz TestCache._test_pycache_fallback!  s    
   "OO 	qww'9'9N!Q+1a  !!#!Q+1a  	1r.   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    rP  rR  )rR   	old_permss     r'   test_non_creatable_pycachez$TestCache.test_non_creatable_pycache:  sR    
 GGDLL)11	
u%$,,	:##%r.   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   rU  r    rP  rR  )rR   pycacherV  s      r'   test_non_writable_pycachez#TestCache.test_non_writable_pycacheE  sj    
 '',,t||];
GGG$,,	
% '95##%r.   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   rl   re  pr   r   base_cmdinputfnrR   s       r'   execute_with_inputz2TestCache.test_ipython.<locals>.execute_with_inputs      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  )rR   verer&   rm  r   rk  rl  s   `     @@r'   test_ipythonzTestCache.test_ipythonR  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^  r_  r`  r   )compilerr  ra  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rc  rh  ri  s       r'   rm  z4TestCache.test_ipykernel.<locals>.execute_with_input  rn  ro  rp  )r   r   r   rq  rr  rs  r   rt  r   	ipykernelry  get_tmp_directoryr   r   r  )	rR   ru  rv  ry  r&   rm  r   rk  rl  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)rd   rt   ru   r   r   r   r   r   r  r  r  r#  r+  r0  r3  r7  r;  rM  rR  skip_bad_accessunittestskipIfr   namerW  rZ  rw  r{  version_infor}  r?   r.   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5r.   r   c                   .     e Zd Z fdZ fdZd Z xZS )TestSequentialParForsCachec                 @    t         t        |           dt        _        y )NT)superr  r   r   sequential_parfor_loweringrR   rc   s    r'   r   z TestSequentialParForsCache.setUp  s    ($57,0)r.   c                 @    t         t        |           dt        _        y )NF)r  r  r   r   r  r  s    r'   r   z#TestSequentialParForsCache.tearDown  s    ($8:,1)r.   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_globalsrQ   )rR   r   r&   arycresdynamic_globalss         r'   r   z'TestSequentialParForsCache.test_caching  s      "1ggbk#c	C8'({{'9'9';=';t  <<;;'; 	 =5'21=s   =C)rd   rt   ru   r   r   r   __classcell__)rc   s   @r'   r  r    s    1
2
	r.   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)rR   
mtimes_oldmatch_countkrH   s        r'   check_later_mtimesz*TestCacheWithCpuSetting.check_later_mtimes  sh    ))+113DAqJ'':a=9q  4 	;3 	 	5r.   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_indexrQ   keysllget_host_cpu_namer   get_host_cpu_features)rR   r   
cache_sizer.  r2   
cache_filecache_indexkey_akey_bkey_hostkey_generics              r'   test_user_set_cpu_namez.TestCacheWithCpuSetting.test_user_set_cpu_name  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/r.   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  rQ   r  r  r  )rR   r   r  r.  my_cpu_featuressystem_featuresr2   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<r.   N)rd   rt   ru   _numba_parallel_test_r  r  r  r?   r.   r'   r  r    s    !506!=r.   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)NrG   spawnr;   r<   )r   simple_usecase_callerre   rf   AttributeErrorPoolr   imaprangecloserQ   )rR   r   r&   r   rn   poolrT   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)rd   rt   ru   r  r   r   r   r   r   r   r   r   r  r?   r.   r'   r  r  %  s;     "77??8$DGGLL':;M.G0r.   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)rR   foomodfouts      r'   r   zTestCacheFileCollision.setUpV  s   %&;<4<<(5s{{3 LLHHdllE%:;dllDLL9
"'',,v}5s;t$$$40 <"'',,vx0#6$$$$40 76 <;66s   E=F	=F	Fc                     t         j                  j                  | j                  d        t         j                  j                  | j                  d        t         j
                  j                  | j                         y rW   )r   r   r   r  r  r   r   r$   rX   s    r'   r   zTestCacheFileCollision.tearDownd  sF    ))40))40%r.   c                 @    t        | j                        j                  S rW   )r   r  barrX   s    r'   import_bar1z"TestCacheFileCollision.import_bar1i      d//0444r.   c                 @    t        | j                        j                  S rW   )r   r  r  rX   s    r'   import_bar2z"TestCacheFileCollision.import_bar2l  r  r.   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)rR   bar1bar2idxname1idxname2s        r'   test_file_locationz)TestCacheFileCollision.test_file_locationo  s    !!;;**66;;**66Hh/++,?@A++N;<r.   rf   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 contextr_   )r  r  r
   getvaluerQ   countr  re   rf   
ValueErrorr  rg   rh   cache_file_collision_testerr$   r  r  rj   rk   r   )rR   r  r  bufres1cachelogres2mpqrp   log1got1log2got2s                 r'   test_no_collisionz(TestCacheFileCollision.test_no_collisiony  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)rd   rt   ru   r  r   r   r   r   r   r   r   r   r  r  r   r   r  r  r  r  
skipUnlesshasattrre   r  r?   r.   r'   r  r  B  s    !77??8$DGGLL':;M'GMM1&
55= X-@DF97F97r.   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)rR   r  s     r'   r   z)TestCacheMultipleFilesWithSignature.setUp  s    %&;<WW\\$,,
;
$**c"d$((t4 # WW\\$,,
;
$**c"d$((t4 #"	 #" #"s   C;C(C%(C1c                 B    t        j                  | j                         y rW   )r!   r"   r$   rX   s    r'   r   z,TestCacheMultipleFilesWithSignature.tearDown  s    dll#r.   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   rQ   r   r  )rR   r   r   r   r  s        r'   *test_caching_mutliple_files_with_signaturezNTestCacheMultipleFilesWithSignature.test_caching_mutliple_files_with_signature  s      #..$**!=(2(29 $$&S#**,}SZZ\NC))1#6   #..$**!=(2(29 $$&S#**,}SZZ\NC))1c2r.   N)	rd   rt   ru   r  r  r  r   r   r  r?   r.   r'   r  r    s)     "
	5$3r.   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   )rR   r   r   r   r   s        r'   r   z&TestCFuncCache.run_in_separate_process  s     t||T\\BC   #..$!=(2
Q$$&Sq  #<U=M=M<N O447JJL>"E F F !r.   c                 $    |j                          y rW   )r  rR   r   s     r'   check_modulezTestCFuncCache.check_module  s    r.   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   rQ   r   r   r   r   div_usecaser  r   r  s     r'   r   zTestCFuncCache.test_caching  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#$$&r.   N)rd   rt   ru   r   r   r   r   r   r   r   r   r   r  r   r?   r.   r'   r
  r
    s>    77??8$DGGLL'@AM)GF0'r.   r
  __main__)>rC  llvmlite.bindingbindingr  re   numpyr   r   r   r!   r   r   r~   r  r  numbar   
numba.corer   numba.core.cachingr   numba.core.errorsr   numba.parforsr   numba.tests.supportr   r	   r
   r   r   r   r   r   r   r{  ImportErrorr(   _access_preventable_access_msgr  r~  r-   r9   r@   rE   rI   rL   rN   ri   r   r   r   r  r  r  r  r  r  r
  rd   mainr?   r.   r'   <module>r      s       	    
      4 *  
 
 

 > 23 I%(%%&9;G

7
+x 7
t,84H 4n$.- $.N[5+ [5| !<  .I=9 I=X0M 0: q7X.. q7 q7h13(*;*; 13h6'] 6'r zHMMO e   Is   (E EE