o
    i4                    @   sN  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Zd dlmZmZmZmZmZ d dlmZmZmZmZ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' d d
l(m)Z) d dl*Z*d dl+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z= d dl>m?Z?m@Z@mAZAmBZBmCZC ddlDmEZEmFZF zd dlGmH  mIZJ W n eKy   dZJY nw G dd deLZMG dd dejNZOeOdZPeM ZQe,ReMdd ZSe/eOejTdd ZUdd ZVe4eOe3jW e6eOdd ZXdd ZYeYd\ZZZ[Z\e6eZdd  Z]dd!d"Z^d#d$ Z_e-e^e_Z`e.e^e.e^ejad%d& Zbe.e^ejcd'd( ZdeAG d)d* d*e?Zee.d+eOd,d- Zfd.d/ Zgd0d1 Zhd2d3 Zie0egd4d5 Zje0egd6d7 Zke0eld8d9 Zme0ejnd:d; Zoe0ejpd<d= Zqe0ejrd>d? Zse0ejtd@dA ZudBdC ZvdDdE Zwe0ejxdFdG ZydHdI ZzdJdK Z{dLdM Z|dNdO Z}dPdQ Z~e2eOdRdSdT ZdUdV ZdWdX ZdYdZ Zd[d\ Zd]d^ Zd_d` Zdadb Zdcdd Zdedf Zdgdh Zdidj Zdkdl Zdmdn Ze0edodp Zdqdr Zdsdt Zdudv ZeBeG dwdx dxeCZdydz ZG d{d| d|eLZG d}d~ d~ejZe4eG dd de3jWZeBeee e edd Ze0e
jdd ZG dd de"ZG dd de"ZG dd de"Zdd Ze'G dd de"Zdd ZG dd de"ZG dd de*j"ZG dd de*j"Ze2eOdddiddd Ze2eOdddiddd Ze1eOdddiddd Ze1eOdddiddd ZG dd de"ZG dd de"Zdd Ze'G dd de"ZG dd de"ZG dd de"ZG dd de"ZG dd de"Zedkr%e*  dS dS )    N)ir)njitjit	vectorizeguvectorizeobjmode)typeserrorstypingcompilercgutils)type_inference_stage)
cpu_target)compile_isolated)lower_constant)TestCasecaptured_stdouttemp_directoryoverride_configrun_in_new_process_in_cache_dirskip_if_typeguard)LoweringError)typeof_impltype_callablelower_builtin
lower_castoverloadoverload_attributeoverload_methodmodelsregister_modelboxunboxNativeValue	intrinsic
_Intrinsicregister_jitableget_cython_function_address	is_jittedoverload_classmethod)ConcreteTemplate	signatureinferinfer_globalAbstractTemplate   )IndexSeriesc                   @      e Zd ZdS )MyDummyN__name__
__module____qualname__ r8   r8   i/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/tests/test_extending.pyr3   K       r3   c                   @      e Zd Zdd ZdS )MyDummyTypec                 C   s"   t |tjrddlm} |jS d S )Nr   )
Conversion)
isinstancer   NumberZnumba.core.typeconvr=   safe)selfcontexttotyr=   r8   r8   r9   can_convert_toP   s   zMyDummyType.can_convert_toN)r5   r6   r7   rD   r8   r8   r8   r9   r<   O       r<   mydummyc                 C      t S N)mydummy_typevalcr8   r8   r9   typeof_mydummy[      rM   c                 C   s   |  |dS )z2
    Implicit conversion from MyDummy to int.
    *   )get_constant)rB   builderZfromtyrC   rK   r8   r8   r9   mydummy_to_number`   s   rR   c                   C   rG   rH   )rF   r8   r8   r8   r9   	get_dummyh      rS   c                 C      t |j S rH   r#   rB   get_dummy_valuetypobjrL   r8   r8   r9   unbox_indexo      r[   c                    sX   G dd dt }G dd dtj}||  t| fdd}t|tj || fS )Nc                   @   r2   )z(base_dummy_type_factory.<locals>.DynTypeNr4   r8   r8   r8   r9   DynTypey   r:   r]   c                   @   r2   )z,base_dummy_type_factory.<locals>.DynTypeTypeNr4   r8   r8   r8   r9   DynTypeType|   r:   r^   c                        S rH   r8   rJ   dyn_type_typer8   r9   rM      rN   z/base_dummy_type_factory.<locals>.typeof_mydummy)objectr   Opaquer   registerr    r   OpaqueModel)namer]   r^   rM   r8   r`   r9   base_dummy_type_factoryx   s   
rg   mydummy2c                 C   rU   rH   rV   rX   r8   r8   r9   unbox_index2   r\   ri   c                 C      t rH   NotImplementedErrorxr8   r8   r9   func1   rT   ro   c                 C   s   ddd}|S )Nc                 S   s(   | d t jfv r
t jS t| t jr| S d S rH   )r   noneint32r>   Floatrm   r8   r8   r9   typer   s
   ztype_func1_.<locals>.typerrH   r8   )rB   rs   r8   r8   r9   type_func1_   s   
rt   c                 C   s   |  |jdS NrO   )rP   return_type)rB   rQ   sigargsr8   r8   r9   func1_nullary   s   ry   c                 C   s   dd }|  ||||S )Nc                 S   s   t d|  S )N   )mathsqrtrm   r8   r8   r9   
func1_impl      zfunc1_unary.<locals>.func1_impl)Zcompile_internal)rB   rQ   rw   rx   r}   r8   r8   r9   func1_unary   s   r   c                   @   s   e Zd ZdZeejegZdS )
PrintDummy
print_itemN)	r5   r6   r7   keyr+   r   rp   rI   casesr8   r8   r8   r9   r      s    r   r   c                 C   s<   |\}|  |}||d}|| || |  S )Nzhello!)Zget_python_apiZunserializeZserialize_objectZprint_objectZdecrefrW   )rB   rQ   rw   rx   rn   pyapiZstrobjr8   r8   r9   print_dummy   s   


r   c                 C   rj   rH   rk   condrn   yr8   r8   r9   where   rT   r   c                 C   s   t | ||S )z8
    Wrap np.where() to allow for keyword arguments
    )npr   r   r8   r8   r9   np_where   s   r   c                 C   s   t | ||dS )N)r   rn   )r   r   r8   r8   r9   
call_where   r~   r   c                 C   sV   t | tjr)|j|jkrtdtdd | ||fD r#dd }|S dd }|S dS )z'
    Implement where() for arrays.
    "x and y should have the same dtypec                 s   s    | ]}|j d kV  qdS )CN)Zlayout).0tyr8   r8   r9   	<genexpr>   s    z(overload_where_arrays.<locals>.<genexpr>c           
      S   st   | j }|j |ks|j |krtdt|}| j}|j}|j}|j}t| jD ]}	||	 r1||	 n||	 ||	< q'|S )zM
                Fast implementation for C-contiguous arrays
                %all inputs should have the same shape)shape
ValueErrorr   
empty_likeZflatrangesize)
r   rn   r   r   rescfZxfZyfrfir8   r8   r9   
where_impl   s   
z)overload_where_arrays.<locals>.where_implc                 S   s\   | j }|j |ks|j |krtdt|}t| D ]\}}|r%|| n|| ||< q|S )zI
                Generic implementation for other arrays
                r   )r   r   r   r   Zndenumerate)r   rn   r   r   r   idxrL   r8   r8   r9   r      s   
N)r>   r   Arraydtyper	   TypingErrorallr   rn   r   r   r8   r8   r9   overload_where_arrays   s   
r   c                 C   s.   t | tjs||krtddd }|S dS )z(
    Implement where() for scalars.
    z!x and y should have the same typec                 S   s"   | r|n|}t |}||d< |S )z@
            Scalar where() => return a 0-dim array
            r8   )r   r   )r   rn   r   Zscalarrr8   r8   r9   r     s   
z*overload_where_scalars.<locals>.where_implN)r>   r   r   r	   r   r   r8   r8   r9   overload_where_scalars  s   

r   c                 C      t | trdd }|S d S )Nc                 S      dS )N   r8   argr8   r8   r9   len_impl.  rT   z$overload_len_dummy.<locals>.len_implr>   r<   )r   r   r8   r8   r9   overload_len_dummy*  s   
r   c                 C   0   t | ttfrt |ttfrdd }|S d S d S )Nc                 S   r   ru   r8   arg1arg2r8   r8   r9   dummy_add_impl:  rT   z*overload_add_dummy.<locals>.dummy_add_implr>   r<   MyDummyType2)r   r   r   r8   r8   r9   overload_add_dummy4     r   c                 C   *   t | trt |tjrdd }|S d S d S )Nc                 S   s   t d| | d S )NdelprintrZ   r   r8   r8   r9   dummy_delitem_implD     z2overload_dummy_delitem.<locals>.dummy_delitem_implr>   r<   r   Integer)rZ   r   r   r8   r8   r9   overload_dummy_delitem@     r   c                 C   r   )Nc                 S   s   |d S )N{   r8   r   r8   r8   r9   dummy_getitem_implN     z2overload_dummy_getitem.<locals>.dummy_getitem_implr   )rZ   r   r   r8   r8   r9   overload_dummy_getitemJ  r   r   c                 C   s4   t t| tt|tjt|tjgrdd }|S d S )Nc                 S   s   t || d S rH   r   rZ   r   rK   r8   r8   r9   dummy_setitem_impl^  r~   z2overload_dummy_setitem.<locals>.dummy_setitem_impl)r   r>   r<   r   r   )rZ   r   rK   r   r8   r8   r9   overload_dummy_setitemT  s   

r   c                 C      t | |S rH   operatoraddr   r8   r8   r9   call_add_operatord     r   c                 C      | | S rH   r8   r   r8   r8   r9   call_add_binoph  r   r   c                 C   r   )Nc                 S   r   ru   r8   r   r8   r8   r9   dummy_iadd_implr  rT   z,overload_iadd_dummy.<locals>.dummy_iadd_implr   )r   r   r   r8   r8   r9   overload_iadd_dummyl  r   r   c                 C   r   rH   r   r   r8   r8   r9   call_iadd_operatorx  r   r   c                 C   s   | |7 } | S rH   r8   r   r8   r8   r9   call_iadd_binop|  s   r   c                 C   s
   | |= d S rH   r8   r   r8   r8   r9   call_delitem     
r   c                 C      | | S rH   r8   r   r8   r8   r9   call_getitem  r   r   c                 C   s   || |< d S rH   r8   r   r8   r8   r9   call_setitem  r   r   lengthc                 C      dd }|S )Nc                 S      t | S rH   lenr   r8   r8   r9   imp  r   z#overload_method_length.<locals>.impr8   )r   r   r8   r8   r9   overload_method_length     r   c                 C      |   S rH   )r   rm   r8   r8   r9   cache_overload_method_usecase  r   r   c                   C      t  S rH   ro   r8   r8   r8   r9   call_func1_nullary     r   c                 C   r   rH   r   rm   r8   r8   r9   call_func1_unary  r   r   c                 C   r   rH   r   rm   r8   r8   r9   len_usecase  r   r   c                 C   s   t |  d S rH   r   rm   r8   r8   r9   print_usecase  r   r   c                 C   r   rH   r8   )rn   r   r8   r8   r9   getitem_usecase  r   r   c                 C   s   t t | S rH   )r   cossinrm   r8   r8   r9   npyufunc_usecase  r   r   c                 C      | j S rH   )_datarm   r8   r8   r9   get_data_usecase  r   r   c                 C   r   rH   )_indexrm   r8   r8   r9   get_index_usecase  r   r   c                 C   r   rH   )Zis_monotonic_increasingrm   r8   r8   r9   is_monotonic_usecase  r   r   c                 C   s
   t | |S rH   )r1   )dataindexr8   r8   r9   make_series_usecase  r   r   c                 C   s   |  ||S rH   )Zclip)rn   lohir8   r8   r9   clip_usecase  r   r   c                   C   rG   rH   r   r8   r8   r8   r9   return_non_boxable  rT   r   c                  C      dd } | S )Nc                   S   rG   rH   r   r8   r8   r8   r9   r     rT   z(overload_return_non_boxable.<locals>.impr8   )r   r8   r8   r9   overload_return_non_boxable  r   r   c                 C   s   t  }|| S rH   )r   arange)szmodr8   r8   r9   non_boxable_ok_usecase  s   
r  c                   C   r   rH   )r   r8   r8   r8   r9   non_boxable_bad_usecase  r   r  c                 C      d S rH   r8   )fr8   r8   r9   mk_func_input  rT   r  c                   @   r;   )MkFuncTypingc                 C   s&   t |d tjs
J ttjg|R  S Nr   )r>   r   MakeFunctionLiteralr+   rp   )rA   rx   kwsr8   r8   r9   generic  s   zMkFuncTyping.genericN)r5   r6   r7   r  r8   r8   r8   r9   r        r  c                   C   s   t dd  d S )Nc                 S      | S rH   r8   ar8   r8   r9   <lambda>      z#mk_func_test_impl.<locals>.<lambda>)r  r8   r8   r8   r9   mk_func_test_impl  r   r  c                   @   r2   )MyClassNr4   r8   r8   r8   r9   r    r:   r  c                       s4   e Zd Z fddZdd Zdd Zdd Z  ZS )	CallableTypeRefc                    s.   || _ i | _g | _tt| d| j  d S )Nzcallable_type_ref[{}])instance_typesig_to_impl_keycompiled_templatessuperr  __init__format)rA   r  	__class__r8   r9   r    s   zCallableTypeRef.__init__c              	   C   st   d }|j t|  D ].}z|||}W n	 ty   Y q	w t|di }||v r7|| | j|< | j|  |S q	|S )NZ_compiled_overloads)Z
_functionstypeapply	Exceptiongetattrr  r  append)rA   rB   rx   r
  Zres_sigtemplateZcompiled_ovldsr8   r8   r9   get_call_type  s   zCallableTypeRef.get_call_typec                 C   s   t | j }|dfS )NT)listr  keys)rA   Zsigsr8   r8   r9   get_call_signatures  s   z#CallableTypeRef.get_call_signaturesc                 C   s
   | j | S rH   )r  )rA   rw   r8   r8   r9   get_impl_key  r   zCallableTypeRef.get_impl_key)r5   r6   r7   r  r#  r&  r'  __classcell__r8   r8   r  r9   r    s
    r  c                   @   r;   )CallableTypeModelc                 C   s   t j| || d S rH   )r   re   r  )rA   ZdmmZfe_typer8   r8   r9   r       zCallableTypeModel.__init__N)r5   r6   r7   r  r8   r8   r8   r9   r)    s    r)  c                 C   r   rH   )rW   )rB   rQ   r   Zpyvalr8   r8   r9   constant_callable_typeref"     r+  c                 C   r   )Nc                 S   r   )N   >[= r8   rZ   r8   r8   r9   r   .  rN   zoverload_np_exp.<locals>.impr   rZ   r   r8   r8   r9   overload_np_exp*  s   
r0  c                   @   s8   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdS )TestLowLevelExtendingz4
    Test the low-level two-tier extension API.
    c                 C   sV   t }tdd|}| | d t}tdd|}| |d d | |dd d S )NTnopythonrO         2@      @)r   r   assertPreciseEqualr   rA   pyfunccfuncr8   r8   r9   
test_func1=  s   z TestLowLevelExtending.test_func1c                 C   sF   t }t|d}| | d t}t|tjf}| |dd d S )Nr8   rO   r4  r5  )r   r   r6  entry_pointr   r   float64rA   r8  crr8   r8   r9   test_func1_isolatedF  s   
z)TestLowLevelExtending.test_func1_isolatedc                 C   s   |  tt | t d S rH   )assertIs
type_func1rt   assertIsNotNonerA   r8   r8   r9   !test_type_callable_keeps_functionN  s   z7TestLowLevelExtending.test_type_callable_keeps_functionc                 C   s&   t }t|dtj}| | d d S )Nr8   g      E@)rS   r   r   r<  r6  r;  r=  r8   r8   r9   test_cast_mydummyR  s   z'TestLowLevelExtending.test_cast_mydummyc                 C   sX   t t}tj}tj}|  |  t|||dd}| t	dd |j
 D  dS )zFmake sure make_function is passed to typer class as a literal
        r8   Nc                 s   s    | ]	}t |tjV  qd S rH   )r>   r   r	  )r   r  r8   r8   r9   r   b  s
    

z=TestLowLevelExtending.test_mk_func_literal.<locals>.<genexpr>)r   Zrun_frontendr  r   Ztyping_contextZtarget_contextrefreshr   
assertTrueanytypemapvalues)rA   Ztest_ir	typingctxZ	targetctxZ
typing_resr8   r8   r9   test_mk_func_literalW  s   

z*TestLowLevelExtending.test_mk_func_literalN)	r5   r6   r7   __doc__r:  r?  rD  rE  rL  r8   r8   r8   r9   r1  5  s    	r1  c                   @   s`   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd ZdS )TestPandasLikeze
    Test implementing a pandas-like Index object.
    Also stresses most of the high-level API.
    c                 C   s0   t td}tddt}| ||d d S )N   Tr2  )r0   r   r   r   r   r6  )rA   r   r9  r8   r8   r9   test_index_leno  s   zTestPandasLike.test_index_lenc                 C   sf   t tg d}tddt}| ||dd ||tdd }| |t  | t	|ddg d S )NrO      Tr2  r/   rR  rS  )
r0   r   rq   r   r   r6  sliceassertIsInstanceassertEqualr$  rA   r   r9  iir8   r8   r9   test_index_getitemt  s   z!TestPandasLike.test_index_getitemc                 C   sT   t tg d}tddt}||}| |t  | |jtt	|j dS )z7
        Check Numpy ufunc on an Index object.
        rQ  Tr2  N)
r0   r   rq   r   r   rU  r6  r   r   r   rW  r8   r8   r9   test_index_ufunc|  s
    zTestPandasLike.test_index_ufuncc                 C   s:   t tg d}tddt}||}| ||j d S )NrQ  Tr2  )r0   r   rq   r   r   r@  r   )rA   r   r9  r   r8   r8   r9   test_index_get_data  s   z"TestPandasLike.test_index_get_datac                 C   s\   t ddt}g ddfg ddfg dffD ]\}}tt|}||}| || qd S )NTr2  )rR  rO      F)r\  rR  rO   )r   r   r0   r   rq   rV  )rA   r9  rJ  expectedr   gotr8   r8   r9   test_index_is_monotonic  s   

z&TestPandasLike.test_index_is_monotonicc                 C   sH   t tg d}ttg d|}tddt}| ||d d S )Nrz      rO  g      ?g      @      @Tr2  rO  )r0   r   rq   r1   r<  r   r   r6  )rA   r   sr9  r8   r8   r9   test_series_len  s   zTestPandasLike.test_series_lenc                 C   s\   t tg d}ttg d|}tddt}||}| |t  | |j	|j	 d S )Nr`  rb  Tr2  )
r0   r   rq   r1   r<  r   r   rU  r@  r   )rA   r   rd  r9  r^  r8   r8   r9   test_series_get_index  s   z$TestPandasLike.test_series_get_indexc                 C   s   t tg d}ttg d|}tddt}||}| |t | |jt  | 	|jj
|j
 | |jtt|j dS )z8
        Check Numpy ufunc on an Series object.
        rQ  r/   rz   rO  Tr2  N)r0   r   rq   r1   int64r   r   rU  r   r@  r   r6  _valuesr   r   rA   r   rd  r9  ssr8   r8   r9   test_series_ufunc  s    z TestPandasLike.test_series_ufuncc                 C   sv   t tg d}tg d}tddt}|||}| |t | |jt  | 	|jj
|j
 | 	|j| d S )NrQ  rb  Tr2  )r0   r   rq   r<  r   r   rU  r1   r   r@  r   ri  )rA   r   dr9  r^  r8   r8   r9   test_series_constructor  s   
z&TestPandasLike.test_series_constructorc                 C   s   t tg d}ttg d|}tddt}||dd}| |t | |jt  | 	|jj
|j
 | |jtg d d S )NrQ  rb  Tr2  皙?      @)ro  rp  rc  )r0   r   rq   r1   r<  r   r   rU  r   r@  r   r6  ri  rj  r8   r8   r9   test_series_clip  s   zTestPandasLike.test_series_clipN)r5   r6   r7   rM  rP  rY  rZ  r[  r_  re  rf  rl  rn  rq  r8   r8   r8   r9   rN  i  s    

rN  c                   @   s   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*S )+TestHighLevelExtendingz+
    Test the high-level combined API.
    c                    s   t }tdd|  fdd}|dddd |ddd |tg dtg d	tg d
 tj} tg tg tg  W d   n1 sRw   Y  	dt
|j dS )z>
        Test implementing a function with @overload.
        Tr2  c                     s,   t | i |} | i |}|| d S rH   )r   r6  )rx   kwargsr]  r^  r9  rA   r8   r9   check  s   z0TestHighLevelExtending.test_where.<locals>.checkrO  rR  )rn   r   r   )TFTrg  )ra  r\  r\  Nr   )r   r   r   Zbool_rq   assertRaisesr	   r   rh  assertInstr	exception)rA   r8  ru  raisesr8   rt  r9   
test_where  s     
z!TestHighLevelExtending.test_wherec                 C   s8   t ddt}| |t d | |ddgd dS )zN
        Test re-implementing len() for a custom type with @overload.
        Tr2  r   ra  r\  rz   N)r   r   r6  r3   rA   r9  r8   r8   r9   test_len  s   zTestHighLevelExtending.test_lenc                 C   sV   t ddt}t  |t  | tj d W d   dS 1 s$w   Y  dS )zP
        Test re-implementing print() for a custom type with @overload.
        Tr2  zhello!
N)r   r   r   r3   rV  sysstdoutgetvaluer|  r8   r8   r9   
test_print  s
   
"z!TestHighLevelExtending.test_printc                 C   T   t }tdd|}| |ddd | |t t d | |t t d dS 	zW
        Test re-implementing operator.add() for a custom type with @overload.
        Tr2  r/   rz   rO  rO   T   N)r   r   r6  MyDummy2r3   r7  r8   r8   r9   test_add_operator  
   z(TestHighLevelExtending.test_add_operatorc                 C   r  	zY
        Test re-implementing '+' for a custom type via @overload(operator.add).
        Tr2  r/   rz   rO  rO   r  N)r   r   r6  r  r3   r7  r8   r8   r9   test_add_binop	  r  z%TestHighLevelExtending.test_add_binopc                 C   r  r  )r   r   r6  r  r3   r7  r8   r8   r9   test_iadd_operator  r  z)TestHighLevelExtending.test_iadd_operatorc                 C   r  r  )r   r   r6  r  r3   r7  r8   r8   r9   test_iadd_binop%  r  z&TestHighLevelExtending.test_iadd_binopc                 C   s   t }tdd|}t }d }t $}z||d W n ty- } z|}W Y d }~nd }~ww W d    n1 s8w   Y  |d urC|| | d d S )NTr2  A  zdel hello! 321
)r   r   r3   r   r  rV  r  rA   r8  r9  rZ   eoutexcr8   r8   r9   test_delitem3  s    z#TestHighLevelExtending.test_delitemc                 C   s*   t }tdd|}| |t dd d S )NTr2  r  i  )r   r   r6  r3   r7  r8   r8   r9   test_getitemC  s   z#TestHighLevelExtending.test_getitemc                 C   s   t }tdd|}t }d }t %}z||dd W n ty. } z|}W Y d }~nd }~ww W d    n1 s9w   Y  |d urD|| | d d S )NTr2  r  r   z321 123
)r   r   r3   r   r  rV  r  r  r8   r8   r9   test_setitemH  s    z#TestHighLevelExtending.test_setitemc           	      C   s   t ddt}d}||}t|}tj|| t ddt}| t}|  W d   n1 s2w   Y  t|j	}d}| 
|| dS )zW
        Test overloading whose return value cannot be represented in CPython.
        Tr2  
   Nzcannot convert native Module)r   r  r   testingZassert_equalr  rv  	TypeErrorrx  ry  rw  )	rA   Zok_cfuncnr^  expectZ	bad_cfuncrz  errmsgZ	expectmsgr8   r8   r9   test_no_cpython_wrapperX  s   
z.TestHighLevelExtending.test_no_cpython_wrapperc                 C   s  d'dd}d}d(dd}|  tj}||dd	d
d W d   n1 s&w   Y  t|j}| || | d| | d| | d| d'dd}|  tj}||dd	d
d W d   n1 sgw   Y  t|j}| || | d| | d| | d| d'dd}|  tj}||dd	d
d W d   n1 sw   Y  t|j}| || | d| | d|v  | d| | d| ddlm}m	}	 |  tj}||dd	d
d W d   n1 sw   Y  t|j}| || | d| | d|v  | d| |  tj}||	dd	d
d W d   n	1 s1w   Y  t|j}| || | d| | d|v  | d| | d| d'dd}
|  tj}||
dd	d
d W d   n	1 szw   Y  t|j}| || | d| | d|v  | d| | d| d'dd }|  tj}||dd	d
d W d   n	1 sw   Y  t|j}| || | d| | d|v  | d!| d)d"d#}|  tj}||dd	d
d W d   n	1 sw   Y  t|j}| || | d| | d$| d%d& }|  tj}||dd	d
d W d   n	1 sAw   Y  t|j}| || | d| | d| dS )*z}
        Tests that an overload which has a differing typing and implementing
        signature raises an exception.
        Nc                    s:   d	ddt d	 fdd	}tddfdd}|S )
Nc                 S   r  rH   r8   r  brL   kwr8   r8   r9   
myoverloadr  rT   zjTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_ol.<locals>.myoverloadc                    r_   rH   r8   r  )implr8   r9   _myoverload_implu  rN   zpTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_ol.<locals>._myoverload_implTr2  c                    s    | |||d d S )Nr  r8   )r  r  rL   rm  r  r8   r9   fooy  r*  zcTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_ol.<locals>.foorH   r   r   )r  r  r  r8   )r  r  r9   gen_olq  s   
zVTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_olz-Typing and implementation arguments differ in   c                 S      | dkrdS dS Nr  r/   r8   r  r8   r8   r9   impl1     zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl1r/   rz   rO  ra  zkeyword argument default valuesz<Parameter "kw=12">z<Parameter "kw=None">c                 S   r  r  r8   )r  r  rL   kwargr8   r8   r9   impl2  r  zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl2zkeyword argument namesz<Parameter "kwarg=None">c                 S   s   t dkrdS dS r  r  )zr  rL   r  r8   r8   r9   impl3  r  zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl3zargument nameskeywordz<Parameter "a">z<Parameter "z">)impl4impl5zFirst difference: 'z'c                 S   r  r  r8   )r  r  rL   rm  r  r  r8   r8   r9   impl6  r  zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl6z<Parameter "d">z<Parameter "e">c                 S   r  r  r8   )r  r  r  r8   r8   r9   impl7  r  zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl7z<Parameter "c">c                 S   r  r  r8   )r  r  rL   r  Zextra_kwargr8   r8   r9   impl8  r  zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl8z<Parameter "extra_kwarg=None">c                 S   r  r  r8   )r  r  rL   r8   r8   r9   impl9  r  zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl9rH   )r  )NN)
rv  r	   r   rx  ry  rw  assertFalseoverload_usecasesr  r  )rA   r  sentinelr  r  msgr  r  r  r  r  r  r  r  r8   r8   r9   /test_typing_vs_impl_signature_mismatch_handlingk  s   















zFTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handlingc                    s   ddd ddl m} t | tdd fdd	}| tj}|dd
 W d   n1 s2w   Y  t|j}| 	d| | 	d| dS )z
        Tests that an overload which has a differing typing and implementing
        signature raises an exception and uses VAR_POSITIONAL (*args) in typing
        Nc                 S   r  rH   r8   r  r  r8   r8   r9   r    rT   ziTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positional.<locals>.myoverloadr/   )var_positional_implTr2  c                    s    | |dddS )N	      r  r8   r  r  r  r8   r9   r  
     zbTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positional.<locals>.foor\  z)VAR_POSITIONAL (e.g. *args) argument kindz-offending argument name is '*star_args_token'rH   )
r  r  r   r   rv  r	   r   rx  ry  rw  )rA   r  r  r  r  r8   r  r9   >test_typing_vs_impl_signature_mismatch_handling_var_positional  s   

zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positionalc                 C   s   ddd}dd }||ddd |  tj}||dd W d	   n1 s)w   Y  t|j}| d
| | d| ddd}|  tj}||dd W d	   n1 s\w   Y  t|j}| d
| | d| d	S )zJ
        Tests that an overload which uses **kwargs (VAR_KEYWORD)
        Tc                    s4   ddd t  |d|  tdd fdd}|S )	Nc                 S   r  rH   r8   r  r8   r8   r9   r    rT   zvTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.gen_ol.<locals>.myoverloadstrictTr2  c                    s    | ddS )Nr  r  r8   r  r  r8   r9   r       zoTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.gen_ol.<locals>.foorH   r  )r  r  r  r8   r  r9   r    s
   
zbTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.gen_olc                 [      ddd}|S )Nr  c                 S   r  rH   r8   r  r8   r8   r9   r  '  rT   zmTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol1.<locals>.implr  r8   )r  r
  r  r8   r8   r9   ol1&  s   
z_TestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol1Fr/   rz   Nz1use of VAR_KEYWORD (e.g. **kwargs) is unsupportedz"offending argument name is '**kws'r   c                 S      dd }|S )Nc                 [   r  rH   r8   )r  r
  r8   r8   r9   r  5  rT   zmTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol2.<locals>.implr8   )r  r  r  r8   r8   r9   ol24  s   z_TestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol2)T)r   )rv  r	   r   rx  ry  rw  )rA   r  r  r  r  r  r8   r8   r9   ;test_typing_vs_impl_signature_mismatch_handling_var_keyword  s    



zRTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keywordc                 C   s@   t tjdd	dd}tdd }td}| ||d d S )
Nr  r  c                 S   r  )Nr  c                 S      |S rH   r8   )r   a_kwargr8   r8   r9   r  D  rT   zQTestHighLevelExtending.test_overload_method_kwargs.<locals>.fooimpl.<locals>.implr  r8   )r   r  r  r8   r8   r9   fooimplB  s   
zCTestHighLevelExtending.test_overload_method_kwargs.<locals>.fooimplc                 S   s   |   |  d| j ddfS )N      )r  r  Ar8   r8   r9   barI  s   z?TestHighLevelExtending.test_overload_method_kwargs.<locals>.barr\  )r  r  r  r  )r   r   r   r   r   r   rV  )rA   r  r  Zr8   r8   r9   test_overload_method_kwargs@  s   


z2TestHighLevelExtending.test_overload_method_kwargsc                 C   sF   t tjddd }tdd }td}|| | ||d d S )Nlitfooc                 S   s,   t |tjrt |tjsdd }|S d S d S )Nc                 S   r  rH   r8   )r   rK   r8   r8   r9   r  Z  rT   zXTestHighLevelExtending.test_overload_method_literal_unpack.<locals>.litfoo.<locals>.impl)r>   r   r   Literal)r   rK   r  r8   r8   r9   r  S  s   zJTestHighLevelExtending.test_overload_method_literal_unpack.<locals>.litfooc                 S   s
   |  dS N  )r  r  r8   r8   r9   r  _     
zGTestHighLevelExtending.test_overload_method_literal_unpack.<locals>.barr/   r  )r   r   r   r   r   ZzerosrV  )rA   r  r  r  r8   r8   r9   #test_overload_method_literal_unpackQ  s   



z:TestHighLevelExtending.test_overload_method_literal_unpackc                 C   s   t dd }| | d d S )Nc                   S   s
   t tS rH   )r   exprF   r8   r8   r8   r9   testk  r  z8TestHighLevelExtending.test_overload_ufunc.<locals>.testr-  )r   rV  )rA   r  r8   r8   r9   test_overload_ufuncg  s   
z*TestHighLevelExtending.test_overload_ufuncc                 C   s   t tddd }tdd }t }| ||ddd | ||ddd	d
 | ||ddd	dd tdd }| ||d | ||dddd | ||dddd | ||dddd d S )Nmethod_starargc                 W   s   dd }|S )Nc                 W   s
   |||fS rH   r8   )rZ   rK   val2rx   r8   r8   r9   gett  r   z\TestHighLevelExtending.test_overload_method_stararg.<locals>._ov_method_stararg.<locals>.getr8   )rZ   rK   r  rx   r  r8   r8   r9   _ov_method_starargr  r   zOTestHighLevelExtending.test_overload_method_stararg.<locals>._ov_method_starargc                 W   s
   | j | S rH   r  )rZ   rx   r8   r8   r9   r  y  s   
z@TestHighLevelExtending.test_overload_method_stararg.<locals>.foor/   rz   r/   rz   r8   rO  r/   rz   rO  ra  r/   rz   rO  ra  c                 S   s(   |  dd|  ddd|  ddddfS )Nr/   rz   rO  ra  r  r.  r8   r8   r9   r    s   
z@TestHighLevelExtending.test_overload_method_stararg.<locals>.bar)r  r  r  r  )r/   rz   )r  r  )r/   rz   )r  rO  )ra  r\  )r/   rz   )r  r   r<   r   r3   rV  )rA   r  r  rZ   r  r8   r8   r9   test_overload_method_starargq  s,   


z3TestHighLevelExtending.test_overload_method_starargc                    s   G dd dt j t ddd }t fdd}d}| ||t| td	d
 }| tj	}|| W d    n1 sBw   Y  | 
dt|j d S )Nc                   @   r2   )zATestHighLevelExtending.test_overload_classmethod.<locals>.MyArrayNr4   r8   r8   r8   r9   MyArray  r:   r  array_allocc                 S   r  )Nc                 S   s   t |}|S rH   r   r   )clsnitemsr   r8   r8   r9   r    s   
zVTestHighLevelExtending.test_overload_classmethod.<locals>.ol_array_alloc.<locals>.implr8   )r  r  r  r8   r8   r9   ol_array_alloc  r   zHTestHighLevelExtending.test_overload_classmethod.<locals>.ol_array_allocc                    s
     | S rH   )r  r  r  r8   r9   r    r  z=TestHighLevelExtending.test_overload_classmethod.<locals>.foor   c                 S   s   t j| S rH   )r   r   r  r  r8   r8   r9   no_classmethod_in_base  r  zPTestHighLevelExtending.test_overload_classmethod.<locals>.no_classmethod_in_basez"Unknown attribute 'array_alloc' of)r   r   r)   r   r6  r   r   rv  r	   r   rw  rx  ry  )rA   r  r  r  r  rz  r8   r  r9   test_overload_classmethod  s    


z0TestHighLevelExtending.test_overload_classmethodc                 C   s\   t tdd }t tdd }tdd }d}d|d	  d
|d  f}| || | d S )Nc                 S      t | tjrdd }|S d S )Nc                 S      d|  S )N     @E@r8   rm   r8   r8   r9   r    r   zeTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld1.<locals>.impl)r>   r   r   rn   r  r8   r8   r9   callable_type_call_ovld1  s   zWTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld1c                 S   r  )Nc                 S   r  )N42.5r8   rm   r8   r8   r9   r    r   zeTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld2.<locals>.impl)r>   r   ZUnicodeTyper  r8   r8   r9   callable_type_call_ovld2  s   zWTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld2c                 S   s   t | t |fS rH   )r  r  r8   r8   r9   r    r  zBTestHighLevelExtending.test_overload_callable_typeref.<locals>.foo)ra  4r  r   r  r/   )r   r  r   r6  )rA   r  r  r  rx   r]  r8   r8   r9   test_overload_callable_typeref  s   


z5TestHighLevelExtending.test_overload_callable_typerefN)r5   r6   r7   rM  r{  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r8   r8   r8   r9   rr    s.    	 ,
+ rr  c                 C   sD   | j | jd  }||krtd| j| jd  }||kr tdd S )Nr   zcache not used)Z_cache_hitsZ
signaturesAssertionErrorZ_cache_misses)r9  Z
expect_hitZexpect_misseshitZmissr8   r8   r9   _assert_cache_stats  s   r  c                   @   s    e Zd ZdZdd Zdd ZdS )TestOverloadMethodCachingFc                 C   sH   t | jj| _td| j |   W d    d S 1 sw   Y  d S )N	CACHE_DIR)r   r  r5   
_cache_dirr   run_caching_overload_methodrC  r8   r8   r9   test_caching_overload_method  s   
"z6TestOverloadMethodCaching.test_caching_overload_methodc                 C   s   t dddt}| |t d t|dd |tf}dd | D }| t	|d zt
d}W n ty@   t
}Y nw | }|jt|| jfd	}|  |t  |  | |jd |jdd
}| |d d S )NTr3  cacher   r   r/   c                 S   s"   g | ]}| d rd|v r|qS )Zdeclarer   )
startswith)r   lnr8   r8   r9   
<listcomp>  s    zITestOverloadMethodCaching.run_caching_overload_method.<locals>.<listcomp>Zspawn)targetrx   )timeout)r   r   r6  r3   r  Zinspect_llvmrI   
splitlinesrV  r   multiprocessingZget_contextAttributeErrorQueueProcessr  r  startputjoinexitcoder  )rA   r9  ZllvmirZdeclsctxqpr   r8   r8   r9   r    s.   
z5TestOverloadMethodCaching.run_caching_overload_methodN)r5   r6   r7   _numba_parallel_test_r  r  r8   r8   r8   r9   r    s    r  c                 C   sf   t d|$ |  }tdddt}||}| | t|dd W d   dS 1 s,w   Y  dS )zH
    Used by TestOverloadMethodCaching.test_caching_overload_method
    r  Tr  r/   r   N)r   r  r   r   r  r  )r  	cache_dirr   r9  r   r8   r8   r9   r    s   
"r  c                   @   4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestIntrinsicc                    s   t dd t dd  tddfdd}tdd fd	d
}| | d | t}|  W d   n1 s<w   Y  | d|jj dS )z
        Verify that returning a None from codegen function is handled
        automatically for void functions, otherwise raise exception.
        c                 S      t t j}dd }||fS )Nc                 S   r  rH   r8   rB   rQ   r+   rx   r8   r8   r9   codegen   rT   zBTestIntrinsic.test_void_return.<locals>.void_func.<locals>.codegenr   voidrq   rK  r  rw   r!  r8   r8   r9   	void_func  s   z1TestIntrinsic.test_void_return.<locals>.void_funcc                 S   s   t t j}dd }||fS )Nc                 S   r  rH   r8   r   r8   r8   r9   r!  *  rT   zFTestIntrinsic.test_void_return.<locals>.non_void_func.<locals>.codegen)r   rq   r$  r8   r8   r9   non_void_func&  s   z5TestIntrinsic.test_void_return.<locals>.non_void_funcTr2  c                          d dS Nr/   r   r8   r8   )r%  r8   r9   call_void_func/     z6TestIntrinsic.test_void_return.<locals>.call_void_funcc                      r'  r(  r8   r8   )r&  r8   r9   call_non_void_func4  r*  z:TestIntrinsic.test_void_return.<locals>.call_non_void_funcr   Nznon-void function returns None)r$   r   rV  rv  r   rw  ry  r  )rA   r)  r+  r  r8   )r&  r%  r9   test_void_return  s   
	
zTestIntrinsic.test_void_returnc                    s   ddl m}m}m}m} fdddd tddfd	d
}dd }|d|||}|| tdd fdd}tjdtj	d}	t
 }
||	 |
  }W d   n1 s^w   Y  |
  ttt|	}|| dS )zV
        Usecase test: custom reinterpret cast to turn int values to pointers
        r   )	CFUNCTYPEPOINTERc_floatc_intc                    s&   t  tjsJ t fdd}|S )Nc                    s8    | tj t|tjr tj}dd }||fS d S )Nc                 S   s"   |\}|j }| |}|||S rH   )rv   get_value_typeZinttoptr)rB   rQ   r+   rx   srcZrtypellrtyper8   r8   r9   r!  Q  s   
z_TestIntrinsic.test_ll_pointer_cast.<locals>.unsafe_caster.<locals>.unsafe_cast.<locals>.codegen)rU  r
   Contextr>   r   r   Zuintp)rK  r2  rw   r!  )result_typerA   r8   r9   unsafe_castJ  s   
zNTestIntrinsic.test_ll_pointer_cast.<locals>.unsafe_caster.<locals>.unsafe_cast)r>   r   CPointerr$   )r5  r6  rC  )r5  r9   unsafe_casterG  s   z9TestIntrinsic.test_ll_pointer_cast.<locals>.unsafe_casterc                 S   s   t d)Nznot callable from pythonrk   r2  r8   r8   r9   unsafe_get_ctypes_pointer]  r   zETestIntrinsic.test_ll_pointer_cast.<locals>.unsafe_get_ctypes_pointerFr  c                    s0   t | tjrt| j  fdd}|S d S )Nc                    s    | j jdS )Nr9  )ctypesr   )r   r6  r8   r9   
array_imple  r~   zdTestIntrinsic.test_ll_pointer_cast.<locals>.array_impl_unsafe_get_ctypes_pointer.<locals>.array_impl)r>   r   r   r7  r   )Zarrtyper=  )r8  r<  r9   $array_impl_unsafe_get_ctypes_pointer`  s
   zPTestIntrinsic.test_ll_pointer_cast.<locals>.array_impl_unsafe_get_ctypes_pointerc                 S   s   t |D ]}t| |  qd S rH   )r   r   )ptrr  r   r8   r8   r9   my_c_fun_rawk  s   z8TestIntrinsic.test_ll_pointer_cast.<locals>.my_c_fun_rawNTr2  c                    s   | } || j  d S rH   )r   )r   r?  )my_c_funr:  r8   r9   r  t  s   z/TestIntrinsic.test_ll_pointer_cast.<locals>.foor  )r   )r;  r-  r.  r/  r0  r   r   r   r   Zfloat32r   r  r  closer$  maprx  rV  )rA   r-  r.  r/  r0  r>  r@  Z	prototyper  r   bufr^  r  r8   )rA  rA   r8  r:  r9   test_ll_pointer_cast@  s$   

z"TestIntrinsic.test_ll_pointer_castc                    s   t dd  tdd fdd}| |dd tj}t|}t|}|d7 }| |t| t|}| |t| | |d|d t }| |t| t|}| 	 | | |t| dS )	z9
        Test serialization of intrinsic objects
        c                 S   s   dd }||}||fS )Nc                 S      |d S r  r8   r   r8   r8   r9   r!    r   zCTestIntrinsic.test_serialization.<locals>.identity.<locals>.codegenr8   )rB   rn   r!  rw   r8   r8   r9   identity  s   z2TestIntrinsic.test_serialization.<locals>.identityTr2  c                        | S rH   r8   rm   rG  r8   r9   r    r,  z-TestIntrinsic.test_serialization.<locals>.foor/   N)
r$   r   rV  r%   _memor   pickledumpsloadsr@  )rA   r  memo	memo_sizeZserialized_fooZfoo_rebuiltZserialized_identityZidentity_rebuiltr8   rI  r9   test_serialization  s$   




z TestIntrinsic.test_serializationc                 C   s   dd }t j}t|}t d|}| |j| t|}|d7 }| |t| ~| |t| t j	  |d8 }| |t| t
|}| |j| t
|}| |j|j dS )z3
        Test deserialization of intrinsic
        c                 S   s   dd }|||fS )Nc                 S   rF  r  r8   r   r8   r8   r9   r!    r   zATestIntrinsic.test_deserialization.<locals>.defn.<locals>.codegenr8   )rB   rn   r!  r8   r8   r9   defn  s   z0TestIntrinsic.test_deserialization.<locals>.defnr  r/   N)r%   rJ  r   r@  Z_defnrK  rL  rV  Z_recentclearrM  ZassertIsNot)rA   rQ  rN  rO  originalZpickledZrebuiltsecondr8   r8   r9   test_deserialization  s"   




z"TestIntrinsic.test_deserializationc                 C   s`   t dtfdd}| d|j | d|j | d|j | dti|j | d|j d S )Nr  c                 S   r  )void_func docstringc                 S   r  rH   r8   r   r8   r8   r9   r!    rT   z@TestIntrinsic.test_docstring.<locals>.void_func.<locals>.codegenr"  r$  r8   r8   r9   r%    s   z/TestIntrinsic.test_docstring.<locals>.void_funcznumba.tests.test_extendingr%  rV  )	r$   intrV  r6   r5   r7   ZassertDictEqual__annotations__rM  )rA   r%  r8   r8   r9   test_docstring  s   
zTestIntrinsic.test_docstringN)r5   r6   r7   r,  rE  rP  rU  rY  r8   r8   r8   r9   r    s    *B.(r  c                   @      e Zd Zdd Zdd ZdS )TestRegisterJitablec                    sJ   t dd   fdd}tdd|}|dd}|dd}| || d S )	Nc                 S   r   rH   r8   rn   r   r8   r8   r9   r    r,  z.TestRegisterJitable.test_no_flags.<locals>.fooc                    s
    | |S rH   r8   r\  r  r8   r9   r    r   z.TestRegisterJitable.test_no_flags.<locals>.barTr2  r/   rz   )r&   r   rV  )rA   r  cbarr  r^  r8   r  r9   test_no_flags  s   


z!TestRegisterJitable.test_no_flagsc                    s   t dddd   fdd}| |d g d td	d
|}| tj}|d W d    n1 s7w   Y  d}| |t|j	 d S )NF)_nrtc                 S   s
   t | S rH   r  r  r8   r8   r9   r    r  z2TestRegisterJitable.test_flags_no_nrt.<locals>.fooc                    rH  rH   r8   r`  r  r8   r9   r    r   z2TestRegisterJitable.test_flags_no_nrt.<locals>.barrO  )r   r/   rz   Tr2  rz   COnly accept returning of array passed into the function as argument)
r&   rV  tolistr   rv  r	   r   rw  rx  ry  )rA   r  r]  rz  r  r8   r  r9   test_flags_no_nrt  s   

z%TestRegisterJitable.test_flags_no_nrtN)r5   r6   r7   r^  rc  r8   r8   r8   r9   r[    s    r[  c                   @   sD   e Zd Zeedu ddd Zdd Zeedu ddd ZdS )	TestImportCythonFunctionNz&Only run if SciPy >= 0.19 is installedc                    sL   t dd}ttjtj}|| tdd fdd}| |dd d S )	Nscipy.special.cython_specialj0Tr2  c                    rH  rH   r8   rm   Z_j0r8   r9   r    s    z@TestImportCythonFunction.test_getting_function.<locals>.<lambda>r   r/   )r'   r;  r-  Zc_doubler   rV  )rA   addrZfunctyperf  r8   rg  r9   test_getting_function  s   z.TestImportCythonFunction.test_getting_functionc                 C   sX   |  t}tdd W d    n1 sw   Y  d}t|t|j}| | d S )NZ
fakemoduleZfakefunctionzNo module named '?fakemodule'?)rv  ImportErrorr'   rematchrx  ry  rB  )rA   rz  r  rl  r8   r8   r9   test_missing_module  s   z,TestImportCythonFunction.test_missing_modulec                 C   sN   |  t}tdd W d    n1 sw   Y  d}| |t|j d S )Nre  r  zINo function 'foo' found in __pyx_capi__ of 'scipy.special.cython_special')rv  r   r'   rV  rx  ry  )rA   rz  r  r8   r8   r9   test_missing_function#  s   z.TestImportCythonFunction.test_missing_function)	r5   r6   r7   unittestZskipIfscri  rm  rn  r8   r8   r8   r9   rd    s    
	rd  method_jit_option_check_nrtr_  TZjit_optionsc                 C   r   )Nc                 S   
   t dS Nr  r  r.  r8   r8   r9   r   4  r   z+ov_method_jit_option_check_nrt.<locals>.impr8   r/  r8   r8   r9   ov_method_jit_option_check_nrt0     ru  method_jit_option_check_no_nrtFc                 C   r   )Nc                 S   rs  rt  r  r.  r8   r8   r9   r   >  r   z.ov_method_jit_option_check_no_nrt.<locals>.impr8   r/  r8   r8   r9   !ov_method_jit_option_check_no_nrt:  rv  rx  attr_jit_option_check_nrtc                 C   r   )Nc                 S   rs  rt  r  r.  r8   r8   r9   r   H  r   z)ov_attr_jit_option_check_nrt.<locals>.impr8   r/  r8   r8   r9   ov_attr_jit_option_check_nrtD  rv  rz  attr_jit_option_check_no_nrtc                 C   r   )Nc                 S   rs  rt  r  r.  r8   r8   r9   r   R  r   z,ov_attr_jit_option_check_no_nrt.<locals>.impr8   r/  r8   r8   r9   ov_attr_jit_option_check_no_nrtN  rv  r|  c                   @   r  )TestJitOptionsNoNRTc                 O   sT   d}|  tj}||i | W d    n1 sw   Y  | |t|j d S )Nra  )rv  r	   r   rw  rx  ry  )rA   funcrx   rs  r  rz  r8   r8   r9   check_error_no_nrt[  s   z&TestJitOptionsNoNRT.check_error_no_nrtc                    sZ   dd  t  d|iddd }t fdd}|r&| | td	 d S | | d S )
Nc                   S   rs  rt  r  r8   r8   r8   r9   dummyf  r   z8TestJitOptionsNoNRT.no_nrt_overload_check.<locals>.dummyr_  rr  c                  S   r   )Nc                   S   rs  rt  r  r8   r8   r8   r9   r  k  r   zJTestJitOptionsNoNRT.no_nrt_overload_check.<locals>.ov_dummy.<locals>.dummyr8   r  r8   r8   r9   ov_dummyi  r   z;TestJitOptionsNoNRT.no_nrt_overload_check.<locals>.ov_dummyc                      s     S rH   r8   r8   r  r8   r9   r  p     z6TestJitOptionsNoNRT.no_nrt_overload_check.<locals>.foor  )r   r   r6  r   r   r  )rA   flagr  r  r8   r  r9   no_nrt_overload_checke  s   
z)TestJitOptionsNoNRT.no_nrt_overload_checkc                 C   s   |  d |  d d S )NTF)r  rC  r8   r8   r9   test_overload_no_nrty  s   
z(TestJitOptionsNoNRT.test_overload_no_nrtc                 C   >   t dd }| |ttd t dd }| |t d S )Nc                 S   r   rH   )rq  rm   r8   r8   r9   udt~  r,  z<TestJitOptionsNoNRT.test_overload_method_no_nrt.<locals>.udtr  c                 S   r   rH   )rw  rm   r8   r8   r9   r    r,  r   r6  rF   r   r   r  rA   r  r8   r8   r9   test_overload_method_no_nrt}     

z/TestJitOptionsNoNRT.test_overload_method_no_nrtc                 C   r  )Nc                 S   r   rH   )ry  rm   r8   r8   r9   r    r  z?TestJitOptionsNoNRT.test_overload_attribute_no_nrt.<locals>.udtr  c                 S   r   rH   )r{  rm   r8   r8   r9   r    r  r  r  r8   r8   r9   test_overload_attribute_no_nrt  r  z2TestJitOptionsNoNRT.test_overload_attribute_no_nrtN)r5   r6   r7   r  r  r  r  r  r8   r8   r8   r9   r}  X  s    
r}  c                       s<   e Zd Z fddZdd Zdd Zdd Zd	d
 Z  ZS )TestBoxingCallingJITc                    s0   t    td}|\| _| _| _|  | _d S )Nrh   )r  setUprg   r^   r]   ra   dyn_type)rA   manyr  r8   r9   r    s   
zTestBoxingCallingJIT.setUpc                    sl   dd t  fddt| jfdd}t| jdd }td	d
 }|| j}| |   d S )Nr  r   c                        fdd}t ||}||fS )Nc                       |\}| || S rH   r   r  rB   rQ   rw   rx   rK   magic_offsetr8   r9   r       zKTestBoxingCallingJIT.test_unboxer_basic.<locals>.my_intrinsic.<locals>.implr+   rK  rK   r  rw   r  r8   r9   my_intrinsic     
z=TestBoxingCallingJIT.test_unboxer_basic.<locals>.my_intrinsicc                    sL   fdd}|j tj g}ttjtj}|j|||\}}t||dS )Nc                    rH  rH   r8   rm   r  r8   r9   bridge  s   zHTestBoxingCallingJIT.test_unboxer_basic.<locals>.unboxer.<locals>.bridgeis_error	rB   rP   r   intpr+   Zvoidptrr   call_jit_coder#   rY   rZ   rL   r  rx   rw   r  r   magic_tokenr  r8   r9   unboxer  s
   z8TestBoxingCallingJIT.test_unboxer_basic.<locals>.unboxerc                 S      |j |tj}|j|S rH   rQ   Zptrtointr   Zintp_tr   long_from_ssize_trY   rK   rL   r   r8   r8   r9   boxer     z6TestBoxingCallingJIT.test_unboxer_basic.<locals>.boxerc                 S   r  rH   r8   rm   r8   r8   r9   passthru  rN   z9TestBoxingCallingJIT.test_unboxer_basic.<locals>.passthrur$   r"   r^   r!   r   r  rV  )rA   r  r  r  r  r8   r  r  r  r9   test_unboxer_basic  s   	


z'TestBoxingCallingJIT.test_unboxer_basicc                 C   z   t | jdd }t| jdd }tdd }| t}|| j W d    n1 s-w   Y  | dt|j	 d S )Nc                 S   sH   dd }|j tjdg}ttjtj}|j|||\}}t||dS )Nc                 S      | dkrt d| S )Nr   cannot be x > 0r   rm   r8   r8   r9   r       zHTestBoxingCallingJIT.test_unboxer_raise.<locals>.unboxer.<locals>.bridger/   r  r  r  r8   r8   r9   r    s
   z8TestBoxingCallingJIT.test_unboxer_raise.<locals>.unboxerc                 S   r  rH   r  r  r8   r8   r9   r    r  z6TestBoxingCallingJIT.test_unboxer_raise.<locals>.boxerc                 S   r  rH   r8   rm   r8   r8   r9   r    rN   z9TestBoxingCallingJIT.test_unboxer_raise.<locals>.passthrur  
r"   r^   r!   r   rv  r   r  rw  rx  ry  rA   r  r  r  rz  r8   r8   r9   test_unboxer_raise  s   



z'TestBoxingCallingJIT.test_unboxer_raisec                    sl   dd t  fddt| jdd }t| jfdd}td	d
 }|| j}| |   d S )Nr  i8  c                    r  )Nc                    r  rH   r  r  r  r8   r9   r    r  zCTestBoxingCallingJIT.test_boxer.<locals>.my_intrinsic.<locals>.implr  r  r  r8   r9   r    r  z5TestBoxingCallingJIT.test_boxer.<locals>.my_intrinsicc                 S   rU   rH   rV   rX   r8   r8   r9   r    r\   z0TestBoxingCallingJIT.test_boxer.<locals>.unboxerc                    sL   fdd}|j tj g}ttjtj}|j|||\}}|j|S )Nc                    rH  rH   r8   rm   r  r8   r9   r    r   z>TestBoxingCallingJIT.test_boxer.<locals>.boxer.<locals>.bridge)rB   rP   r   r  r+   r   r  r  )rY   rK   rL   r  rx   rw   r  r   r  r8   r9   r    s
   z.TestBoxingCallingJIT.test_boxer.<locals>.boxerc                 S   r  rH   r8   rm   r8   r8   r9   r    rN   z1TestBoxingCallingJIT.test_boxer.<locals>.passthrur  )rA   r  r  r  rr8   r  r9   
test_boxer  s   	



zTestBoxingCallingJIT.test_boxerc                 C   r  )Nc                 S   rU   rH   rV   rX   r8   r8   r9   r    r\   z6TestBoxingCallingJIT.test_boxer_raise.<locals>.unboxerc           
      S   s   dd }|j tjdg}ttjtj}|j|||\}}tj|j	|jj
dd}|j	|j	| |j|}	|j	|	| W d    n1 sKw   Y  |j	|S )Nc                 S   r  )Nr   cannot do x > 0r  rm   r8   r8   r9   r    r  zDTestBoxingCallingJIT.test_boxer_raise.<locals>.boxer.<locals>.bridger/   T)zfill)rB   rP   r   r  r+   r   r  r   Zalloca_oncerQ   ZpyobjZif_thennot_r  storeload)
rY   rK   rL   r  rx   rw   r  r   retvalrZ   r8   r8   r9   r    s   z4TestBoxingCallingJIT.test_boxer_raise.<locals>.boxerc                 S   r  rH   r8   rm   r8   r8   r9   r  $  rN   z7TestBoxingCallingJIT.test_boxer_raise.<locals>.passthrur  r  r  r8   r8   r9   test_boxer_raise  s   



z%TestBoxingCallingJIT.test_boxer_raise)	r5   r6   r7   r  r  r  r  r  r(  r8   r8   r  r9   r    s    *%r  c                 C   r  rH   r8   rm   r8   r8   r9   with_objmode_cache_ov_example/  s   r  c                   @   sX   e Zd ZdZdZdd Zdd Zdd Zed	d
 Z	edd Z
edd Zdd ZdS )TestCachingOverloadObjmodezTTest caching of the use of overload implementations that use
    `with objmode`
    Fc                 C   s   t dtj d S )Nerror)warningssimplefilterr	   ZNumbaWarningrC  r8   r8   r9   r  <  s   z TestCachingOverloadObjmode.setUpc                 C   s   t   d S rH   )r  resetwarningsrC  r8   r8   r9   tearDown?  r   z#TestCachingOverloadObjmode.tearDownc                    s   t | jj}td|F dd fdd tt fdd}tdd	d
d }d}|d}| || tdd	|j}|d}| || W d    d S 1 sTw   Y  d S )Nr  c                 S   s   t | |  }t j|S rH   )r   r   ZlinalgZnorm)rn   r   r8   r8   r9   realworkF  s   zJTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>.realworkc                    rH  rH   r8   rm   )r  r8   r9   python_codeK  r,  zMTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>.python_codec                    s    fdd}|S )Nc                    s8   t dd  | }W d    |S 1 sw   Y  |S Nr<  )r   r   r\  r  r8   r9   r  Q  s   

zqTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>._ov_with_objmode_cache_ov_example.<locals>.implr8   r  r  r8   r9   !_ov_with_objmode_cache_ov_exampleO  s   zcTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>._ov_with_objmode_cache_ov_exampleTr	  c                 S   r   rH   )r  rm   r8   r8   r9   testcaseX  r,  zJTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>.testcaser   )	r   r  r5   r   r   r  r   rV  py_func)rA   r  r  r  r  r^  Ztestcase_cachedr8   )r  r  r9   test_caching_overload_objmodeB  s   
"z8TestCachingOverloadObjmode.test_caching_overload_objmodec                    s\   dd dd  t  fdd}tdd fd	d
}| }| }||kr,td|S )Nc                 S      t | | S rH   r   sumr  r8   r8   r9   do_thisf  r~   zGTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.do_thisc                 S   r  rH   r  r  r8   r8   r9   do_somethingi  r~   zLTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.do_somethingc                    s    fdd}|S )Nc                    s:   t dd  | |}W d    |S 1 sw   Y  |S r  r  )r  r  r   r  r8   r9   _do_something_impln  s   
zqTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.overload_do_something.<locals>._do_something_implr8   )r  r  r  r  r8   r9   overload_do_somethingl  s   zUTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.overload_do_somethingTr  c                     s   t d} t d} | |S )Nr  r  r  )r  r8   r9   test_cachingt  s   


zLTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.test_cachingzincorrect result)r   r   r  r   )r  r  r  r^  r  r8   )r  r  r9   check_objmode_cache_ndarrayd  s   z6TestCachingOverloadObjmode.check_objmode_cache_ndarrayc                 C   s   |    d S rH   )r  )r  r8   r8   r9   *populate_objmode_cache_ndarray_check_cache  r  zETestCachingOverloadObjmode.populate_objmode_cache_ndarray_check_cachec                 C   s<   |   }t|jjdkrtdt|jjdkrtdd S )Nr   zunexpected cache misszunexpected missing cache hit)r  r   statsZcache_missesr   Z
cache_hits)r  Zdispr8   r8   r9   'check_objmode_cache_ndarray_check_cache  s   zBTestCachingOverloadObjmode.check_objmode_cache_ndarray_check_cachec                 C   sb   t | jj}td| t| j| t| j|}W d    n1 s"w   Y  | |d d d S )Nr  r  r   )r   r  r5   r   r   r  r  rV  )rA   r  r   r8   r8   r9    test_check_objmode_cache_ndarray  s   	z;TestCachingOverloadObjmode.test_check_objmode_cache_ndarrayN)r5   r6   r7   rM  r  r  r  r  classmethodr  r  r  r  r8   r8   r8   r9   r  5  s    "


r  c                   @   rZ  )TestMiscc                 C   sn   dd }|  t| | tt| |  tt| |  ttdd| |  ttdd| d S )Nc                 S   r  rH   r8   rm   r8   r8   r9   r    rT   z$TestMisc.test_is_jitted.<locals>.fooT)Zparallelzvoid(float64[:])z(m))r  r(   rG  r   r   r   )rA   r  r8   r8   r9   test_is_jitted  s   zTestMisc.test_is_jittedc                 C   s@   t dd }t dd }| | |  | | |  d S )Nc                   S      t jdddjS )Nr     )r   
fill_valuer   fullr   r8   r8   r8   r9   standard_order  r  z:TestMisc.test_overload_arg_binding.<locals>.standard_orderc                   S   r  )Nr  r   )r  r   r  r8   r8   r8   r9   reversed_order  r  z:TestMisc.test_overload_arg_binding.<locals>.reversed_order)r   r6  r  )rA   r  r  r8   r8   r9   test_overload_arg_binding  s   

z"TestMisc.test_overload_arg_bindingN)r5   r6   r7   r  r  r8   r8   r8   r9   r    s    r  c                   @   rZ  )TestOverloadPreferLiteralc                    s   dd dd  dd }t dd| t  | tfd	d
}|d\}}}| |d | |d | |d t fdd}|d\}}}| |d | |d | |d d S )Nc                 S   r  rH   r8   rm   r8   r8   r9   
prefer_lit  rT   z;TestOverloadPreferLiteral.test_overload.<locals>.prefer_litc                 S   r  rH   r8   rm   r8   r8   r9   non_lit  rT   z8TestOverloadPreferLiteral.test_overload.<locals>.non_litc                 S   s8   t | tjr| jdkrdd }|S tddd }|S )Nr/   c                 S   r   r  r8   rm   r8   r8   r9   r    rT   zATestOverloadPreferLiteral.test_overload.<locals>.ov.<locals>.implliteral valuec                 S   s   | d S Nd   r8   rm   r8   r8   r9   r    r   r>   r   IntegerLiteralliteral_valuer	   r   r  r8   r8   r9   ov     

z3TestOverloadPreferLiteral.test_overload.<locals>.ovTZprefer_literalc                        d d | fS Nr/   rz   r8   rm   r  r8   r9   check_prefer_lit     zATestOverloadPreferLiteral.test_overload.<locals>.check_prefer_litrO  r     ,  c                    r  r  r8   rm   r  r8   r9   check_non_lit  r  z>TestOverloadPreferLiteral.test_overload.<locals>.check_non_litr  )r   r   rV  rA   r  r  r  r  rL   r  r8   r  r  r9   test_overload  s"   z'TestOverloadPreferLiteral.test_overloadc                 C   s   dd }t tddd| t tddd| tdd	 }|t d
\}}}| |d | |d | |d tdd }|t d
\}}}| |d | |d | |d d S )Nc                 S   s8   t |tjr|jdkrdd }|S tddd }|S )Nr/   c                 S   r   r  r8   rA   rn   r8   r8   r9   r    rT   zHTestOverloadPreferLiteral.test_overload_method.<locals>.ov.<locals>.implr  c                 S   s   |d S r  r8   r  r8   r8   r9   r    r   r  )rA   rn   r  r8   r8   r9   r    r  z:TestOverloadPreferLiteral.test_overload_method.<locals>.ovmethod_prefer_literalTr  method_non_literalFc                 S      |  d|  d|  |fS r  )r  r  rn   r8   r8   r9   r       zHTestOverloadPreferLiteral.test_overload_method.<locals>.check_prefer_litrO  r  r  r  c                 S   r  r  )r  r	  r8   r8   r9   r    r
  zETestOverloadPreferLiteral.test_overload_method.<locals>.check_non_litr  r  r  r8   r8   r9   test_overload_method  s2   

z.TestOverloadPreferLiteral.test_overload_methodN)r5   r6   r7   r  r  r8   r8   r8   r9   r    s    *r  c                   @   r;   )TestIntrinsicPreferLiteralc                    s   dd }t dd|t dd| tfdd}|d\}}}| |d	 | |d
 | |d t fdd}|d\}}}| |d | |d
 | |d d S )Nc                 S   sL   t tj|}t|tjr|jdkrdd }||fS tddd }||fS )Nr/   c                 S   s    |j d }| |}t|dS )Nr   r  )rx   r1  r   Constant)rB   rQ   r+   rx   atyper3  r8   r8   r9   r!  !  s   

zJTestIntrinsicPreferLiteral.test_intrinsic.<locals>.intrin.<locals>.codegenr  c                 S   s,   |j }| |}t|d}||d |S )Nr  r   )rv   r1  r   r  mul)rB   rQ   r+   rx   r  r3  Zint_100r8   r8   r9   r!  )  s   
)r+   r   r  r>   r  r  r	   r   )rB   rn   rw   r!  r8   r8   r9   intrin  s   

z9TestIntrinsicPreferLiteral.test_intrinsic.<locals>.intrinTr  Fc                    r  r  r8   rm   r  r8   r9   r  3  r  zCTestIntrinsicPreferLiteral.test_intrinsic.<locals>.check_prefer_litrO  r  r  r  c                    r  r  r8   rm   r   r8   r9   r  <  r  z@TestIntrinsicPreferLiteral.test_intrinsic.<locals>.check_non_litr  )r$   r   rV  )rA   r  r  r  r  rL   r  r8   r  r9   test_intrinsic  s   z)TestIntrinsicPreferLiteral.test_intrinsicN)r5   r6   r7   r  r8   r8   r8   r9   r    rE   r  c                   @   r;   )TestNumbaInternalOverloadsc                    s   d  fdd fdd}t dd d tjjj }|  |j}|	 D ]\}}|j
D ]}t|d	d
r@||j|j q1q*d S )Nr   c              	      s   |  ds[ttj| }t|r]|j }|D ]G}|rZ| }| | rZ| dkr*d}d| d}i }zt	|i | W n t
yM    dkrItd Y  d S w || }t|  S qd S d S d S )N_seedz
seed(seed)zdef z:
	passrz   z%... skipped as cannot parse signature)r
  r   r   randominspect	isbuiltinrM  r  stripexecSyntaxErrorr   r  r+   )rn   thingZdocstrlslZ	fake_implfn)DEBUGr8   r9   sig_from_np_randomW  s2   




z[TestNumbaInternalOverloads.test_signatures_match_overloaded_api.<locals>.sig_from_np_randomc                    s   dkrt d|   dd }d }zt| }W n" ty:   t| dd }r8ttj|d }r8|| kr8|}Y nw |d urt|}t|j	 }t|j	 }t
|d t| |D ]m\}	}
|	|
kr|j|	 }|j|jkr{ dkrxt d  d S dt|v r dkrt d	  d S | jr| jd
s|| |||} dkr dkrt d |rt |  d S t| dkr| jst d  d S t d  d S q] dkrt d d S d S d S )Nrz   z
Checking: c           
      S   s   g }|  dt | d d}|| |d|  |d|  t|d }t|}|tjd d}|d	| d
|  dd| }	|	S )Nz from module 'r6   z' has mismatched sig.z    - expected: z    -      got: r/   r    zfrom :
)	r   r!  r  getsourcelinesgetfilereplacenumba__path__r  )
r~  overload_funcfunc_sigol_sigr  rd  linenoZ
tmpsrcfilesrcfilemsgstrr8   r8   r9   create_messagex  s   

zhTestNumbaInternalOverloads.test_signatures_match_overloaded_api.<locals>.checker.<locals>.create_messager5   Fz.... skipped as positional only arguments found*z... skipped as contains *argsr'  r   z... INVALIDz$... skipped as no __module__ presentz... skipped as Numba internalz... OK)r   r  r+   r   r   r   r  r$  
parametersr%  zipr   kindPOSITIONAL_ONLYrx  r6   r
  )r~  r)  r/  r*  fnameZ
maybe_funcr+  rn   r   r  r  r  r.  r  r   r8   r9   checkert  sh   


$)zPTestNumbaInternalOverloads.test_signatures_match_overloaded_api.<locals>.checkerc                   S   r  rH   r8   r8   r8   r8   r9   r    r  zQTestNumbaInternalOverloads.test_signatures_match_overloaded_api.<locals>.<lambda>r8   _overload_funcF)r   compiler'  corer
   rB   r4  rF  _registriesitemsZ	functionsr   r   r8  )rA   r7  Ztyctxregskvitemr8   r6  r9   $test_signatures_match_overloaded_apiH  s   K
z?TestNumbaInternalOverloads.test_signatures_match_overloaded_apiN)r5   r6   r7   rA  r8   r8   r8   r9   r  F  r  r  __main__rH   )r  r{   r   r~  rK  r  r;  r  rk  numpyr   Zllvmliter   r'  r   r   r   r   r   Z
numba.corer   r	   r
   r   r   Znumba.core.typed_passesr   Znumba.core.registryr   Znumba.core.compilerr   Znumba.core.imputilsr   Znumba.tests.supportr   r   r   r   r   r   Znumba.core.errorsr   ro  Znumba.extendingr   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   Znumba.core.typing.templatesr*   r+   r,   r-   r.   Zpdlike_usecaser0   r1   Zscipy.special.cython_specialZspecialZcython_specialrp  rj  rb   r3   rc   r<   rI   rF   rd   rM   r?   rR   rS   re   r[   rg   r   r  Zmydummy_type_2ri   ro   rt   rA  rp   ry   rr   r   r   r   r   r   r   r   r   r   r   r   r   delitemr   getitemr   setitemr   r   r   iaddr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  Callabler  r)  r+  r  r0  r1  rN  rr  r  r  r  r  r[  rd  ru  rx  rz  r|  r}  r  r  r  r  r  r  r  r5   mainr8   r8   r8   r9   <module>   s@    P






	




1

	

	
	



"


4a    	( Y# 







@ h_- 
