U
    id                     @   s  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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,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z: d dl;m<Z<m=Z=m>Z>m?Z?m@Z@ ddlAmBZBmCZC zd dlDmE  mFZG W n eHk
r   dZGY nX G dd deIZJG dd dejKZLeLdZMeJ ZNe)OeJdd ZPe,eLejQdd ZRdd ZSe1eLe0jT e3eLdd ZUdd ZVeVd\ZWZXZYe3eWdd ZZdd d!Z[d"d# Z\e*e[e\Z]e+e[e+e[ej^d$d% Z_e+e[ej`d&d' Zae>G d(d) d)e<Zbe+d*eLd+d, Zcd-d. Zdd/d0 Zed1d2 Zfe-edd3d4 Zge-edd5d6 Zhe-eid7d8 Zje-ejkd9d: Zle-ejmd;d< Zne-ejod=d> Zpe-ejqd?d@ ZrdAdB ZsdCdD Zte-ejudEdF ZvdGdH ZwdIdJ ZxdKdL ZydMdN ZzdOdP Z{e/eLdQdRdS Z|dTdU Z}dVdW Z~dXdY ZdZd[ Zd\d] Zd^d_ Zd`da Zdbdc Zddde Zdfdg Zdhdi Zdjdk Zdldm Ze-edndo Zdpdq Zdrds Zdtdu Ze?eG dvdw dwe@Zdxdy ZG dzd{ d{eIZG d|d} d}ejZe1eG d~d de0jTZe?eee eedd Ze-e	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Ze/eLdddiddd Ze/eLdddiddd Ze.eLdddiddd Ze.eLdddiddd ZG dd deZG dd deZdd Ze$G dd deZG dd deZG dd deZedkre'  dS )    N)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                   @   s   e Zd ZdS )MyDummyN__name__
__module____qualname__ r6   r6   S/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/tests/test_extending.pyr1   H   s   r1   c                   @   s   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.typeconvr9   safe)selfcontexttotyr9   r6   r6   r7   can_convert_toM   s    zMyDummyType.can_convert_toN)r3   r4   r5   r@   r6   r6   r6   r7   r8   L   s   r8   mydummyc                 C   s   t S N)mydummy_typevalcr6   r6   r7   typeof_mydummyX   s    rG   c                 C   s   |  |dS )z2
    Implicit conversion from MyDummy to int.
    *   )get_constant)r>   builderZfromtyr?   rE   r6   r6   r7   mydummy_to_number]   s    rK   c                   C   s   t S rB   )rA   r6   r6   r6   r7   	get_dummye   s    rL   c                 C   s   t |j S rB   r"   r>   get_dummy_valuetypobjrF   r6   r6   r7   unbox_indexl   s    rR   c                    sX   G dd dt }G dd dtj}||  t| fdd}t|tj || fS )Nc                   @   s   e Zd ZdS )z(base_dummy_type_factory.<locals>.DynTypeNr2   r6   r6   r6   r7   DynTypev   s   rS   c                   @   s   e Zd ZdS )z,base_dummy_type_factory.<locals>.DynTypeTypeNr2   r6   r6   r6   r7   DynTypeTypey   s   rT   c                    s    S rB   r6   rD   dyn_type_typer6   r7   rG   ~   s    z/base_dummy_type_factory.<locals>.typeof_mydummy)objectr   Opaquer   registerr   r   OpaqueModel)namerS   rT   rG   r6   rU   r7   base_dummy_type_factoryu   s    r\   mydummy2c                 C   s   t |j S rB   rM   rO   r6   r6   r7   unbox_index2   s    r^   c                 C   s   t d S rB   NotImplementedErrorxr6   r6   r7   func1   s    rc   c                 C   s   ddd}|S )Nc                 S   s(   | d t jfkrt jS t| t jr$| S d S rB   )r   noneint32r:   Floatra   r6   r6   r7   typer   s    ztype_func1_.<locals>.typer)Nr6   )r>   rg   r6   r6   r7   type_func1_   s    
rh   c                 C   s   |  |jdS NrH   )rI   return_type)r>   rJ   sigargsr6   r6   r7   func1_nullary   s    rm   c                 C   s   dd }|  ||||S )Nc                 S   s   t d|  S )N   )mathsqrtra   r6   r6   r7   
func1_impl   s    zfunc1_unary.<locals>.func1_impl)Zcompile_internal)r>   rJ   rk   rl   rq   r6   r6   r7   func1_unary   s    rr   c                   @   s   e Zd ZdZeejegZdS )
PrintDummy
print_itemN)	r3   r4   r5   keyr*   r   rd   rC   Zcasesr6   r6   r6   r7   rs      s   rs   rt   c                 C   s<   |\}|  |}||d}|| || |  S )Nzhello!)Zget_python_apiZunserializeZserialize_objectZprint_objectZdecrefrN   )r>   rJ   rk   rl   rb   pyapiZstrobjr6   r6   r7   print_dummy   s    


rw   c                 C   s   t d S rB   r_   condrb   yr6   r6   r7   where   s    r{   c                 C   s   t | ||S )z8
    Wrap np.where() to allow for keyword arguments
    )npr{   rx   r6   r6   r7   np_where   s    r}   c                 C   s   t | ||dS )N)rz   rb   )r{   rx   r6   r6   r7   
call_where   s    r~   c                 C   sT   t | tjrP|j|jkr"tdtdd | ||fD rDdd }n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tyr6   r6   r7   	<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 ] }	||	 rb||	 n||	 ||	< qN|S )zM
                Fast implementation for C-contiguous arrays
                %all inputs should have the same shape)shape
ValueErrorr|   
empty_likeZflatrangesize)
ry   rb   rz   r   rescfZxfZyfrfir6   r6   r7   
where_impl   s    
z)overload_where_arrays.<locals>.where_implc                 S   s\   | j }|j |ks|j |kr"tdt|}t| D ] \}}|rJ|| n|| ||< q6|S )zI
                Generic implementation for other arrays
                r   )r   r   r|   r   Zndenumerate)ry   rb   rz   r   r   idxrF   r6   r6   r7   r      s    
N)r:   r   Arraydtyper   TypingErrorallry   rb   rz   r   r6   r6   r7   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
            r6   )r|   r   )ry   rb   rz   Zscalarrr6   r6   r7   r     s    
z*overload_where_scalars.<locals>.where_implN)r:   r   r   r   r   r   r6   r6   r7   overload_where_scalars  s
    

r   c                 C   s   t | trdd }|S d S )Nc                 S   s   dS )N   r6   argr6   r6   r7   len_impl+  s    z$overload_len_dummy.<locals>.len_implr:   r8   )r   r   r6   r6   r7   overload_len_dummy'  s    
r   c                 C   s,   t | ttfr(t |ttfr(dd }|S d S )Nc                 S   s   dS ri   r6   arg1arg2r6   r6   r7   dummy_add_impl7  s    z*overload_add_dummy.<locals>.dummy_add_implr:   r8   MyDummyType2)r   r   r   r6   r6   r7   overload_add_dummy1  s     r   c                 C   s&   t | tr"t |tjr"dd }|S d S )Nc                 S   s   t d| | d S )NdelprintrQ   r   r6   r6   r7   dummy_delitem_implA  s    z2overload_dummy_delitem.<locals>.dummy_delitem_implr:   r8   r   Integer)rQ   r   r   r6   r6   r7   overload_dummy_delitem=  s    r   c                 C   s&   t | tr"t |tjr"dd }|S d S )Nc                 S   s   |d S )N{   r6   r   r6   r6   r7   dummy_getitem_implK  s    z2overload_dummy_getitem.<locals>.dummy_getitem_implr   )rQ   r   r   r6   r6   r7   overload_dummy_getitemG  s    r   c                 C   s4   t t| tt|tjt|tjgr0dd }|S d S )Nc                 S   s   t || d S rB   r   rQ   r   rE   r6   r6   r7   dummy_setitem_impl[  s    z2overload_dummy_setitem.<locals>.dummy_setitem_impl)r   r:   r8   r   r   )rQ   r   rE   r   r6   r6   r7   overload_dummy_setitemQ  s    

r   c                 C   s   t | |S rB   operatoraddr   r6   r6   r7   call_add_operatora  s    r   c                 C   s   | | S rB   r6   r   r6   r6   r7   call_add_binope  s    r   c                 C   s,   t | ttfr(t |ttfr(dd }|S d S )Nc                 S   s   dS ri   r6   r   r6   r6   r7   dummy_iadd_implo  s    z,overload_iadd_dummy.<locals>.dummy_iadd_implr   )r   r   r   r6   r6   r7   overload_iadd_dummyi  s     r   c                 C   s   t | |S rB   r   r   r6   r6   r7   call_iadd_operatoru  s    r   c                 C   s   | |7 } | S rB   r6   r   r6   r6   r7   call_iadd_binopy  s    r   c                 C   s
   | |= d S rB   r6   r   r6   r6   r7   call_delitem  s    r   c                 C   s   | | S rB   r6   r   r6   r6   r7   call_getitem  s    r   c                 C   s   || |< d S rB   r6   r   r6   r6   r7   call_setitem  s    r   lengthc                 C   s   dd }|S )Nc                 S   s   t | S rB   lenr   r6   r6   r7   imp  s    z#overload_method_length.<locals>.impr6   )r   r   r6   r6   r7   overload_method_length  s    r   c                 C   s   |   S rB   )r   ra   r6   r6   r7   cache_overload_method_usecase  s    r   c                   C   s   t  S rB   rc   r6   r6   r6   r7   call_func1_nullary  s    r   c                 C   s   t | S rB   r   ra   r6   r6   r7   call_func1_unary  s    r   c                 C   s   t | S rB   r   ra   r6   r6   r7   len_usecase  s    r   c                 C   s   t |  d S rB   r   ra   r6   r6   r7   print_usecase  s    r   c                 C   s   | | S rB   r6   )rb   ru   r6   r6   r7   getitem_usecase  s    r   c                 C   s   t t | S rB   )r|   cossinra   r6   r6   r7   npyufunc_usecase  s    r   c                 C   s   | j S rB   )_datara   r6   r6   r7   get_data_usecase  s    r   c                 C   s   | j S rB   )_indexra   r6   r6   r7   get_index_usecase  s    r   c                 C   s   | j S rB   )Zis_monotonic_increasingra   r6   r6   r7   is_monotonic_usecase  s    r   c                 C   s
   t | |S rB   )r0   )dataindexr6   r6   r7   make_series_usecase  s    r   c                 C   s   |  ||S rB   )Zclip)rb   lohir6   r6   r7   clip_usecase  s    r   c                   C   s   t S rB   r|   r6   r6   r6   r7   return_non_boxable  s    r   c                  C   s   dd } | S )Nc                   S   s   t S rB   r   r6   r6   r6   r7   r     s    z(overload_return_non_boxable.<locals>.impr6   )r   r6   r6   r7   overload_return_non_boxable  s    r   c                 C   s   t  }|| S rB   )r   arange)szmodr6   r6   r7   non_boxable_ok_usecase  s    r   c                   C   s   t  S rB   )r   r6   r6   r6   r7   non_boxable_bad_usecase  s    r   c                 C   s   d S rB   r6   )fr6   r6   r7   mk_func_input  s    r   c                   @   s   e Zd Zdd ZdS )MkFuncTypingc                 C   s$   t |d tjstttjf| S Nr   )r:   r   MakeFunctionLiteralAssertionErrorr*   rd   )r=   rl   kwsr6   r6   r7   generic  s    zMkFuncTyping.genericN)r3   r4   r5   r   r6   r6   r6   r7   r     s   r   c                   C   s   t dd  d S )Nc                 S   s   | S rB   r6   ar6   r6   r7   <lambda>      z#mk_func_test_impl.<locals>.<lambda>)r   r6   r6   r6   r7   mk_func_test_impl  s    r   c                   @   s   e Zd ZdS )MyClassNr2   r6   r6   r6   r7   r     s   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)r=   r   	__class__r6   r7   r     s    zCallableTypeRef.__init__c              	   C   st   d }|j t|  D ]\}z|||}W n tk
r:   Y qX t|di }||kr|| | j|< | j|  qpq|S )NZ_compiled_overloads)Z
_functionstypeapply	Exceptiongetattrr   r   append)r=   r>   rl   r   Zres_sigtemplateZcompiled_ovldsr6   r6   r7   get_call_type  s    zCallableTypeRef.get_call_typec                 C   s   t | j }|dfS )NT)listr   keys)r=   Zsigsr6   r6   r7   get_call_signatures  s    z#CallableTypeRef.get_call_signaturesc                 C   s
   | j | S rB   )r   )r=   rk   r6   r6   r7   get_impl_key  s    zCallableTypeRef.get_impl_key)r3   r4   r5   r   r   r  r  __classcell__r6   r6   r   r7   r     s   r   c                   @   s   e Zd Zdd ZdS )CallableTypeModelc                 C   s   t j| || d S rB   )r   rZ   r   )r=   ZdmmZfe_typer6   r6   r7   r     s    zCallableTypeModel.__init__N)r3   r4   r5   r   r6   r6   r6   r7   r    s   r  c                 C   s   |   S rB   )rN   )r>   rJ   r   Zpyvalr6   r6   r7   constant_callable_typeref  s    r  c                 C   s   t | trdd }|S d S )Nc                 S   s   dS )N   >[= r6   rQ   r6   r6   r7   r   +  s    zoverload_np_exp.<locals>.impr   rQ   r   r6   r6   r7   overload_np_exp'  s    
r	  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nopythonrH         2@      @)r   r   assertPreciseEqualr   r=   pyfunccfuncr6   r6   r7   
test_func1:  s    z TestLowLevelExtending.test_func1c                 C   sF   t }t|d}| | d t}t|tjf}| |dd d S )Nr6   rH   r  r  )r   r   r  entry_pointr   r   float64r=   r  crr6   r6   r7   test_func1_isolatedC  s    
z)TestLowLevelExtending.test_func1_isolatedc                 C   s   |  tt | t d S rB   )assertIs
type_func1rh   assertIsNotNoner=   r6   r6   r7   !test_type_callable_keeps_functionK  s    z7TestLowLevelExtending.test_type_callable_keeps_functionc                 C   s&   t }t|dtj}| | d d S )Nr6   g      E@)rL   r   r   r  r  r  r  r6   r6   r7   test_cast_mydummyO  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
        r6   Nc                 s   s   | ]}t |tjV  qd S rB   )r:   r   r   )r   r   r6   r6   r7   r   _  s   z=TestLowLevelExtending.test_mk_func_literal.<locals>.<genexpr>)r
   Zrun_frontendr   r   Ztyping_contextZtarget_contextrefreshr   
assertTrueanytypemapvalues)r=   Ztest_ir	typingctxZ	targetctxZ
typing_resr6   r6   r7   test_mk_func_literalT  s    

z*TestLowLevelExtending.test_mk_func_literalN)	r3   r4   r5   __doc__r  r  r  r  r%  r6   r6   r6   r7   r
  2  s   	r
  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   Tr  )r/   r|   r   r   r   r  )r=   r   r  r6   r6   r7   test_index_lenl  s    zTestPandasLike.test_index_lenc                 C   sh   t tdddg}tddt}| ||dd ||tdd }| |t  | t	|ddg d S )NrH      Tr  r.   )
r/   r|   re   r   r   r  sliceassertIsInstanceassertEqualr   r=   r   r  iir6   r6   r7   test_index_getitemq  s    z!TestPandasLike.test_index_getitemc                 C   sV   t tdddg}tddt}||}| |t  | |jtt	|j dS )z7
        Check Numpy ufunc on an Index object.
        rH   r*  r+  Tr  N)
r/   r|   re   r   r   r-  r  r   r   r   r/  r6   r6   r7   test_index_ufuncy  s
    zTestPandasLike.test_index_ufuncc                 C   s<   t tdddg}tddt}||}| ||j d S )NrH   r*  r+  Tr  )r/   r|   re   r   r   r  r   )r=   r   r  r   r6   r6   r7   test_index_get_data  s    z"TestPandasLike.test_index_get_datac                 C   s`   t ddt}dddgdfdddgdfg dffD ]*\}}tt|}||}| || q0d S )NTr  r*  rH      F)r   r   r/   r|   re   r.  )r=   r  r#  expectedr   gotr6   r6   r7   test_index_is_monotonic  s    z&TestPandasLike.test_index_is_monotonicc                 C   sL   t tdddg}ttdddg|}tddt}| ||d d S 	Nrn      r(        ?      @      @Tr  )r/   r|   re   r0   r  r   r   r  )r=   r   sr  r6   r6   r7   test_series_len  s    zTestPandasLike.test_series_lenc                 C   s`   t tdddg}ttdddg|}tddt}||}| |t  | |j	|j	 d S r8  )
r/   r|   re   r0   r  r   r   r-  r  r   )r=   r   r=  r  r6  r6   r6   r7   test_series_get_index  s    z$TestPandasLike.test_series_get_indexc                 C   s   t tdddg}ttdddg|}tddt}||}| |t | |jt  | 	|jj
|j
 | |jtt|j d	S )
z8
        Check Numpy ufunc on an Series object.
        rH   r*  r+  r.   rn   r(  Tr  N)r/   r|   re   r0   int64r   r   r-  r   r  r   r  _valuesr   r   r=   r   r=  r  ssr6   r6   r7   test_series_ufunc  s    z TestPandasLike.test_series_ufuncc                 C   sz   t tdddg}tdddg}tddt}|||}| |t | |jt  | 	|jj
|j
 | 	|j| d S )	NrH   r*  r+  r:  r;  r<  Tr  )r/   r|   re   r  r   r   r-  r0   r   r  r   rA  )r=   r   dr  r6  r6   r6   r7   test_series_constructor  s    
z&TestPandasLike.test_series_constructorc                 C   s   t tdddg}ttdddg|}tddt}||d	d
}| |t | |jt  | 	|jj
|j
 | |jtd	d
dg d S )NrH   r*  r+  r:  r;  r<  Tr  g?g      @)r/   r|   re   r0   r  r   r   r-  r   r  r   r  rA  rB  r6   r6   r7   test_series_clip  s    zTestPandasLike.test_series_clipN)r3   r4   r5   r&  r)  r1  r2  r3  r7  r>  r?  rD  rF  rG  r6   r6   r6   r7   r'  f  s   

r'  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dddgtd	d
dgtdddg tj$} tg tg tg  W 5 Q R X 	dt
|j dS )z>
        Test implementing a function with @overload.
        Tr  c                     s$   t | |} | |}|| d S rB   )r}   r  )rl   kwargsr5  r6  r  r=   r6   r7   check  s    

z0TestHighLevelExtending.test_where.<locals>.checkr(  r*  )rb   ry   rz   Fr.   rn   r9  r4  r   N)r~   r   r|   Zbool_re   assertRaisesr   r   r@  assertInstr	exception)r=   r  rK  raisesr6   rJ  r7   
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.
        Tr  r   r9  r4  rn   N)r   r   r  r1   r=   r  r6   r6   r7   test_len  s    zTestHighLevelExtending.test_lenc              	   C   s@   t ddt}t " |t  | tj d W 5 Q R X dS )zP
        Test re-implementing print() for a custom type with @overload.
        Tr  zhello!
N)r   r   r   r1   r.  sysstdoutgetvaluerR  r6   r6   r7   
test_print  s    
z!TestHighLevelExtending.test_printc                 C   sT   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.
        Tr  r.   rn   r(  rH   T   N)r   r   r  MyDummy2r1   r  r6   r6   r7   test_add_operator  s
    z(TestHighLevelExtending.test_add_operatorc                 C   sT   t }tdd|}| |ddd | |t t d | |t t d dS 	zY
        Test re-implementing '+' for a custom type via @overload(operator.add).
        Tr  r.   rn   r(  rH   rY  N)r   r   r  rZ  r1   r  r6   r6   r7   test_add_binop  s
    z%TestHighLevelExtending.test_add_binopc                 C   sT   t }tdd|}| |ddd | |t t d | |t t d dS rX  )r   r   r  rZ  r1   r  r6   r6   r7   test_iadd_operator  s
    z)TestHighLevelExtending.test_iadd_operatorc                 C   sT   t }tdd|}| |ddd | |t t d | |t t d dS r\  )r   r   r  rZ  r1   r  r6   r6   r7   test_iadd_binop"  s
    z&TestHighLevelExtending.test_iadd_binopc                 C   s   t }tdd|}t }d }t <}z||d W n& tk
rX } z|}W 5 d }~X Y nX W 5 Q R X |d k	rp|| | d d S )NTr  A  zdel hello! 321
)r   r   r1   r   r   r.  rV  r=   r  r  rQ   eoutexcr6   r6   r7   test_delitem0  s     z#TestHighLevelExtending.test_delitemc                 C   s*   t }tdd|}| |t dd d S )NTr  r`  i  )r   r   r  r1   r  r6   r6   r7   test_getitem@  s    z#TestHighLevelExtending.test_getitemc                 C   s   t }tdd|}t }d }t >}z||dd W n& tk
rZ } z|}W 5 d }~X Y nX W 5 Q R X |d k	rr|| | d d S )NTr  r`  r   z321 123
)r   r   r1   r   r   r.  rV  ra  r6   r6   r7   test_setitemE  s     z#TestHighLevelExtending.test_setitemc           	   	   C   sx   t ddt}d}||}t|}tj|| t ddt}| t}|  W 5 Q R X t|j	}d}| 
|| dS )zW
        Test overloading whose return value cannot be represented in CPython.
        Tr  
   zcannot convert native ModuleN)r   r   r|   testingZassert_equalr   rL  	TypeErrorrN  rO  rM  )	r=   Zok_cfuncnr6  expectZ	bad_cfuncrP  errmsgZ	expectmsgr6   r6   r7   test_no_cpython_wrapperU  s    
z.TestHighLevelExtending.test_no_cpython_wrapperc              	   C   s   d'dd}d}d(dd}|  tj}||dd	d
d W 5 Q R X t|j}| || | d| | d| | d| d)dd}|  tj}||dd	d
d W 5 Q R X t|j}| || | d| | d| | d| d*dd}|  tj}||dd	d
d W 5 Q R X t|j}| || | d| | d|k | d| | d| ddlm}m	}	 |  tj}||dd	d
d W 5 Q R X t|j}| || | d| | d|k | d| |  tj}||	dd	d
d W 5 Q R X t|j}| || | d| | d|k | d| | d| d+dd}
|  tj}||
dd	d
d W 5 Q R X t|j}| || | d| | d|k | d| | d| d,dd }|  tj}||dd	d
d W 5 Q R X t|j}| || | d| | d|k | d!| d-d"d#}|  tj}||dd	d
d W 5 Q R X t|j}| || | d| | d$| d%d& }|  tj}||dd	d
d W 5 Q R X 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   s   d S rB   r6   r   brF   kwr6   r6   r7   
myoverloado  s    zjTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_ol.<locals>.myoverloadc                    s    S rB   r6   ro  )implr6   r7   _myoverload_implr  s    zpTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_ol.<locals>._myoverload_implTr  c                    s    | |||d d S )Nrq  r6   )r   rp  rF   rE  rr  r6   r7   foov  s    zcTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_ol.<locals>.foo)N)Nr   r   )rs  rt  rw  r6   )rs  rr  r7   gen_oln  s    
zVTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.gen_olz-Typing and implementation arguments differ in   c                 S   s   | dkrdS dS d S Nrh  r.   r6   ro  r6   r6   r7   impl1  s    zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl1r.   rn   r(  r9  zkeyword argument default valuesz<Parameter "kw=12">z<Parameter "kw=None">c                 S   s   | dkrdS dS d S r{  r6   )r   rp  rF   kwargr6   r6   r7   impl2  s    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 d S r{  r   )zrp  rF   rq  r6   r6   r7   impl3  s    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   s   | dkrdS dS d S r{  r6   )r   rp  rF   rE  rb  rq  r6   r6   r7   impl6  s    zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl6z<Parameter "d">z<Parameter "e">c                 S   s   | dkrdS dS d S r{  r6   )r   rp  rq  r6   r6   r7   impl7  s    zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl7z<Parameter "c">c                 S   s   | dkrdS dS d S r{  r6   )r   rp  rF   rq  Zextra_kwargr6   r6   r7   impl8  s    zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl8z<Parameter "extra_kwarg=None">c                 S   s   | dkrdS dS d S r{  r6   )r   rp  rF   r6   r6   r7   impl9  s    zUTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling.<locals>.impl9)N)rz  )N)N)N)N)NN)
rL  r   r   rN  rO  rM  assertFalseoverload_usecasesr  r  )r=   ry  sentinelr}  rb  msgr  r  r  r  r  r  r  r  r6   r6   r7   /test_typing_vs_impl_signature_mismatch_handlingh  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 5 Q R X 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   s   d S rB   r6   r   rq  r6   r6   r7   rr     s    ziTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positional.<locals>.myoverloadr.   )var_positional_implTr  c                    s    | |dddS )N	      ru  r6   r   rp  rv  r6   r7   rw    s    zbTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_positional.<locals>.foor4  z)VAR_POSITIONAL (e.g. *args) argument kindz-offending argument name is '*star_args_token')N)
r  r  r   r   rL  r   r   rN  rO  rM  )r=   r  rw  rb  r  r6   rv  r7   >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 5 Q R X t|j}| d	| | d
| ddd}|  tj}||dd W 5 Q R X 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   s   d S rB   r6   r  r6   r6   r7   rr    s    zvTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.gen_ol.<locals>.myoverloadstrictTr  c                    s    | ddS )Nr  ru  r6   r  rv  r6   r7   rw    s    zoTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.gen_ol.<locals>.foo)Nrx  )rs  r  rw  r6   rv  r7   ry    s
    
zbTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.gen_olc                 [   s   ddd}|S )Nrh  c                 S   s   | S rB   r6   r  r6   r6   r7   rs  $  s    zmTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol1.<locals>.impl)rh  r6   )r   r   rs  r6   r6   r7   ol1#  s    
z_TestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol1Fr.   rn   z1use of VAR_KEYWORD (e.g. **kwargs) is unsupportedz"offending argument name is '**kws'r   c                 S   s   dd }|S )Nc                 [   s   | S rB   r6   )r   r   r6   r6   r7   rs  2  s    zmTestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol2.<locals>.implr6   )r   rq  rs  r6   r6   r7   ol21  s    z_TestHighLevelExtending.test_typing_vs_impl_signature_mismatch_handling_var_keyword.<locals>.ol2N)T)r   )rL  r   r   rN  rO  rM  )r=   ry  r  rb  r  r  r6   r6   r7   ;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 )
Nrw  rh  c                 S   s   ddd}|S )Nrh  c                 S   s   |S rB   r6   )r   a_kwargr6   r6   r7   rs  A  s    zQTestHighLevelExtending.test_overload_method_kwargs.<locals>.fooimpl.<locals>.impl)rh  r6   )r   r  rs  r6   r6   r7   fooimpl?  s    
zCTestHighLevelExtending.test_overload_method_kwargs.<locals>.fooimplc                 S   s   |   |  d| j ddfS )N      )r  rw  Ar6   r6   r7   barF  s    z?TestHighLevelExtending.test_overload_method_kwargs.<locals>.barr4  )rh  r  r  )rh  )r   r   r   r   r|   r   r.  )r=   r  r  Zr6   r6   r7   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 )Nc                 S   s   |S rB   r6   )r   rE   r6   r6   r7   rs  W  s    zXTestHighLevelExtending.test_overload_method_literal_unpack.<locals>.litfoo.<locals>.impl)r:   r   r   Literal)r   rE   rs  r6   r6   r7   r  P  s    zJTestHighLevelExtending.test_overload_method_literal_unpack.<locals>.litfooc                 S   s
   |  dS N  )r  r  r6   r6   r7   r  \  s    zGTestHighLevelExtending.test_overload_method_literal_unpack.<locals>.barr.   r  )r   r   r   r   r|   Zzerosr.  )r=   r  r  r  r6   r6   r7   #test_overload_method_literal_unpackN  s    



z:TestHighLevelExtending.test_overload_method_literal_unpackc                 C   s   t dd }| | d d S )Nc                   S   s
   t tS rB   )r|   exprA   r6   r6   r6   r7   testh  s    z8TestHighLevelExtending.test_overload_ufunc.<locals>.testr  )r   r.  )r=   r  r6   r6   r7   test_overload_ufuncd  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 rB   r6   )rQ   rE   val2rl   r6   r6   r7   getq  s    z\TestHighLevelExtending.test_overload_method_stararg.<locals>._ov_method_stararg.<locals>.getr6   )rQ   rE   r  rl   r  r6   r6   r7   _ov_method_starargo  s    zOTestHighLevelExtending.test_overload_method_stararg.<locals>._ov_method_starargc                 W   s
   | j | S rB   r  )rQ   rl   r6   r6   r7   rw  v  s    z@TestHighLevelExtending.test_overload_method_stararg.<locals>.foor.   rn   r.   rn   r6   r(  r.   rn   r(  r9  r.   rn   r(  r9  c                 S   s(   |  dd|  ddd|  ddddfS )Nr.   rn   r(  r9  r  r  r6   r6   r7   r    s    
z@TestHighLevelExtending.test_overload_method_stararg.<locals>.bar)r  r  r  r  )r.   rn   )r  r  )r.   rn   )r  r(  )r9  r4  )r.   rn   )r  r   r8   r   r1   r.  )r=   r  rw  rQ   r  r6   r6   r7   test_overload_method_starargn  s4    


    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 5 Q R X | 
dt|j d S )Nc                   @   s   e Zd ZdS )zATestHighLevelExtending.test_overload_classmethod.<locals>.MyArrayNr2   r6   r6   r6   r7   MyArray  s   r  array_allocc                 S   s   dd }|S )Nc                 S   s   t |}|S rB   r|   r   )clsnitemsr   r6   r6   r7   rs    s    
zVTestHighLevelExtending.test_overload_classmethod.<locals>.ol_array_alloc.<locals>.implr6   )r  r  rs  r6   r6   r7   ol_array_alloc  s    zHTestHighLevelExtending.test_overload_classmethod.<locals>.ol_array_allocc                    s
     | S rB   )r  r  r  r6   r7   rw    s    z=TestHighLevelExtending.test_overload_classmethod.<locals>.foor   c                 S   s   t j| S rB   )r   r   r  r  r6   r6   r7   no_classmethod_in_base  s    zPTestHighLevelExtending.test_overload_classmethod.<locals>.no_classmethod_in_basez"Unknown attribute 'array_alloc' of)r   r   r(   r   r  r|   r   rL  r   r   rM  rN  rO  )r=   r  rw  r  r  rP  r6   r  r7   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   s   t | tjrdd }|S d S )Nc                 S   s   d|  S )N     @E@r6   ra   r6   r6   r7   rs    s    zeTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld1.<locals>.impl)r:   r   r   rb   rs  r6   r6   r7   callable_type_call_ovld1  s    zWTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld1c                 S   s   t | tjrdd }|S d S )Nc                 S   s   d|  S )N42.5r6   ra   r6   r6   r7   rs    s    zeTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld2.<locals>.impl)r:   r   ZUnicodeTyper  r6   r6   r7   callable_type_call_ovld2  s    zWTestHighLevelExtending.test_overload_callable_typeref.<locals>.callable_type_call_ovld2c                 S   s   t | t |fS rB   )r   r  r6   r6   r7   rw    s    zBTestHighLevelExtending.test_overload_callable_typeref.<locals>.foo)r9  4r  r   r  r.   )r   r   r   r  )r=   r  r  rw  rl   r5  r6   r6   r7   test_overload_callable_typeref  s    


z5TestHighLevelExtending.test_overload_callable_typerefN)r3   r4   r5   r&  rQ  rS  rW  r[  r]  r^  r_  re  rf  rg  rn  r  r  r  r  r  r  r  r  r  r6   r6   r6   r7   rH    s,   	 ,
+ rH  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
signaturesr   Z_cache_misses)r  Z
expect_hitZexpect_misseshitZmissr6   r6   r7   _assert_cache_stats  s    r  c                   @   s    e Zd ZdZdd Zdd ZdS )TestOverloadMethodCachingFc              	   C   s2   t | jj| _td| j |   W 5 Q R X d S )N	CACHE_DIR)r   r   r3   
_cache_dirr   run_caching_overload_methodr  r6   r6   r7   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k
r   t
}Y nX | }|jt|| jfd	}|  |t  |  | |jd |jdd
}| |d d S )NTr  cacher   r   r.   c                 S   s"   g | ]}| d rd|kr|qS )Zdeclarer   )
startswith)r   lnr6   r6   r7   
<listcomp>  s   
 zITestOverloadMethodCaching.run_caching_overload_method.<locals>.<listcomp>spawn)targetrl   )timeout)r   r   r  r1   r  Zinspect_llvmrC   
splitlinesr.  r   multiprocessingZget_contextAttributeErrorQueueProcessr  r  startputjoinexitcoder  )r=   r  ZllvmirZdeclsctxqpr   r6   r6   r7   r    s.    
 z5TestOverloadMethodCaching.run_caching_overload_methodN)r3   r4   r5   _numba_parallel_test_r  r  r6   r6   r6   r7   r    s   r  c              	   C   sP   t d|< |  }tdddt}||}| | t|dd W 5 Q R X dS )zH
    Used by TestOverloadMethodCaching.test_caching_overload_method
    r  Tr  r.   r   N)r   r  r   r   r  r  )r  	cache_dirr   r  r   r6   r6   r7   r    s    
r  c                   @   s4   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 5 Q R X | 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   s   t t j}dd }||fS )Nc                 S   s   d S rB   r6   r>   rJ   r*   rl   r6   r6   r7   codegen  s    zBTestIntrinsic.test_void_return.<locals>.void_func.<locals>.codegenr   voidre   r$  r   rk   r  r6   r6   r7   	void_func  s    z1TestIntrinsic.test_void_return.<locals>.void_funcc                 S   s   t t j}dd }||fS )Nc                 S   s   d S rB   r6   r  r6   r6   r7   r  '  s    zFTestIntrinsic.test_void_return.<locals>.non_void_func.<locals>.codegen)r   re   r  r6   r6   r7   non_void_func#  s    z5TestIntrinsic.test_void_return.<locals>.non_void_funcTr  c                      s    d dS Nr.   r   r6   r6   )r  r6   r7   call_void_func,  s    z6TestIntrinsic.test_void_return.<locals>.call_void_funcc                      s    d dS r  r6   r6   )r  r6   r7   call_non_void_func1  s    z:TestIntrinsic.test_void_return.<locals>.call_non_void_funcr   znon-void function returns NoneN)r#   r   r.  rL  r   rM  rO  r  )r=   r  r  rb  r6   )r  r  r7   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 5 Q R X |
  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tt fdd}|S )Nc                    s8    | tj t|tjr4 tj}dd }||fS d S )Nc                 S   s"   |\}|j }| |}|||S rB   )rj   Zget_value_typeZinttoptr)r>   rJ   r*   rl   srcZrtypeZllrtyper6   r6   r7   r  N  s    
z_TestIntrinsic.test_ll_pointer_cast.<locals>.unsafe_caster.<locals>.unsafe_cast.<locals>.codegen)r-  r	   Contextr:   r   r   Zuintp)r$  r  rk   r  )result_typer=   r6   r7   unsafe_castG  s
    
zNTestIntrinsic.test_ll_pointer_cast.<locals>.unsafe_caster.<locals>.unsafe_cast)r:   r   CPointerr   r#   )r  r  r  )r  r7   unsafe_casterD  s    z9TestIntrinsic.test_ll_pointer_cast.<locals>.unsafe_casterc                 S   s   t dd S )Nznot callable from pythonr_   r  r6   r6   r7   unsafe_get_ctypes_pointerZ  s    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 )Nr  )ctypesr   )r   r  r6   r7   
array_implb  s    zdTestIntrinsic.test_ll_pointer_cast.<locals>.array_impl_unsafe_get_ctypes_pointer.<locals>.array_impl)r:   r   r   r  r   )Zarrtyper   )r  r  r7   $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 rB   )r   r   )ptrrk  r   r6   r6   r7   my_c_fun_rawh  s    z8TestIntrinsic.test_ll_pointer_cast.<locals>.my_c_fun_rawNTr  c                    s   | } || j  d S rB   )r   )r   r  )my_c_funr  r6   r7   rw  q  s    z/TestIntrinsic.test_ll_pointer_cast.<locals>.foorh  )r   )r  r  r  r  r  r   r   r|   r   Zfloat32r   rV  r  closer   maprN  r.  )r=   r  r  r  r  r  r  Z	prototyperw  r   bufr6  rl  r6   )r  r=   r  r  r7   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   s   |d S r   r6   r  r6   r6   r7   r    s    zCTestIntrinsic.test_serialization.<locals>.identity.<locals>.codegenr6   )r>   rb   r  rk   r6   r6   r7   identity  s    z2TestIntrinsic.test_serialization.<locals>.identityTr  c                    s    | S rB   r6   ra   r	  r6   r7   rw    s    z-TestIntrinsic.test_serialization.<locals>.foor.   N)
r#   r   r.  r$   _memor   pickledumpsloadsr  )r=   rw  memo	memo_sizeZserialized_fooZfoo_rebuiltZserialized_identityZidentity_rebuiltr6   r
  r7   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   s   |d S r   r6   r  r6   r6   r7   r    s    zATestIntrinsic.test_deserialization.<locals>.defn.<locals>.codegenr6   )r>   rb   r  r6   r6   r7   defn  s    z0TestIntrinsic.test_deserialization.<locals>.defnrw  r.   N)r$   r  r   r  Z_defnr  r  r.  Z_recentclearr  ZassertIsNot)r=   r  r  r  originalZpickledZrebuiltsecondr6   r6   r7   test_deserialization  s"    




z"TestIntrinsic.test_deserializationc                 C   s`   t tddd}| d|j | d|j | d|j | dti|j | d|j d S )Nr   c                 S   s   t t j}dd }||fS )void_func docstringc                 S   s   d S rB   r6   r  r6   r6   r7   r    s    z@TestIntrinsic.test_docstring.<locals>.void_func.<locals>.codegenr  r  r6   r6   r7   r    s    z/TestIntrinsic.test_docstring.<locals>.void_funcznumba.tests.test_extendingr  r   r  )	r#   intr.  r4   r3   r5   ZassertDictEqual__annotations__r&  )r=   r  r6   r6   r7   test_docstring  s    
zTestIntrinsic.test_docstringN)r3   r4   r5   r  r  r  r  r  r6   r6   r6   r7   r    s
   *B.(r  c                   @   s   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   s   | | S rB   r6   rb   rz   r6   r6   r7   rw    s    z.TestRegisterJitable.test_no_flags.<locals>.fooc                    s
    | |S rB   r6   r  r  r6   r7   r    s    z.TestRegisterJitable.test_no_flags.<locals>.barTr  r.   rn   )r%   r   r.  )r=   r  cbarrl  r6  r6   r  r7   test_no_flags  s    


z!TestRegisterJitable.test_no_flagsc              	      s   t dddd   fdd}| |d dd	d
g tdd|}| tj}|d
 W 5 Q R X d}| |t|j	 d S )NF)_nrtc                 S   s
   t | S rB   r  rk  r6   r6   r7   rw    s    z2TestRegisterJitable.test_flags_no_nrt.<locals>.fooc                    s    | S rB   r6   r   r  r6   r7   r    s    z2TestRegisterJitable.test_flags_no_nrt.<locals>.barr(  r   r.   rn   Tr  COnly accept returning of array passed into the function as argument)
r%   r.  tolistr   rL  r   r   rM  rN  rO  )r=   r  r  rP  r  r6   r  r7   test_flags_no_nrt  s    
z%TestRegisterJitable.test_flags_no_nrtN)r3   r4   r5   r  r#  r6   r6   r6   r7   r    s   r  c                   @   sD   e Zd Zeedkddd Zdd Zeedk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j0Tr  c                    s    | S rB   r6   ra   Z_j0r6   r7   r     r   z@TestImportCythonFunction.test_getting_function.<locals>.<lambda>r   r.   )r&   r  r  Zc_doubler   r.  )r=   addrZfunctyper&  r6   r'  r7   test_getting_function  s     z.TestImportCythonFunction.test_getting_functionc              	   C   sD   |  t}tdd W 5 Q R X d}t|t|j}| | d S )NZ
fakemoduleZfakefunctionzNo module named '?fakemodule'?)rL  ImportErrorr&   rematchrN  rO  r  )r=   rP  r  r,  r6   r6   r7   test_missing_module  s
    z,TestImportCythonFunction.test_missing_modulec              	   C   s:   |  t}tdd W 5 Q R X d}| |t|j d S )Nr%  rw  zINo function 'foo' found in __pyx_capi__ of 'scipy.special.cython_special')rL  r   r&   r.  rN  rO  )r=   rP  r  r6   r6   r7   test_missing_function   s     z.TestImportCythonFunction.test_missing_function)	r3   r4   r5   unittestZskipIfscr)  r-  r.  r6   r6   r6   r7   r$    s
   
	r$  method_jit_option_check_nrtr  TZjit_optionsc                 C   s   dd }|S )Nc                 S   s
   t dS Nrh  r  r  r6   r6   r7   r   1  s    z+ov_method_jit_option_check_nrt.<locals>.impr6   r  r6   r6   r7   ov_method_jit_option_check_nrt-  s    r4  method_jit_option_check_no_nrtFc                 C   s   dd }|S )Nc                 S   s
   t dS r3  r  r  r6   r6   r7   r   ;  s    z.ov_method_jit_option_check_no_nrt.<locals>.impr6   r  r6   r6   r7   !ov_method_jit_option_check_no_nrt7  s    r6  attr_jit_option_check_nrtc                 C   s   dd }|S )Nc                 S   s
   t dS r3  r  r  r6   r6   r7   r   E  s    z)ov_attr_jit_option_check_nrt.<locals>.impr6   r  r6   r6   r7   ov_attr_jit_option_check_nrtA  s    r8  attr_jit_option_check_no_nrtc                 C   s   dd }|S )Nc                 S   s
   t dS r3  r  r  r6   r6   r7   r   O  s    z,ov_attr_jit_option_check_no_nrt.<locals>.impr6   r  r6   r6   r7   ov_attr_jit_option_check_no_nrtK  s    r:  c                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestJitOptionsNoNRTc              	   O   s<   d}|  tj}||| W 5 Q R X | |t|j d S )Nr!  )rL  r   r   rM  rN  rO  )r=   funcrl   rI  r  rP  r6   r6   r7   check_error_no_nrtX  s
    z&TestJitOptionsNoNRT.check_error_no_nrtc                    sX   dd  t  d|iddd }t fdd}|rJ| | td	 n
| | d S )
Nc                   S   s
   t dS r3  r  r6   r6   r6   r7   dummyc  s    z8TestJitOptionsNoNRT.no_nrt_overload_check.<locals>.dummyr  r2  c                  S   s   dd } | S )Nc                   S   s
   t dS r3  r  r6   r6   r6   r7   r>  h  s    zJTestJitOptionsNoNRT.no_nrt_overload_check.<locals>.ov_dummy.<locals>.dummyr6   r>  r6   r6   r7   ov_dummyf  s    z;TestJitOptionsNoNRT.no_nrt_overload_check.<locals>.ov_dummyc                      s     S rB   r6   r6   r?  r6   r7   rw  m  s    z6TestJitOptionsNoNRT.no_nrt_overload_check.<locals>.foorh  )r   r   r  r|   r   r=  )r=   flagr@  rw  r6   r?  r7   no_nrt_overload_checkb  s    
z)TestJitOptionsNoNRT.no_nrt_overload_checkc                 C   s   |  d |  d d S )NTF)rB  r  r6   r6   r7   test_overload_no_nrtv  s    
z(TestJitOptionsNoNRT.test_overload_no_nrtc                 C   s>   t dd }| |ttd t dd }| |t d S )Nc                 S   s   |   S rB   )r1  ra   r6   r6   r7   udt{  s    z<TestJitOptionsNoNRT.test_overload_method_no_nrt.<locals>.udtrh  c                 S   s   |   S rB   )r5  ra   r6   r6   r7   rD    s    r   r  rA   r|   r   r=  r=   rD  r6   r6   r7   test_overload_method_no_nrtz  s    

z/TestJitOptionsNoNRT.test_overload_method_no_nrtc                 C   s>   t dd }| |ttd t dd }| |t d S )Nc                 S   s   | j S rB   )r7  ra   r6   r6   r7   rD    s    z?TestJitOptionsNoNRT.test_overload_attribute_no_nrt.<locals>.udtrh  c                 S   s   | j S rB   )r9  ra   r6   r6   r7   rD    s    rE  rF  r6   r6   r7   test_overload_attribute_no_nrt  s    

z2TestJitOptionsNoNRT.test_overload_attribute_no_nrtN)r3   r4   r5   r=  rB  rC  rG  rH  r6   r6   r6   r7   r;  U  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 )Nr]   )r   setUpr\   rT   rS   rV   dyn_type)r=   manyr   r6   r7   rJ    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                    s    fdd}t ||}||fS )Nc                    s   |\}| || S rB   r   r   r>   rJ   rk   rl   rE   magic_offsetr6   r7   rs    s    zKTestBoxingCallingJIT.test_unboxer_basic.<locals>.my_intrinsic.<locals>.implr*   r$  rE   rs  rk   rO  r6   r7   my_intrinsic  s    
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                    s    | S rB   r6   ra   rS  r6   r7   bridge  s    zHTestBoxingCallingJIT.test_unboxer_basic.<locals>.unboxer.<locals>.bridgeis_error	r>   rI   r   intpr*   Zvoidptrrv   call_jit_coder"   rP   rQ   rF   rU  rl   rk   rW  r   magic_tokenrS  r6   r7   unboxer  s
    z8TestBoxingCallingJIT.test_unboxer_basic.<locals>.unboxerc                 S   s   |j |tj}|j|S rB   rJ   Zptrtointr   Zintp_trv   long_from_ssize_trP   rE   rF   r   r6   r6   r7   boxer  s    z6TestBoxingCallingJIT.test_unboxer_basic.<locals>.boxerc                 S   s   | S rB   r6   ra   r6   r6   r7   passthru  s    z9TestBoxingCallingJIT.test_unboxer_basic.<locals>.passthrur#   r!   rT   r    r   rK  r.  )r=   r^  rb  rc  rc  r6   rP  r]  rS  r7   test_unboxer_basic  s    	


z'TestBoxingCallingJIT.test_unboxer_basicc              	   C   sf   t | jdd }t| jdd }tdd }| t}|| j W 5 Q R X | 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   s   | dkrt d| S )Nr   cannot be x > 0r   ra   r6   r6   r7   rU    s    zHTestBoxingCallingJIT.test_unboxer_raise.<locals>.unboxer.<locals>.bridger.   rV  rX  r[  r6   r6   r7   r^    s
    z8TestBoxingCallingJIT.test_unboxer_raise.<locals>.unboxerc                 S   s   |j |tj}|j|S rB   r_  ra  r6   r6   r7   rb    s    z6TestBoxingCallingJIT.test_unboxer_raise.<locals>.boxerc                 S   s   | S rB   r6   ra   r6   r6   r7   rc    s    z9TestBoxingCallingJIT.test_unboxer_raise.<locals>.passthrurg  
r!   rT   r    r   rL  r   rK  rM  rN  rO  r=   r^  rb  rc  rP  r6   r6   r7   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                    s    fdd}t ||}||fS )Nc                    s   |\}| || S rB   rM  rN  rO  r6   r7   rs    s    zCTestBoxingCallingJIT.test_boxer.<locals>.my_intrinsic.<locals>.implrQ  rR  rO  r6   r7   rS    s    
z5TestBoxingCallingJIT.test_boxer.<locals>.my_intrinsicc                 S   s   t |j S rB   rM   rO   r6   r6   r7   r^    s    z0TestBoxingCallingJIT.test_boxer.<locals>.unboxerc                    sL   fdd}|j tj g}ttjtj}|j|||\}}|j|S )Nc                    s    | S rB   r6   ra   rT  r6   r7   rU    s    z>TestBoxingCallingJIT.test_boxer.<locals>.boxer.<locals>.bridge)r>   rI   r   rY  r*   rv   rZ  r`  )rP   rE   rF   rU  rl   rk   rW  r   r\  r6   r7   rb    s
    z.TestBoxingCallingJIT.test_boxer.<locals>.boxerc                 S   s   | S rB   r6   ra   r6   r6   r7   rc    s    z1TestBoxingCallingJIT.test_boxer.<locals>.passthrurd  )r=   r^  rb  rc  rr6   re  r7   
test_boxer  s    	



zTestBoxingCallingJIT.test_boxerc              	   C   sf   t | jdd }t| jdd }tdd }| t}|| j W 5 Q R X | dt|j	 d S )Nc                 S   s   t |j S rB   rM   rO   r6   r6   r7   r^    s    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 5 Q R X |j	|S )Nc                 S   s   | dkrt d| S )Nr   cannot do x > 0rh  ra   r6   r6   r7   rU    s    zDTestBoxingCallingJIT.test_boxer_raise.<locals>.boxer.<locals>.bridger.   T)zfill)r>   rI   r   rY  r*   rv   rZ  r   Zalloca_oncerJ   ZpyobjZif_thennot_r`  storeload)
rP   rE   rF   rU  rl   rk   rW  r   retvalrQ   r6   r6   r7   rb    s    z4TestBoxingCallingJIT.test_boxer_raise.<locals>.boxerc                 S   s   | S rB   r6   ra   r6   r6   r7   rc  !  s    z7TestBoxingCallingJIT.test_boxer_raise.<locals>.passthrurn  ri  rj  r6   r6   r7   test_boxer_raise
  s    


 z%TestBoxingCallingJIT.test_boxer_raise)	r3   r4   r5   rJ  rf  rk  rm  rt  r  r6   r6   r   r7   rI    s
   *%rI  c                 C   s   d S rB   r6   ra   r6   r6   r7   with_objmode_cache_ov_example,  s    ru  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NumbaWarningr  r6   r6   r7   rJ  9  s    z TestCachingOverloadObjmode.setUpc                 C   s   t   d S rB   )rx  resetwarningsr  r6   r6   r7   tearDown<  s    z#TestCachingOverloadObjmode.tearDownc              	      s   t | jj}td| dd fdd tt fdd}tdd	d
d }d}|d}| || tdd	|j}|d}| || W 5 Q R X d S )Nr  c                 S   s   t | |  }t j|S rB   )r|   r   ZlinalgZnorm)rb   r   r6   r6   r7   realworkC  s    zJTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>.realworkc                    s    | S rB   r6   ra   )r|  r6   r7   python_codeH  s    zMTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>.python_codec                    s    fdd}|S )Nc              	      s"   t dd  | }W 5 Q R X |S Nr  )rz   r   r  r}  r6   r7   rs  N  s    zqTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>._ov_with_objmode_cache_ov_example.<locals>.implr6   r  r  r6   r7   !_ov_with_objmode_cache_ov_exampleL  s    zcTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>._ov_with_objmode_cache_ov_exampleTr  c                 S   s   t | S rB   )ru  ra   r6   r6   r7   testcaseU  s    zJTestCachingOverloadObjmode.test_caching_overload_objmode.<locals>.testcaser   )	r   r   r3   r   r   ru  r   r.  py_func)r=   r  r  r  rl  r6  Ztestcase_cachedr6   )r}  r|  r7   test_caching_overload_objmode?  s    
z8TestCachingOverloadObjmode.test_caching_overload_objmodec                    s\   dd dd  t  fdd}tdd fd	d
}| }| }||krXtd|S )Nc                 S   s   t | | S rB   r|   sumr  r6   r6   r7   do_thisc  s    zGTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.do_thisc                 S   s   t | | S rB   r  r  r6   r6   r7   do_somethingf  s    zLTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.do_somethingc                    s    fdd}|S )Nc              	      s$   t dd  | |}W 5 Q R X |S r~  r  )r   rp  rz   r  r6   r7   _do_something_implk  s    zqTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.overload_do_something.<locals>._do_something_implr6   )r   rp  r  r  r6   r7   overload_do_somethingi  s    zUTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.overload_do_somethingTr  c                     s   t d} t d} | |S )Nr  r  r  )r  r6   r7   test_cachingq  s    

zLTestCachingOverloadObjmode.check_objmode_cache_ndarray.<locals>.test_cachingzincorrect result)r   r   r  r   )r  r  r  r6  rl  r6   )r  r  r7   check_objmode_cache_ndarraya  s    z6TestCachingOverloadObjmode.check_objmode_cache_ndarrayc                 C   s   |    d S rB   )r  )r  r6   r6   r7   *populate_objmode_cache_ndarray_check_cache  s    zETestCachingOverloadObjmode.populate_objmode_cache_ndarray_check_cachec                 C   s<   |   }t|jjdkr tdt|jjdkr8tdd S )Nr   zunexpected cache misszunexpected missing cache hit)r  r   statsZcache_missesr   Z
cache_hits)r  Zdispr6   r6   r7   'check_objmode_cache_ndarray_check_cache  s
    zBTestCachingOverloadObjmode.check_objmode_cache_ndarray_check_cachec              	   C   sN   t | jj}td| t| j| t| j|}W 5 Q R X | |d d d S )Nr  r  r   )r   r   r3   r   r   r  r  r.  )r=   r  r   r6   r6   r7    test_check_objmode_cache_ndarray  s      z;TestCachingOverloadObjmode.test_check_objmode_cache_ndarrayN)r3   r4   r5   r&  r  rJ  r{  r  classmethodr  r  r  r  r6   r6   r6   r7   rv  2  s   "


rv  c                   @   s   e Zd Zdd Zdd ZdS )TestMiscc                 C   sn   dd }|  t| | tt| |  tt| |  ttdd| |  ttdd| d S )Nc                 S   s   d S rB   r6   ra   r6   r6   r7   rw    s    z$TestMisc.test_is_jitted.<locals>.fooT)parallelzvoid(float64[:])z(m))r  r'   r   r   r   r   )r=   rw  r6   r6   r7   test_is_jitted  s    zTestMisc.test_is_jittedc                 C   s@   t dd }t dd }| | |  | | |  d S )Nc                   S   s   t jdddjS )Nr     )r   
fill_valuer|   fullr   r6   r6   r6   r7   standard_order  s    z:TestMisc.test_overload_arg_binding.<locals>.standard_orderc                   S   s   t jdddjS )Nr  r   )r  r   r  r6   r6   r6   r7   reversed_order  s    z:TestMisc.test_overload_arg_binding.<locals>.reversed_order)r   r  r  )r=   r  r  r6   r6   r7   test_overload_arg_binding  s    

z"TestMisc.test_overload_arg_bindingN)r3   r4   r5   r  r  r6   r6   r6   r7   r    s   r  c                   @   s   e Zd Zdd Zdd ZdS )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   s   d S rB   r6   ra   r6   r6   r7   
prefer_lit  s    z;TestOverloadPreferLiteral.test_overload.<locals>.prefer_litc                 S   s   d S rB   r6   ra   r6   r6   r7   non_lit  s    z8TestOverloadPreferLiteral.test_overload.<locals>.non_litc                 S   s>   t | tjr.| jdkr"dd }|S tdndd }|S d S )Nr.   c                 S   s   dS r  r6   ra   r6   r6   r7   rs    s    zATestOverloadPreferLiteral.test_overload.<locals>.ov.<locals>.implliteral valuec                 S   s   | d S Nd   r6   ra   r6   r6   r7   rs    s    r:   r   ZIntegerLiteralZliteral_valuer   r   r  r6   r6   r7   ov  s    
z3TestOverloadPreferLiteral.test_overload.<locals>.ovTZprefer_literalc                    s    d d | fS Nr.   rn   r6   ra   )r  r6   r7   check_prefer_lit  s    zATestOverloadPreferLiteral.test_overload.<locals>.check_prefer_litr(  r     ,  c                    s    d d | fS r  r6   ra   )r  r6   r7   check_non_lit  s    z>TestOverloadPreferLiteral.test_overload.<locals>.check_non_litr  )r   r   r.  r=   r  r  r   rp  rF   r  r6   )r  r  r7   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   s>   t |tjr.|jdkr"dd }|S tdndd }|S d S )Nr.   c                 S   s   dS r  r6   r=   rb   r6   r6   r7   rs    s    zHTestOverloadPreferLiteral.test_overload_method.<locals>.ov.<locals>.implr  c                 S   s   |d S r  r6   r  r6   r6   r7   rs    s    r  )r=   rb   rs  r6   r6   r7   r    s    
z:TestOverloadPreferLiteral.test_overload_method.<locals>.ovmethod_prefer_literalTr  method_non_literalFc                 S   s   |  d|  d|  |fS r  )r  r>  rb   r6   r6   r7   r    s    zHTestOverloadPreferLiteral.test_overload_method.<locals>.check_prefer_litr(  r  r  r  c                 S   s   |  d|  d|  |fS r  )r  r  r6   r6   r7   r    s    zETestOverloadPreferLiteral.test_overload_method.<locals>.check_non_litr  r  r  r6   r6   r7   test_overload_method  s6      

z.TestOverloadPreferLiteral.test_overload_methodN)r3   r4   r5   r  r  r6   r6   r6   r7   r    s   *r  __main__)N)ro   r   rT  r  r  r  rx  r+  numpyr|   Znumbar   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   r/  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_usecaser/   r0   Zscipy.special.cython_specialspecialZcython_specialr0  r*  rW   r1   rX   r8   rC   rA   rY   rG   r;   rK   rL   rZ   rR   r\   r   rZ  Zmydummy_type_2r^   rc   rh   r  rd   rm   rf   rr   rs   rw   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  r	  r
  r'  rH  r  r  r  r  r  r$  r4  r6  r8  r:  r;  rI  ru  rv  r  r  r3   mainr6   r6   r6   r7   <module>   s>   P






	





1

	

	
	



"


4a    	( Y#   
  
  
  
@ h_
