U
|
P ±dN ã @ s² d dl Z d dlZd dlZd dlmZmZmZ G dd dejj j
|
Z dd Z d ddZ ejejejejejejejejejejejejeeej ej ej ej ej gZ!G d d
|
d
|
Z"dS ) é N)Úassert_Ú assert_equalÚ assert_raisesc @ s0 e Zd Zdd ZejejfZdd Z dd Z
|
dS )Ú ArrayLikec C s t |¡| _d S ©N)ÚnpZasarrayÚvalue)Úselfr © r
|
úRd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_mixins.pyÚ__init__ s zArrayLike.__init__c s® | dd¡}|| D ] }t| jtf st S qtdd |D }|r`tdd |D |d< t||||}t|tkrt fdd|D S |dkrd S t |S d S )NÚoutr
|
c s s" | ]}t |tr|jn|V qd S r ©Ú
|
isinstancer r ©Ú.0Úxr
|
r
|
r Ú <genexpr> s ÿz,ArrayLike.__array_ufunc__.<locals>.<genexpr>c s s" | ]}t |tr|jn|V qd S r r r r
|
r
|
r r ! s ÿc 3 s | ]}t |V qd S r )Útyper ©r r
|
r r ( s Úat)Úgetr Ú_HANDLED_TYPESr ÚNotImplementedÚtupleÚgetattrr )r ZufuncÚmethodÚinputsÚkwargsr r Úresultr
|
r r Ú__array_ufunc__ s"
|
ÿþ zArrayLike.__array_ufunc__c C s dt | j| jf S )Nz%s(%r))r Ú__name__r r r
|
r
|
r Ú__repr__0 s zArrayLike.__repr__N) r! Ú
|
__module__Ú __qualname__r r ZndarrayÚnumbersÚNumberr r r" r
|
r
|
r
|
r r s r c C s* t | tkr tdd | D S t| S d S )Nc s s | ]}t |V qd S r )r )r Úrr
|
r
|
r r 6 s z"wrap_array_like.<locals>.<genexpr>)r r r ©r r
|
r
|
r Úwrap_array_like4 s r) c C s t t| t||d t| trXt t| t||d t| |D ]\}}t||| q@n4t | j|j|d t t| jdd t|jdd |d d S )N©Úerr_msgÚdtype) r r r r ÚlenÚzipÚ _assert_equal_type_and_valuer r )r Úexpectedr+ Z result_itemZ expected_itemr
|
r
|
r r/ ; s
|
ÿr/ c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚTestNDArrayOperatorsMixinc C s¼ dd }|t dd |dt d |t dt d¡ |t d¡t d |t t d¡d |dt t d¡ |t t d¡t d¡ |t d¡t t d¡ d S )Nc S s t | td d S ©Nr )r/ r r( r
|
r
|
r Úcheckb s z<TestNDArrayOperatorsMixin.test_array_like_add.<locals>.checkr )r r Úarray)r r3 r
|
r
|
r Útest_array_like_add` s z-TestNDArrayOperatorsMixin.test_array_like_addc C s` t t dg¡}|d7 }t|t t dg¡ t dg¡}|t d7 }t|t t dg¡ d S )Nr é )r r r4 r/ )r Ú
|
array_liker4 r
|
r
|
r Ú test_inplaceq s z&TestNDArrayOperatorsMixin.test_inplacec C s G dd d}t d}| }t|| |k t|| |k tt ||7 }W 5 Q R X tt || W 5 Q R X tt || W 5 Q R X d S )Nc @ s$ e Zd ZdZdZdd Zdd ZdS )z6TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOutz(Object that opts out of __array_ufunc__.Nc S s | S r r
|
©r Úotherr
|
r
|
r Ú__add__ s z>TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__add__c S s | S r r
|
r9 r
|
r
|
r Ú__radd__ s z?TestNDArrayOperatorsMixin.test_opt_out.<locals>.OptOut.__radd__)r! r# r$ Ú__doc__r r; r< r
|
r
|
r
|
r ÚOptOut| s r> r6 )r r r Ú TypeError)r r> r7 Zopt_outr
|
r
|
r Ú test_opt_outz s
|
|
|
|
z&TestNDArrayOperatorsMixin.test_opt_outc C s@ G dd dt }t d}|d}t|| | t|| | d S )Nc @ s e Zd ZdZdS )z=TestNDArrayOperatorsMixin.test_subclass.<locals>.SubArrayLikez&Should take precedence over ArrayLike.N)r! r# r$ r= r
|
r
|
r
|
r Ú SubArrayLike s rA r r6 )r r/ )r rA r Úyr
|
r
|
r Ú test_subclass s
|
z'TestNDArrayOperatorsMixin.test_subclassc C sf t d}t }tt || W 5 Q R X tt || W 5 Q R X tt ||7 }W 5 Q R X d S r2 )r Úobjectr r? )r r Úobjr
|
r
|
r Ú test_object s
|
|
|
z%TestNDArrayOperatorsMixin.test_objectc C sL t ddddg¡}t|}tjtjttjfD ]}t||t|| q,d S )Néÿÿÿÿr r6 é ) r r4 r ÚoperatorÚnegÚposÚabsÚinvertr/ )r r4 r7 Úopr
|
r
|
r Útest_unary_methodsª s ýz,TestNDArrayOperatorsMixin.test_unary_methodsc C sX t ddddg¡}t|}tD ]4}t||d}||d}d |¡}t|||d q d S )NrG r r6 rH úfailed for operator {}r* )r r4 r Ú_ALL_BINARY_OPERATORSr) Úformatr/ )r r4 r7 rN r0 Úactualr+ r
|
r
|
r Útest_forward_binary_methods³ s
|
|
z5TestNDArrayOperatorsMixin.test_forward_binary_methodsc C sB t D ]8}t|dd}|dtd}d |¡}t|||d qd S )NrH r6 rP r* )rQ r) r rR r/ )r rN r0 rS r+ r
|
r
|
r Ú test_reflected_binary_methods¼ s
|
|
z7TestNDArrayOperatorsMixin.test_reflected_binary_methodsc C sd t jddgt jd}t|}tt d¡}t|t ||¡ t|t ||¡ t|t ||¡ d S )Nr6 rH )r, é )r r4 Úfloat64r r/ ÚmatmulrI )r r4 r7 r0 r
|
r
|
r Ú test_matmulà s
|
ÿ
|
ÿz%TestNDArrayOperatorsMixin.test_matmulc C sP t t ddddg¡}ttj |t ddg¡¡d k t|t ddddg d S )Nr6 rH é é r rG éþÿÿÿ)r r r4 r Únegativer r/ )r r4 r
|
r
|
r Ú test_ufunc_atÍ s z'TestNDArrayOperatorsMixin.test_ufunc_atc C sP t d¡\}}t|t|f}tt td¡| tt tt d¡¡| d S )Ng À?)r Úfrexpr r/ r4 )r ZmantissaÚexponentr0 r
|
r
|
r Útest_ufunc_two_outputsÒ s ÿ ÿz0TestNDArrayOperatorsMixin.test_ufunc_two_outputsN)r! r# r$ r5 r8 r@ rC rF rO rT rU rY r^ ra r
|
r
|
r
|
r r1 ^ s
|
|
|
r1 )N)#r% rI Únumpyr Z numpy.testingr r r ÚlibZmixinsZNDArrayOperatorsMixinr r) r/ ÚltÚleÚeqÚneÚgtÚgeÚaddÚsubÚmulÚtruedivÚfloordivÚmodÚdivmodÚpowÚlshiftÚrshiftÚand_ÚxorÚor_rQ r1 r
|
r
|
r
|
r Ú<module> s6 )
|
í
|
