U
|
O ±dDu ã @ s\ d dl Z d dlZd dlZd dlmZmZ d dlmZ d dl m
|
Z
|
d dl m Z m Z mZmZ dZdZeege Zd ZdZd d
|
Zd d Zd d Zdd Zdd Zejj dd Z dd Z dd Z dd Z d;d d Z!d d Z"ejj d!d" Z#ejj d#d$ Z$d%d& Z%d'd( Z&d)d* Z'd+d, Z(d<d-d.Z)d/d0 Z*d1d2 Z+d3d4 Z,d5d6 Z-d7d8 Z.G d9d: d:Z/dS )=é N)Úsolve_diophantineÚinternal_overlap)Ú _umath_tests)Ú
|
as_strided)Úassert_Ú assert_raisesÚ assert_equalÚassert_array_equalé é
|
éÿÿÿÿc C sl | dkrt d gS g }dD ]L}dD ]B}t d | | | d }|| | | }| t|||| ¡ q"q|S )zGReturns slices of length nelems, from start onwards, in direction sign.r r
|
©é r
|
)r r )ÚsizeÚappendÚslice)ÚnelemsÚresÚstepÚsignÚstartÚstop© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_mem_overlap.pyÚ_indices_for_nelems s
|
r c C s, g } dD ] }t |}| t ||¡¡ q| S )z$Returns (src, dst) pairs of indices.)r r
|
é )r ÚextendÚ itertoolsÚproduct)r r Úindr r r Ú_indices_for_axis% s
|
r c C s t }tj|| dS )zKReturns ((axis0_src, axis0_dst), (axis1_src, axis1_dst), ... ) index pairs.)Úrepeat)r r r )Úndimsr r r r Ú_indices0 s r# c C sV t t t¡¡ t¡}| ¡ }|| ||< || ||< tt ||k¡d|| f dS )z1Check assignment arr[dstidx] = arr[srcidx] works.zassigning arr[%s] = arr[%s]N)ÚnpÚaranger ÚshapeÚreshapeÚcopyr Úall)ÚsrcidxÚdstidxZarrÚcpyr r r Ú_check_assignment7 s
|
ÿr- c C sD t t} | D ]2}tdd |D }tdd |D }t|| q d S )Nc S s g | ] }|d qS ©r r ©Ú.0Úar r r Ú
|
<listcomp>K s z0test_overlapping_assignments.<locals>.<listcomp>c S s g | ] }|d qS ©r r r/ r r r r2 L s )r# r" Útupler- )Zindsr r* r+ r r r Ú test_overlapping_assignmentsE s
|
r5 c s6 t j d¡t t j¡j} tdD ] }d}d}d|d }t|||k r"djddt jdd jddt jdd t| t| d t fd d
|
t|D }t fd d
|
t|D }t| d t
|
d d
|
t ||D }t jd|d t jd|dkr ||k r dt ||}|d krÄt ||dd} t| d k||| f t dd
|
t ||D }
|
d} |
|
D ]} | t| 9 } qz| dk r.ttfdd
|
tj|
|
D |d7 }q<t ||dd} t| d k ||| f tt
|
dd
|
t ||D k ttdd
|
t ||D |d7 }q<q"d S )NéÒ r r éô r é ©Údtypeé c 3 s( | ] }t jd d tjdV qdS )r r9 N©ÚintÚrandintr$ Úintp©r0 Új)ÚA_maxÚrngr r Ú <genexpr>f s ÿz(test_diophantine_fuzz.<locals>.<genexpr>c 3 s( | ] }t jd d tjdV qdS )r r
|
r9 Nr< r@ )ÚU_maxrC r r rD h s ÿr
|
c s s | ]\}}|| V qd S ©Nr ©r0 r1 Úubr r r rD k s r )Zsimplifyc s s& | ] \}}t d || d |V qdS )r r N©ÚrangerG r r r rD { s i c 3 s | ]}t | kV qd S rF )Úsum)r0 Úw)Úbr r rD s c s s | ]\}}|| V qd S rF r ©r0 r1 Úxr r r rD s c s s* | ]"\}}d | ko |kn V qdS ©r Nr )r0 rO rH r r r rD s )r$ ÚrandomÚ RandomStateÚiinfor? ÚmaxrJ Úminr> r4 rK Úzipr= r r ÚlenÚanyr r r) ) Zmax_intÚndimZfeasible_countZinfeasible_countÚ min_countÚAÚUZb_ubÚXZ X_simplifiedÚrangesr Úrr )rB rE rM rC r Útest_diophantine_fuzzQ sJ
|
ÿÿ"
|
|
"
|
r` c C sr t t j¡j} t t j¡j}|| krn|d |d d f}|d |d d f}d|d d }tt|||d d S )Nr
|
r )r r )r$ rS r? rT Úint64r r )Zmax_intpZ max_int64r[ r\ rM r r r Útest_diophantine_overflow s rb c C sÜ t j| |td}tt | |¡t j| |td | d¡ | d¡ | d¡ | ¡ }d}||krÊdd d| jd d |jd d f d | j f d
|
|j f d | j
|
f d |j
|
f d | j f d|j f g¡ }t|||d d S )N©Zmax_workr r Ú z z
|
zbase_a - base_b = %rÚdataz shape_a = %rz shape_b = %rzstrides_a = %rzstrides_b = %rz size_a = %rz size_b = %r©Úerr_msg) r$ Úmay_share_memoryÚMAY_SHARE_EXACTr ÚMAY_SHARE_BOUNDSÚfillrX ÚjoinÚ__array_interface__r& Ústridesr )r1 rM ÚgotÚexactrg r r r Ú check_may_share_memory_exact s( ÿ
|
|
|
|
|
|
|
|
|
ù
|
rq c
|
C s¢ t jddddgt jdt jddddgt jdd d
d d
d d d
d d
f g} g }| D ]<}tjtd td d dgfd D ]}|| }| |¡ qzqV|D ]Ø}tt |d d
d
|
d d
f |d d
d d d
f ¡ tt j|d d
d
|
d d
f |d d
d d d
f d d t |d d
d
|
d d
f |d d
d d d
f t |d d
d d d
f |d d
dd d
f z&|
|
¡ }|j j r~W q| t j¡}W n tk
|
r¦ Y qY nX t |
|
¡ dd
| dddd¡d d
d d d
f t |d d
d d d
f | dddd¡ t |d d
d d d
f | dddd¡d d
dd d
f t |
|
¡ dd
| dddd¡d d
d d d
f qt jd gt jd}t || t || ¡ d S )Né é é é r9 é. r
|
r é r r rc é r r; r8 )r$ ÚzerosÚint8r r r r r rh rq ZravelÚflagsZowndataÚviewÚint16Ú
|
ValueErrorr' r( )Zxs0ÚxsrO ÚssÚxpr r r Ú test_may_share_memory_manual· sF 4þ$
|
48..
|
|
ÿÿ ÿ ÿ
|
r TFc # sR t j d¡|r|rt fdd fdd| | fV tdddD ]D}| |d
| d |
fV | d |d
f | d d |
f fV qHt| j}d
|
|d
|
< t| | j|d }| |fV ||fV t| j}|d
|
dkrâd|d
|
< t| | j|d }| |fV ||fV t fd d t| j
|
D }t fdd t | j|D }t | j
|
¡} |¡ |rb|} nt | j
|
¡} | ¡ | | }
|
|rÜ|
|
jd
|
krqt fdd t | j||
|
jD } t fdd t | j| |
|
jD } n(|rè|} n t fdd t| j
|
D } |s(t fdd t | j| D } |
|
|¡}
|
| | | ¡} |
|
| fV qd S )Nr6 c s` j d| d tjd} j || d tjd} j ddtjddkrT|| }}|d9 }t|||S ©Nr r r9 r
|
r ©r> r$ r? r ©Únr r r ©rC r r Ú random_sliceð s
|
z,iter_random_view_pairs.<locals>.random_slicec sn d| d || ¡}||d | d } dd¡dkrb|d |d }}|dk rZd }|d9 }t|||S )Nr r r
|
r )r> r )r r r r r r r r Úrandom_slice_fixed_sizeø s z7iter_random_view_pairs.<locals>.random_slice_fixed_sizer rx r .r )r& rn c 3 s: | ]2} j d dtjdd kr. j ddtjdndV qdS ©r é r9 r r8 N©r> r$ r? r@ r r r rD s ÿÿz)iter_random_view_pairs.<locals>.<genexpr>c 3 s | ]\}} ||V qd S rF r ©r0 ÚpÚs©r r r rD s c 3 sB | ]:\}}} d d¡d kr6 dtd|d| ¡ndV qdS )r r r r
|
N)r> rT ©r0 r r Úpar r r rD - s ÿÿ c 3 s | ]\}}} |||V qd S rF r r )r r r rD 0 s ÿc 3 s: | ]2} j d dtjdd kr. j ddtjdndV qdS r r r@ r r r rD 5 s ÿÿc 3 s | ]\}} ||V qd S rF r r r r r rD : s )r$ rQ rR r~ rJ Úlistrn r r& r4 rY rV r% Úshuffler Ú transpose)rO Ú
|
same_stepsÚ
|
equal_sizerA rn r ÚstepsÚs1Út1Út2r1 Zsteps2Ús2rM r )r r rC r Úiter_random_view_pairsê sd
|
&
|
|
|
|
|
|
þ
|
|
þ ÿ
|
þ
|
r c C sÆ t jddddgt jd}d}d}t||}t|||k rÂt|\}}t ||¡} t ||¡}
|
t j||| ||d} t j||td} | | krt| | |
|
| kr¦t|
|
| | r*| r¸|d7 }q*|d7 }q*d S ) Né é" éG éa r9 r rc r ) r$ ry r} r rU Únextrh ri r ) Ú get_max_workr rZ rO ZfeasibleZ
|
infeasibleZ pair_iterr1 rM Zbounds_overlapZmay_share_answerZ easy_answerZ exact_answerr r r Ú check_may_share_memory_easy_fuzzB s$
|
|
|
|
r¤ c C s t dd ddd d S )Nc S s dS ©Nr r ©r1 rM r r r Ú<lambda>g ó z1test_may_share_memory_easy_fuzz.<locals>.<lambda>TéÐ ©r£ r rZ ©r¤ r r r r Ú test_may_share_memory_easy_fuzzb s þr¬ c C s t dd ddd d S )Nc S s t | j|jd S )Nr
|
)rT r r¦ r r r r§ u r¨ z3test_may_share_memory_harder_fuzz.<locals>.<lambda>Fr© rª r« r r r r Ú!test_may_share_memory_harder_fuzzl s þr c C s¼ t jdddgt jd} tt | | ¡d tt | | ¡ ¡d | d d
d d d
d d d
f }| d d
d d d
d d d
f }tt ||¡d tt j||d d d tt jt j||d
|
d d S ) Nrw r r; r9 TFr
|
r rc r )r$ ry rz r Ú shares_memoryr( r Z TooHardError)rO r1 rM r r r Útest_shares_memory_apiz s r¯ c C s@ t dg¡} ttt j| | dd d ttt j| | dd d d S )Nr r éd rc )r$ ry r Ú OverflowErrorrh r® )rO r r r Ú"test_may_share_memory_bad_max_work s r² c C s* d
|
dd} | dddd | ddd d d S ) Nc S s t | |ddd}|d kr |d k }|d k rttdd t| |D tdd t| |D k ttdd t||D ttdd t||D |r¬t|d k t| nt|d kt| d S ) Nr r )Zrequire_ub_nontrivialc s s | ]\}}|| V qd S rF r rN r r r rD s zCtest_internal_overlap_diophantine.<locals>.check.<locals>.<genexpr>c s s | ]\}}|| d V qdS ©r
|
Nr )r0 r1 Úur r r rD s c s s* | ]"\}}d | ko |kn V qdS rP r ©r0 rO r´ r r r rD s c s s | ]\}}||d kV qdS r³ r rµ r r r rD s )r r rK rV r) rX Úrepr)r[ r\ Úexistsr] r r r Úcheck s 4 z0test_internal_overlap_diophantine.<locals>.check)r r
|
)rw r; T)r· )r; r
|
)é rw F)Nr )r¸ r r r Ú!test_internal_overlap_diophantine s
|
rº c sº t jddddgt jd} t j d¡fdd d }d
|
}||k r¶tfd d t| jD }t | j¡} |¡ t fd d t
|
| j |D }| | |¡}t t| |d7 }q8d S )Nr r r r¡ r9 r6 c s` j d| d tjd} j || d tjd} j ddtjddkrT|| }}|d9 }t|||S r r r
r r r r ª s
|
z2test_internal_overlap_slices.<locals>.random_slicer é c 3 s: | ]2} j d dtjdd kr. j ddtjdndV qdS r r r@ r r r rD ¶ s ÿÿz/test_internal_overlap_slices.<locals>.<genexpr>c 3 s | ]\}} ||V qd S rF r r r r r rD » s r )r$ ry r} rQ rR r4 rJ rY r% r rV r& r r r )rO ZcasesrZ r r r r1 r )r rC r Ú test_internal_overlap_slices£ s þ
|
r¼ c C s¬ t | }t }tdd | jD }tj| D ]:}tdd t| j|D }||kr\d} qlq,| |¡ q,d}||krt
|
||t | j| jfd |d k r¨||kr¨t
|
|| |S )Nc s s | ]}t |V qd S rF rI )r0 r r r r rD Ç s z)check_internal_overlap.<locals>.<genexpr>c s s | ]\}}|| V qd S rF r )r0 r rL r r r rD É s TFrf ) r Úsetr4 r& r r rK rV rn Úaddr r¶ )r1 Zmanual_expectedro Úmr^ ÚvÚoffsetÚexpectedr r r Úcheck_internal_overlap s
|
rà c C sð t d¡ t j¡} t| d t| g ¡d t| ddd}t|d t| ddd}t|d t| ddd}t|d t| dd d}t|d t| dd
|
d}t|d t| d d d}t|d t| d d d}t|d t| d dd}t|d d S )Nr F)r rw )rw rw ©rn r& )r rw Tr. r3 )r
|
)r i÷Øÿÿ)éW ru )r ru )r ru )r$ r% Úastyperz rà r' r )rO r1 r r r Ú test_internal_overlap_manualÚ s&
|
|
|
|
|
|
|
|
rÇ c s° t d¡ t j¡} d}d}d}t j d¡ t|||k r¬ jddt jd}t fddt
|
|D }t fd dt
|
|D }t | ||d
|
}t |}|r¢|d7 }q*|d7 }q*d S ) Nr r r° r6 rw r9 c 3 s | ]} j d dtjdV qdS )iüÿÿéè r9 Nr r@ r r r rD s ÿz-test_internal_overlap_fuzz.<locals>.<genexpr>c 3 s | ]} j d dtjdV qdS )r r¹ r9 Nr r@ r r r rD s ÿrÄ ) r$ r% rÆ rz rQ rR rU r> r? r4 rJ r rà ) rO ÚoverlapZ
|
no_overlaprZ rY rn r& r1 Úresultr r r Útest_internal_overlap_fuzzÿ s$ ÿ ÿ
|
rË c C sÚ G dd d} G dd d}| |fD ]°}t d¡}tt ||d d d
|dd d
¡ tt ||d d d
|dd d
¡ tt ||dd d
|d d d
¡ tt ||dd d
|d d d
¡ q$d S ) Nc @ s e Zd Zdd Zedd ZdS )z(test_non_ndarray_inputs.<locals>.MyArrayc S s
|
|| _ d S rF ©re ©Úselfre r r r Ú__init__ s z1test_non_ndarray_inputs.<locals>.MyArray.__init__c S s | j jS rF )re rm ©rÎ r r r rm " s z<test_non_ndarray_inputs.<locals>.MyArray.__array_interface__N)Ú__name__Ú
|
__module__Ú __qualname__rÏ Úpropertyrm r r r r ÚMyArray s rÕ c @ s e Zd Zdd Zdd ZdS )z)test_non_ndarray_inputs.<locals>.MyArray2c S s
|
|| _ d S rF rÌ rÍ r r r rÏ ' s z2test_non_ndarray_inputs.<locals>.MyArray2.__init__c S s | j S rF rÌ rÐ r r r Ú __array__* s z3test_non_ndarray_inputs.<locals>.MyArray2.__array__N)rÑ rÒ rÓ rÏ rÖ r r r r ÚMyArray2& s r× r r
|
r r )r$ r% r rh r® )rÕ r× ÚclsrO r r r Útest_non_ndarray_inputs s
|
(*(rÙ c C s8 ddl m} t| j}d|d< dg|d< t ||| ¡S )z@Construct an array viewing the first byte of each element of `x`r )Ú
|
DummyArrayú|b1Ztypestr)rd rÛ Údescr)Únumpy.lib.stride_tricksrÚ Údictrm r$ Zasarray)rO rÚ Z interfacer r r Úview_element_first_byte7 s
|
|
|
rß c K sj ||d< t |}| ¡ |d< | ¡ }| |||d< | ¡ }||d< | || ¡ }||k ¡ rft|| dS )z}
|
Check that operation(*args, out=out) produces results
|
equivalent to out[...] = operation(*args, out=out.copy())
|
Úout.N)rÞ r( rX r )Ú operationÚargsrà ÚkwargsZkwargs2Zout_origr ro r r r Úassert_copy_equivalent@ s rä c @ s e Zd ZdZejdfddZejj dd Z
|
ejj dd Z d d
|
Z d d Z d d Zdd Zejj dd Zdd Zdd Zdd Zejj dd Zdd Zd S ) Ú TestUFuncz1
|
Test ufunc call memory overlap handling
|
r» c C s ddddddg}t j d¡}tdd D ]Þ}|jd
|
d |d |
d |¡}t|d dd} ||d d }
|
d
|
} | |
|
k r&t| \} } | ¡ }| ¡ }|d kr¾t || g| d t
|
| | ¡r¼| d7 } qlt t|d g¡D ]0}|| d< || d< || | |\}}|dkrqÐt d g| }|d kr\|d kr@t d
|
dgd
|
g|d }nt d
|
|gd
|
g|d }nb|d kr| j| d }|dkrt ||d ||< n|||< n"| j| |ks°tt d
|
|||< | t| }|rÚ| g ¡}t
|
| |¡rð| d7 } t || g||d qÐqlq&d S )Nrx rr é rs é é r6 r r; r i )r FT©r r r
|
©rà .Úskip©rà Úaxis)r$ rQ rR rJ r> rÆ r r¢ r( rä r® r Úchainr r& ÚAssertionErrorr4 r' )rÎ rá Úget_out_axis_sizer: ÚcountÚshapesrC rY rO ÚitrZ Ú overlappingr1 rM Úa_origÚb_origrí ZoutsizeZ scalarizeÚslÚkZb_outr r r Úcheck_unary_fuzzZ sN
|
|
|
|
|
|
|
|
zTestUFunc.check_unary_fuzzc C s | tjd tj¡ d S rF )rù r$ Úinvertr} rÐ r r r Útest_unary_ufunc_call_fuzz s z$TestUFunc.test_unary_ufunc_call_fuzzc C s | j tjd tjdd d S )Nr7 )rñ )rù r$ ÚnegativeÚ
|
complex128rÐ r r r Ú"test_unary_ufunc_call_complex_fuzz s z,TestUFunc.test_unary_ufunc_call_complex_fuzzc C s$ dd }| j tjj|tjdd d S )Nc S s4 |d kr"| j dkr | jdfS dS n| j| dfS d S ©Nr F)rë F©rY r r& ©r1 rM rí r r r rð £ s
|
|
|
zFTestUFunc.test_binary_ufunc_accumulate_fuzz.<locals>.get_out_axis_sizer7 ©r: rñ )rù r$ r¾ Ú
|
accumulater} ©rÎ rð r r r Ú!test_binary_ufunc_accumulate_fuzz¢ s
|
ÿz+TestUFunc.test_binary_ufunc_accumulate_fuzzc C s$ dd }| j tjj|tjdd d S )Nc S s d |d kp| j dkfS r¥ )rY r r r r rð ° s zBTestUFunc.test_binary_ufunc_reduce_fuzz.<locals>.get_out_axis_sizer7 r )rù r$ r¾ Úreducer} r r r r Ú test_binary_ufunc_reduce_fuzz¯ s
|
ÿz'TestUFunc.test_binary_ufunc_reduce_fuzzc C s( dd }dd }| j ||tjdd d S )Nc S s4 |d kr"| j dkr | jdfS dS n| j| dfS d S rÿ r r r r r rð · s
|
|
|
zDTestUFunc.test_binary_ufunc_reduceat_fuzz.<locals>.get_out_axis_sizec S s^ |d kr t | }|t | }n | j| }| j| |j| }t d||¡}tjj| |||dS )Nr rì )rW r& r$ r% r¾ Úreduceat)r1 rà rí r r Úidxr r r Ú do_reduceatÀ s
|
z>TestUFunc.test_binary_ufunc_reduceat_fuzz.<locals>.do_reduceatr7 r )rù r$ r} )rÎ rð r
|
r r r Ú test_binary_ufunc_reduceat_fuzz¶ s
|
ÿz)TestUFunc.test_binary_ufunc_reduceat_fuzzc C sd dd }t jdt jd}|t j||d d d
¡ | t jdt jd}|t j||d d d
| d S )Nc S s: | j | ¡ | ¡ | ¡ d}| j |||d}t|| d S ©Nrê )r r( r )Úufuncr1 r rà Úc1Úc2r r r r¸ Î s z:TestUFunc.test_binary_ufunc_reduceat_manual.<locals>.checké' r9 r )r$ r% r} r¾ r( )rÎ r¸ r1 r r r Ú!test_binary_ufunc_reduceat_manualÍ s
|
z+TestUFunc.test_binary_ufunc_reduceat_manualc
|
C sÒ ddddddg}t j}tj d¡}tdd D ]}|j|d |
}t|d
|
d d }d |d d }d}||k r,t|\} }
|
t | j
|
dd
dk sft |
|
j
|
dd
dk sf| j
|
d dk r¶qf|
|
j
|
d |
|
j
|
d krÞ|
|
ddd d
f }
|
n|
|
dd d
df }
|
| j
|
d } | | d d } | |
|
j
|
d kr6| dkr6|
|
dd |
f }
|
nTt dt t |
|
j
|
d ¡d } | | d d } | dd |
d d
f } |
|
dd |
f }
|
t | |
|
¡r |d7 }tjddd t|| g|
|
d W 5 Q R X qfq,d S )Nrx rr ræ rs rç rè r6 r
|
r; FTré r7 r r éþÿÿÿr .Úignore)ZoverÚinvalidrê )r Zeuclidean_pdistr$ rQ rR rJ Úrandr r¢ rU r& rT r= Úsqrtr® Úerrstaterä ) rÎ rò ZgufuncrC rY rO ró rZ rô r1 rM r r r r r Útest_unary_gufunc_fuzzÛ s6 :
|
z TestUFunc.test_unary_gufunc_fuzzc C sr d dd}t jdt jd}|t j|d d d
| t jdt jd}t jdddt jd}|t j|||d d
|
d S ) Nc S s | ¡ }|d kr>| || ¡ ¡ | ¡ }| ||¡ | ¡ }n4| || ¡ | ¡ ¡ | ¡ }| |||¡ | ¡ }t|| d S rF )r( Úatr )r r1 r rM Za0r r r r r r¸ s
|
z-TestUFunc.test_ufunc_at_manual.<locals>.checkr r9 r r° r r
|
é éK )N)r$ r% r} rú r¾ )rÎ r¸ r1 r r r r Útest_ufunc_at_manual s
|
z TestUFunc.test_ufunc_at_manualc sÔ fdd}t jt jt jt jt jt jt jt jg}dd |D }|D ]}t |t j
|
¡r^t j nt j d}d}t j d |
t j |||
t j |d d d
t j || d |d d
t j d d | d
t j ||d | d
t j d | d d d
|
t j |d | d |d d
|
g}t ||¡D ]¬\}}t jdd|d | |d } | | }
|
| | } t jd d d ||
|
| ||
|
d d
| ||
|
dd
| ||
|
d d
g ¡| ||
|
dd
g ¡| W 5 Q R X q qBd S )Nc sÔ | ¡ }| ¡ }| ¡ } | |d} | |d}t|| t| tj¡}|| d< ||d< | | ¡ | ¡ d ¡ }|| d< ||d< | || ¡ d ¡ }|| d< ||d< | ||d ¡ }t|| t|| d S )Nrê .©rà Úwhere)r( r rß r| r$ Úbool_) r1 rM rõ rö Zb0r r ÚmaskÚc3©r r r r¸ " s$
|
|
z3TestUFunc.test_unary_ufunc_1d_manual.<locals>.checkc S s g | ]}t |¡qS r )r$ r: )r0 rO r r r r2 @ s z8TestUFunc.test_unary_ufunc_1d_manual.<locals>.<listcomp>rÈ r r r r
|
r r9 r )r) )r$ rz r} Úint32ra Úfloat32Úfloat64Ú complex64rý Z
|
issubdtypeÚintegerrú Z
|
reciprocalÚ index_expr r r% r r' ) rÎ r¸ Zdtypesr: r rø ÚindicesÚxiÚyirÀ rO Úyr r" r Útest_unary_ufunc_1d_manual s@ ÿ
|
ø
|
z$TestUFunc.test_unary_ufunc_1d_manualc sT t j fdd}t d¡ t j¡}|||| ||| ¡ | |||| ¡ d S )Nc s2 | | ¡ | ¡ d} | ||d}t|| d S )Nr ©r( r )r1 rà r r r r" r r r¸ g s z4TestUFunc.test_unary_ufunc_where_same.<locals>.checkr° )r$ rú r% rÆ r r( )rÎ r¸ rO r r" r Útest_unary_ufunc_where_samec s z%TestUFunc.test_unary_ufunc_where_samec sÌ t j fdd}t jt jt jt jt jt jt jt j fD ]}d}d}g }dD ]n}|
|
t j d || |
t j |||| |
t j || d d |
t j ||| d |d |
g¡ qLt |||¡D ]ú\}}} t d| ¡ |¡}
|
|
|
| }|
|
| }|
|
| } |||| ||d d
|| ||dd
|| ||d d
g ¡|| ||dd
g ¡|| |||d d
| |||dd
| |||d d
g ¡| |||dd
g ¡| qÊq6d S ) Nc s2 | ¡ } | ||d} | ||d}t|| d S r r. )r1 rM ÚcZc0r r r" r r r¸ v s z4TestUFunc.test_binary_ufunc_1d_manual.<locals>.checkrÈ r r r r; r )r$ r¾ rz r} r# ra r$ r% r& rý r r( r r r% rÆ r' ) rÎ r¸ r: r rø r) r rO r, ÚzrÀ r r" r Útest_binary_ufunc_1d_manualr s@ ÿ
|
ü z%TestUFunc.test_binary_ufunc_1d_manualc C s6 t j d¡}| dd¡}||j7 }t||j d d S )Nr6 éÈ r )r$ rQ rR r ÚTr )rÎ rC rO r r r Ú test_inplace_op_simple_manual s
|
z'TestUFunc.test_inplace_op_simple_manualN)rÑ rÒ rÓ Ú__doc__r$ r} rù ÚpytestÚmarkÚslowrû rþ r r r r r r r- r/ r2 r5 r r r r rå U s( ÿ
|
?
|
|
|
+D
|
+rå )TF)N)0r r7 Únumpyr$ Z numpy.core._multiarray_testsr r Z
|
numpy.corer rÝ r Z numpy.testingr r r r r" r r4 r& rj ri r r r# r- r5 r8 r9 r` rb rq r r r¤ r¬ r r¯ r² rº r¼ rà rÇ rË rÙ rß rä rå r r r r Ú<module> sL
|
<3
|
X
|
|
|
%
|
