Edit File: test_arraypad.cpython-37.pyc
B ��Fd�� � @ sl d Z ddlmZmZmZ ddlZddlmZm Z m Z m Z ddl m Z G dd� de �ZG dd � d e �ZG d d � d e �ZG d d � d e �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG d d � d e �ZG d d � d e �ZG d d!� d!e �ZG d"d#� d#e �Z G d$d%� d%e �Z G d&d'� d'e �Z e d(k�rhej �!� dS ))z(Tests for the array padding functions. � )�division�absolute_import�print_functionN)�assert_array_equal� assert_raises�assert_allclose�TestCase)�padc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestConditionalShortcutsc C sb t �d��ddd�}dd� |jD �}ddd d d d d dddg }x |D ]}t|t|||d�� qBW d S )N�x � � � c S s g | ]}d �qS ))r r � )�.0�axisr r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_arraypad.py� <listcomp> s zHTestConditionalShortcuts.test_zero_padding_shortcuts.<locals>.<listcomp>�constant�edge� linear_ramp�maximum�mean�median�minimum�reflect� symmetric�wrap)�mode)�np�arange�reshape�shaper r )�self�test�pad_amt�modesr r r r �test_zero_padding_shortcuts s z4TestConditionalShortcuts.test_zero_padding_shortcutsc C sb t �d��ddd�}dd� |jD �}ddd d g}x,|D ]$}tt||d d �t|||d d�� q6W d S )Nr r r r c S s g | ]}d �qS ))� r( r )r r r r r r s zITestConditionalShortcuts.test_shallow_statistic_range.<locals>.<listcomp>r r r r r )r r( )r � stat_length)r r r! r" r r )r# r$ r% r& r r r r � test_shallow_statistic_range s z5TestConditionalShortcuts.test_shallow_statistic_rangec C s` t �d��dd�}dd� |jD �}dddd g}x,|D ]$}tt|||d �t|||dd �� q4W d S ) N� r r c S s g | ]}d �qS ))� r, r )r r r r r r , s zFTestConditionalShortcuts.test_clip_statistic_range.<locals>.<listcomp>r r r r )r )r r) )r r r! r" r r )r# r$ r% r& r r r r �test_clip_statistic_range* s z2TestConditionalShortcuts.test_clip_statistic_rangeN)�__name__� __module__� __qualname__r' r* r- r r r r r s r c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) � TestStatisticc � C sZ t �d��d�}t|dddd�}t �ddddddddddddddddddddddddddd 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`dadbdcdddedfdgdhdidjdkdjdjdjdjdjdjdjdjdjdjdjdjdjdjdjdjdjdjdjdjg��}t||� d S )lN�d �f))� � 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@)r r �astyper �arrayr )r# �a�br r r �test_check_mean_stat_length8 s& z)TestStatistic.test_check_mean_stat_lengthc � C sP t �d�}t|dd�}t �dddddddddddddddddddddddddddddd 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`dadbdcdddedfdgdddddddddddddddddddddg��}t||� d S )hNr2 )r4 r5 r �c r r( r6 r, r r r � � � � � � � � � � � � � r5 � � � � r4 � � � � r+ � � �! �" �# �$ �% �& �' �( �) �* �+ �, �- �. �/ �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 r8 r )r# r9 r: r r r �test_check_maximum_1P s& z"TestStatistic.test_check_maximum_1c � C sT t �d�d }t|dd�}t �ddddddddddddddddddddddddddddddd 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`dadbdcdddedfdddddddddddddddddddddg��}t||� d S )gNr2 r( )r4 r5 r r6 r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r �test_check_maximum_2h s& z"TestStatistic.test_check_maximum_2c � C sX t �d�d }t|dddd�}t �ddddddddddddddddddddddddddddd d d d d dddddddddddddddd 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`dadbdcdddedfdgdddddddddddddddddddddg��}t||� d S )hNr2 r( )r4 r5 r r@ )r) r6 r, r r r r= r> r? rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r � test_check_maximum_stat_length� s& z,TestStatistic.test_check_maximum_stat_lengthc � C sP t �d�}t|dd�}t �ddddddddddddddddddddddddddddddd 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`dadbdcdddedfdgddddddddddddddddddddg��}t||� d S )hNr2 )r4 r5 r r r( r6 r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r �test_check_minimum_1� s& z"TestStatistic.test_check_minimum_1c � C sT t �d�d }t|dd�}t �ddddddddddddddddddddddddddddddd 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`dadbdcdddeddfddddddddddddddddddddg��}t||� d S )gNr2 r6 )r4 r5 r r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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< �e )r r r r8 r )r# r9 r: r r r �test_check_minimum_2� s& z"TestStatistic.test_check_minimum_2c � C sX t �d�d }t|dddd�}t �ddddddddddddddddddddddddddddd d d d d dddddddddddddddd 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`dadbdcdddedfdgdd_d_d_d_d_d_d_d_d_d_d_d_d_d_d_d_d_d_d_d_g��}t||� d S )hNr2 r( )r4 r5 r r@ )r) r6 r, r r r r= r> r? rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r � test_check_minimum_stat_length� s& z,TestStatistic.test_check_minimum_stat_lengthc � C sV t �d��d�}t|dd�}t �ddddddddddddddddddddddddddddd 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`dadbdcdddedfdgdhdiddddddddddddddddddddg��}t||� d S )jNr2 r3 )r4 r5 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@)r r r7 r r8 r )r# r9 r: r r r �test_check_median� s& z TestStatistic.test_check_medianc C s� t �dddgdddgdddgg�}t|dd�}t �dddddgdddddgdddddgdddddgdddddgg�}t||� d S ) Nr, r( r r r? r> r6 r )r r8 r r )r# r9 r: r r r �test_check_median_01� s " z"TestStatistic.test_check_median_01c C s� t �dddgdddgdddgg�}t|jdd�j}t �dddddgdddddgdddddgdddddgdddddgg�}t||� d S ) Nr, r( r r r? r> r6 r )r r8 r �Tr )r# r9 r: r r r �test_check_median_02 s " z"TestStatistic.test_check_median_02c � C sj t �d��d�}d|d< d|d< t|ddd d �}t �dddddddddddddddddddddddddd ddd 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`dadbdcdddedfdgdhdddidjddddddddddddddddddddg��}t||� d S )kNr2 r3 g @r( g X@r� )r4 r5 r )r, r )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@)r r r7 r r8 r )r# r9 r: r r r � test_check_median_stat_length s* z+TestStatistic.test_check_median_stat_lengthc C s� dddgg}t |dddd�}t�dddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg �}t||� d S )Nr r r )r r= r r6 )r) )r r r8 r )r# r9 r: r r r �test_check_mean_shape_one. s" &z'TestStatistic.test_check_mean_shape_onec � C sV t �d��d�}t|dd�}t �ddddddddddddddddddddddddddddd 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`dadbdcdddedfdgdhdiddddddddddddddddddddg��}t||� d S )jNr2 r3 )r4 r5 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@)r r r7 r r8 r )r# r9 r: r r r �test_check_mean_2D s& z TestStatistic.test_check_mean_2N)r. r/ r0 r; r� r� r� r� r� r� r� r� r� r� r� r� r r r r r1 7 s r1 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� TestConstantc � C sT t �d�}t|dddd�}t �dddddddddddddddddddddddddddd d d d d dddddddddddddddd 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`dadbdcdddedfdgdhdiddddddddddddddddddddg��}t||� d S )jNr2 )r4 r5 r )r@ r5 )�constant_valuesr@ r r( r6 r, r r r r= r> r? rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r �test_check_constant^ s& z TestConstant.test_check_constantc � C sP t �d�}t|dd�}t �ddddddddddddddddddddddddddddddd 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`dadbdcdddedfdgddddddddddddddddddddg��}t||� d S )hNr2 )r4 r5 r r r( r6 r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r �test_check_constant_zerosv s& z&TestConstant.test_check_constant_zerosc C s� t �d��dd�}t|dddd�}t �dddddddddg dd dd d d dddg ddd ddddddg dddddddddg dddddd d ddg dd d d d!d"d#ddg dddddddddg dddddddddg g�}t||� d S )$Nr+ r r )r( r6 r g�������?)r r� r( r r6 r, r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ )r r r! r r8 r )r# �arrr$ �expectedr r r �test_check_constant_float� s z&TestConstant.test_check_constant_floatc C s� t �d��dd�}|�t j�}t|dddd�}t �dddddddddg ddd d d d d ddg dddddddddg dddddddddg dddd d d d ddg dd d!d"d#d$d%ddg dddddddddg dddddddddg g�}t||� d S )&Nr+ r r ))r( r6 )r( r6 r g�������?)r 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 =@)r r r! r7 �float64r r8 r )r# r� Z arr_floatr$ r� r r r �test_check_constant_float2� s z'TestConstant.test_check_constant_float2c � C sX t jdtd�}t|dddd�}t �ddddddddddddddddddddddddddd 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`dadbdcdddedfdgdhdidjdkdldldldldldldldldldldldldldldldldldldldlg��}t||� d S )mNr2 )�dtype)r4 r5 r )g�������g333333�)r� g�������r r( r6 r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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< g333333�)r r �floatr r8 r )r# r9 r: r r r �test_check_constant_float3� s& z'TestConstant.test_check_constant_float3c C s� t �d��dd�}t|dddd�}t �ddddddddddg dddd d dd dddg dddd d dddddg ddddddddddg dddddddd ddg ddd d d d d!d"ddg ddddddddddg g�}t||� d S )#Nr+ r r ))r( )r6 r r, )r r� r r( r6 r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ )r r r! r r8 r )r# r� r$ r� r r r �"test_check_constant_odd_pad_amount� s z/TestConstant.test_check_constant_odd_pad_amountc C s� t �d��dd�}t jj|dddd�}t �ddddddgdd ddddgddddddgddddddgddddddgg�}t||� d S ) Nr r6 ))r( r6 )r( r, r ))r( r6 )r, r )r r� r, r( r )r r r! �libr r8 r )r# r� r$ r� r r r �test_check_constant_pad_2d� s z'TestConstant.test_check_constant_pad_2dN) r. r/ r0 r� r� r� r� r� r� r� r r r r r� ] s r� c @ s e Zd Zdd� Zdd� ZdS )�TestLinearRampc � C s` t �d��d�}t|dddd�}t �ddd d d d d ddddddddddddddd d d d d d!d"d#dd$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�dSd�d�d�d�d�d�d�d�d�d$g��}t||d�d�d�� d S )�Nr2 r3 )r4 r5 r )r r )� end_valuesg @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㈵��>)�rtol�atol)r r r7 r r8 r )r# r9 r: r r r �test_check_simple� s& z TestLinearRamp.test_check_simplec C s� t �d��dd��t j�}t|dddd�}t �dddddddddg 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 )%Nr5 r r )r6 r6 r )r r )r 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! r7 r� r r8 r )r# r� r$ r� r r r � test_check_2d s z TestLinearRamp.test_check_2dN)r. r/ r0 r� r� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestReflectc � C sP t �d�}t|dd�}t �dddddd d d d d ddddddddddddddd d d ddddddddddddddd d d d d dddddd 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`dadbdcdddedfdgdfdedddcdbdad`d_d^d]d\d[dZdYdXdWdVdUdTdSg��}t||� d S )hNr2 )r4 r5 r r4 rM rL rK rJ r5 rI rH rG rF rE rD rC rB rA r@ r? r> r= r r r r, r6 r( r rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r r� s& z TestReflect.test_check_simplec � C sT t �d�}t|dddd�}t �dddd 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`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dd�d�d�dd�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�g��}t||� d S )�Nr2 )r4 r5 r �odd)� reflect_typei��i��i��i��i��i��i��i��i��i�i�i�i�i�i����i����i����i����i����i�������������������r r( r6 r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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� �f �g �h �i �j �k �l �m �n �o �p �q �r �s �t �u �v �w )r r r r8 r )r# r9 r: r r r �test_check_odd_method/ s& z!TestReflect.test_check_odd_methodc C s� dddgdddgg}t |dd�}t�dddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg�}t||� d S )Nr r r r= r> )r r= r )r r r8 r )r# r9 r: r r r �test_check_large_padG s$ &z TestReflect.test_check_large_padc C s� dddgg}t |dd�}t�dddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg �}t||� d S )Nr r r )r r= r )r r r8 r )r# r9 r: r r r �test_check_shape^ s" &z TestReflect.test_check_shapec C s8 t dddgdd�}t�dddddddg�}t||� d S )Nr( r6 r, r )r r r8 r )r# r9 r: r r r � test_check_01t s zTestReflect.test_check_01c C s< t dddgdd�}t�dddddddddg �}t||� d S )Nr( r6 r, r )r r r8 r )r# r9 r: r r r � test_check_02y s zTestReflect.test_check_02c C s@ t dddgdd�}t�dddddddddddg �}t||� d S )Nr( r6 r, r r )r r r8 r )r# r9 r: r r r � test_check_03~ s zTestReflect.test_check_03c C s, t t�d�ddd�}t�d�}t||� d S )N)r r, ))r )r( r )r )r r )r r �zerosr )r# r9 r: r r r �!test_check_padding_an_empty_array� s z-TestReflect.test_check_padding_an_empty_arrayN) r. r/ r0 r� r� r� r� r� r� r� r� r r r r r� s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestSymmetricc � C sP t �d�}t|dd�}t �dddddd d d d d ddddddddddddddd d ddddddddddddddd d d d d dddddd 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`dadbdcdddedfdgdgdfdedddcdbdad`d_d^d]d\d[dZdYdXdWdVdUdTg��}t||� d S )hNr2 )r4 r5 r rM rL rK rJ r5 rI rH rG rF rE rD rC rB rA r@ r? r> r= r r r r, r6 r( r r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r r� � s& z TestSymmetric.test_check_simplec � C sT t �d�}t|dddd�}t �dddd 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�dd�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�g��}t||� d S )�Nr2 )r4 r5 r r� )r� i��i��i��i��i��i��i��i��i�i�i�i�i�i����i����i����i����i����i����r� r� r� r� r� r r( r6 r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs 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 r8 r )r# r9 r: r r r r� � s& z#TestSymmetric.test_check_odd_methodc C s� dddgdddgg}t |dd�}t�dddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg�}t||� d S )Nr r r r= r> )r r= r )r r r8 r )r# r9 r: r r r r� � s$ &z"TestSymmetric.test_check_large_padc C s� dddgdddgg}t |dddd �}t�d d d d d d dddddddddgd d d d d d dddddddddgd d d ddddddddddddgd d d ddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg�}t||� d S )Nr r r r= r> )r r= r r� )r� r� r� r� r r( r6 r, r? r@ rA rB rC rD rE rF rG rH )r r r8 r )r# r9 r: r r r �test_check_large_pad_odd� s$ &z&TestSymmetric.test_check_large_pad_oddc C s� dddgg}t |dd�}t�dddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgg �}t||� d S )Nr r r )r r= r )r r r8 r )r# r9 r: r r r r� � s" &z TestSymmetric.test_check_shapec C s8 t dddgdd�}t�dddddddg�}t||� d S )Nr( r6 r, r )r r r8 r )r# r9 r: r r r r� � s zTestSymmetric.test_check_01c C s< t dddgdd�}t�dddddddddg �}t||� d S )Nr( r6 r, r )r r r8 r )r# r9 r: r r r r� s zTestSymmetric.test_check_02c C sH t dddgdd�}t�dddddddddddddddg�}t||� d S )Nr( r6 r, r r )r r r8 r )r# r9 r: r r r r� s (zTestSymmetric.test_check_03N) r. r/ r0 r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestWrapc � C sP t �d�}t|dd�}t �dddddd 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`dadbdcdddedfdgdddddd d d d d ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0g��}t||� d S )hNr2 )r4 r5 r r~ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r< r r( r6 r, r r r r= r> r? r@ rA rB rC rD rE rF rG rH rI r5 rJ rK rL rM r4 rN rO rP rQ r+ rR rS rT rU rV rW rX rY rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rt ru rv rw rx ry rz r{ r| r} )r r r r8 r )r# r9 r: r r r r� s& zTestWrap.test_check_simplec 4 C s� t �d�}t �|d�}t|dd�}t �ddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgd 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 gd dddd dddd dddd dddd dddd dddd dgddddddddddddddddddddddddddgg�}t||� d S )NrB )r, r )r@ rB r r@ rA r> r? r6 r, r r( r r= r r )r r r! r r8 r )r# r9 r: r r r r� ( sn """""""""""""""""""""""""z TestWrap.test_check_large_padc C s< t dddgdd�}t�dddddddddg �}t||� d S )Nr( r6 r, r )r r r8 r )r# r9 r: r r r r� d s zTestWrap.test_check_01c C s@ t dddgdd�}t�dddddddddddg �}t||� d S )Nr( r6 r, r r )r r r8 r )r# r9 r: r r r r� i s zTestWrap.test_check_02N)r. r/ r0 r� r� r� r� r r r r r� s <r� c @ s e Zd Zdd� ZdS )� TestStatLenc C s0 t �d�}t �|d�}t|dddd�}t �dddddd d d d d g dddddd d d d d g d d d d d dddddg dddddd d d d d g ddddddddddg ddddddddddg dddd dd d d d d g d d d d!d d"d#d$d#d#g dddd dd d d d d g dddd dd d d d d g dddd dd d d d d g g �}t||� d S )%Nr+ )r r ))r6 r, )r, r6 r )r, )r r) r r r= r> r? r( r r6 r, r rA r@ rB rC rD rF rE rG rH rI rJ r5 rK rL rM rN r4 rO rP rQ )r r r! r r8 r )r# r9 r: r r r r� p s z TestStatLen.test_check_simpleN)r. r/ r0 r� r r r r r� o s r� c @ s e Zd Zdd� Zdd� ZdS )�TestEdgec C s� t �d�}t �|d�}t|dd�}t �ddddddddgddddddddgddddddddgddddd d d d gd d d d d d d d gddddddddgddddddddgddddddddgddddddddgg �}t||� d S )NrB )r r, ))r6 r, )r, r6 r r r( r6 r, r r r r= r> r? r@ rA )r r r! r r8 r )r# r9 r: r r r r� � s zTestEdge.test_check_simplec C s� t �dddg�}t|dd�}t �ddddddg�}t||� t �dddgdddgg�}t|dd�}t|d d�}t||� t �d ��ddd�}t|dd�}t|d d�}t||� d S ) Nr( r6 r, ))r( r6 r r r r ))r( r6 )r( r6 rM ))r( r6 )r( r6 )r( r6 )r r8 r r r r! )r# r9 �paddedr� r r r �test_check_width_shape_1_2� s z#TestEdge.test_check_width_shape_1_2N)r. r/ r0 r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestZeroPadWidthc C s< t �d�}t �|d�}x dD ]}t|t||dd�� q W d S )Nr+ )r r )r )r r ))r r )r r r )r )r r r! r r )r# r� � pad_widthr r r �test_zero_pad_width� s z$TestZeroPadWidth.test_zero_pad_widthN)r. r/ r0 r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestLegacyVectorFunctionc C s� dd� }t �d��dd�}t|d|�}t �dddddddgdddddddgdddddddgdddd d ddgdddddddgdddddddgg�}t||� d S ) Nc S s& d| d |d �< d| |d d �< | S )Nr@ r r( r )Zvectorr� �iaxis�kwargsr r r � _padwithtens� s zOTestLegacyVectorFunction.test_legacy_vector_functionality.<locals>._padwithtensr r6 r, r@ r r( r r )r r r! r r8 r )r# r� r9 r: r r r � test_legacy_vector_functionality� s z9TestLegacyVectorFunction.test_legacy_vector_functionalityN)r. r/ r0 r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestNdarrayPadWidthc C s� t �d�}t �|d�}t|t �d�d�}t �ddddddddgddddddddgddddddddgddddd d d d gd d d d d d d d gddddddddgddddddddgddddddddgddddddddgg �}t||� d S )NrB )r r, ))r6 r, )r, r6 r r r( r6 r, r r r r= r> r? r@ rA )r r r! r r8 r )r# r9 r: r r r r� � s z%TestNdarrayPadWidth.test_check_simpleN)r. r/ r0 r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestUnicodeInputc C s8 d}t jdgd|d�}t �dddddg�}t||� d S )Nr r( r6 )r r )r r r8 r )r# Z constant_moder9 r: r r r �test_unicode_mode� s z"TestUnicodeInput.test_unicode_modeN)r. r/ r0 r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � ValueError1c C s8 t �d�}t �|d�}tddd�}ttt|df|� d S )Nr+ )r r r )r, )r r) ))r6 r, )r, r6 )r r )r r r! �dictr � ValueErrorr )r# r� r� r r r r� � s z ValueError1.test_check_simplec C s8 t �d�}t �|d�}tddd�}ttt|df|� d S )Nr+ )r r r )r� )r r) ))r6 r, )r, r6 )r r r! r� r r� r )r# r� r� r r r � test_check_negative_stat_length� s z+ValueError1.test_check_negative_stat_lengthc C s8 t �d�}t �|d�}tddd�}ttt|df|� d S )Nr+ )r r r )r, )r r) ))r� r, )r, r6 )r r r! r� r r� r )r# r� r� r r r � test_check_negative_pad_width� s z)ValueError1.test_check_negative_pad_widthc C sF t ttg ddd� t ttt�d�ddd� t ttt�d�ddd� d S )Nr r )r r )r r, ))r( )r )r r� r r �ndarrayr� )r# r r r �test_check_empty_array� s z"ValueError1.test_check_empty_arrayN)r. r/ r0 r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� ValueError2c C s8 t �d�}t �|d�}tddd�}ttt|df|� d S )Nr+ )r r r )r, )r r) ))r� r, )r, r6 )r r r! r� r r� r )r# r� r� r r r � test_check_negative_pad_amount s z*ValueError2.test_check_negative_pad_amountN)r. r/ r0 r� r r r r r� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � ValueError3c C s* t �d��dd�}ttt|dddd� d S )Nr+ r r r r r� )r r� )r r r! r r� r )r# r� r r r � test_check_kwarg_not_allowed s z(ValueError3.test_check_kwarg_not_allowedc C s$ t �d��dd�}ttt|d� d S )Nr+ r r r )r r r! r � TypeErrorr )r# r� r r r �test_mode_not_set s z ValueError3.test_mode_not_setc C s( t �d��dd�}ttt|ddd� d S )Nr+ r r )r r r r= r )r )r r r! r r� r )r# r� r r r �test_malformed_pad_amount s z%ValueError3.test_malformed_pad_amountc C s( t �d��dd�}ttt|ddd� d S )Nr+ r r ))r, r r )r r( r6 r )r )r r r! r r� r )r# r� r r r �test_malformed_pad_amount2 s z&ValueError3.test_malformed_pad_amount2c C s, t �d��dd�}d}ttt||dd� d S )Nr+ r r )))r, )r )r ))r )r( )r6 r )r )r r r! r r� r )r# r� Z bad_shaper r r �test_pad_too_many_axes! s z"ValueError3.test_pad_too_many_axesN)r. r/ r0 r� r� r� r� r� r r r r r� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TypeError1c C s0 t �d�}ttt|d� ttt|t �d�� d S )Nr+ ))g������ �r, )r, r6 )r r r r� r r8 )r# r� r r r � test_float+ s zTypeError1.test_floatc C s0 t �d�}ttt|d� ttt|t �d�� d S )Nr+ �foo)r r r r� r r8 )r# r� r r r �test_str0 s zTypeError1.test_strc C s. G dd� dt �}t�d�}ttt||� � d S )Nc @ s e Zd ZdS )z&TypeError1.test_object.<locals>.FooBarN)r. r/ r0 r r r r �FooBar6 s r r+ )�objectr r r r� r )r# r r� r r r � test_object5 s zTypeError1.test_objectc C s< t �d�}ttt|tdd�� ttt|t �tdd��� d S )Nr+ r( r� )r r r r� r �complexr8 )r# r� r r r � test_complex; s zTypeError1.test_complexc C s8 t �d�}t �|d�}tddd�}ttt|df|� d S )Nr+ )r r r )r, )r r) ))r6 r, r )r, r6 )r r r! r� r r� r )r# r� r� r r r �test_check_wrong_pad_amount@ s z&TypeError1.test_check_wrong_pad_amountN)r. r/ r0 r� r r r r r r r r r� * s r� �__main__)"�__doc__� __future__r r r �numpyr Z numpy.testingr r r r � numpy.libr r r1 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r. �testing�run_module_suiter r r r �<module> s6 + ( (s `'