yzHTestConditionalShortcuts.test_zero_padding_shortcuts..w ...a6...r, rW r! r; r< shaper padr% rW testpad_amts r* test_zero_padding_shortcutsz4TestConditionalShortcuts.test_zero_padding_shortcutst s] y~~%%aA....4:...4gD!A!A!ABBBBBr, )r r r r c t j d ddd }d |j D }t t j ||d t j |||d d S )
NrY r. rZ r: c g | ]}d S )rC rC rT r] s r* r` zITestConditionalShortcuts.test_shallow_statistic_range..} ra r, r rb rC rW r
rc rf s r* test_shallow_statistic_rangez5TestConditionalShortcuts.test_shallow_statistic_rangez s y~~%%aA....4:...26$f=== 6$dJJJ L L L L Lr, c t j d dd }d |j D }t t j ||| t j |||d d S )N rZ r: c g | ]}d S ))r r rT r] s r* r` zFTestConditionalShortcuts.test_clip_statistic_range.. ra r, rb rm rc rf s r* test_clip_statistic_rangez2TestConditionalShortcuts.test_clip_statistic_range s~ y}}$$Q**..4:...26$d;;; 6$dKKK M M M M Mr, N) rQ rR rS rI mark parametrize
_all_modeskeysri rn rr rT r, r* rV rV s s [ VZ__%6%677C C 87C
[ V%N%N%NOOL L POL [ V%N%N%NOOM M POM M Mr, rV c X e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zej dg d d Zej dg d ej dg d d Zd Zej d ej d ej dddg d Zej dddg d Zd S )
TestStatisticc t j d d }t j |ddd }t j g d }t || d S )Nd f) r )rN 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 ? @ @ @ @ @ @ @ "@ $@ &@ (@ *@ ,@ .@ 0@ 1@ 2@ 3@ 4@ 5@ 6@ 7@ 8@ 9@ :@ ;@ <@ =@ >@ ?@ @@ @@ A@ A@ B@ B@ C@ C@ D@ D@ E@ E@ F@ F@ G@ G@ H@ H@ I@ I@ J@ J@ K@ K@ L@ L@ M@ M@ N@ N@ O@ O@ P@ @P@ P@ P@ Q@ @Q@ Q@ Q@ R@ @R@ R@ R@ S@ @S@ S@ S@ T@ @T@ T@ T@ U@ @U@ U@ U@ V@ @V@ V@ V@ W@ @W@ W@ W@ X@ @X@ X@ X@r r r r r r r r r r r r r r r r r r r r r! r; astypere r" r r% r/ bs r* test_check_mean_stat_lengthz)TestStatistic.test_check_mean_stat_length sm IcNN ! !# & & F1lF
C C C H
& 1a r, c t j d }t j |dd }t j g d }t || d S )Nrz r| r )c 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 rC r0 r r. rZ r: rD r r9
r~ r} rp ! " # $ % & ' ( ) * + , - . / 0 1 2 3 4 5 6 7 8 9 : ; < = > ? @ A B C D E F G H I J K L M N O P Q R S T U V W X Y Z [ \ ] ^ _ ` a b r r r r r r r r r r r r r r r r r r r r r r! r; re r" r r s r* test_check_maximum_1z"TestStatistic.test_check_maximum_1 Z IcNN F1h * * H
5
5
5
& 1a r, c t j d dz }t j |dd }t j g d }t || d S )Nrz rC r| r )rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rB r s r* test_check_maximum_2z"TestStatistic.test_check_maximum_2 s_ IcNNQ F1h * * H
?
?
?
& 1a r, c t j d dz }t j |ddd }t j g d }t || d S )Nrz rC 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 rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rz rB r s r* test_check_maximum_stat_lengthz,TestStatistic.test_check_maximum_stat_length sd IcNNQ F1h r : : : H
?
?
?
& 1a r, c t j d }t j |dd }t j g d }t || d S )Nrz 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 rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r r r r r r r r r r r r r r r r r r r r r rB r s r* test_check_minimum_1z"TestStatistic.test_check_minimum_1 sZ IcNN F1h * * H
+
+
+
& 1a r, c t j d dz }t j |dd }t j g d }t || d S )Nrz r0 r| r )r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r rz e r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 r0 rB r s r* test_check_minimum_2z"TestStatistic.test_check_minimum_2 s_ IcNNQ F1h * * H
+
+
+
& 1a r, c t j d dz }t j |ddd }t j g d }t || d S )Nrz rC r| r r r )rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r rz r: r: r: r: r: r: r: r: r: r: r: r: r: r: r: r: r: r: r: r: rB r s r* test_check_minimum_stat_lengthz,TestStatistic.test_check_minimum_stat_length sd IcNNQ F1h r : : : H
5
5
5
& 1a r, c t j d d }t j |dd }t j g d }t || d S )Nrz r{ r| r H@rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS 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 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 r r r r r r r r r r r r r r r r r r r r r r r r rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS r r s r* test_check_medianz TestStatistic.test_check_median2 sk IcNN ! !# & & F1h ) ) H
I
I
I
& 1a r, c t j g dg dg dg }t j |dd }t j g dg dg dg d g dg }t || d S )
Nr rC r. r. rZ rD rD r r0 rC r )r. r. rZ r. r. r r rC r. r rZ r. rZ rD rZ r rD r r0 r )r! r" re r r s r* test_check_median_01z"TestStatistic.test_check_median_01J s HiiiIII6 7 7 F1a " " H
__
__
__
__
__
1a r, c t j g dg dg dg }t j |j dd j }t j g dg dg dg d g dg }t || d S )
NrV rW rX rC r )rZ r. rZ r. rZ rY rZ r[ )r! r" re Tr r s r* test_check_median_02z"TestStatistic.test_check_median_02X s HiiiIII6 7 7 F138 $ $ & H
__
__
__
__
__
1a r, c t j d d }d|d< d|d< t j |ddd
}t j g d }t || d S ) Nrz r{ r rC r r@ r| r r rZ 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 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 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 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 s r* test_check_median_stat_lengthz+TestStatistic.test_check_median_stat_lengthf s IcNN ! !# & &!" F1hf = = = H
?
?
?
& 1a r, c g dg}t j |ddd }t j g dg dg dg dg dg dg dg dg dg dg dg dg dg
}t || d S )Nr. rZ r: rZ r r r0 r )r. r. r. r. r. r. rZ r: r: r: r: r: r: r: r: r! re r" r r s r* test_check_mean_shape_onez'TestStatistic.test_check_mean_shape_one s
YYK F1ff! 4 4 4 H
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
<
" 1a r, c t j d d }t j |dd }t j g d }t || d S )Nrz r{ r| r rR r r s r* test_check_mean_2z TestStatistic.test_check_mean_2 sk IcNN ! !# & & F1h ' ' H
I
I
I
& 1a r, rW )r r r r c t j g d t j g dt j z }t j |d| }t |d |d dS )z3 Test that appended and prepended values are equal )r0 rk )r g-q=r r? rl r rk N)r! r" float64re r )r% rW r/ s r* test_same_prepend_appendz&TestStatistic.test_same_prepend_append sb
H[[[ ! !BH]]]"*$M$M$M M F1fd # #QqT1R5!!!!!r, r
rE )rE )r rk ))rZ r0 )rE r ))rB r0 c t j d d }d}t j t
| 5 t j |d|| d d d d S # 1 swxY w Y d S )Nrp r: rZ #index can't contain negative valuesrF r0 r r! r; r< rI rJ rK re )r% rW r
arrrG s r* test_check_negative_stat_lengthz-TestStatistic.test_check_negative_stat_length s
imm##F++5
]:U
3
3
3 : : F34[ 9 9 9 9 : : : : : : : : : : : : : : : : : :s A++A/2A/c t j d }t j |d }t j |ddd }t j g dg dg dg dg d g d
g d g d g d g d g d g }t || d S )
Nrp rq rN r r0 r r rm )
r: r: r: rZ r: r r rD r r )
rC rC rC r rC r0 r r. r r )
r r r r r r9 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< re r" r r s r* test_simple_stat_lengthz%TestStatistic.test_simple_stat_length s IbMM Jq& ! ! F1&V F F F H
+
+
+
+
+
+
+
+
+
+
+
+
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
7
1a r, z)ignore:Mean of empty slice:RuntimeWarningzCignore:invalid value encountered in( scalar)? divide:RuntimeWarningr r c t j ddgd|d }t j t j ddt j t j g }t || d S )Nr r rC r0 r r )r! re r" nanr )r% rW rt r& s r* test_zero_stat_length_validz)TestStatistic.test_zero_stat_length_valid sV fb"Xvt;;;8RVRRVRV <==S(#####r, r r c t d}t j t | 5 t j ddgd|d d d d n # 1 swxY w Y t j t | 5 t j ddgd|d d d d n # 1 swxY w Y t j t | 5 t j ddgd|d d d d n # 1 swxY w Y t j t | 5 t j ddgd|d d d d d S # 1 swxY w Y d S ) Nz,stat_length of 0 yields no value for paddingrF r r r r )rC r rC )rI rJ rK r! re r% rW rG s r* test_zero_stat_length_invalidz+TestStatistic.test_zero_stat_length_invalid s2 >
]:U
3
3
3 5 5 FB8Q! 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
]:U
3
3
3 : : FB8Q& 9 9 9 9 : : : : : : : : : : : : : : :
]:U
3
3
3 5 5 FB8Q! 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
]:U
3
3
3 : : FB8Q& 9 9 9 9 : : : : : : : : : : : : : : : : : :sG AA A +BBB8C C#&C#D--D14D1N)rQ rR rS r rC rF rH rK rN rP rT r\ r_ rb rg ri rI rs rt rm ru r{ filterwarningsr r rT r, r* rx rx s/ ! ! !0! ! !0! ! !0! ! !0! ! !0! ! !0! ! !0! ! !0 ! ! ! ! ! ! ! ! !4! ! !,! ! !0 [ V & & & " "
" [ V%M%M%MNN [ MMM : : ON: ! ! !* [ KLL [ M [ Vfh%788$ $ 98 ML
$
[ Vi%;<< : : =< : : :r, rx c J e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d S ) TestConstantc t j d }t j |ddd }t j g d }t || d S )Nrz 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 rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ r~ rB r s r* test_check_constantz TestConstant.test_check_constant s_ IcNN F1h
H E E E H
5
5
5
& 1a r, c t j d }t j |dd }t j g d }t || d S )Nrz r| r rJ rB r s r* test_check_constant_zerosz&TestConstant.test_check_constant_zeros sZ IcNN F1h
+ + H
5
5
5
& 1a r, c t j d dd }t j |ddd }t j g dg d g d
g d g d g d
g dg dg }t || d S )Nrp rZ r: r} r 皙?rW r ) rC rC rC rC rC rC rC rC rC ) rC r rC r0 r r. rZ rC rC ) rC r: r r rD r r rC rC ) rC r9 r r r r r rC rC ) rC r r r~ r r r rC rC ) rC r r} r r r r rC rC r! r; r< re r" r r% rt rg r& s r* test_check_constant_floatz&TestConstant.test_check_constant_float# s imm##Aq))vc6
#&( ( (8
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
3 h'''''r, c < t j d dd }| t j }t j |ddd }t j g dg d g d
g d g d g d
g dg dg }t || d S )Nrp rZ 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 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 rl re r" r )r% rt arr_floatrg r& s r* test_check_constant_float2z'TestConstant.test_check_constant_float27 s imm##Aq))JJrz** vi!1
#&( ( (8
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L
N h'''''r, c t j dt }t j |ddd }t j g d }t || d S )Nrz r? r| r )皙333333r )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 rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r r r r r r r r r r r r r r r r r r r r r )r! r; r re r" r r s r* test_check_constant_float3z'TestConstant.test_check_constant_float3L si Ic ' ' ' F1h
L I I I H
I
I
I
& 1 r, c
t j d dd }t j |ddd }t j g dg d g d
g d g d g d
g dg }t || d S )Nrp rZ r: )rC ro r r r )
r r r r r r r r r r )
r r r rC r0 r r. rZ r r )
r r r: r r rD r r r r )
r r r9 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 s r* "test_check_constant_odd_pad_amountz/TestConstant.test_check_constant_odd_pad_amountd s imm##Aq))vc)rtolatol)r! r; r re r" r r s r* test_check_simplez TestLinearRamp.test_check_simple sw IcNN ! !# & & F1h
& A A A H
F
F
F
& 14d333333r, c 8 t j d dd t j }t j |ddd }t j g dg d g d
g d g d g d
g dg dg }t || d S )Nr~ r. rZ )r0 r0 r r\ rW r ) r r r r r r r r r ) r r r r r g ?r r r ) r r r r r r r r r ) r g @r r r r r g @r ) r r r r r r r r r ) r @r r r r r #@r ) r g @r r g !@r r g @r r r s r*
test_check_2dz TestLinearRamp.test_check_2d s imm##Aq))00<<vc6
&III8
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
B
DE E h'''''r, )
exceptionsc x ddl m} t j |dd |dd g }t j |ddd }t j |dd |d
d |d d |d d |d
d |dd |dd g }t || d S )Nr )FractionrC r0 rk rN r r r9 r r: rB rE ) fractionsr r! r" re r )r% r rt actualr& s r* test_object_arrayz TestLinearRamp.test_object_array s &&&&&&hAQ899V-AFFF 8 Ha Ha Ha HR HR HR HR
VX&&&&&r, c P t j t j d dd dd }t |dddf d t |ddd
f d t |dddf d t |d
ddf d dS ) z!Ensure that end values are exact.r r0 rZ ) { r rb Nr r rk )r! re rO r< r )r% r/ s r* test_end_valuesz TestLinearRamp.test_end_values s F272;;&&q!,,j} M M MQqqq!tWb!!!Qqqq"uXr"""Qq!!!tWb!!!Qr111uXr"""""r, r@ c b t j dg| }t j |ddd }t j g d| }t || t j dg| }t j |ddd }t j g d| }t || dS ) a<
Check correct behavior of unsigned dtypes if there is a negative
difference between the edge to pad and `end_values`. Check both cases
to be independent of implementation. Test behavior for all other dtypes
in case dtype casting interferes with complex dtypes. See gh-14191.
r r? r r r )r rC r0 r r0 rC r )r r0 rC r rC r0 r N)r! r" re r )r% r@ r' r( r& s r* test_negative_differencez'TestLinearRamp.test_negative_difference s
HaS & & &1=Q???8 1 1 1???VX&&& HaS & & &1=Q???8 1 1 1???VX&&&&&r, N)rQ rR rS r r rI rs xfailAssertionErrorr r rt _numeric_dtypesr rT r, r* r r s 4 4 40 ( ( ( [>"344' ' 54'$# # # [ Wo66' ' 76' ' 'r, r c 8 e Zd Zd Zd Zd Zd Zd Zd Zd Z dS ) TestReflectc t j d }t j |dd }t j g d }t || d S )Nrz r| r )r} r r r r r~ r r r r r r r r9 r r rD r r r: rZ r. r r0 rC r rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r rA r@ r? r> r= r< r; r: r9 r8 r7 r6 r5 r4 r3 r2 r1 r0 r/ r. rB r s r* r z TestReflect.test_check_simple rD r, c t j d }t j |ddd }t j g d }t || d S )Nrz r| r oddr )ir rB rA rE rk r rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r rz rM f g h i j k l m n o p q r s t u v w rB r s r* test_check_odd_methodz!TestReflect.test_check_odd_method s_ IcNN F1h > > > H
?
?
?
& 1a r, c g dg dg}t j |dd }t j g dg dg dg dg dg dg dg dg dg dg dg dg dg dg }t || d S )Nrd r: r r re r )r r: r r r r: r r r r: r r r r: r rZ r. rZ r: rZ r. rZ r: rZ r. rZ r: rZ r. rZ rf r s r* test_check_large_padz TestReflect.test_check_large_pad+ s
YY " F1fi ( ( H
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
<
$ 1a r, c g dg}t j |dd }t j g dg dg dg dg dg dg dg dg dg dg dg dg dg
}t || d S )Nrd re r r rf r s r* test_check_shapez TestReflect.test_check_shapeB s
YYK F1fi ( ( H
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
<
" 1a r, c t j g ddd }t j g d }t || d S )NrC r0 r r0 r )r r0 rC r0 r r0 rC rf r s r*
test_check_01zTestReflect.test_check_01X sE F999a + + H*** + +1a r, c t j g ddd }t j g d }t || d S )Nr r r ) r0 r r0 rC r0 r r0 rC r0 rf r s r*
test_check_02zTestReflect.test_check_02] sE F999a + + H000 1 11a r, c t j g ddd }t j g d }t || d S )Nr r. r ) rC r0 r r0 rC r0 r r0 rC r0 r rf r s r*
test_check_03zTestReflect.test_check_03b sE F999a + + H666 7 71a r, N)
rQ rR rS r r r r r r
r rT r, r* r r s} ! ! !0! ! !0! ! !.! ! !,! ! !
! ! !
! ! ! ! !r, r c e Zd ZdZej d ee ddhz
d Z
ej de d Z dS )TestEmptyArrayz > F28F##\ = = = = > > > > > > > > > > > > > > > > > >s5 AA A(*B B"%B"*C;;C?C?c r t j t j d d| }|j dk sJ d S )N)r0 r r0 )ry r r rb )r r r. )r! re rO rd )r% rW r( s r* test_pad_non_empty_dimensionz+TestEmptyArray.test_pad_non_empty_dimensionz s< **,>TJJJ|y((((((r, N) rQ rR rS __doc__rI rs rt sortedru rv r r rT r, r* r r h s FF [ z((J+@@AA
> >
> [ VZ__%6%677) ) 87) ) )r, r c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestSymmetricc t j d }t j |dd }t j g d }t || d S )Nrz r| r )r r r r r~ r r r r r r r r9 r r rD r r r: rZ r. r r0 rC r r rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r r rA r@ r? r> r= r< r; r: r9 r8 r7 r6 r5 r4 r3 r2 r1 r0 r/ rB r s r* r z TestSymmetric.test_check_simple sZ IcNN F1h , , H
5
5
5
& 1a r, c t j d }t j |ddd }t j g d }t || d S )Nrz r| r r r )r r r r r r r r r r r r r r r r r r r r rB rA rE rk r r rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r r rz rM r r r r r r r r r r r r r r r r r rB r s r* r z#TestSymmetric.test_check_odd_method s_ IcNN F1h % @ @ @ H
?
?
?
& 1a r, c g dg dg}t j |dd }t j g dg dg dg dg dg dg dg dg dg dg dg dg dg dg }t || d S )Nrd r re r rZ r: r: rZ r. r. rZ r: r: rZ r. r. rZ r: r: )r r r r r: r: r r r r r: r: r r r rf r s r* r z"TestSymmetric.test_check_large_pad s
YY " F1fk * * H
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
<
& 1a r, c g dg dg}t j |ddd }t j g dg dg dg dg d g d g d
g d
g d g d g d g d g d
g d
g }t || d S )Nrd r re r r r )rA rE rE rk r r rC r0 r0 r r. r. rZ r: r: )rk r r rC r0 r0 r r. r. rZ r: r: r r r )rC r0 r0 r r. r. rZ r: r: r r r rD r r )r r. r. rZ r: r: r r r rD r r r r9 r9 )rZ r: r: r r r rD r r r r9 r9 r r r )r r r rD r r r r9 r9 r r r r r r )rD r r r r9 r9 r r r r r r r r r rf r s r* test_check_large_pad_oddz&TestSymmetric.test_check_large_pad_odd s
YY " F1fk > > > H
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
I
K
$ 1a r, c g dg}t j |dd }t j g dg dg dg dg dg dg dg dg dg dg dg dg dg
}t || d S )Nrd re r r rf r s r* r z TestSymmetric.test_check_shape s
YYK F1fk * * H
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
:
<
" 1a r, c t j g ddd }t j g d }t || d S )Nr r0 r )r0 rC rC r0 r r r0 rf r s r* r zTestSymmetric.test_check_01 sE F999a - - H*** + +1a r, c t j g ddd }t j g d }t || d S )Nr r r ) r r0 rC rC r0 r r r0 rC rf r s r* r
zTestSymmetric.test_check_02 sE F999a - - H000 1 11a r, c t j g ddd }t j g d }t || d S )Nr r: r )rC r0 r r r0 rC rC r0 r r r0 rC rC r0 r rf r s r* r zTestSymmetric.test_check_03 sE F999a - - HBBB C C1a r, N) rQ rR rS r r r r r r r
r rT r, r* r r s ! ! !0! ! !0! ! !0! ! !.! ! !,! ! !
! ! !
! ! ! ! !r, r c 8 e Zd Zd Zd Zd Zd Zd Zd Zd Z dS ) TestWrapc t j d }t j |dd }t j g d }t || d S )Nrz r| r )r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r r rC r0 r r. rZ r: r r rD r r r9 r r r r r r r r~ r r r r r} r r r r rp 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# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ rA r r rC r0 r r. rZ r: r r rD r r r9 r r r r r r r rB r s r* r zTestWrap.test_check_simple sZ IcNN F1h ' ' H
5
5
5
& 1a r, c b t j d }t j |d }t j |dd }t j g dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg }t || d S )Nr9 r )r r9 r )r r r rD r r r rD r r r rD r r r rD r r r rD r r r rD r r )r0 r r rC r0 r r rC r0 r r rC r0 r r rC r0 r r rC r0 r r rC r0 r )r: r r. rZ r: r r. rZ r: r r. rZ r: r r. rZ r: r r. rZ r: r r. rZ r: r rz r s r* r z TestWrap.test_check_large_pad s IbMM Jq& ! ! F1h ' ' H. . .) ) )) ) ). . .) ) )) ) ). . .) ) )) ) ). . .) ) )) ) ). . .) ) )) ) ). . .) ) )) ) ). . .) ) )) ) ). . .) ) )) ) ). . .e3
/5
5
l 1a r, c t j g ddd }t j g d }t || d S )Nr r r ) rC r0 r rC r0 r rC r0 r rf r s r* r zTestWrap.test_check_01[ sE F999a ( ( H000 1 11a r, c t j g ddd }t j g d }t || d S )Nr r. r ) r rC r0 r rC r0 r rC r0 r rC rf r s r* r
zTestWrap.test_check_02` sE F999a ( ( H666 7 71a r, c t j d }t j |dd }t ||d dd df d S )Nra )r rZ r rb r )r! rO re r r s r* test_pad_with_zerozTestWrap.test_pad_with_zeroe sK GFOO F1f6 * * *1aSbSk*****r, c b t j d }t j |dd }t t j ||||f dd | t j d }t j |dd }t t j ||||f dd | dS ) zy
Check wrapping on each side individually if the wrapped area is longer
than the original array.
rZ )r9 r r rb r N)r r9 rA )r! r; re r r_r s r* test_repeated_wrappingz TestWrap.test_repeated_wrappingj s
IaLL F1gF + + +25Aq!,QRR0!444 IaLL F1gF + + +25Aq!,SbS1155555r, c t j d dd }t j |ddgd }t j g dg dg dg dg dg dg }t || d S )
z
Assert that 'wrap' pads only with multiples of the original area if
the pad width is larger than the original array.
r. r0 r r rC r rb )r r0 r r0 r r0 )rC r rC r rC r Nrz r s r* &test_repeated_wrapping_multiple_originz/TestWrap.test_repeated_wrapping_multiple_originw s
IaLL A & & F1vv&V 4 4 4 H
!
1a r, N)
rQ rR rS r r r r
r( r+ r. rT r, r* r" r" s~ ! ! !0:! :! :!x! ! !
! ! !
+ + +
6 6 6! ! ! ! !r, r" c e Zd Zd Zd ZdS )TestEdgec t j d }t j |d }t j |dd }t j g dg dg dg dg dg dg dg dg dg }t || d S Nr9 r. r rw r )r r r r rC r0 r0 r0 )r r r r r. rZ rZ rZ )r: r: r: r: r r r r )rD rD rD rD r r r r rz r s r* r zTestEdge.test_check_simple s IbMM Jq& ! ! F1& / / H
%
%
%
%
%
%
%
%
%
%
%
%
%
%
%
)
)
)
)
)
)
)
)
)
)
)
)
+
1a r, c t j g d }t j |dd }t j g d }t || t j g dg dg }t j |dd }t j |dd }t || t j d dd d
}t j |dd }t j |d d }t || d S ) Nr )r} r )rC rC r0 r r r rd r r r0 r r. )r} r} r} )r! r" re r r; r< )r% r/ paddedr& s r* test_check_width_shape_1_2z#TestEdge.test_check_width_shape_1_2 s
HYYY 9f--8 . . .//68,,, Hiii+ , ,9f--6! -v6668,,, IbMM ! !!Q * *9f--6! 5v>>68,,,,,r, N)rQ rR rS r r6 rT r, r* r0 r0 s2 ! ! !&- - - - -r, r0 c e Zd Zd Zd ZdS ) TestEmptyc t j d dd }t j |ddgd }|j dk sJ t ||d d
d d f d S )
Nr r. r: rN r- r rb )rD r r0 rA r rk r! r; r< re rd r r s r* test_simplezTestEmpty.test_simple st imm##Aq))ff -G<<<|w&&&&S&2qt,-----r, c z t j d }t j |g dd }|j dk sJ d S )Nr r r rb r r r s r* r z"TestEmpty.test_pad_empty_dimension sG hy!! / / /g>>>|y((((((r, N)rQ rR rS r; r rT r, r* r8 r8 s2 . . . ) ) ) ) )r, r8 c d } t j d dd }t j |d| }t j g dg dg dg dg dg dg }t || d S )Nc < d| d |d < d| |d d < d S )Nr r rC rT )vectorr iaxiskwargss r* _padwithtensz6test_legacy_vector_functionality.._padwithtens s. "} ! } !# ! }~~ r, r: r0 r )r r r r r r r )r r r rC r0 r r )r r r r. rZ r r rz )rB r/ r s r* test_legacy_vector_functionalityrC s $ $ $ ! Q""A
q!\""A
% % % % % % % % % % % % % % % % % % '
A q! r, c t j dgdd } t j g d }t | | d S )NrC r0 r rb )r r rC r r rf )r/ r s r* test_unicode_moderE sC
sAJ'''A
!!Aq! r, rW )r r r r c t j dd }d}t j dd }t t j ||| | d S )Nr3 )
fill_valuerw )rD r rb )r! r r re )rW r/ rh r s r* test_object_inputrH sX 4(((A G
4(((Arvat444a88888r, c
e Zd Zej dg d ej de d Zej de d Z ej dg d ej de d Z
ej dd d
d e d fd
de f e dd d fdg ej de d Z
d Zej dg d ej de d Zd S ) TestPadWidthr ))r. rZ r: r )r ro ry )r} r )rZ r: ))r r. rZ )r rC r0 rW c t j d d }d}t j t
| 5 t j ||| d d d d S # 1 swxY w Y d S )Nrp rq z(operands could not be broadcast togetherrF rs r% r rW rt rG s r* test_misshaped_pad_widthz%TestPadWidth.test_misshaped_pad_width s imm##F++:
]:U
3
3
3 ) ) F3 4 ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) A))A-0A-c t j d d }d}t j t
| 5 t j |d| d d d d S # 1 swxY w Y d S )Nrp rq zDinput operand has more dimensions than allowed by the axis remappingrF ))ry )r. )rZ )r r ro rs )r% rW rt rG s r* test_misshaped_pad_width_2z'TestPadWidth.test_misshaped_pad_width_2 s imm##F++
]:U
3
3
3 H H F3@$ G G G H H H H H H H H H H H H H H H H H HrN rn c t j d d }d}t j t
| 5 t j ||| d d d d S # 1 swxY w Y d S )Nrp rq rr rF rs rL s r* test_negative_pad_widthz$TestPadWidth.test_negative_pad_width s imm##F++5
]:U
3
3
3 ) ) F3 4 ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )rN zpad_width, dtype)3N)wordNr3 N)g333333 @N))r0 r r. rx rC rk )))g r rx Nc f t j d d }d}|`t j t
| 5 t j |t j || | d d d d S # 1 swxY w Y d S t j t
| 5 t j ||| d d d n # 1 swxY w Y t j t
| 5 t j |t j | | d d d d S # 1 swxY w Y d S )Nrp rq z%`pad_width` must be of integral type.rF r? )r! r; r< rI rJ TypeErrorre r" )r% r r@ rW rt rG s r*
test_bad_typezTestPadWidth.test_bad_type s imm##F++7 y666
D
DsBHYe < < r{ s
K K K K K K K K K K ( ( ( ( ( ( Jvjj j #A& !$ t$
D ! d# t$ ' &)
P* P* P* P* P* P* P* P*fM M M M M M M M,f: f: f: f: f: f: f: f:R p) p) p) p) p) p) p) p)fR' R' R' R' R' R' R' R'jk! k! k! k! k! k! k! k!\) ) ) ) ) ) ) )0C! C! C! C! C! C! C! C!L@! @! @! @! @! @! @! @!F$- $- $- $- $- $- $- $-N
)
)
)
)
)
)
)
) ( !I!I!IJJ9 9 KJ9NC NC NC NC NC NC NC NCb !2!233
7
7 43
7 ; ; ;
!WffhhdE!JKK( ( LK( !2!233* * 43* !2!2334 4 434 /22 !2!233! ! 43 32! ! !r,