HEX
Server: Apache/2.4.52 (Ubuntu)
System: Linux spn-python 5.15.0-89-generic #99-Ubuntu SMP Mon Oct 30 20:42:41 UTC 2023 x86_64
User: arjun (1000)
PHP: 8.1.2-1ubuntu2.20
Disabled: NONE
Upload Files
File: //usr/local/lib/python3.10/dist-packages/numpy/ma/tests/__pycache__/test_old_ma.cpython-310.pyc
o

���g��@s�ddlmZddlZddlZddlmmZddlmm	Z	ddl
mZmZm
Z
ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYddlZm[Z[ej\Z\ddd�Z]Gd	d
�d
�Z^Gdd�d�Z_Gd
d�d�Z`dd�ZadS)�)�reduceN)�assert_�
assert_raises�assert_equal)K�MaskType�MaskedArray�absolute�add�all�allclose�allequal�alltrue�arange�arccos�arcsin�arctan�arctan2�array�average�choose�concatenate�	conjugate�cos�cosh�count�divide�equal�exp�filled�getmask�greater�
greater_equal�inner�
isMaskedArray�less�
less_equal�log�log10�	make_mask�masked�masked_array�masked_equal�masked_greater�masked_greater_equal�
masked_inside�masked_less�masked_less_equal�masked_not_equal�masked_outside�masked_print_option�
masked_values�masked_where�maximum�minimum�multiply�nomask�nonzero�	not_equal�ones�outer�product�put�ravel�repeat�resize�shape�sin�sinh�sometrue�sort�sqrt�subtract�sum�take�tan�tanh�	transpose�where�zeros)�pickle�cCs,t||�}|std|�d|�d|���|S)NzNot eq:�
z
----)r�print)�v�w�msg�result�rY�E/usr/local/lib/python3.10/dist-packages/numpy/ma/tests/test_old_ma.py�eqs
r[c@seZdZdd�Zdd�Zej�dddg�dd	��Zd
d�Z	dd
�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd6d7�Zd8d9�Z d:d;�Z!d<d=�Z"d>d?�Z#d@S)A�TestMacCs�t�ddddtddddddddg�}t�gd	��}d}gd
�}gd�}t||d�}t||d�}t�gd
��}t|gd�d�}	t�|d|�}
|j}|�d�|||||||||	|
|f|_dS)N��?g��@�@�@�$��$@�@)r`�rcr^g���rdrarbr]rdrc)�rrrrrrfrrrrr)rrrfrrrfrfrrrrrf��mask)g�rd��?g�������?�rrfrr�@��x�D)�npr�pirOrC�set_fill_value�d��self�x�y�a10�m1�m2�xm�ym�z�zm�xf�srYrYrZ�setup_method&s&
 zTestMa.setup_methodcCs�|j\}}}}}}}}}	}
}tt|��tt|��tt|�|�t|j|�t|j|j�t|jtdd�|��tt|�t	|�tdd�|��tt
||
��tt
t|d�|
��tt
||��dS)NcS�||S�NrY�rrrsrYrYrZ�<lambda>=�z)TestMa.test_testBasic1d.<locals>.<lambda>cS�||SrrYr�rYrYrZr�>r�rk)rorr#rrC�dtype�sizerr�lenr[rrprYrYrZ�test_testBasic1d5s zTestMa.test_testBasic1dr|���)��cCs�|j\}}}}}}}}	}
}}||_||_||_||_||_tt|��tt|��tt|�|�t|j|�t|jtdd�|��tt|�t|�tdd�|��tt	||��tt	t
|d�|��tt	||��dS)NcSr~rrYr�rYrYrZr�Qr�z)TestMa.test_testBasic2d.<locals>.<lambda>cSr�rrYr�rYrYrZr�Rr�rk)rorCrr#rr�rrr�r[r)rqr|rrrsrtrurvrwrxryrzr{rYrYrZ�test_testBasic2dCs zTestMa.test_testBasic2dcCs*|j\}}}}}}}}}	}
}tddgddgg�}t|ddgddgg�}
tt||||
��tt||||
��tt||||
��dD�]K}|�|�}|�|�}|�|�}|�|�}|
�|�}
tt||��tt||||��tt||||��tt||||��tjddd��tt||||��Wd�n1s�wYtt||||��tt||||��tt||||��tjddd��tt||||��Wd�n1s�wYtt||||��tt||||��tt||||��tt||||��tt|d|d��ttt|�dt|�d��tt||||��ttt�	||�t	||���ttt�
||�t
||���ttt�||�t||���tjddd��ttt�||�t||���Wd�n	1�s�wYqFdS)	Nrfr�rr�))�r�)r�r��ignore�r�invalid�@)
rorr*rr[�reshaperl�errstate�absr	rIr8r)rqrrrsrtrurvrwrxryrzr{r|�a2d�a2dmrYrYrZ�test_testArithmeticWsP





�����zTestMa.test_testArithmeticcCs>t�dg�}tdg�}tt||t��tt||t��dS)Nrf)rlrr�
isinstancer)rq�na�marYrYrZ�test_testMixedArithmetic}s
zTestMa.test_testMixedArithmeticcCs
|j\}}}}}}}}}	}
}ttt�|�t|���ttt�|�t|���ttt�|�t|���ttt�|�t|���ttt�|�t|���ttt�	|�t	|���tj
ddd��2ttt�t|��t|���ttt�
t|��t
|���ttt�t|��t|���Wd�n1s�wYttt�|�t|���ttt�|�t|	���ttt�|�t|	���ttt�|�t|	���ttt�||�t||���ttt�|�t|���ttt�||�t||���ttt�||�t||���ttt�||�t||���ttt�||�t||���ttt�||�t||���ttt�||�t||���ttt�|�t|���ttt�||f�t||f���ttt�||f�t||f���ttt�||f�t||f���ttt�|||f�t|||f���dS)Nr�r�)rorr[rlrrrDrErLrMr�rHr�r&r'rrrrrrrr;r$r r%r!rrrprYrYrZ�test_testUfuncs1�s:�   (zTestMa.test_testUfuncs1cCs�tgd�gd�d�}tt|�jjtju�tdt|��tdtd��ttdtddgd���|�	d�}tt|�jjtju�tt
t|d�tj��tt|�jjtju�ttdt|���ttt|d��t
u�ttddgt|d���dS)	N�rdr]r^rc�rfrrrrgr�rfr)r�r�r�)rrrr��typerl�intprr[r�r��ndarrayrr9)rq�ottrYrYrZ�test_xtestCount�s
zTestMa.test_xtestCountcCsb|j\}}}}}}}}}	}
}t�|�}t|�}
ttt|�t�|
���ttt|�t	�|
���dSr)
rorlr@rr[�maxr6r�minr7)rqrrrsrtrurvrwrxryrzr{r|�xr�xmrrYrYrZ�test_testMinMax�s

zTestMa.test_testMinMaxcCs�|j\}}}}}}}}}	}
}tttj�|�t�|���tttj�|�t�|���ttdttd�dd���ttdttd�dd���tttj|dd�t|dd���tttjt	|d�dd�t|dd���ttt�|d�t|d���tttj
|dd�t|dd���ttt�
|d�t|d���tttj
t	|d�dd�t|dd���t|�dkr�ttt�
||fd�t
||fd���tttj�|d�t�|d���ttt�|d�t|d���ttt�
|d�t|d���dSdS)Nr�r��axisrf)rorr[rlr	r�
accumulaterJrr�prodr>r�rrprYrYrZ�test_testAddSumProd�s* & 
��  �zTestMa.test_testAddSumProdcCs�t�gd��}t|gd�d�}t|gd�d�}t|�}t|�t|�ttt�|�t|dd���tt|d�t|d�u�t|d|dk�t|dtu�tt|d|d��tt|dd	�|dd	���tt|dd�|dd���tt|dd�|dd���d
|d<d
|d<tt||��d|dd�<d|dd�<tt||��t|d<tt||��t|dd�<tt||��||dd�<t|d<tt	t
|�tgd
����tgd�gd��|dd�<tt	t
|�tgd����tgd�gd��|dd�<tt	t
|�tgd����tt	|tgd����t�d	�d}t
|d�}tt||��tt	tgd�t�|j��ttd|j��tgd�t�}t�gd�t�}|d}|d}tt|�t�tt|�t�t||�t|dd�jdk�dS)N)rfr�r�r�r�rg)rrfrrfr)�
fill_valuerfr���	�cr�rj�rfr�r�r�)rrfrfrr]rc)rrrrfr)rf�hellor�r�)r)rlr�str�reprrr[rGr�r)rrr*rr4rrhr��objectrrC)rq�x1�x2�x3�x4�s1�s2rYrYrZ�test_testCI�sZ

zTestMa.test_testCIcCsgd�}t|�}t|�}t||u�t|dd�}t||u�t�d�}t||d�}t|j|u�tt||j��t|j|u�t|dd�}t|jj|jjk�t||dd�}t|j|u�t|dt	u�d	|d<t|dt	u�t|j|u�tt|j
d��t||d
d�}	t|	j|u�t|	dt	u�d	|	d<t|	dt	u�t|	j|u�tt|	j
d��t|d|d�}
tt|
�j|dju�td�}t	|d<t
|d
�}ttt||g�|��ttt|�gd���t|ddd�}
tt|
gd���t|ddd�}tt|
|��dS)N)rrrfrrT)�copyr�rgr�rhr�r�r�rfr]r�)�)rrrfrrrrfr)r�r�r�r�r�)rrrfrfr�r�r�r�)r(rrlrr�_datar�_mask�__array_interface__r)rhrr�rBr[rrrA)rq�n�mrv�m3r��y1�y1a�y2�y2a�y3r��y4�y5�y6rYrYrZ�test_testCopySizesP
�
zTestMa.test_testCopySizecCstd�}gd�}t|�}|��}t||d�}t|dtu�t|dtu�ddg|ddg<t|j|u�t|dtu�t|dtu�tt|gd	���t||d
d�}|�gd�gd
��t|j|u�t|dtu�t|dtu�tt|gd���dS)Nr�)rrrrfrfrgr�r��
�(rf)rr�r����r�Tr�)rrfr�)r��d��)r�r�r�rr)	rr(r�rrr)r�r[r?)rqror�r�rvrrrYrYrZ�test_testPut5s$zTestMa.test_testPutcCsVtd�}t|gd�d�}tddgddgd�}t|dtu�t|d	tu�||dd
�<t|dtu�t|d	tu�tt|gd���td�}t|gd�d�}|dd
�}tddgddgd�}t|dtu�t|d	tu�||dd�<t|dtu�t|dtu�tt|ddg��t|dtu�t|d	tu�tt|gd���dS)Nr�)rrrrrrgr�r�rfrr�r�r�)rrfr�r�r�)rrrr)r[)rqrorrryrsrYrYrZ�
test_testPut2Ks,zTestMa.test_testPut2cCs\|j\}}}}}}}}}	}
}gd�}t�|�d}
t||
|	�ttt||
dd�|	k��dS)N)rfrrrrrrfrrrfrrfrr�)rorlr:r?rr
rK)rqrrrsrtrurvrwrxryrzr{r|r��irYrYrZ�test_testMaPutes
zTestMa.test_testMaPutc
Cs*td�}|�dd�}d|jd<t|ddk�|d|}tt|j|��tt|jd|��tt|t|�jd||��d	|jd
<td�}t|d<tt	|d�t	t�k�|dk}tt
t|tt��d
k�ttt|tt��|jk�t||t�}t|j
|j
u�t|dtu�t|dtu�t|dtu�t|dtu�t|dtu�tt||��t|t|�}t|j
|j
u�t|dtu�t|dtu�t|dtu�t|dtu�t|dtu�t||�}t|j
|j
u�t|dtu�t|dtu�t|dtu�t|dtu�t|dtu�tt||��tgd��}tgd��}t|d<t|||�}tt|gd���t|d
<t|||�}tt|gd���t|d
tu�t|dtu�t|dtu�tttt|d�|�t|d���tttt|d�|�t|d���tttt|d�|�t|d���tttt|d�|�t|d���tttt|d�|�t|d���tttt|d�|�t|d���tttt|d�|�t|d���tttttd��dd�gd���tttttd��dd�gd���ttttttd��gd�d�dd�j gd���ttttttd��gd�d�dd�j gd���ttttttd��gd�d�d�j gd���ttttgd�gd�d�d�j gd���tttgd�gd ��gd!���t!d"t"j#d#�}t$|jt%�}t||�}tt||��t&|||f�}tt|gd���t|d
tu�t|dtu�t|dtu�td$�}t|d<td$�d}t|d<tgd%�gd&�d�}|�'d�}t|||�}t|||�}	tt||	��tt(|	�t)u�tt|	gd'���t|td�}tt|gd(���t|dt�}tt|gd)���dS)*N�r�r�r�)rfry$@r��erd.r�r�r�r�)r]r^rcr_r`)rfrfrfrrr�)r]r^rdre���rf)r��r�r�r�)r�rfr�r�r�)rfrrrrrg)rfrfrfrfr)rrfrrr)rfrfrrrf)rfrrfrr)r�r�rfr�rf)rfrrfrrf)rfrfrrr)rfr�r�r�r�)r�r�r�r�r�)r�r�r�)r�r�)rfrfrfrrr)rfrrrrr)rrfr��r��2)r�r�r�rfrfrf)r�rfrfr�r�r�)*rr��flatrr[�real�imagrr)r�rrOrCr�r5rr r,r!r-r$r/r%r0r;r1rr+r.�list�ranger2rhr<rl�float32rPrrrrr9)
rqrrry�c�atest�btest�ctestrs�cmrzrYrYrZ�test_testOddFeaturesls�
 

 �     ""�������������

zTestMa.test_testOddFeaturescCs�tttgd�gd��gd���tttgd�gd��gd���td�}td�d}t|d<t|d<ttt||�tt||�||���ttt||�tt||�||���tt�	|�dk�tt�	|�d	k�dS)
N)rfr�r�)r�rr�)rfrr�)r�r�r�r�r�r�rr�)
rr[r7r6rr)rOr$r r)rqrrrsrYrYrZ�test_testMinMax2�s  ""zTestMa.test_testMinMax2cCstd�}t�d�}t|dd�<|�ddd�}|�ddd�}ttt�|d�t|d���ttt�|dd�t|dd���ttt�t	|d	�t	|d	��t||���ttt�
t	|d	�t	|d	��t
||���tgd
�t�}t|d<t|gd��}t|d	dk�t|ddk�t|ddk�dS)
N�r�r�r�r�r�)r�rrfrfr)�abcrf�defr�r�)rr�r�r�)
rrlr)r�rr[rNrKr"rr=rr�)rqrrrs�trYrYrZ� test_testTakeTransposeInnerOuter�s&
 ��z'TestMa.test_testTakeTransposeInnerOutercCs�td�}td�}td�}t|d<|d7}tt||d��|d7}tt||d��td�}td�}t|d<|d8}tt||d��|d8}tt||d��td�d}td�d}t|d<|d9}tt||d��|d9}tt||d��td�d}td�}t|d<|d}tt||��|d}tt||��td�d}td�d}t|d<|d}tt||d��|td�}tt|td���td��tj�}td�}t|d<|d7}tt||d��dS)Nr�r�rfr]r^)r�)rr)rr[r<�astyperlr�)rqrsrrrwrYrYrZ�test_testInplace�sRzTestMa.test_testInplacecCsbtd�}t|ddd�<|�dd�}tdtjd�D]}tj||d�}t�|�}tt	||��qdS)Nr�r�r�r�r�rf)�protocol)
rr)r�r�rQ�HIGHEST_PROTOCOL�dumps�loadsrr[)rqrr�protor|rsrYrYrZ�test_testPickle$s
�zTestMa.test_testPicklecCsHtd�}t|d<ttt�dk�t|dtu�tt|dd�d�dS)Nr�rfz--r)rr)rr�rr)rq�xxrYrYrZ�test_testMasked.s
zTestMa.test_testMaskedcCs8tgd�gd�d�}ttdt|dd���ttdt|gd�d���t|gd�d	d
�\}}ttd|��t|dk�t|dd�<tt|dd�tu�tgd�gd�d�}|�dd�}t|dd�d
f<ttt|dd�ddg��tt|d
d�dtu�ttddgt|dd���t|dd	d�\}}tt|ddg��dS)Nr�r�rgr^rr�)r]r]r^r]��weightsT�r��returnedr_r�rfrd�r�rr])rrr[rr)r�)rqr�rX�wtsrYrYrZ�test_testAverage16s zTestMa.test_testAverage1cCs�gd�}gd�gd�g}td�}ttt|dd�d��ttt|d|d�d��ttd�dtd�g�}ttt|d�tj�t�d��d	d
��ttt|dd�t�d�d	d��ttt|dd�t|dd�t|dd�dg��ttt|d|d�d
��ttt|d|d�gd���ttt|dd�t|dd�t|dd�dg��td�}gd�}gd�gd�g}t	d�}gd�}	tttt
||�dd�d��tttt
||�dd�d��ttt
||�dd�tu�ttt
||	�dd�d�tt
tt
||�dd��d�t
||�}
ttt|
d�d
��ttt|
dd�gd���ttt|
dd�ddg��ttt|
d|d�gd���td�}td�d}t||g||ggddd�\}
}tt|
�t|��t|
j|j�tt	d�dddgdd�\}}tt|�t|��tt	d�dd�\}}tt|�t|��tt	d�t	d�dd�\}}tt|�t|�k�tddgddggt�}t
|ddgddgg�}t|dd�}tt|dd	g��t|dd�}tt|dd	g��t|dd�}tt|d ��t|dd�}tt|d!d"g��dS)#N)rrfrfrfrfr)rfrrrrrfr�rr�r�)r�r�r^rcg(@rfr�g������
@)rdr]r^rcr_rb)rrrfrfrr)rrfrfrfrfrfrd)rdr]��X@rr_g@r`)rdr]rrr_rbr�Tr)r�r�r�)r�r�r)rr�r�r�rir]g������@g�?r_)rrrrrrlr	rrPr<r*r)rrrC�floatr[)rq�w1�w2rrrsrurvr��m4�m5ry�a�b�r1�r2r�r��a2da�a2dmarYrYrZ�test_testAverage2Isx�$���
��zTestMa.test_testAverage2cCs�tdttd���tdttd���tdttdggg���tdttdgg���ttttddg��ttttddg��ttttddgddgd��dS)Nrfr]rrg)r�intrrr�	TypeError�
ValueError�bool�rqrYrYrZ�test_testToPython�s zTestMa.test_testToPythoncCs�tddd�}tjdd��tdtd�j�Wd�n1s wYtd|j�t|j�t|j�tt||�j�tt||�j�t|��j|j	ju�tddd�}t|��|j	k�t
t|�tt��dS)Nrrfrgr��r)
rrlr�rrhr6r7rr�r�rr�r3)rqrwrrrYrYrZ�test_testScalarArithmetic�s�z TestMa.test_testScalarArithmeticc
Csptgd��}tt|��|j����tt|��|j����tt|��|j����tt|��|j����tt|�ddddd�|j�ddddd���tt|�	gd��|j�	gd����tt|�
�|j�
���tt|��|j����tddgddgg�}tt|��|j����tt|�
�|j�
���tt|�ddg�|j�ddg���tt|��|j����dS)N�rfr�r�rrfr�r�r��rfrrf)rrr[�anyr�r
�argmax�argminr�compress�conjr�diagonalrJrKrN)rqr
r�rYrYrZ�test_testArrayMethods�s �$$zTestMa.test_testArrayMethodscCstgd��}t|jd�dS)Nrrf)rr�ndim�rqr
rYrYrZ�test_testArrayAttributes�szTestMa.test_testArrayAttributescCstdd�ttj�D��dS)NcSs&g|]}|tt�vr|�d�s|�qS)�_)�dirr�
startswith)�.0r�rYrYrZ�
<listcomp>�s��z'TestMa.test_testAPI.<locals>.<listcomp>)rr&rlr�rrYrYrZ�test_testAPI�szTestMa.test_testAPIcCsTtgd��}tgd�gd�d�}t|djd�t|djd�t|djd�dS)NrrrgrrYrf)rrrC)rqr
rrYrYrZ�test_testSingleElementSubscript�s
z&TestMa.test_testSingleElementSubscriptcCs8tgd�gd�d�}|dk}d||<t|dtu�dS)Nr�)rfrrfrrgr�r�r�)rrr))rqr
r�rYrYrZ�test_assignment_by_condition�sz#TestMa.test_assignment_by_conditioncCsTtddgddgd�}tddgddgd�}|dk}||||<ddg}t|j|�dS)NrrfFrgT)r*rrh)rqr
rrh�
expected_maskrYrYrZ�test_assignment_by_condition_2�sz%TestMa.test_assignment_by_condition_2N)$�__name__�
__module__�__qualname__r}r��pytest�mark�parametrizer�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrr!r$r*r+r,r.rYrYrYrZr\$s@
&
23e
2
:	r\c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
TestUfuncscCsZtdddtdgdddgdgdd�tdddtdgdddgdgdd�f|_dS)Nr]rr�r�rfr�rg)rrmrorrYrYrZr}�s((
�zTestUfuncs.setup_methodc	Cs�gd�}dD]m}ztt|�}Wntytt|�}Ynwttj|�}|jd|j�}t���$||vr;tj	dd�|dvrEtj	dd�||�}||�}Wd�n1sWwYt
t|�d�|�d�|��t
t
|j|j��qdS)N)
rH�arctanhrr�arccoshr6r&r'r�true_divide�floor_divide�	remainder�fmod)*rHr&r'rrrDrrLrrrrErrM�arcsinhr7r6r�fabs�negative�floor�ceil�logical_notr	rIr8rr8r9r:r;�hypotrrr;r%r!r$r �logical_and�
logical_or�logical_xorr�)r�)r6r&r'rr)�getattr�umath�AttributeError�fromnumericrlr�ro�ninr��seterrrr[r�eqmaskrh)rq�f_invalid_ignore�f�uf�mf�args�ur�mrrYrYrZ�test_testUfuncRegression�s(�

��z#TestUfuncs.test_testUfuncRegressioncCs\|jd}tt|dd��tt|dd��tt|dd�dd�d�tt|dd�d�dS)Nrr�r�)rorr
rFrrJr>r#rYrYrZ�test_reduce�s

zTestUfuncs.test_reducecCs�tdd��dd�}t|dk|�}t|��|���t|��d�t|�d�|�d�k���t|�d�gd�k���t|�d�dj�t|�d�dj�dS)Nrf�
r�r�r�r)r�r�r�r�)	rr�r5rr�r�rr
rh)rqr
�amaskrYrYrZ�test_minmax�szTestUfuncs.test_minmaxcCs6dD]}tgd�gd�d�}ttt|�dg��qdS)Nz?bhilqpBHILQPfdgFDGO)rfrr�r)rrrfrfrgr)rrr[r:)rqr�rrrYrYrZ�test_nonzeros�zTestUfuncs.test_nonzeroN)r/r0r1r}rTrUrXrYrYrYrYrZr5�s%
r5c@sLeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)�TestArrayMethodscCs�t�gd��}|�dd�}|�dddd�}t�gd��}t||d�}t||�|j�d�}t||�|j�d�}|||||||f|_dS)N)$g� @g�G�z.@g-���!@g!@g�&1��?g��ʡE�@g\��(� @g��Q�@g{�G�#@gP��n�@gh��|?�!@g!�rh��@gj�t�@gJ+�@g�z�Ga@g!�rh��@g�ʡE��@g-���@g)\��(@g�(\�B#@g+��@gu�V@g�~j��@g�"�~�@g��K7��@g
ףp=J#@g�G�z�@g��Mb�@g}?5^��#@g��/�$@g!�rh�m@g-���#@gJ+�@g���Mb@g��ʡ�@g-�����?r�r�r�)$rrfrrfrrrfrrfrfrrfrrrrfrrfrrrrfrfrfrfrrrfrrrrrfrrfr)�datarh)rlrr�rCro)rqrr�X�XXr��mx�mX�mXXrYrYrZr}szTestArrayMethods.setup_methodc	Csf|j\}}}}}}}|��}t|��|�������tt|��|��t|j|��dd���dS)Nrr�)	ror r�trace�
compressedrJrr[rh)	rqrrr\r]r�r^r_r`�mXdiagrYrYrZ�
test_trace&s
�
�zTestArrayMethods.test_tracec	Csh|j\}}}}}}}|�dd�}tt|j|j��tt|j|�dd���tt|j|j�dd���dS)Nr�r�)ro�cliprr[rhr�)	rqrrr\r]r�r^r_r`�clippedrYrYrZ�	test_clip.s
zTestArrayMethods.test_clipcCs�|j\}}}}}}}|j\}}t|��|�����t�|tj�}	t�|tj�}
t|�D]}|dd�|f����|
|<q,t|�D]}||����|	|<qAt	t
|�d�|
��t	t
|�d�|	��dS�Nrrf)rorCr�ptprbrlrP�float_r�rr[)rqrrr\r]r�r^r_r`r��rows�cols�krYrYrZ�test_ptp5s
zTestArrayMethods.test_ptpc
CsZ|j\}}}}}}}|�dd�}tt|d|dd�df��|�dd�}	t|	jd�dS)Nrrfr�r�)r�r�r�r�)ro�swapaxesrr[rrC)
rqrrr\r]r�r^r_r`�	mXswapped�
mXXswappedrYrYrZ�
test_swapaxesBs
zTestArrayMethods.test_swapaxesc	Csd|j\}}}}}}}|�d�}tt|j|�d��d���|�d�}tt|j|�d��d���dSrh)ro�cumprodrr[r�r�	rqrrr\r]r�r^r_r`�mXcprYrYrZ�test_cumprodI�


 zTestArrayMethods.test_cumprodc	Csd|j\}}}}}}}|�d�}tt|j|�d��d���|�d�}tt|j|�d��d���dSrh)ro�cumsumrr[r�rrtrYrYrZ�test_cumsumPrwzTestArrayMethods.test_cumsumcCs(|j\}}}}}}}tt|jdd�|������tt|jdd�|������tt|jdd�j|jdd�j��tt|��j|��j��|jdd�|jdd�}}	td�D]:}
tt|	|
||
������tt||
|dd�|
f������ttt�	||
�|dd�|
f������qWdS)Nr�r�rrfr�)
rorr[�varrb�stdrCr�rlrH)rqrrr\r]r�r^r_r`�mXvar0�mXvar1rmrYrYrZ�test_varstdWs"&��zTestArrayMethods.test_varstdN)r/r0r1r}rdrgrnrrrvryr~rYrYrYrZrZs
rZcCs,|tur|tuS|tur|tuS||k��Sr)r9r
)rurvrYrYrZrLes
rL)rR)b�	functoolsrr2�numpyrl�numpy.core.umath�corerG�numpy.core.fromnumericrI�
numpy.testingrrr�numpy.marrrr	r
rrr
rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrP�numpy.compatrQrmr[r\r5rZrLrYrYrYrZ�<module>s(�6

.AW