
    xKg                         d dl Z d dlZd dlmZ d dlmZ  edd      d        Z edd      d        Z ed      d	        Z	 G d
 de      Z
d Zd Zy)    N)njit)TestCaseT)parallelcachec                 (    | | j                         z  S N)sumarrs    f/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/tests/parfors_cache_usecases.pyarrayexprs_caser   	   s    ?    c                     t        j                  |       }d| j                         z  }t        | j                        D ]  }| |   |z  ||<    |S )N   )np
zeros_liker	   rangesize)r   outcis       r   prange_caser      sG    
--
C	CGGIA388_Q!A Jr   )r   c                 *    t        t        |             S r   )r   r   r
   s    r   caller_caser      s    s+,,r   c                       e Zd ZdZd Zd Zy)_TestModulez
    Tests for functionality of this module's functions.
    Note this does not define any "test_*" method, instead check_module()
    should be called by hand.
    c                 X   d}|j                   |j                  |j                  fD ]n  }t        j                  d      }t        j
                  j                   ||      |j                  |             |t        |j                  j                        z  }p | j                  |dd       y )Nr      z*At least one dispatcher has used the cache)msg)r   r   r   r   onestestingassert_allclosepy_funclenstats
cache_hitsassertGreater)selfmodtotal_cache_hitsfnr   s        r   check_modulez_TestModule.check_module"   s    &&IB''"+CJJ&&3C BHH$7$7 88 J 	a< 	 	
r   c                     |j                   |j                  |j                  fD ]L  }t        j                  d      }t        j
                  j                   ||      |j                  |             N y )Nr   )r   r   r   r   r    r!   r"   r#   )r(   r)   r+   r   s       r   
run_modulez_TestModule.run_module0   sT     &&IB''"+CJJ&&3C Jr   N)__name__
__module____qualname____doc__r,   r.    r   r   r   r      s    

r   r   c                  d    t         j                  t           } t               j	                  |        y r   )sysmodulesr/   r   r,   r)   s    r   	self_testr8   ;   s     
++h
CMs#r   c                  d    t         j                  t           } t               j	                  |        y r   )r5   r6   r/   r   r.   r7   s    r   self_runr:   @   s     
++h
CMS!r   )r5   numpyr   numbar   numba.tests.supportr   r   r   r   r   r8   r:   r3   r   r   <module>r>      sv    
   ( t4  ! t4  ! D- -( >$
"r   