3
���h � � @ s d Z ddlZddlZddlmZmZmZ ddlm Z ej
d ej
d ej
d ej
d Z d dii d
did did did did did d
id d
ii i d� Z G dd� d�Z
G dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd � d �ZG d d � d �ZG d d � d �ZG d!d"� d"�ZG d#d$� d$�Zd%d&� Zd'd(� Zejjd)d*d+d,d-g�d.d/� �Z G d0d1� d1�Z ejjd)e j � �d2d3� �Z d4d5� Z ejjd)d6d7e!� dd8d9g�d:d;� �Z"ejjd)e j � �d<d=� �Z#ejjd)e j � �d>d?� �Z$ejjd@e �ejjd)e j � �dAdB� ��Z%dS )Cz(Tests for the array padding functions.
� N)�assert_array_equal�assert_allclose� assert_equal)� _as_pairsZuint�int�float�complex�constant_values�
end_values� stat_length� reflect_typeZeven) �constant�edge� linear_ramp�maximum�mean�median�minimum�reflect� symmetric�wrap�emptyc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� TestAsPairsc C sl t jddggd �}x,ddgdggfD ]}t|d�}t||� q&W t� }tt|d�t j||ggd �� dS )z Test casting for a single value.� �
N)�np�arrayr r �object)�self�expected�x�result�obj� r# �>/tmp/pip-build-5_djhm0z/numpy/numpy/lib/tests/test_arraypad.py�test_single_value! s
z TestAsPairs.test_single_valuec C s� t jddggd �}x.ddgddggfD ]}t|d�}t||� q(W t� }ttd|gd�t jd|ggd �� ttdgdggd�t jddgddgg�� ttdg|ggd�t jddg||gg�� dS )z-Test proper casting for two different values.r � r �a� N)r r r r r )r r r r! r" r# r# r$ �test_two_values. s
zTestAsPairs.test_two_valuesc C s0 d}t td ddd�|� t td ddd�|� d S ) Nr F)�as_indexT�NN�NN�NN)r+ r, r- )r r )r r r# r# r$ �test_with_noneG s zTestAsPairs.test_with_nonec C s$ t jd�jd�}tt|d�|� dS )z?Test if `x` already matching desired output are passed through.� � r( N)r0 r( )r �arange�reshaper r )r r r# r# r$ �test_pass_throughR s z TestAsPairs.test_pass_throughc
C s� t tddgddd�tjddggd tjd�� t tddgddd�tjdd ggd tjd�� xrddgdggdd gddgdd ggd dggd
d ggd d
dgg fD ],}tjtd
d�� t|ddd� W dQ R X q�W dS )z Test results if `as_index=True`.g������@gffffff
@r T)r* r )�dtypeg�(\��@r& � r( � znegative values)�matchN���r8 r8 r8 ���r8 r8 ���)r r r r Zintp�pytest�raises�
ValueError)r r r# r# r$ �
test_as_indexZ s * zTestAsPairs.test_as_indexc C s� t jtdd�� tdgggd� W dQ R X t jtdd�� tddgdd ggd� W dQ R X t jtdd�� ttjd
�d� W dQ R X dS ) z"Ensure faulty usage is discovered.z more dimensions than allowed)r7 r r Nzcould not be broadcastr5 r( r& )r( r )r; r< r= r r �ones)r r# r# r$ �test_exceptionsi s zTestAsPairs.test_exceptionsN) �__name__�
__module__� __qualname__r% r) r. r3 r> r@ r# r# r# r$ r s
r c @ sb e Zd Zejjdej� �dd� �Zejjdddddg�dd � �Z ejjdddddg�d
d � �Z
d S )
�TestConditionalShortcuts�modec C s> t jd�jddd�}dd� |jD �}t|t j|||d�� d S )N�x r& � r0 c S s g | ]}d�qS )r )r r r# )�.0�_r# r# r$ �
<listcomp>w s zHTestConditionalShortcuts.test_zero_padding_shortcuts.<locals>.<listcomp>)rE )r r1 r2 �shaper �pad)r rE �test�pad_amtr# r# r$ �test_zero_padding_shortcutst s z4TestConditionalShortcuts.test_zero_padding_shortcutsr r r r c C sL t jd�jddd�}dd� |jD �}tt j||dd�t j|||d d
�� d S ) NrF r& rG r0 c S s g | ]}d�qS )r5 )r5 r5 r# )rH rI r# r# r$ rJ } s zITestConditionalShortcuts.test_shallow_statistic_range.<locals>.<listcomp>r )rE r5 )rE r )r r1 r2 rK r rL )r rE rM rN r# r# r$ � test_shallow_statistic_rangez s z5TestConditionalShortcuts.test_shallow_statistic_rangec C sJ t jd�jdd�}dd� |jD �}tt j|||d�t j|||dd�� d S )N� rG r0 c S s g | ]}d�qS )r )r r r# )rH rI r# r# r$ rJ � s zFTestConditionalShortcuts.test_clip_statistic_range.<locals>.<listcomp>)rE )rE r )r r1 r2 rK r rL )r rE rM rN r# r# r$ �test_clip_statistic_range� s z2TestConditionalShortcuts.test_clip_statistic_rangeN) rA rB rC r; �mark� parametrize�
_all_modes�keysrO rP rR r# r# r# r$ rD s s rD c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zejjdd d d d g�d d!� �Zejjdd d d d g�ejjd"d3d5d7d;d?g�d(d)� ��Zd*d+� Zejjd,�ejjd-�ejjdd d g�d.d/� ���Zejjdd d g�d0d1� �Zd2S )@�
TestStatisticc � C s\ t jd�jd�}t j|doddqd�}t jd d d d d d d d d d d d d d d d d d d d d d d d d d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdldldldldldldldldldldldldldldldldldldldlg��}t||� d S )rN�d �f� � r r( r )r g �?g g �?g @g @g @g @g @g @g @g "@g $@g &@g (@g *@g ,@g .@g 0@g 1@g 2@g 3@g 4@g 5@g 6@g 7@g 8@g 9@g :@g ;@g <@g =@g >@g ?@g @@g �@@g A@g �A@g B@g �B@g C@g �C@g D@g �D@g E@g �E@g F@g �F@g G@g �G@g H@g �H@g I@g �I@g J@g �J@g K@g �K@g L@g �L@g M@g �M@g N@g �N@g O@g �O@g P@g @P@g �P@g �P@g Q@g @Q@g �Q@g �Q@g R@g @R@g �R@g �R@g S@g @S@g �S@g �S@g T@g @T@g �T@g �T@g U@g @U@g �U@g �U@g V@g @V@g �V@g �V@g W@g @W@g �W@g �W@g X@g @X@g �X@g �X@�rZ r[ )r\ �r( r )r] )r r1 �astyperL r r )r r' �br# r# r$ �test_check_mean_stat_length� s&
z)TestStatistic.test_check_mean_stat_lengthc � C sR t jd�}t j|dgd�}t jddddddddddddddddddddddddddddd d
d d d
dddddddddddddddd d dd d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdddddddddddddddddddddg��}t||� d S )hNrX rZ r[ r �c r r5 r( r r& rG r0 � � r6 r � r/ �
� � � � � � � � � � � � � � rQ � � �! �" �# �$ �% �& �' �( �) �* �+ �, �- �. �/ �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 )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ �test_check_maximum_1� s&
z"TestStatistic.test_check_maximum_1c � C sV t jd�d }t j|dfd�}t jdddddddddddddddddddddddddddddd d
d d d
ddddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedddddddddddddddddddddg��}t||� d S )gNrX r5 rZ r[ r r( r r& rG r0 rb rc r6 r rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ �test_check_maximum_2� s&
z"TestStatistic.test_check_maximum_2c � C sZ t jd�d }t j|dgddd�}t jdddddddddddddddddddddddddddd d
d d d
ddddddddddddddddd dd d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdddddddddddddddddddddg��}t||� d S )hNrX r5 rZ r[ r r )r r( r r& rG r0 rb rc r6 rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ � test_check_maximum_stat_length� s&
z,TestStatistic.test_check_maximum_stat_lengthc � C sR t jd�}t j|dgd�}t jdddddddddddddddddddddddddddddd d
d d d
dddddddddddddddd dd d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfddddddddddddddddddddg��}t||� d S )hNrX rZ r[ r r r5 r( r r& rG r0 rb rc r6 r rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ �test_check_minimum_1� s&
z"TestStatistic.test_check_minimum_1c � C sV t jd�d }t j|dfd�}t jdddddddddddddddddddddddddddddd d
d d d
ddddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcddddeddddddddddddddddddddg��}t||� d S )gNrX r( rZ r[ r r r& rG r0 rb rc r6 r rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra �e )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ �test_check_minimum_2 s&
z"TestStatistic.test_check_minimum_2c � C sZ t jd�d }t j|dgddd�}t jdddddddddddddddddddddddddddd d
d d d
ddddddddddddddddd dd d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdd^d^d^d^d^d^d^d^d^d^d^d^d^d^d^d^d^d^d^d^g��}t||� d S )hNrX r5 rZ r[ r r )r r( r r& rG r0 rb rc r6 rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ � test_check_minimum_stat_length s&
z,TestStatistic.test_check_minimum_stat_lengthc � C sX t jd�jd�}t j|dkd�}t jdddddddddddddddddddddddddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjddddddddddddddddddddg��}t||� d S )lNrX rY rZ r[ r g �H@g g �?g @g @g @g @g @g @g @g "@g $@g &@g (@g *@g ,@g .@g 0@g 1@g 2@g 3@g 4@g 5@g 6@g 7@g 8@g 9@g :@g ;@g <@g =@g >@g ?@g @@g �@@g A@g �A@g B@g �B@g C@g �C@g D@g �D@g E@g �E@g F@g �F@g G@g �G@g H@g �H@g I@g �I@g J@g �J@g K@g �K@g L@g �L@g M@g �M@g N@g �N@g O@g �O@g P@g @P@g �P@g �P@g Q@g @Q@g �Q@g �Q@g R@g @R@g �R@g �R@g S@g @S@g �S@g �S@g T@g @T@g �T@g �T@g U@g @U@g �U@g �U@g V@g @V@g �V@g �V@g W@g @W@g �W@g �W@g X@g @X@g �X@g �X@)rZ r[ )r r1 r^ rL r r )r r' r_ r# r# r$ �test_check_median2 s&
z TestStatistic.test_check_medianc
C s� t jdddgdddgdddgg�}t j|dd�}t jdddddgdddddgdddddgdddddgdddddgg�}t||� d S ) Nr r5 r& rG r6 rc r( r )r r rL r )r r' r_ r# r# r$ �test_check_median_01J s " z"TestStatistic.test_check_median_01c
C s� t jdddgdddgdddgg�}t j|jdd�j}t jdddddgdddddgdddddgdddddgdddddgg�}t||� d S ) Nr r5 r& rG r6 rc r( r )r r rL �Tr )r r' r_ r# r# r$ �test_check_median_02X s " z"TestStatistic.test_check_median_02c � C sl t jd�jd�}d|d< d|d< t j|dmd dnd �}t jdddddddddddddddddddddddddd
ddddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdddkdlddddddddddddddddddddg��}t||� d S )oNrX rY g @r5 g X@r� rZ r[ r r rG )r g g @g @g @g @g @g @g "@g $@g &@g (@g *@g ,@g .@g 0@g 1@g 2@g 3@g 4@g 5@g 6@g 7@g 8@g 9@g :@g ;@g <@g =@g >@g ?@g @@g �@@g A@g �A@g B@g �B@g C@g �C@g D@g �D@g E@g �E@g F@g �F@g G@g �G@g H@g �H@g I@g �I@g J@g �J@g K@g �K@g L@g �L@g M@g �M@g N@g �N@g O@g �O@g P@g @P@g �P@g �P@g Q@g @Q@g �Q@g �Q@g R@g @R@g �R@g �R@g S@g @S@g �S@g �S@g T@g @T@g �T@g �T@g U@g @U@g �U@g �U@g V@g @V@g �V@g �V@g W@g @W@g �W@g �W@g �X@g �X@)rZ r[ )r rG )r r1 r^ rL r r )r r' r_ r# r# r$ � test_check_median_stat_lengthf s*
z+TestStatistic.test_check_median_stat_lengthc C s� dddgg}t j|dddd�}t jdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg
�}t||� d S ) Nr& rG r0 rb r r( )r )rG rb )r rL r r )r r' r_ r# r# r$ �test_check_mean_shape_one� s" &z'TestStatistic.test_check_mean_shape_onec � C sX t jd�jd�}t j|dkd�}t jdddddddddddddddddddddddddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjddddddddddddddddddddg��}t||� d S )lNrX rY rZ r[ r g �H@g g �?g @g @g @g @g @g @g @g "@g $@g &@g (@g *@g ,@g .@g 0@g 1@g 2@g 3@g 4@g 5@g 6@g 7@g 8@g 9@g :@g ;@g <@g =@g >@g ?@g @@g �@@g A@g �A@g B@g �B@g C@g �C@g D@g �D@g E@g �E@g F@g �F@g G@g �G@g H@g �H@g I@g �I@g J@g �J@g K@g �K@g L@g �L@g M@g �M@g N@g �N@g O@g �O@g P@g @P@g �P@g �P@g Q@g @Q@g �Q@g �Q@g R@g @R@g �R@g �R@g S@g @S@g �S@g �S@g T@g @T@g �T@g �T@g U@g @U@g �U@g �U@g V@g @V@g �V@g �V@g W@g @W@g �W@g �W@g X@g @X@g �X@g �X@)rZ r[ )r r1 r^ rL r r )r r' r_ r# r# r$ �test_check_mean_2� s&
z TestStatistic.test_check_mean_2rE r r r r c C sJ t jdddg�t jdddgt jd� }t j|d |�}t|d |d
� dS ) z3 Test that appended and prepended values are equal r5 r( r g�-���q=)r4 N���r� )r5 r5 r� )r r �float64rL r )r rE r' r# r# r$ �test_same_prepend_append� s
&z&TestStatistic.test_same_prepend_appendr r( r r5 rG r& c
C sD t jd�jd�}d}tjt|d�� t j|d||d� W d Q R X d S ) NrQ r0 rG z#index can't contain negative values)r7 r( )r )r0 rG )r r1 r2 r; r< r= rL )r rE r �arrr7 r# r# r$ � test_check_negative_stat_length� s z-TestStatistic.test_check_negative_stat_lengthc C s2 t jd�}t j|d"�}t j|d%dd&d�}t jddddddd d
d d g
ddddddd d
d d g
d d d d d ddd
ddg
ddddddd d
d d g
ddddddddddg
ddddddddddg
dddddddd ddg
d d d d d d d d!d d g
dddddddd ddg
dddddddd ddg
dddddddd ddg
g �}t||� d S )'NrQ r0 rG r( r r )rE r rb rc r6 r5 r r& rd r r/ re rf rh rg ri rj rk rl r[ rm rn ro rp rZ rq rr rs )r0 rG �r( r �r r( )r� r� )r )r r1 r2 rL r r )r r' r_ r# r# r$ �test_simple_stat_length� s
z%TestStatistic.test_simple_stat_lengthz)ignore:Mean of empty slice:RuntimeWarningzOignore:invalid value encountered in (true_divide|double_scalars):RuntimeWarningc C s> t jddgd|dd�}t jt jddt jt jg�}t||� d S )Ng �?g @r5 r( r )r )r5 r( )r rL r �nanr )r rE r� r r# r# r$ �test_zero_stat_length_valid� s z)TestStatistic.test_zero_stat_length_validc " C s� d}t jt|d�� tjddgd|dd� W d Q R X t jt|d�� tjddgd|dd� W d Q R X t jt|d�� tjddgd|dd� W d Q R X t jt|d�� tjddgd|d d� W d Q R X d S )
Nz,stat_length of 0 yields no value for padding)r7 g �?g @r )r r5 )r5 r )r5 r )r; r< r= r rL )r rE r7 r# r# r$ � test_zero_stat_length_invalid� s z+TestStatistic.test_zero_stat_length_invalidNr: r: )r: r� )r r� �rG r( r: �r: r )r� r� r9 �r9 �r( )r� r� )rA rB rC r` r� r� r� r� r� r� r� r� r� r� r� r� r; rS rT r� r� r� �filterwarningsr� r� r# r# r# r$ rW � s6
rW c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )� TestConstantc � C sV t jd�}t j|dhddid�}t jdddddddddddddddddddddddddddd d
d d d
ddddddddddddddddd d dd d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgddddddddddddddddddddg��}t||� d S )jNrX rZ r[ r
r )r r r5 r( r r& rG r0 rb rc r6 rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r[ )r r1 rL r r )r r' r_ r# r# r$ �test_check_constant� s&
z TestConstant.test_check_constantc � C sR t jd�}t j|dgd�}t jdddddddddddddddddddddddddddddd d
d d d
dddddddddddddddd dd d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfddddddddddddddddddddg��}t||� d S )hNrX rZ r[ r
r r5 r( r r& rG r0 rb rc r6 r rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ �test_check_constant_zeros s&
z&TestConstant.test_check_constant_zerosc C s� t jd�jdd�}t j|d#ddd�}t jdddddddddg dd ddd
d dddg ddd d
dddddg dddddddddg ddddddd ddg dd d d d d!d"ddg dddddddddg dddddddddg g�}t||� d S )$NrQ rG r0 r5 r( r
g�������?)rE r r r r& rb rc r6 r rd r/ re rf rg rh ri rj rk r[ rl rm rn ro rZ rp rq rr rs )r5 r( )r r1 r2 rL r r )r r� rM r r# r# r$ �test_check_constant_float$ s
z&TestConstant.test_check_constant_floatc C s� t jd�jdd�}|jt j�}t j|d)ddd�}t jdddddddddg dd d
d d d
dddg dddddddddg dddddddddg ddd d d d d ddg dd!d"d#d$d%d&ddg dddddddddg dddddddddg g�}t||� d S )*NrQ rG r0 r5 r( r
g�������?)rE r g g �?g @g @g @g @g @g @g @g "@g $@g &@g (@g *@g ,@g .@g 0@g 1@g 2@g 3@g 4@g 5@g 6@g 7@g 8@g 9@g :@g ;@g <@g =@�r5 r( �r5 r( )r� r� )r r1 r2 r^ r� rL r r )r r� Z arr_floatrM r r# r# r$ �test_check_constant_float28 s
z'TestConstant.test_check_constant_float2c � C sZ t jdtd�}t j|dkddnd�}t jdodpdqdrdsdtdudvdwdxdydzd{d|d}d~dd�d�d�d�d�d�d�d�d d
d d d
ddddddddddddddd dd d d d dd!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjd�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�g��}t||� d S )�NrX )r4 rZ r[ r
g�������?g333333�?)r r r5 r( r r& rG r0 rb rc r6 r rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ 皙������333333�)r� r� g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g�������g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�g333333�)r r1 r rL r r )r r' r_ r# r# r$ �test_check_constant_float3M s&
z'TestConstant.test_check_constant_float3c C s� t jd�jdd�}t j|d$ddd�}t jddddddddddg
ddd dddd
dddg
dddd d d
ddddg
ddddddddddg
ddddddddddg
ddd d d d d d!ddg
ddddddddddg
g�}t||� d S )%NrQ rG r0 r5 r( r
r )rE r r r& rb rc r6 r rd r/ re rf rg rh ri rj rk r[ rl rm rn ro rZ rp rq rr rs �r5 �r( )r� r� )r r1 r2 rL r r )r r� rM r r# r# r$ �"test_check_constant_odd_pad_amounte s
z/TestConstant.test_check_constant_odd_pad_amountc C s� t jd�jdd�}t jj|d
dd
d�}t jddddddgddddddgddddddgddddddgddddddgg�}t||� d S )Nr& r( r5 r r
)rE r r �r5 r( �r5 r )r� r� �r5 r( �r r& )r� r� )r r1 r2 �librL r r )r r� rM r r# r# r$ �test_check_constant_pad_2dv s z'TestConstant.test_check_constant_pad_2dc C s� d }t jd|t jd�}t j|dd|j� d�}t jd|t jd�}t||� d
}t jd|t jd�}t j|dd|j� d�}t jd|t jd�}t||� d S )Nr( r� r5 rG )r4 r
)rE r rb r� l l ���� l l ���� )r �fullZuint64rL �minr Zint64)r Z
uint64_maxr� rM r Z int64_maxr# r# r$ �test_check_large_integers� s
z&TestConstant.test_check_large_integersc C sr t jdtd�}t� }||d< t� }t� }t j|dd||fd�}t jdtd�}||d< ||d< ||d< t||� d S ) Nr5 )r4 r r
)� pad_widthrE r r r( )r )r r r rL r )r r� Zobj_aZobj_bZobj_cr r# r# r$ �test_check_object_array� s
z$TestConstant.test_check_object_arrayc C s2 t jd�}t j|d d
d gdd�}|jd ks.t�d S )
Nr r r( r5 r
)rE r& )r r r( )r )r( )r5 )r r& r& )r �zerosrL rK �AssertionError)r r� r! r# r# r$ �test_pad_empty_dimension� s
z%TestConstant.test_pad_empty_dimensionN)
rA rB rC r� r� r� r� r� r� r� r� r� r� r# r# r# r$ r� � s
r� c @ sR e Zd Zdd� Zdd� Zejjefd�dd� �Z dd � Z
ejj d
e �d d � �Z
d
S )�TestLinearRampc � C sb t jd�jd�}t j|d�dd�d�}t jd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dd�d�d�d�d�d�d�d�d�d�d�d�d�d�dUd�d�d�d�d�d�d�d�d�d&g��}t||d�d�d�� d S )�NrX rY rZ r[ r r& rG )r
g @g� ��Q�@gq=
ףp
@g)\��( @g�z�G�
@g������ @gR� ��Q@g
ףp=
@g�(\��@g{�G�z@g333333@g�Q� ��@g�p=
ף @g� ��Q��?g)\��(�?g�������?g
ףp=
�?g{�G�z�?g�Q� ���?g� ��Q��?g�������?g{�G�z�?g� ��Q��?g{�G�z�?g{�G�z�?g g �?g @g @g @g @g @g @g "@g $@g &@g (@g *@g ,@g .@g 0@g 1@g 2@g 3@g 4@g 5@g 6@g 7@g 8@g 9@g :@g ;@g <@g =@g >@g ?@g @@g �@@g A@g �A@g B@g �B@g C@g �C@g D@g �D@g E@g �E@g F@g �F@g G@g �G@g H@g �H@g I@g �I@g J@g �J@g K@g �K@g L@g �L@g M@g �M@g N@g �N@g O@g �O@g P@g @P@g �P@g �P@g Q@g @Q@g �Q@g �Q@g R@g @R@g �R@g �R@g S@g @S@g �S@g �S@g T@g @T@g �T@g �T@g U@g @U@g �U@g �U@g V@g @V@g �V@g �V@g W@g @W@g �W@g �W@g X@g @X@g �X@g �X@g33333�W@gffffffV@g�����9U@g����� T@g �R@g33333�Q@gfffff�P@g33333�N@g�����YL@gfffff�G@g�����LE@g33333�B@g������@@g �<@g������7@g�����3@g������,@gffffff#@g�h㈵��>)ZrtolZatol)rZ r[ )r& rG )r r1 r^ rL r r )r r' r_ r# r# r$ �test_check_simple� s&
z TestLinearRamp.test_check_simplec C s� t jd�jdd�jt j�}t j|d%dd&d�}t jdddddddddg dddd d
d d d
dg dddd
d d
dd dg dddddddddg dddddddddg ddd d d d d d!dg dd"ddd#dd!d$dg dddddddddg g�}t||� d S )'Nr[ r& rG r( r r )rE r
g g �?g �?g �?g @g @g @g @g @g @g @g @g "@g @g $@g &@g (@g *@g ,@g @g .@g 0@g 1@g 2@g 3@g #@g @g !@g @)r( r( )r r )r r1 r2 r^ r� rL r r )r r� rM r r# r# r$ �
test_check_2d� s z TestLinearRamp.test_check_2d)�
exceptionsc
C s� ddl m} tj|dd�|d d�g�}tj|d ddd�}tj|dd�|dd�|d d�|d
d�|dd�|dd�|dd�g�}t||� d S )Nr )�Fractionr5 r( r r )rE r
r/ r0 r& r� )r( r i����r9 r: r )Z fractionsr� r r rL r )r r� r� �actualr r# r# r$ �test_object_array� s z TestLinearRamp.test_object_arrayc C sz t jt jd�jdd�d dd�}t|dd�d f d
� t|dd�d
f d
� t|d dd�f d
� t|ddd�f d
� dS )z!Ensure that end values are exact.r r( rG �� �{ r )rE Nr g r5 )r� r� r� r� )r rL r? r2 r )r r' r# r# r$ �test_end_values� s
z TestLinearRamp.test_end_valuesr4 c C s� t jdg|d�}t j|dddd�}t jdddddddg|d�}t||� t jdg|d�}t j|dddd�}t jdddddddg|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 )r4 r r )rE r
r5 r( N)r r rL r )r r4 r r! r r# r# r$ �test_negative_difference� s
z'TestLinearRamp.test_negative_differenceN)rA rB rC r� r� r; rS Zxfailr� r� r� rT �_numeric_dtypesr� r# r# r# r$ r� � s
r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� TestReflectc � C sR t jd�}t j|dgd�}t jddddddd d
d d d
ddddddddddddddd ddddddddddddddd
d d d
d ddddddd d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdedddcdbdad`d_d^d]d\d[dZdYdXdWdVdUdTdSdRg��}t||� d S )hNrX rZ r[ r ro rn rm rl rk rj ri rh rg rf re r/ rd r r6 rc rb r0 rG r& r r( r5 r rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ r� � s&
z TestReflect.test_check_simplec � C sV t jd�}t j|d|ddd�}t jd}d~dd�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d d d ddddddddddddddd
d d dd
d dddd d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhddidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{g��}t||� d S )�NrX rZ r[ r �odd)r ro rn rm rl rk rj ri rh rg rf re r/ rd r r6 rc rb r0 rG r& r r( r5 r rp rq rr rs rQ rt ru rv rw rx ry 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� ra r� �f �g �h �i �j �k �l �m �n �o �p �q �r �s �t �u �v �w )rZ r[ i��i��i��i��i��i��i��i��i��i�i�i�i�i�i����i����i����i����i����i�������r9 r8 r: r� )r r1 rL r r )r r' r_ r# r# r$ �test_check_odd_method s&
z!TestReflect.test_check_odd_methodc C s� dddgdddgg}t j|dd�}t jdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg�}t||� d S )Nr& rG r0 rb rc r )rG rb )r rL r r )r r' r_ r# r# r$ �test_check_large_pad, s$ &z TestReflect.test_check_large_padc C s� dddgg}t j|dd�}t jdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg
�}t||� d S )Nr& rG r0 rb r )rG rb )r rL r r )r r' r_ r# r# r$ �test_check_shapeC s" &z TestReflect.test_check_shapec C s: t jdddgdd�}t jdddddddg�}t||� d S )Nr5 r( r r )r rL r r )r r' r_ r# r# r$ �
test_check_01Y s zTestReflect.test_check_01c
C s> t jdddgdd�}t jdddddddddg �}t||� d S )Nr5 r( r r )r rL r r )r r' r_ r# r# r$ �
test_check_02^ s zTestReflect.test_check_02c C sB t jdddgdd�}t jdddddddddddg �}t||� d S )Nr5 r( r r& r )r rL r r )r r' r_ r# r# r$ �
test_check_03c s zTestReflect.test_check_03N)
rA rB rC r� r r r r r r r# r# r# r$ r� � s r� c @ sP e Zd ZdZejjdeej � ddh ��dd� �Z
ejjdej � �dd� �Z d S )
�TestEmptyArrayz<Check how padding behaves on arrays with an empty dimension.rE r
r c C s� d}t jt|d�� tjg d|d� W d Q R X t jt|d�� tjtjd�d|d� W d Q R X t jt|d�� tjtjd�d |d� W d Q R X d S ) NzFcan't extend empty axis 0 using modes other than 'constant' or 'empty')r7 r& )rE r r r5 )r r �r5 �r )r r )r; r< r= r rL Zndarrayr� )r rE r7 r# r# r$ r� l s z'TestEmptyArray.test_pad_empty_dimensionc C s( t jt jd�d |d�}|jd
ks$t�d S )Nr( r r r5 )rE rc r& )r( r r( �r �r �r5 )r r r )rc r r& )r rL r? rK r� )r rE r! r# r# r$ � test_pad_non_empty_dimension{ s z+TestEmptyArray.test_pad_non_empty_dimensionN) rA rB rC �__doc__r; rS rT �sortedrU rV r� r r# r# r# r$ r i s r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�
TestSymmetricc � C sR t jd�}t j|dgd�}t jdddddd d
d d d
ddddddddddddddd d ddddddddddddddd
d d d
d ddddddd d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdfdedddcdbdad`d_d^d]d\d[dZdYdXdWdVdUdTdSg��}t||� d S )hNrX rZ r[ r ro rn rm rl rk rj ri rh rg rf re r/ rd r r6 rc rb r0 rG r& r r( r5 r rp rq rr rs rQ rt ru rv rw rx ry 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� ra )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ r� � s&
z TestSymmetric.test_check_simplec � C sV t jd�}t j|d{ddd�}t jd|d}d~dd�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d d d d ddddddddddddddd
d d dd
d dddd d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdhddidjdkdldmdndodpdqdrdsdtdudvdwdxdydzg��}t||� d S )�NrX rZ r[ r r� )r ro rn rm rl rk rj ri rh rg rf re r/ rd r r6 rc rb r0 rG r& r r( r5 r rp rq rr rs rQ rt ru rv rw rx ry 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� ra r� r� r r r r r r r r r r r
r r r
r r )rZ r[ i��i��i��i��i��i��i��i��i�i�i�i�i�i����i����i����i����i����i����r r9 r8 r: r� )r r1 rL r r )r r' r_ r# r# r$ r � s&
z#TestSymmetric.test_check_odd_methodc C s� dddgdddgg}t j|dd�}t jdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg�}t||� d S )Nr& rG r0 rb rc r )rG rb )r rL r r )r r' r_ r# r# r$ r � s$ &z"TestSymmetric.test_check_large_padc C s� dddgdddgg}t j|dddd�}t jddddd d d d
d
d dddddgd d d d d d d d
d
d dddddgd d d d d
d
d ddddddddgd!d d d d
d
d ddddddddgd d
d
d ddddddddd
ddgd d
d
d ddddddddd
ddgd ddddddddd
dddddgd ddddddddd
dddddgddddddd
ddddddddgddddddd
ddddddddgdddd
dddddddddddgdddd
dddddddddddgd
ddddddddddddddgd
ddddddddddddddgg�}t||� d S )"Nr& rG r0 rb rc r r� )r r r( r5 r r6 r rd r/ re rf rg rh ri rj )rG rb r8 r: r: r� r8 r: r: r� r� r� )r rL r r )r r' r_ r# r# r$ �test_check_large_pad_odd� s$ &z&TestSymmetric.test_check_large_pad_oddc C s� dddgg}t j|dd�}t jdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg
�}t||� d S )Nr& rG r0 rb r )rG rb )r rL r r )r r' r_ r# r# r$ r � s" &z TestSymmetric.test_check_shapec C s: t jdddgdd�}t jdddddddg�}t||� d S )Nr5 r( r r )r rL r r )r r' r_ r# r# r$ r � s zTestSymmetric.test_check_01c
C s> t jdddgdd�}t jdddddddddg �}t||� d S )Nr5 r( r r )r rL r r )r r' r_ r# r# r$ r � s zTestSymmetric.test_check_02c C sJ t jdddgdd�}t jdddddddddddddddg�}t||� d S )Nr5 r( r r0 r )r rL r r )r r' r_ r# r# r$ r s (zTestSymmetric.test_check_03N) rA rB rC r� r r r" 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 )�TestWrapc � C sR t jd�}t j|dgd�}t jddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1dd2d3d4d5dd6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1g��}t||� d S )hNrX 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� ra r r5 r( r r& rG r0 rb rc r6 r rd r/ re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rQ rt ru rv rw rx ry 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� )rZ r[ )r r1 rL r r )r r' r_ r# r# r$ r� s&
zTestWrap.test_check_simplec 3 C s� t jd�}t j|d�}t j|dd�}t jddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgd dd
d d dd
d d dd
d d dd
d d dd
d d dd
d d dgd d
ddd d
ddd d
ddd d
ddd d
ddd d
ddd d
gddddddddddddddddddddddddddgg�}t||� d S )Nr/ r r& r r rd rc r6 r( r r5 r0 rb rG )r r& )r r/ )r r1 r2 rL r r )r r' r_ r# r# r$ r sn
"""""""""""""""""""""""""z TestWrap.test_check_large_padc
C s> t jdddgdd�}t jdddddddddg �}t||� d S )Nr5 r( r r )r rL r r )r r' r_ r# r# r$ r \ s zTestWrap.test_check_01c C sB t jdddgdd�}t jdddddddddddg �}t||� d S )Nr5 r( r r& r )r rL r r )r r' r_ r# r# r$ r a s zTestWrap.test_check_02c C s8 t jd�}t j|ddd�}t||d d�d d �f � d S )
Nr rG r r )rE )r rG )r rG r r )r r? rL r )r r' r_ r# r# r$ �test_pad_with_zerof s
zTestWrap.test_pad_with_zeroc C sx t jd�}t j|ddd�}tt j||||f dd� |� t jd�}t j|d dd�}tt j||||f dd
� |� dS ) zy
Check wrapping on each side individually if the wrapped area is longer
than the original array.
rG r/ r r )rE r N)r/ r )r r/ r8 )r r1 rL r Zr_)r r' r_ r# r# r$ �test_repeated_wrappingk s
z TestWrap.test_repeated_wrappingN) rA rB rC r� r r r r$ r% r# r# r# r$ r# s <r# c @ s e Zd Zdd� Zdd� ZdS )�TestEdgec C s� t jd�}t j|d�}t j|dd�}t jddddddddgddddddddgddddddddgddddddddgd d d d d
d d d gd d d d d
dddgd d d d d
dddgd d d d d
dddgd d d d d
dddgg �}t||� d S )Nr/ r& r r( r r r5 rG r0 rb rc r6 r rd )r& r �r( r �r r( )r' r( )r r1 r2 rL r r )r r' r_ r# r# r$ r� z s
zTestEdge.test_check_simplec C s� t jdddg�}t j|d
d�}t jddddddg�}t||� t jdddgdddgg�}t j|d d�}t j|dd�}t||� t jd�jddd�}t j|dd�}t j|dd�}t||� d S )Nr5 r( r r r& rG r0 ro �r5 r( )r) �r5 r( )r* �r5 r( �r5 r( )r+ r, �r5 r( )r- �r5 r( �r5 r( �r5 r( )r. r/ r0 )r r rL r r1 r2 )r r' Zpaddedr r# r# r$ �test_check_width_shape_1_2� s
z#TestEdge.test_check_width_shape_1_2N)rA rB rC r� r1 r# r# r# r$ r& y s r& c @ s e Zd Zdd� Zdd� ZdS )� TestEmptyc C sR t jd�jdd�}t j|d d gdd�}|jd
ks4t�t||dd�dd�f � d S )Nro r& r0 r( r r5 r )rE r6 r )r( r )r r5 )r6 r r8 r� )r r1 r2 rL rK r� r )r r� r! r# r# r$ � test_simple� s zTestEmpty.test_simplec C s2 t jd�}t j|d d
d gdd�}|jd ks.t�d S )
Nr r r( r5 r )rE r& )r r r( )r )r( )r5 )r r& r& )r r� rL rK r� )r r� r! r# r# r$ r� � s
z"TestEmpty.test_pad_empty_dimensionN)rA rB rC r3 r� r# r# r# r$ r2 � s r2 c
C s� dd� } t jd�jdd�}t j|d| �}t jdddddddgdddddddgdddddddgdddd d
ddgdddddddgdddddddgg�}t||� d S ) Nc S s& d| d |d �< d| |d d �< d S )Nr r r5 r# )Zvectorr� Ziaxis�kwargsr# r# r$ � _padwithtens� s z6test_legacy_vector_functionality.<locals>._padwithtensr0 r( r r r r5 r& rG )r r1 r2 rL r r )r5 r' r_ r# r# r$ � test_legacy_vector_functionality� s r6 c C s4 t jdgddd�} t jdddddg�}t| |� d S )Nr5 r( r
)rE r )r rL r r )r' r_ r# r# r$ �test_unicode_mode� s r7 rE r r r r c C s: t jdd d�}d }t jd d d�}tt j||| d�|� d S )
Nr& r )Z
fill_valuer( r6 rc )rE )r& r �r( r �r r( )r8 r9 )r6 rc )r r� r rL )rE r' rN r_ r# r# r$ �test_object_input� s r: c @ s
e Zd Zejjdd d!d%d(g�ejjd
ej� �d d � ��Zejjd
ej� �d
d� �Z ejjdd)d+d-d1d5g�ejjd
ej� �dd� ��Z
ejjdd6d7d8e � dfd9d<e fe dd=�dfdBg�ejjd
ej� �dd� ��Z
dd� Zejjdd dCdFg�ejjd
ej� �dd � ��ZdS )G� TestPadWidthr� r& rG r0 rb r5 r( r r rE c C s@ t jd�jd�}d}tjt|d�� t j|||� W d Q R X d S )NrQ r0 rG z(operands could not be broadcast together)r7 )r0 rG )r r1 r2 r; r< r= rL )r r� rE r� r7 r# r# r$ �test_misshaped_pad_width� s z%TestPadWidth.test_misshaped_pad_widthc
C s@ t jd�jd �}d}tjt|d�� t j|d|� W d Q R X d S )NrQ r0 rG zDinput operand has more dimensions than allowed by the axis remapping)r7 r r& r r5 r( )r0 rG �r �r& �rG �r= r> r? �r �r5 �r( �rA rB rC )r@ rD )r r1 r2 r; r< r= rL )r rE r� r7 r# r# r$ �test_misshaped_pad_width_2� s z'TestPadWidth.test_misshaped_pad_width_2c C s@ t jd�jd�}d}tjt|d�� t j|||� W d Q R X d S )NrQ r0 rG z#index can't contain negative values)r7 )r0 rG )r r1 r2 r; r< r= rL )r r� rE r� r7 r# r# r$ �test_negative_pad_width� s z$TestPadWidth.test_negative_pad_widthzpad_width, dtype�3N�word�333333 @g������ @c C s� t jd�jd�}d}|d k rPtjt|d�� t j|t j||d�|� W d Q R X nVtjt|d�� t j|||� W d Q R X tjt|d�� t j|t j|�|� W d Q R X d S )NrQ r0 rG z%`pad_width` must be of integral type.)r7 )r4 )r0 rG )r r1 r2 r; r< � TypeErrorrL r )r r� r4 rE r� r7 r# r# r$ �
test_bad_type� s $zTestPadWidth.test_bad_typec C s� t jd�}t j|d�}t j|t jd�d�}t jddddddddgddddddddgddddddddgddddddddgd d d d d
d d d gd d d d d
dddgd d d d d
dddgd d d d d
dddgd d d d d
dddgg �}t||� d S )Nr/ r& r r( r r r5 rG r0 rb rc r6 r rd )r& r �r( r �r r( )rL rM )r r1 r2 rL r r )r r' r_ r# r# r$ �test_pad_width_as_ndarray s
z&TestPadWidth.test_pad_width_as_ndarrayc C s, t jd�jdd�}t|t j|||d�� d S )NrQ r0 rG )rE )r r1 r2 r rL )r r� rE r� r# r# r$ �test_zero_pad_width s z TestPadWidth.test_zero_pad_width)r& rG r0 rb �r5 �r( �r )rP rQ rR �r5 r( �r r& �rG r0 )rS rT rU �r r& rG �r r5 r( )rV rW r: r: )r: r� )r r� �rG r( r: �r: r )rX rY r9 �r9 �r( )rZ r[ )rG N)rH N)NN)rI N�r( r r& �r r( )r\ r] r� ������� ��r^ r �r r( �r_ r` )ra N)r r �r r �r r )rb rc )rA rB rC r; rS rT rU rV r< rE rF r r rK rN rO r# r# r# r$ r; � s. r; c C s� t | }i }x"t j� D ]}||kr|j|� qW tjdddgd| f|� xT|j� D ]H\}}dj| �}tjt |d��" tjdddgd| f||i� W dQ R X qRW dS )z1Test behavior of pad's kwargs for the given mode.r5 r( r z+unsupported keyword arguments for mode '{}')r7 N)
rU �values�updater rL �items�formatr; r< r= )rE �allowed� not_allowedr4 �key�valuer7 r# r# r$ � test_kwargs! s
rl c C s0 t jddg�} tt j| d�ddddddg� d S )Nr5 r( r )r r r rL )r� r# r# r$ �test_constant_zero_default2 s rm r5 �constTFc C s> dj | �}tjt|d�� tjdddgd| d� W d Q R X d S )Nzmode '{}' is not supported)r7 r5 r( r r& )rE )rg r; r< r= r rL )rE r7 r# r# r$ �test_unsupported_mode7 s
ro c C s` t jd�jdd�d d d�d d d�f }t j|d| �}|jd ksBt�t|dd
�dd �f |� d S ) Nro r& r0 r( r rb rc )r( r )rb rc r8 r8 )r r1 r2 rL rK r� r )rE r� r! r# r# r$ �test_non_contiguous_array> s &rp c C sP t jd dd�}t j|d| �jd s&t�t jd
dd�}t j|d| �jd sLt�dS ) z5Test if C and F order is preserved for all pad modes.rG r �C)�orderZ C_CONTIGUOUS�FZ F_CONTIGUOUSN)rG r )rG r )r r? rL �flagsr� )rE r r# r# r$ � test_memory_layout_persistenceF s ru r4 c C s0 t jd| d�}t j|d|d�}|j| ks,t�d S )Nr r( r5 )r4 )rE )r r( r5 )r r� rL r4 r� )r4 rE r� r! r# r# r$ �test_dtype_persistenceO s rv )&r r; Znumpyr Z
numpy.testingr r r Znumpy.lib.arraypadr Zsctypesr� rU r rD rW r� r� r� r r! r# r& r2 r6 r7 rS rT r: r; rV rl rm r ro rp ru rv r# r# r# r$ �<module> sT (S l 4Un r'
Q$ |