U
|
P ±dÆ[ ã @ sB d dl Zd dlmZ d dlmZmZmZmZm Z m
|
Z
|
d dl m Z m Z mZmZmZmZ d dlZdd Zdd Zd8d
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Z dd Z d d Z d d Z d d! Z d"d# Z!d$d% Z"d&d' Z#G d(d) d)Z$d*d+ Z%G d,d- d-ej&Z'G d.d/ d/e'Z(d0d1 Z)d2d3 Z*d4d5 Z+d6d7 Z,dS )9é N©Úrational)Ú assert_equalÚassert_array_equalÚ assert_raisesÚassert_Úassert_raises_regexÚ assert_warns)Ú
|
as_stridedÚbroadcast_arraysÚ_broadcast_shapeÚ broadcast_toÚbroadcast_shapesÚsliding_window_viewc C s@ dd | D }t | }dd |D }|gt| }t|| d S )Nc S s g | ]}t |¡qS © ©ÚnpÚzeros©Ú.0Úsr r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_stride_tricks.pyÚ
|
<listcomp> s z)assert_shapes_correct.<locals>.<listcomp>c S s g | ]
|
}|j qS r )Úshape)r Úar r r r s )r Úlenr )Ú input_shapesÚexpected_shapeÚinarraysZ outarraysZ outshapesÚexpectedr r r Úassert_shapes_correct s
|
r c C s" dd | D }t ttf| d S )Nc S s g | ]}t |¡qS r r r r r r r s z4assert_incompatible_shapes_raise.<locals>.<listcomp>)r Ú
|
ValueErrorr )r r r r r Ú assert_incompatible_shapes_raise s r" Fc
|
C s t j| td}tt j |¡}t |¡ |¡}|r>|j}|j}|r^|d d d
}|d d d
}|| }t||\}} t || d S )N©Údtypeéÿÿÿÿ)
|
r r ÚintÚmultiplyÚreduceÚarangeÚreshapeÚTr r )
|
Zshape0Zshape1Z
|
transposedZflippedZx0ÚnÚx1ÚyZb0Zb1r r r Úassert_same_as_ufunc! s r/ c C s: t d¡} t d¡}t| |\}}t| | t|| d S )Né
|
)r r) r r )Úxr. ÚbxÚbyr r r Ú test_same7 s
|
|
|
|
r4 c C s< t d¡} t d¡}ttd t| |dd W 5 Q R X d S )Nr0 zgot an unexpected keywordÚfloat64r# )r r) r Ú TypeErrorr )r1 r. r r r Útest_broadcast_kwargs> s
|
|
r7 c C sv t dddgg¡} t dgdgdgg¡}t| |\}}t dddgdddgdddgg¡}|j}t|| t|| d S )Né é é )r Úarrayr r+ r )r1 r. r2 r3 Zbx0Zby0r r r Ú test_one_offI s "
|
r< c
|
C s\ ddddddddd d
|
g
|
} | D ]:}|g}t || ||g}t || |||g}t || q d S ) Nr ©r8 ©r: ©r r8 )r r: ©r8 r )r: r ©r8 r: ©r: r8 ©r: r: ©r )Údatar r Z input_shapes2Z input_shapes3r r r Útest_same_input_shapesS s$ ö
|
|
|
rF c C s¶ ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gg } | D ]&\}}t || t |d d d
|
| qd S ) Nr= r> rA rC rB ©r8 r8 r@ ©r r r? r% rD ©rE r r r r r Ú(test_two_compatible_by_ones_input_shapesn s"
|
|
|
|
|
|
|
|
|
|
|
|
|
ó
|
rJ c C sü ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgdd gd gdd
|
gd gdd gd gdd gd gddgd gdd
|
gd
|
gdd gd gdd gd gdd
|
gd
|
gg} | D ]&\}}t || t |d d d
| qÐd S ) Nr r> rC rB r= rG rA ©r rH r? r@ r% rD rI r r r Ú3test_two_compatible_by_prepending_ones_input_shapes s0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ì
|
rL c C sF ddgddgdddgddgg} | D ] }t | t |d d d
q"d S )Nr> ©é ©r9 r: ©r9 ©r8 r: rN ©r9 r: r: r% )r" )rE r r r r Ú)test_incompatible_shapes_raise_valueerror§ s ürS c " C sâ ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gd
|
dgdgddgdgddgdgddgdgd
|
dgdgddgdgddgdgddgdgd
|
dgdgd
|
dgdgd
|
d gd gd dgdgd d gdgddgdgd
|
dgdgdd gdgdd gd gddgdgd
|
dgdgd
|
d gd gg!} | D ]\}}t |d |d d|d |d f t |d |d t |d |d d d
|
|krRt |d |d dd t |d |d dd qRd S )Nr= r> rA rC rB rG r@ rH r? r rK r r8 z Shapes: %s %sTF)r/ rI r r r Útest_same_as_ufunc¶ sV
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ß# ÿ
|
rT c C sr t d¡dt d¡gt d¡dt d¡gt d¡dt d¡gt d¡dt d¡gt d¡dt d¡gt d¡d t d ¡gt d¡dt d¡gt d¡d
|
t d¡ dd ¡gt d¡d t dddgdddgg¡gt d¡dt d¡gt d¡dt d¡gt d¡dt d¡gt d¡dt d¡gt d ¡dt d¡gt d¡dt d¡gg} | D ] \}}}t||}t|| qLd S )Nr rK r= r8 r> r: rP r9 )r8 r9 r: rA r% rO ©r8 r9 )r r9 ©r9 r8 )r9 r )r r; r Úonesr) r* r r )rE Ú input_arrayr r Úactualr r r Útest_broadcast_to_succeedsê s& $î
|
rZ c sv ddgddgddgddgddgddgddgd dgdd
|
gdd gdd gg } | D ]&\}t |¡ tt fd d qJd S )NrK r r= r> rP rM rU rV rG r% )r% )r% r9 c s
|
t S ©N)r r ©ÚarrÚ target_shaper r Ú<lambda> ó z*test_broadcast_to_raises.<locals>.<lambda>)r r r r! )rE Z
|
orig_shaper r\ r Útest_broadcast_to_raises s õ
|
ra c sØ t t d t tddgd t tt d¡d t tt d¡t d¡d t tt d¡gd d t tt d¡gd d t tt d¡gd dg d t d¡gd t d
|
¡gd tt fd d d S ) Nr r8 r9 rP rG ©r: rN rU é éd r: c s t S r[ )r r ©Úbad_argsr r r_ % r` z&test_broadcast_shape.<locals>.<lambda>)r r r rW r r! r r re r Útest_broadcast_shape s rg c C sh g dgdgdgdgdgddgdgdgdgddgdgdddd gd
|
gdgdgd d gd gddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgg} | D ]\}}t t| | qt tdgd d t tdgd d t tdgd d d S )Nr )é rU rP rG rb )é rh )é ri r8 )rj r8 rh )rj ri rh rA rB rC r@ rH r? rK r= r> r9 ©r: r9 rc rd )r r )rE r r^ r r r Ú test_broadcast_shapes_succeeds( sB
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
å rl c st ddgddgdddgddgddd d
|
gd dgg} | D ]t tfd d q2dgd dgd t t fdd d S )Nr> rM rO rP rQ rR rU rB rk )r0 rj r9 c s t S r[ ©r r )r r r r_ \ r` z.test_broadcast_shapes_raises.<locals>.<lambda>rc c s t S r[ rm r re r r r_ _ r` )r r! )rE r )rf r r Ú test_broadcast_shapes_raisesQ s
|
úrn c C s" t d g¡} t| }t d g¡}t|t d g¡ t ddddg¡} t| dd| j fd}t ddg¡}t|| t ddddg¡} t| ddd| j fd}t ddddgddddgddddgg¡}t|| t d d
|
g¡}t jd |d } t dd ¡| d< t| dd| jfd}ddddggd }d gd gd }t|j| t||d t||d t jd dd } t| dd| jfd}t| j|j t dgdgd¡}t jd |d } t| dd| jfd}t| j|j dd t dD }t j|t
|
d } t| dd| jfd}t| j|j t|gd | d S )Nr8 r9 r: rN rP ©r Ústridesrb r )ÚnumÚi4)ÚobjÚOrM r# rj rq rs ZV4Ú )ÚnamesÚformatsc S s g | ] }t |qS r r )r Úir r r r s z#test_as_strided.<locals>.<listcomp>) r r; r
|
r Úitemsizer$ Úemptyr) r Úranger )r Úa_viewr ÚdtZ expected_numZ expected_objÚrr r r Útest_as_stridedb sB
|
(
|
r c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestSlidingWindowViewc C sD t d¡}t|d}t ddgddgddgddgg¡}t|| d S )Nrj r9 r r8 r: rN )r r) r r; r )Úselfr] Úarr_viewr r r r Útest_1d s
|
|
|
ýz TestSlidingWindowView.test_1dc C s¤ t jd d
d d
f \}}d| | }d}t||}t ddgddggddgdd ggddgd d
|
gggddgd d ggdd gd d ggd d
|
gd dgggg¡}t|| d S )Nr: rN r0 )r9 r9 r r8 é r9 é é é é é é ©r Úogridr r; r ©r rx Újr] r r r r r r Útest_2d s
|
þþýz TestSlidingWindowView.test_2dc C sl t jd d
d d
f \}}d| | }t|dd}t dddgdddgd d
|
d gdd d ggg¡}t|| d S )Nr: rN r0 r r r8 r r r9 r
r r r r ©r rx r r] r r r r r Útest_2d_with_axis§ s ýz'TestSlidingWindowView.test_2d_with_axisc C s t jd d
d d
f \}}d| | }t|dd}t dddgdddgggdd d
|
gd d
|
d gggd d dgd ddgggg¡}t|| d S )Nr: rN r0 rO rG r r8 r9 r r
r r r r r r r r r r Útest_2d_repeated_axis± s ÿÿÿüz+TestSlidingWindowView.test_2d_repeated_axisc C s¾ t jd d
d d
f \}}d| | }d}t||}t dddgdddggddd gddd
|
gggdddgd d d ggddd
|
gd d dgggd d d gdddggd d dgdddgggg¡}t|| d S )NrN r0 rO r r8 r9 r r
r: r r r r r é é rc é! r r r r r Útest_2d_without_axis½ s
|
ÿÿÿüz*TestSlidingWindowView.test_2d_without_axisc C s¾ t jd d
d d
f \}}d| | }tjtdd t|d W 5 Q R X tjtdd t|d W 5 Q R X tjtdd t|d d
|
d W 5 Q R X tjtd d t|d W 5 Q R X d S )NrN r0 z cannot contain negative values©Úmatch)r% r: z3must provide window_shape for all dimensions of `x`r= z2Must provide matching length window_shape and axisrQ r? )Zaxisz.window shape cannot be larger than input array)rj rj )r r ÚpytestÚraisesr! r )r rx r r] r r r Ú test_errorsÊ s& þþþz!TestSlidingWindowView.test_errorsc C s t d¡}t|ddd}t|jj tjtdd d|d< W 5 Q R X t|dd d}t|jj d|d
|
< t |t
|
d dddd g¡ d S ) Nrj r9 F©Ú writeablez#assignment destination is read-onlyr r: rH Tr? r rN ) r r) r r Úflagsr r r r! r r; )r r] Úviewr r r Útest_writeableÜ s
|
þ z$TestSlidingWindowView.test_writeablec C sn G dd dt j}t d¡ |¡}ttt|ddd| ttt|ddd| ttt|d| d S )Nc @ s e Zd ZdS )z1TestSlidingWindowView.test_subok.<locals>.MyArrayN)Ú__name__Ú
|
__module__Ú __qualname__r r r r ÚMyArrayê s r¤ rj r9 F©ÚsubokT)r Úndarrayr) r r Ú
|
isinstancer )r r¤ r] r r r Ú
|
test_suboké s
|
ÿþz TestSlidingWindowView.test_subokN) r¡ r¢ r£ r r r r r r r r© r r r r r s
|
r c C s t d¡} t| dd}t|jj t| dd}t|jj d|d< t| t | d¡ d| j_t| dd}t| dd}t|jj d S )Nr0 Fr Tr: .)r rW r
|
r r r r Z full_like)r] r r r r Úas_strided_writeableö s
|
rª c @ s e Zd Zdd ZdS )ÚVerySimpleSubClassc O s t j|ddi| | ¡S )Nr¦ T)r r; r )ÚclsÚargsÚkwargsr r r Ú__new__ s zVerySimpleSubClass.__new__N)r¡ r¢ r£ r¯ r r r r r« s r« c @ s e Zd Zdd Zdd ZdS )ÚSimpleSubClassc O s$ t j|ddi| | ¡}d|_|S )Nr¦ TÚsimple)r r; r Úinfo)r¬ r r® r r r r r¯ s zSimpleSubClass.__new__c C s t |ddd | _d S )Nr² ru z
|
finalized)Úgetattrr² )r rs r r r Ú__array_finalize__ s z!SimpleSubClass.__array_finalize__N)r¡ r¢ r£ r¯ r´ r r r r r° s r° c C sÌ t ddddg} tt| t k t| dd| j fd}tt|tjk t| dd| j fdd}tt|t k tddddg} t| dd| j fdd}tt|tk t|jd k t t
|
| ¡ d
|
d¡}t | |\}}tt|tjk tt|tjk t|j |j k t | |dd \}}tt|tk t|jd k tt|tjk t|j |j k d }t| |}tt|tjk t|j |k t| |dd }tt|tk t|jd k t|j |k d S ) Nr8 r9 r: rN rP ro T)r rp r¦ zsimple finalizedr% r¥ )r9 rN )r« r Útyper
|
ry r r§ r° r² r) r r* r r r )r r| ÚbZb_viewr r r r Útest_subclasses s8
|
r· c C s~ t dddg¡} t| d}t|jjd tt|jt d d dt
|
| fdt
|
d| ffD ]v\}}|D ]h}|r¼t t t|jjd W 5 Q R X t t d|d d
< W 5 Q R X t|jjd qbt|jjd qbqVt
|
| t
|
d| fD ]0}|D ]&}d|j_t|jjd d|d d
< qèqàd| j_t
|
d| \}}t|jjd d}dg}tt d¡||}t d ¡}t
|
||\} }
|
t| j|
|
jk d S )
|
Nr8 r9 r: rO Fr TrP r= )r r; r r r r r r! Ú __setitem__Úslicer r Ú FutureWarningÚDeprecationWarningr
|
r r r ) ÚoriginalÚresultÚ is_broadcastÚresultsÚ_r rp Z tricky_arrayÚotherÚfirstÚsecondr r r r < s>
|
|
ÿ
|
|
ÿ
|
r c C sd t dddg¡} dt| fdtd| ffD ]4\}}|D ]&}|rNt|js\tq6t|jr6tq6q*d S )Nr8 r9 r: FTr )r r; r Ú
|
memoryviewÚreadonlyÚAssertionError)r¼ r¾ r¿ r½ r r r Útest_writeable_memoryviewk s
|
ÿ rÇ c C sX t jdtd} t jdgd td}t| d}t|| t| t d¡\}}t|| d S )Nr r# r: r> )r r; Úobjectr r r rW )rX r rY rÀ r r r Útest_reference_types~ s
|
|
rÉ )FF)-Únumpyr Znumpy.core._rational_testsr Z numpy.testingr r r r r r Znumpy.lib.stride_tricksr
|
r r r r r r r r" r/ r4 r7 r< rF rJ rL rS rT rZ ra rg rl rn r r rª r§ r« r° r· r rÇ rÉ r r r r Ú<module> s:
|
|
4).f
|
%/
|
