U
|
P ±dk ã @ sz d dl Zd dlZd dlZd dlZd dlmZmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZ d dlmZmZmZmZmZ ejdkZdd Z G dd dZ G d d
|
d
|
Z G d d d Z G d d dZ G dd dZ!G dd dZ"G dd dZ#G dd dZ$G dd dZ%G dd dZ&G dd d Z'G d d d Z(G d d d Z)G d!d" d"Z*G d#d$ d$Z+G d%d& d&Z,d'd( Z-dS ))é N)Úapply_along_axisÚapply_over_axesÚ array_splitÚsplitÚhsplitÚdsplitÚvsplitÚdstackÚ column_stackÚkronÚtileÚ expand_dimsÚtake_along_axisÚput_along_axis)Úassert_Ú assert_equalÚassert_array_equalÚ assert_raisesÚ assert_warnsì c s t ¡ fdd}|S )z: hack in keepdims behavior into a function taking an axis c s. | fd|i|}|d kr d}t j||dS )NÚaxisr ©r )Únpr )Úar ÚkwargsÚres©Úfunc© úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_shape_base.pyÚwrapped s z _add_keepdims.<locals>.wrapped)Ú functoolsÚwraps)r r r r r Ú _add_keepdims s r# c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestTakeAlongAxisc
|
C sÒ ddl m} |ddd}tjtjt fttjttjt fttj ttj
|
t ftj tj tddfg}|D ]`\}}}t t|jdg D ]@}||fd |i|}||fd |i|} t|t|| |d
|
qqldS ) z- Test it translates from arg<func> to <func> r ©Úrandé é é é )ZkthNr r )Ú numpy.randomr& r ÚsortZargsortÚdictr# ÚminÚargminÚmaxÚargmaxÚ partitionZ argpartitionÚlistÚrangeÚndimr r )
|
Úselfr& r Úfuncsr Zargfuncr r Za_funcZai_funcr r r Útest_argequivalent s üz$TestTakeAlongAxis.test_argequivalentc C s t d¡}t jdt jd}t||dd ttt|t d¡dd ttt|| t ¡dd ttt|| t
|
¡dd tt j t||dd dS )z4 Test it errors when indices has too few dimensions )é
|
r9 )r9 r* ©Údtypeé r r9 N) r ÚonesÚintpr r Ú
|
ValueErrorÚarrayÚ
|
IndexErrorZastypeÚboolÚfloatÚ AxisError©r6 r Úair r r Ú test_invalid1 s
|
z TestTakeAlongAxis.test_invalidc C s: t d¡}t jdt jd}t||dd}t|j|j dS )zC Test everything is ok with empty results, even with inserted dims )r' r( r) )r' r r) r: r< r N©r r= r> r r Úshape©r6 r rF Úactualr r r Ú
|
test_emptyB s
|
z TestTakeAlongAxis.test_emptyc C s8 t d¡}t jdt jd}t||dd}t|jd dS )úD Test that non-indexing dimensions are broadcast in both directions ©r' r( r< ©r< r* r) r: r< r ©r' r* r) NrH rJ r r r Útest_broadcastJ s
|
z TestTakeAlongAxis.test_broadcastN)Ú__name__Ú
|
__module__Ú __qualname__r8 rG rL rQ r r r r r$ s r$ c @ s e Zd Zdd Zdd ZdS )ÚTestPutAlongAxisc C s~ t dddgdddgg¡}tt|jd g D ]J}| ¡ }tt j||d}t||d|d tt j ||d}t
|
|| q.d S ) Nr9 é é é< é( é2 r iÿÿÿ) r r@ r3 r4 r5 Úcopyr# r1 r r/ r )r6 Za_baser r Zi_maxZi_minr r r Útest_replace_maxS s z!TestPutAlongAxis.test_replace_maxc C sL t d¡}t jdt jd d¡d }t||ddd tt||ddd d S )
|
rM rN r9 r: rO r( rW r< r N)r r= Úaranger> Úreshaper r r rE r r r rQ c s
|
z TestPutAlongAxis.test_broadcastN)rR rS rT r\ rQ r r r r rU R s rU c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej fd d Z
|
ej fd dZ ej fddZ dd Z dd Zdd Zdd Zdd ZdS ) ÚTestApplyAlongAxisc C s6 t dd¡}tttd|t|t |jd ¡ d S )N©rW r9 Údr r< ©r r= r r ÚlenrI ©r6 r r r r Ú test_simplel s
|
|
ÿz TestApplyAlongAxis.test_simplec C s6 t dd¡}tttd|t|t |jd ¡ d S )N)r9 ée ra r r< rb rd r r r Útest_simple101q s
|
|
ÿz!TestApplyAlongAxis.test_simple101c C s@ t d¡ d¡}ttt jd|dddgdddgd d
|
d gg d S ) Né ©r' r' r' r rV é! é$ é' é* é- é0 é3 )r r] r^ r r Úsumrd r r r Útest_3dv s ÿzTestApplyAlongAxis.test_3dc C s dd }G dd dt j}t ddgddgg¡ |¡}t ddgd d
|
gg¡ |¡}t|d|}tt|| t|| t|d|}tt|| t|| d S ) Nc S s | d S )Nr* r )Úrowr r r Údouble| s z9TestApplyAlongAxis.test_preserve_subclass.<locals>.doublec @ s e Zd ZdS )z<TestApplyAlongAxis.test_preserve_subclass.<locals>.MyNDArrayN©rR rS rT r r r r Ú MyNDArray s rv r r< r* r' r( é )r Úndarrayr@ Úviewr r Ú
|
isinstancer )r6 rt rv ÚmÚexpectedÚresultr r r Útest_preserve_subclass{ s
|
z)TestApplyAlongAxis.test_preserve_subclassc C sL G dd dt j}dd }t d¡ |¡}tt|d|t dddg¡ d S )Nc @ s e Zd ZdZdS )z9TestApplyAlongAxis.test_subclass.<locals>.MinimalSubclassr< N)rR rS rT Údatar r r r ÚMinimalSubclass s r c S s | j S ©N)r )r@ r r r Úminimal_function s z:TestApplyAlongAxis.test_subclass.<locals>.minimal_function©rw r' r r< )r rx Úzerosry r r r@ )r6 r r r r r r Ú test_subclass s
|
ÿz TestApplyAlongAxis.test_subclassc C sL t d¡ |¡}tt jd|}tt|| t|t dddg¡ |¡ d S )Nr r rw ) r r= ry r rq r rz r r@ )r6 Úclsr r r r r Útest_scalar_array s z$TestApplyAlongAxis.test_scalar_arrayc
|
C s dd }t d¡ |¡}t|d|}tt|| t|t dddg¡ |¡ t|d|}tt|| t|t ddddddg¡ |¡ d S )Nc S s t | jd t tj| dd¡S )z/ Sum x, returning a 0d array of the same class r< T)Zkeepdims)r r5 r Úsqueezerq ©Úxr r r Ú sum_to_0d¡ s z3TestApplyAlongAxis.test_0d_array.<locals>.sum_to_0dr r rw r< r' )r r= ry r r rz r r@ )r6 r r r r r r r Ú test_0d_array s z TestApplyAlongAxis.test_0d_arrayc s2 fddt d¡ d¡ td }t j fddt jd D d d
|
¡}tt |t | t|| td }t j fd dt jd D dd
|
¡}tt |t | t|| t d ¡ d ¡td}t jfddtjd D d d
|
¡}tt |t | t|| d S )Nc s0 t | jd | ddd
| dd
df ¡S )ú/produces an asymmetric non-square matrix from xr< Néÿÿÿÿ)r r5 ry r )r r r Úf1to2¯ s z5TestApplyAlongAxis.test_axis_insertion.<locals>.f1to2é r r c s g | ]} d d
|f qS r r ©Ú.0Úi©Úa2dr r r Ú
|
<listcomp>¸ s z:TestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>r< r r c s g | ]} |d d
f qS r r r r r r r À s éZ )rw r) r' c s6 g | ]. t j fd dtjd D ddqS )c s" g | ]} |d d
f qS r r r )Úa3dr Újr r r Ë s zETestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>.<listcomp>r r )r Ústackr4 rI )r )r r )r r r Ê s ý ÿþr* )
|
r r] r^ r r r4 rI ry r Útype)r6 r rK r| r )r r r r r Útest_axis_insertion® sB ÿþþ
|
ÿþþ
|
üûûz&TestApplyAlongAxis.test_axis_insertionc C s4 G dd dt j}| |¡ | |¡ | |¡ d S )Nc @ s e Zd ZdS )zFTestApplyAlongAxis.test_subclass_preservation.<locals>.MinimalSubclassNru r r r r r Ô s r )r rx r r r )r6 r r r r Útest_subclass_preservationÓ s
|
|
z-TestApplyAlongAxis.test_subclass_preservationc C sÖ dd }t d¡ d¡}t|d|}tt|t jj t|j d t
|
|d d
d d
df j ||d d
df j t
|
|d d
d d
df j ||d d
df j t
|
|d d
d d
df j ||d d
df j d S ) Nc S s@ t | jd | ddd
| dd
df }tj |d dk|¡S )r r< Nr r) r )r r5 r ÚmaZ masked_where)r r r r r r Û s z8TestApplyAlongAxis.test_axis_insertion_ma.<locals>.f1to2r r r r' r< r* ) r r] r^ r r rz r Z masked_arrayr r5 r Úmask)r6 r r r r r r Útest_axis_insertion_maÚ s 00z)TestApplyAlongAxis.test_axis_insertion_mac C sH dd }t |dt ddgddgg¡¡}t|t ddgddgg¡ d S )Nc S s | d | d fS )Nr< r r r r r r Ú sample_1dé s z7TestApplyAlongAxis.test_tuple_func1d.<locals>.sample_1dr< r* r' r( )r r r@ r )r6 r¡ r r r r Útest_tuple_func1dè s z$TestApplyAlongAxis.test_tuple_func1dc C s| dd }t d¡}ttt j|d| ttt j|d| dd }t d¡}t |d|¡}t|t d ¡ ttt j|d| d S )
|
Nc S s t d d S )NF)r r r r r Ú
|
never_callð s z1TestApplyAlongAxis.test_empty.<locals>.never_call)r r r r< c S s t t| dk dS )Nr r< )r rc r r r r Ú
|
empty_to_1ø s z1TestApplyAlongAxis.test_empty.<locals>.empty_to_1)r9 r r9 )r Úemptyr r? r r r= )r6 r£ r r¤ rK r r r rL î s
|
|
z TestApplyAlongAxis.test_emptyc C s¦ t ddhddhddhgddhd d
|
hd d hgg¡}t d d d|¡}t ddddhddd d
|
hddd d hg¡}t|| t |j¡D ] }tt|| t|| qd S )Nr< é r* é r' rj r( é, r) é7 rw éB c S s
|
t j| S r )ÚsetÚunion)r r r r Ú<lambda> ó z>TestApplyAlongAxis.test_with_iterable_object.<locals>.<lambda>r )r r@ r r ZndindexrI r )r6 ra rK r| r r r r Útest_with_iterable_object s þ(
|
z,TestApplyAlongAxis.test_with_iterable_objectN)rR rS rT re rg rr r~ r
r rx r r r r r r¢ rL r¯ r r r r r_ k s %r_ c @ s e Zd Zdd ZdS )ÚTestApplyOverAxesc C sH t d¡ ddd¡}tt j|ddg}t|t dgdgdggg¡ d S ) Né r* r' r( r rX é\ é| )r r] r^ r rq r r@ )r6 r Zaoa_ar r r re s z TestApplyOverAxes.test_simpleN)rR rS rT re r r r r r° s r° c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestExpandDimsc C sR d}t |¡}tddD ]4}t||}t|j| dk tt |¡j|k qd S )N©r* r' r( r) éûÿÿÿr( r< )r r¥ r4 r r rI r )r6 Úsr r Úbr r r Útest_functionality s
|
|
z!TestExpandDims.test_functionalityc C sn t d¡}t j|ddjdks"tt j|ddjdks:tt j|ddjdksRtt j|d djd
|
ksjtd S ) Nri )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' r< )r éýÿÿÿr¶ )r< r< r' r< r' r' )r r¥ r rI ÚAssertionErrorrd r r r Útest_axis_tuple! s
|
|
z TestExpandDims.test_axis_tuplec C s\ d}t |¡}tt jt|d tt jt|d t d¡}tt jt|d tt jt|d d S )Nrµ éúÿÿÿr) ri )r r¾ )r r) )r r¥ r rD r )r6 r· r r r r Útest_axis_out_of_range( s
|
|
z%TestExpandDims.test_axis_out_of_rangec C s t d¡}ttt|dd d S )Nri ©r< r< r )r r¥ r r? r rd r r r Útest_repeated_axis2 s
|
z!TestExpandDims.test_repeated_axisc C sf t d¡ d¡}t jj||d dkd}t j|dd}tt|t jj t |j
|
d t |j j
|
d d S ) Nr9 )r* r) r' r ©r r< r )r* r< r) ) r r] r^ r r@ r r rz Z MaskedArrayr rI r )r6 r Úexpandedr r r Útest_subclasses6 s z TestExpandDims.test_subclassesN)rR rS rT r¹ r½ r¿ rÁ rÄ r r r r r´ s
|
|
r´ c @ sf e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej j
|
e d d d d Z dd Z dd Zdd ZdS )ÚTestArraySplitc C s t d¡}ttt|d d S )Nr9 r )r r] r r? r rd r r r Útest_integer_0_splitA s
|
z#TestArraySplit.test_integer_0_splitc C s t d¡}t|d}t d¡g}t|| t|d}t d¡t dd¡g}t|| t|d}t d¡t dd¡t dd¡g}t|| t|d}t d¡t dd¡t dd ¡t d d¡g}t|| t|d}t d¡t dd¡t dd¡t dd ¡t d d¡g}t|| t|d}t d¡t dd¡t dd¡t dd ¡t d d
|
¡t d
|
d¡g}t|| t|d}t d¡t dd¡t dd¡t dd¡t dd ¡t d d
|
¡t d
|
d¡g}t|| t|d }t d¡t dd¡t dd¡t dd¡t dd¡t dd ¡t d d
|
¡t d
|
d¡g}t|| t|d
|
}t d¡t dd¡t dd¡t dd¡t dd¡t dd¡t dd ¡t d d
|
¡t d
|
d¡g }t|| t|d}t d¡t dd¡t dd¡t dd¡t dd¡t dd¡t dd¡t dd ¡t d d
|
¡t d
|
d¡g
|
}t|| t|d }t d¡t dd¡t dd¡t dd¡t dd¡t dd¡t dd¡t dd ¡t d d
|
¡t d
|
d¡t g ¡g }t|| d S ) Nr9 r< r* r) r' r( é rw é é r¦ )r r] r Úcompare_resultsr@ ©r6 r r Údesiredr r r Útest_integer_splitE s
|
|
|
|
|
|
|
|
|
ÿ
|
|
|
|
ÿ
|
|
|
|
|
ÿ
|
|
|
|
|
|
þ
|
|
|
|
|
|
|
þ
|
|
|
|
|
|
|
|
þ
|
|
|
|
|
|
|
|
|
ý
|
|
|
|
|
|
|
|
|
ýz!TestArraySplit.test_integer_splitc C sÚ t t d¡t d¡g¡}t|ddd}t t d¡g¡t t d¡g¡t d¡g}t|| t|jj|d jjk t|ddgdd}t d¡t t d¡g¡t t d¡g¡g}t|| t|jj|d jjk d S )Nr9 r' r r ©r r9 r r< © r r@ r] r r rÊ r r; r ©r6 r r Útgtr r r Útest_integer_split_2D_rows s ÿ
|
ÿ
|
z)TestArraySplit.test_integer_split_2D_rowsc C s t t d¡t d¡g¡}t|ddd}t t d¡t d¡g¡t t dd¡t dd¡g¡t t dd¡t dd¡g¡g}t|| d S )Nr9 r' r r r( rÇ )r r@ r] r rÊ rË r r r Útest_integer_split_2D_cols s þz)TestArraySplit.test_integer_split_2D_colsc C sv t t d¡t d¡g¡}t|d}t t d¡g¡t t d¡g¡t d¡g}t|| t|jj|d jjk dS )z2 This will fail if we change default axis
|
r9 r' rÎ r NrÏ rÐ r r r Ú test_integer_split_2D_default s
|
ÿ
|
z,TestArraySplit.test_integer_split_2D_defaultzNeeds 64bit platform)Úreasonc C s\ t dgd¡}t|d}t dgd¡}|gd }tt|D ]}t|| j|| j q<d S )Nr )r r* r( )i @r* )r Z broadcast_tor r4 rc r rI )r6 r r ÚchunkrÑ r r r r Ú,test_integer_split_2D_rows_greater_max_int32¤ s
|
|
z;TestArraySplit.test_integer_split_2D_rows_greater_max_int32c C s\ t d¡}dddg}t||dd}t dd¡t dd¡t dd¡t dd¡g}t|| d S )Nr9 r< r) rÇ r r r )r r] r rÊ ©r6 r Úindicesr rÌ r r r Útest_index_split_simple s
|
|
|
ÿz&TestArraySplit.test_index_split_simplec C sZ t d¡}dddg}t||dd}t g ¡t dd¡t dd¡t dd¡g}t|| d S )Nr9 r r) rÇ r r ©r r] r r@ rÊ rØ r r r Útest_index_split_low_boundµ s
|
|
|
ÿz)TestArraySplit.test_index_split_low_boundc C sn t d¡}dddddg}t||dd}t g ¡t dd¡t dd¡t dd¡t g ¡t g ¡g}t|| d S )Nr9 r r) rÇ é r r rÛ rØ r r r Útest_index_split_high_bound½ s
|
|
ÿz*TestArraySplit.test_index_split_high_boundN)rR rS rT rÆ rÍ rÒ rÓ rÔ ÚpytestÚmarkZskipifÚIS_64BITr× rÚ rÜ rÞ r r r r rÅ @ s =
|
rÅ c @ s e Zd Zdd Zdd ZdS )Ú TestSplitc C s8 t d¡}t|d}t d¡t dd¡g}t|| d S )Nr9 r* r) )r r] r rÊ rË r r r Útest_equal_splitË s
|
|
zTestSplit.test_equal_splitc C s t d¡}ttt|d d S )Nr9 r' )r r] r r? r rd r r r Útest_unequal_splitÑ s
|
z TestSplit.test_unequal_splitN)rR rS rT rã rä r r r r râ Æ s râ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestColumnStackc C s t ttd d S ©Nr< )r Ú TypeErrorr
|
©r6 r r r Útest_non_iterable× s z!TestColumnStack.test_non_iterablec C sL t d¡}t d¡}t ddgddgddgg¡}t ||f¡}t|| d S )N©r< r* r' )r* r' r( r< r* r' r( ©r r@ r
|
r ©r6 r r¸ r| rK r r r Útest_1D_arraysÚ s
|
|
|
þz TestColumnStack.test_1D_arraysc C sd t dgdgdgg¡}t dgdgdgg¡}t ddgddgddgg¡}t ||f¡}t|| d S ©Nr< r* r' r( rë rì r r r Útest_2D_arraysä s
|
þz TestColumnStack.test_2D_arraysc C s. t t tdd tdD W 5 Q R X d S )Nc s s | ]}t d ¡V qdS ©r' N©r r] ©r Ú_r r r Ú <genexpr>ð s z1TestColumnStack.test_generator.<locals>.<genexpr>r* )r Ú FutureWarningr
|
r4 rè r r r Útest_generatorî s
|
z TestColumnStack.test_generatorN)rR rS rT ré rí rï rö r r r r rå Ö s
|
|
rå c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú
|
TestDstackc C s t ttd d S ræ )r rç r rè r r r ré ô s z TestDstack.test_non_iterablec C s@ t d¡}t d¡}t||g}t ddggg¡}t|| d S ©Nr< r* ©r r@ r r ©r6 r r¸ r rÌ r r r Ú test_0D_array÷ s
|
|
|
zTestDstack.test_0D_arrayc C sD t dg¡}t dg¡}t||g}t ddggg¡}t|| d S rø rù rú r r r Ú test_1D_arrayþ s
|
zTestDstack.test_1D_arrayc C sX t dgdgg¡}t dgdgg¡}t||g}t ddggddggg¡}t|| d S rø rù rú r r r Ú test_2D_array s
|
zTestDstack.test_2D_arrayc C sN t ddg¡}t ddg¡}t||g}t ddgddggg¡}t|| d S rø rù rú r r r Útest_2D_array2 s
|
zTestDstack.test_2D_array2c C s. t t tdd tdD W 5 Q R X d S )Nc s s | ]}t d ¡V qdS rð rñ rò r r r rô s z,TestDstack.test_generator.<locals>.<genexpr>r* )r rõ r r4 rè r r r rö s
|
zTestDstack.test_generatorN) rR rS rT ré rû rü rý rþ rö r r r r r÷ ó s r÷ c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
|
S ) Ú
|
TestHsplitú&Only testing for integer splits.
|
|
c C s t ttdd d S ræ )r r? r rè r r r ré s z TestHsplit.test_non_iterablec C s: t d¡}zt|d td W n tk
|
r4 Y nX d S )Nr< r* r )r r@ r r r? rd r r r rû ! s
|
|
zTestHsplit.test_0D_arrayc C sF t ddddg¡}t|d}t ddg¡t ddg¡g}t|| d S rî ©r r@ r rÊ rË r r r rü ) s
|
zTestHsplit.test_1D_arrayc C sb t ddddgddddgg¡}t|d}t ddgddgg¡t ddgddgg¡g}t|| d S rî r rË r r r rý / s
|
ÿ
|
,zTestHsplit.test_2D_arrayN©rR rS rT Ú__doc__ré rû rü rý r r r r rÿ s
|
rÿ c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
|
S ) Ú
|
TestVsplitr c C s t ttdd d S ræ )r r? r rè r r r ré ; s z TestVsplit.test_non_iterablec C s t d¡}ttt|d d S rø )r r@ r r? r rd r r r rû > s
|
zTestVsplit.test_0D_arrayc C sB t ddddg¡}zt|d td W n tk
|
r< Y nX d S ©Nr< r* r' r( r )r r@ r r r? rd r r r rü B s
|
zTestVsplit.test_1D_arrayc C s^ t ddddgddddgg¡}t|d}t ddddgg¡t ddddgg¡g}t|| d S rî )r r@ r rÊ rË r r r rý J s
|
ÿ
|
(zTestVsplit.test_2D_arrayNr r r r r r 7 s
|
r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestDsplitc C s t ttdd d S ræ )r r? r rè r r r ré T s z TestDsplit.test_non_iterablec C s t d¡}ttt|d d S rø ©r r@ r r? r rd r r r rû W s
|
zTestDsplit.test_0D_arrayc C s$ t ddddg¡}ttt|d d S rî r rd r r r rü [ s zTestDsplit.test_1D_arrayc C sN t ddddgddddgg¡}zt|d td W n tk
|
rH Y nX d S r )r r@ r r r? rd r r r rý _ s
|
ÿ
|
zTestDsplit.test_2D_arrayc C s t ddddgddddggddddgddddggg¡}t|d}t ddgddggddgddggg¡t ddgddggddgddggg¡g}t|| d S rî )r r@ r rÊ rË r r r Ú test_3D_arrayh s
|
ÿ
|
|
ÿþ
|
$$ÿzTestDsplit.test_3D_arrayN)rR rS rT ré rû rü rý r r r r r r R s
|
r c @ s e Zd Zdd ZdS )Ú TestSqueezec C sÀ ddl m} |ddddd}|ddddd}|dddd}tt |¡t |d¡ tt |¡t |d¡ tt |¡t |d¡ d ggg}t |¡}t|d t|jd tt|tj d S )
|
Nr r% rW r9 r< )rW r9 r9 )rW r9 rW r` g ø?)
|
r+ r& r r r r^ r r5 r rx )r6 r& r r¸ Úcr r r r Ú
|
test_basict s
|
|
|
zTestSqueeze.test_basicN©rR rS rT r r r r r r s s r c @ sd e Zd Zdd Zdd Zej dej ej
|
g¡dd Z dd Z ej d
|
d d d dddg¡dd Z dS )ÚTestKronc C sÄ t d¡}t ddgddgg¡}t ddgddgg¡}tt ||¡| t ddgddgg¡}t d¡}tt ||¡| t dg¡}t ddgddgg¡}t ddgddgg¡}tt ||¡| t ddgddgg¡}t dg¡}tt ||¡| t dggdggg¡}t ddgddgg¡}t ddgddggddgddggg¡}tt ||¡| t ddgddgg¡}t dggdggg¡}t ddgddggddgddggg¡}tt ||¡| d S ) Nr< r* r' r( rw rÉ rÝ rÈ )r r@ r r )r6 r r¸ Úkr r r r s,
|
|
&&zTestKron.test_basicc C s G dd dt j}t ddg¡}||j|j|j}ttt||t j ttt||| ttt||| ttt||| d S )Nc @ s e Zd ZdZdS )z*TestKron.test_return_type.<locals>.myarrayg ð?N)rR rS rT Z__array_priority__r r r r Úmyarray¥ s r r* ) r rx r= rI r; r r r r )r6 r r r r r r Útest_return_type¤ s zTestKron.test_return_typeÚ array_classc C sL |t ddg¡}|t ddg¡}|t ddg¡}tt ||¡| d S )Nr' rÉ )r r= r r )r6 r r r¸ r r r r Útest_kron_smoke¯ s zTestKron.test_kron_smokec C sn t jjddgddggddgddggd}t jjt ddddg¡t jt d¡td d}t|t ||¡ d S ) Nr< r* r' r( r r é r: )r r r@ ZdiagÚidentityrB r r )r6 r r r r r Ú test_kron_ma¹ s
|
(ÿzTestKron.test_kron_mazshape_a,shape_b)rÀ rÀ )rê )r( r) rw )©r* r* )r* r* r* ))r< r rÀ )©r* r r* r ))r* r r r* r c C s t |¡}t |¡}dtdt|t| | }dtdt|t| | }t ||¡}t ||¡}t |j|¡s~tdd S )N)r< r zUnexpected shape from kron) r r= r0 rc Úmultiplyr Z array_equalrI r¼ ) r6 Zshape_aZshape_br r¸ Znormalised_shape_aZnormalised_shape_bZexpected_shaper r r r Útest_kron_shapeÀ s
|
|
|
ÿÿzTestKron.test_kron_shapeN)rR rS rT r r rß rà Z parametrizer ZasarrayÚmatr r r r r r r r s&
|
ÿ
|
úÿ r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestTilec C s t dddg¡}ddgddgg}tt|dddddddg tt|dddddddgddddddgg tt|dddddddgg tt|dddddgddddgg tt|dddgddgddgddgg tt|dddddgddddgddddgddddgg d S ) Nr r< r* r' r( r ©r< r* ©r* r< )r r@ r r ©r6 r r¸ r r r r Ö s , $(
|
|
ÿzTestTile.test_basicc C s0 t d¡}t|d}|d7 }t|t d¡ d S )Nr) r< r* )r r] r r r r r r Ú(test_tile_one_repetition_on_array_gh4679á s
|
|
z1TestTile.test_tile_one_repetition_on_array_gh4679c C sL t g gg¡}t g g g¡}t|dj}t|dj}t|d t|d d S )Nr* rP )r* r )r' r* r )r r@ r rI r )r6 r r¸ r
|
ra r r r rL ç s
|
zTestTile.test_emptyc
|
C s~ ddl m} ddddddg}d d
|
d d d dg}|D ]H}|dd|d}|D ]0}t ||j¡}t||}t||} t|| qFq0d S )Nr )Úrandint)r* r r r )r* r' r* )r' r* )r' )r* r' )r' r( r' )r' r* r' )r( r' r* r( r9 )Úsize)r+ r r r= r; r r r )
|
r6 r ZrepsrI r· r¸ Úrr ZlargeZklarger r r Útest_kroncompareï s
|
|
zTestTile.test_kroncompareN)rR rS rT r r rL r# r r r r r Õ s r c @ s e Zd Zdd ZdS )ÚTestMayShareMemoryc C sü t d¡}t d¡}tt ||¡ tt ||d d d
¡ tt ||d d d
¡ tt ||dd
d d d
f ¡ tt |d d d
|¡ tt |d d d
|¡ tt |dd
d d d
f |¡ tt |dd
d d d
f |¡ d S )N)rZ rX )rV rX rw r r* r< )r r= r Zmay_share_memory)r6 ra Zd2r r r r þ s
|
|
" $z TestMayShareMemory.test_basicNr r r r r r$ ý s r$ c C s: t | t |krtdt| |D ]\}}t|| q"dS )zCompare lists of arrays.z Iterables have different lengthsN)rc r? Úzipr )r rÌ r Úyr r r rÊ s rÊ ).Únumpyr r! Úsysrß Znumpy.lib.shape_baser r r r r r r r r
|
r r r r r Z numpy.testingr r r r r Úmaxsizerá r# r$ rU r_ r° r´ rÅ râ rå r÷ rÿ r r r r r r$ rÊ r r r r Ú<module> s4 @
|
4 '( ' !O(
|
