HOME


sh-3ll 1.0
DIR:/usr/local/lib64/python3.6/site-packages/numpy/ma/tests/__pycache__/
Upload File :
Current File : //usr/local/lib64/python3.6/site-packages/numpy/ma/tests/__pycache__/test_old_ma.cpython-36.pyc
3

���h~�@s�ddlmZddlZddljjZddljjZddl	m
Z
mZmZddl
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXddlYmZZZej[Z[dfdd�Z\Gd	d
�d
�Z]Gdd�d�Z^Gd
d�d�Z_dd�Z`dS)�)�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||�}|s(td|t|�t|�f�|S)NzNot eq:%s
%s
----%s)r�print�str)�v�w�msg�result�rY�;/tmp/pip-build-5_djhm0z/numpy/numpy/ma/tests/test_old_ma.py�eqs
r[c@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Zd7d8�Zd9S):�TestMac
Cstjddddtddddddddg�}tjddddddddddddg�}d}dd	d	d	d	d	dd	d	d	d	d	g}d	d	dd	d	ddd	d	d	d	dg}t||d
�}t||d
�}tjddddg�}t|d	dd	d	gd
�}	tj|d
|�}
|j}|jd
�|||||||||	|
|f|_dS)Ng�?g@g@g@g$@g@g�r)�maskg�?g�������?g@��x�Dg�g$�g�g�g$�g�)�npr�pirOrCZset_fill_value�d)�self�x�y�a10�m1�m2�xm�ym�z�zm�xf�srYrYrZ�setup$s&"
zTestMa.setupcCs�|j\}}}}}}}}}	}
}tt|��tt|��tt|�|�t|j|�t|j|j�t|jtdd�|��tt|�t	|�tdd�|��tt
||
��tt
t|d�|
��tt
||��dS)NcSs||S)NrY)rcrdrYrYrZ�<lambda>;sz)TestMa.test_testBasic1d.<locals>.<lambda>cSs||S)NrY)rcrdrYrYrZro<sg@��x�D)rarr#rrC�dtype�sizerr�lenr[r)rbrcrdrerfrgrhrirjrkrlrmrYrYrZ�test_testBasic1d3s zTestMa.test_testBasic1dcCs�x�dD]�}|j\}}}}}}}}	}
}}||_||_||_||_||_tt|��tt|��tt|�|�t|j|�t|jtdd�|��tt|�t|�tdd�|��tt	||��tt	t
|d�|��tt	||��|j�qWdS)N����cSs||S)NrY)rcrdrYrYrZroOsz)TestMa.test_testBasic2d.<locals>.<lambda>cSs||S)NrY)rcrdrYrYrZroQsg@��x�D�rtru�rvrw)rxry)rarCrr#rrqrrrrr[rrn)rbrmrcrdrerfrgrhrirjrkrlrYrYrZ�test_testBasic2dAs$
zTestMa.test_testBasic2dcCs�|j\}}}}}}}}}	}
}tddgddgg�}t|ddgddgg�}
tt||||
��tt||||
��tt||||
���xbdD�]X}|j|�}|j|�}|j|�}|j|�}|
j|�}
tt||��tt||||��tt||||��tt||||��tjddd	��tt||||��WdQRXtt||||��tt||||��tt||||��tjddd	��tt||||��WdQRXtt||||��tt||||��tt||||��tt||||��tt|d|d��ttt|�d
t|�d
��tt||||��tttj	||�t	||���tttj
||�t
||���tttj||�t||���tjddd	��"tttj||�t||���WdQRXq�WdS)Nr]rwrrt�rurv�ignore)r�invalidg@�r{�rtru�rwrv)r~rr�)
rarr*rr[�reshaper_�errstate�absr	rIr8r)rbrcrdrerfrgrhrirjrkrlrm�a2d�a2dmrYrYrZ�test_testArithmeticWsF




  zTestMa.test_testArithmeticcCs>tjdg�}tdg�}tt||t��tt||t��dS)Nr])r_rr�
isinstancer)rb�na�marYrYrZ�test_testMixedArithmetic}s
zTestMa.test_testMixedArithmeticcCs�|j\}}}}}}}}}	}
}tttj|�t|���tttj|�t|���tttj|�t|���tttj|�t|���tttj|�t|���tttj	|�t	|���tj
ddd��Ztttjt|��t|���tttj
t|��t
|���tttjt|��t|���WdQRXtttj|�t|���tttj|�t|	���tttj|�t|	���tttj|�t|	���tttj||�t||���tttj|�t|���tttj||�t||���tttj||�t||���tttj||�t||���tttj||�t||���tttj||�t||���tttj||�t||���tttj|�t|���tttj||f�t||f���tttj||f�t||f���tttj||f�t||f���tttj|||f�t|||f���dS)Nr|)rr})rarr[r_rrrDrErLrMr�rHr�r&r'rrrrrrrr;r$r r%r!rr)rbrcrdrerfrgrhrirjrkrlrmrYrYrZ�test_testUfuncs1�s8&   zTestMa.test_testUfuncs1cCs�tddddgddddgd�}tt|�jjtjk�tdt|��tdtd��ttdtddgd���|j	d
�}tt|�jjtjk�tt
t|d�tj��tt|�jjtjk�ttdt|���ttt|d��t
k�ttdd	gt|d���dS)Ngg�?g@g@r]r)r^rurw)rwrw)rrrrp�typer_Zintprr[r�r��ndarrayrr9)rb�ottrYrYrZ�test_xtestCount�s
zTestMa.test_xtestCountcCsb|j\}}}}}}}}}	}
}tj|�}t|�}
ttt|�tj|
���ttt|�t	j|
���dS)N)
rar_r@rr[�maxr6r�minr7)rbrcrdrerfrgrhrirjrkrlrmZxrZxmrrYrYrZ�test_testMinMax�s

zTestMa.test_testMinMaxcCs�|j\}}}}}}}}}	}
}tttjj|�tj|���tttjj|�tj|���ttdttd�dd���ttdttd�dd���tttj|dd�t|dd���tttjt	|d�dd�t|dd���tttj|d�t|d���tttj
|dd�t
|dd���tttj
|d�t
|d���tttj
t	|d�dd�t
|dd���t|�dk�r�tttj||fd�t||fd���tttjj|d�tj|d���tttj|d�t|d���tttj
|d�t
|d���dS)Nrtr)�axisr])
rarr[r_r	r�
accumulaterJrrr>rrr)rbrcrdrerfrgrhrirjrkrlrmrYrYrZ�test_testAddSumProd�s$ &  zTestMa.test_testAddSumProdcCs*tjddddg�}t|ddddgd�}t|ddddgd�}t|�}t|�t|�tttj|�t|dd���tt|d�t|d�k�t|d|dk�t|dtk�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
|�tddddg���tddddgddddg�|dd�<tt	t
|�tddddg���tddddgddddg�|dd�<tt	t
|�tddddg���tt	|tddddg���tjd�d}t
|d�}tt||��tt	tdddddgt�|j��ttd|j��tdd
ddgt�}tjdd
ddgt�}|d}|d}tt|�t�tt|�t�t||�t|dd�jdk�dS)Nr]rwrtrur)r^)�
fill_value��	�cg�?g@Zhello)r)r_rrT�reprrr[rGr�r)rrr*rr4rr^r��objectrrC)rb�x1Zx2Zx3�x4�s1�s2rYrYrZ�test_testCI�sZ""
 
zTestMa.test_testCIcCs.dddddg}t|�}t|�}t||k�t|dd�}t||k	�tjd�}t||d�}t|j|k	�tt||j��t|j|k�t|dd�}t|jj|jjk�t||dd�}t|j|k�t|dt	k�d	|d<t|dt	k	�t|j|k�tt|j
d��t||dd�}	t|	j|k	�t|	dt	k�d	|	d<t|	dt	k	�t|	j|k	�tt|	j
d��t|d
|d�}
tt|
�j|d
jk�td�}t	|d<t
|d�}ttt||g�|��ttt|�ddddddddg��t|ddd
�}
tt|
ddddddddg��t|ddd
�}tt|
|��dS)Nrr]T)�copyr�)r^)r^r�rwr�g�?rt�)r�ru)r�)rwrwrwrw)r(rr_rr�_datar�_maskZ__array_interface__r)r^rrprBr[rrrA)rb�n�mrg�m3r��y1Zy1a�y2Zy2aZy3r�Zy4Zy5Zy6rYrYrZ�test_testCopySizesN

"zTestMa.test_testCopySizecCstd�}dddddg}t|�}|j�}t||d�}t|dtk�t|dtk�ddg|ddg<t|j|k�t|dtk�t|dtk	�tt|ddd	ddg��t||d
d�}|jddd	gddd
g�t|j|k	�t|dtk�t|dtk�tt|ddd
ddg��dS)Nr�rr])r^rurt�
�(rwT)r^r��d�����r�r�)	rr(r�rrr)r�r[r?)rbrar�r�rgrcrYrYrZ�test_testPut5s$zTestMa.test_testPutcCsntd�}t|dddddgd�}tddgddgd�}t|dtk	�t|dtk	�||dd	�<t|dtk�t|dtk	�tt|ddddd	g��td�}t|dddddgd�}|dd	�}tddgddgd�}t|dtk	�t|dtk	�||dd�<t|dtk�t|dtk	�tt|ddg��t|dtk�t|dtk	�tt|ddddd	g��dS)
Nr�r)r^r�r�r]rwrurt)rrrr)r[)rbrarcrjrdrYrYrZ�
test_testPut2Ks,zTestMa.test_testPut2cCsp|j\}}}}}}}}}	}
}ddddddddddddg}tj|�d}
t||
|	�ttt||
dd�|	k��dS)Nr]r)r�)rar_r:r?rr
rK)rbrcrdrerfrgrhrirjrkrlrmr��irYrYrZ�test_testMaPutes
zTestMa.test_testMaPutc

Cs�td�}|jdd�}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��dk�ttt|tt��|jk�t||t�}t|j
|j
k�t|dtk�t|dtk�t|dtk�t|d
tk	�t|dtk	�tt||��t|t|�}t|j
|j
k�t|dtk�t|dtk	�t|dtk	�t|d
tk�t|dtk�t||�}t|j
|j
k�t|dtk�t|dtk	�t|dtk	�t|d
tk�t|dtk�tt||��tdddddg�}tdddddg�}t|d<t|||�}tt|ddd
dd g��t|d<t|||�}tt|ddd
d!d"g��t|dtk�t|dtk	�t|dtk�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�dddddg��tttttd��dd�dddddg��ttttttd��dddddgd�dd�j dddddg��ttttttd��dddddgd�dd�j dddddg��ttttttd��dddddgd�d�j dddddg��ttttdddddgdddddgd�d�j dddddg��tttdddddgdddddg�dddddg��t!d#t"j#d�}t$|jt%�}t||�}tt||��t&|||f�}tt|ddd
d$d%g��t|dtk�t|dtk	�t|dtk�td�}t|d<td�d}t|d<tddddddgddddddgd�}|j'd�}t|||�}t|||�}	tt||	��tt(|	�t)k�tt|	ddddddg��t|td�}tt|ddddddg��t|dt�}tt|ddddddg��dS)&N�rtr�r{r]ry$@r��eg.rur��r�g�?g@g@g@g@rw��)r^r�)rprv�r��2)r]rg����g�r�)r�r�r�g�r�)*rr�Zflatrr[�real�imagrr)rTrrOrCrpr5rr r,r!r-r$r/r%r0r;r1rr+r.�list�ranger2r^r<r_�float32rPrrrrr9)
rbrcrj�cZatestZbtestZctestrd�cmrkrYrYrZ�test_testOddFeaturesls�
 

      (( 
$
zTestMa.test_testOddFeaturescCs�tttdddgdddg�dddg��tttdddgdddg�dddg��td�}td�d}t|d<t|d<ttt||�tt||�||���ttt||�tt||�||���ttj	|�dk�ttj	|�dk�dS)Nr]rwrurtrr�r�)
rr[r7r6rr)rOr$r r)rbrcrdrYrYrZ�test_testMinMax2�s&&""zTestMa.test_testMinMax2cCs&td�}tjd�}t|dd�<|jddd�}|jddd�}tttj|d�t|d���tttj|d
d�t|dd���tttjt	|d�t	|d��t||���tttj
t	|d�t	|d��t
||���td	dd
ddgt�}t|d<t|dddg�}t|dd	k�t|ddk�t|ddk�dS)N�r�rvrwrurtrr]�abc�def)rwrr])rwrr])rwrr])rwrr])
rr_r)r�rr[rNrKr"rr=rr�)rbrcrd�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�jtj�}td�}t|d<|d7}tt||d��dS)Nr�rwr]g�?g@)r�)rr)rr[r<Zastyper_r�)rbrdrcrhrYrYrZ�test_testInplace�sRzTestMa.test_testInplacecCsftd�}t|ddd�<|jdd�}x>tdtjd�D]*}tj||d�}tj|�}tt	||��q4WdS)Nr{rtr�rwrur])�protocol)
rr)r�r�rQ�HIGHEST_PROTOCOL�dumps�loadsrr[)rbrc�protormrdrYrYrZ�test_testPickle$s
zTestMa.test_testPicklecCsHtd�}t|d<ttt�dk�t|dtk�tt|dd�d�dS)Nrvr]z--r)rr)rrTrr)rbZxxrYrYrZ�test_testMasked.s
zTestMa.test_testMaskedc	CsPtddddgddddgd�}ttdt|dd���ttdt|ddddgd	���t|ddddgd
d�\}}ttd|��t|dk�t|dd�<tt|dd�tk�tddddgddddgd�}|jd
d
�}t|dd�df<ttt|dd�ddg��tt|dd�dtk�ttddgt|dd���t|dd
d�\}}tt|ddg��dS)Ngg�?g@g@r]r)r^)r�)�weightsT)r��returnedg@rw)r�r�)rrr[rr)r�)rbr�rXZwtsrYrYrZ�test_testAverage16s zTestMa.test_testAverage1c	Cs0ddddddg}ddddddgddddddgg}td�}ttt|dd�d��ttt|d|d�d��ttd�dtd�g�}ttt|d�tjjtjd��dd	��ttt|dd�tjd�dd��ttt|dd�t|dd�t|dd�dg��ttt|d|d
�d��ttt|d|d�d
dddddg��ttt|dd�t|dd�t|dd�dg��td�}ddddddg}ddddddgddddddgg}t	d�}ddddddg}	tttt
||�dd�d��tttt
||�dd�d��ttt
||�dd�tk�ttt
||	�dd�d
�tt
tt
||�dd��d�t
||�}
ttt|
d�d ��ttt|
dd�d
dddddg��ttt|
dd�ddg��ttt|
d|d�d
dddddg��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|ddg��t|dd�}tt|ddg��t|dd�}tt|d%��t|dd�}tt|ddg��dS)&Nrr]rv)r�g@)r�r�g@g@g(@)r�g4@g@gg�?g@g$@g�X@g@g@ruT)r�r�rw)r�r�r�)r�)r�r�rtg�?g@g�?g������
@g������
@)rwrwru)rwrwru)rwrwru)rwrwrug������@)rrrrrr_r	rrPr<r*r)rrrC�floatr[)rbZw1Zw2rcrdrfrgr�Zm4Zm5rj�a�bZr1Zr2r�r�Za2daZa2dmarYrYrZ�test_testAverage2Isl $   
zTestMa.test_testAverage2cCs�tdttd���tdttd���tdttdggg���tdttdgg���ttttddg��ttttddg��ttttddgddgd��dS)Nr]g�?r)r^)r�intrr�r�	TypeError�
ValueError�bool)rbrYrYrZ�test_testToPython�szTestMa.test_testToPythoncCs�tddd�}tjdd��tdtd�j�WdQRXtd|j�t|j�t|j�tt||�j�tt||�j�t|j�j|j	jk�tddd�}t|j�|j	k�t
t|�tt��dS)Nrr])r^r|)r)
rr_r�rr^r6r7rrpr�rrTr3)rbrhrcrYrYrZ�test_testScalarArithmetic�sz TestMa.test_testScalarArithmeticc	Csvtdddg�}tt|j�|jj���tt|j�|jj���tt|j�|jj���tt|j�|jj���tt|jddddd�|jjddddd���tt|j	dddg�|jj	dddg���tt|j
�|jj
���tt|j�|jj���tddgddgg�}tt|j�|jj���tt|j
�|jj
���tt|jddg�|jjddg���tt|j�|jj���dS)Nr]rurwrrt)rrr[�anyr�r
ZargmaxZargminr�compressZconjr�diagonalrJrKrN)rbr�r�rYrYrZ�test_testArrayMethods�s($zTestMa.test_testArrayMethodscCstdddg�}t|jd�dS)Nr]rurw)rr�ndim)rbr�rYrYrZ�test_testArrayAttributes�szTestMa.test_testArrayAttributescCstdd�ttj�D��dS)NcSs(g|] }|tt�kr|jd�r|�qS)�_)�dirr�
startswith)�.0r�rYrYrZ�
<listcomp>�sz'TestMa.test_testAPI.<locals>.<listcomp>)rr�r_r�)rbrYrYrZ�test_testAPI�szTestMa.test_testAPIcCsZtdddg�}tdddgdddgd�}t|djf�t|djf�t|djf�dS)Nr]rurwr)r^)rrrC)rbr�r�rYrYrZ�test_testSingleElementSubscript�s
z&TestMa.test_testSingleElementSubscriptN)�__name__�
__module__�__qualname__rnrsrzr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rYrYrYrZr\"s8&
23e
2
:	r\c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
TestUfuncscCsZtdddtdgdddgdgdd�tdddtdgdddgdgdd�f|_dS)	Ng�?rr]rwrv)r^r�r�)rr`ra)rbrYrYrZrn�s(zTestUfuncs.setupc*Cs�dddddddddd	d
ddg
}�x�d/D]�}ytt|�}Wntk
rVtt|�}YnXttj|�}|jd|j�}tj��@||kr�tj	d+d,�|d0k�r�tj	d+d-�||�}||�}WdQRXt
t|jd.�|jd.�|��t
t
|j|j��q&WdS)1NrH�arctanhrr�arccoshr&r'r�true_divide�floor_divide�	remainder�fmodrrrDrrLrrErrM�arcsinhr�fabs�negative�floor�ceil�logical_notr	rIr8�hypotrrr;r%r!r$r �logical_and�
logical_or�logical_xorr|)r})rr)*rHr&r'rrrDrrLrrrrErrMr�r�r�rr�r�r�r�r�r	rIr8rr�r�r�r�r�rrr;r%r!r$r r�r�r�)r�r&r')�getattr�umath�AttributeError�fromnumericr_r�raZninr�Zseterrrr[r�eqmaskr^)rbZf_invalid_ignore�fZufZmf�args�ur�mrrYrYrZ�test_testUfuncRegression�sD


z#TestUfuncs.test_testUfuncRegressioncCs\|jd}tt|dd��tt|dd��tt|dd�dd�d�tt|dd�d�dS)Nr)r�ru)rarr
rFrrJr>)rbr�rYrYrZ�test_reduce�s

zTestUfuncs.test_reducecCs�tdd�jdd�}t|dk|�}t|j�|j��t|j�d�t|jd�|jd�kj��t|jd�dddd	gkj��t|jd�dj�t|jd�dj�dS)
Nr]�
rurtr�rrvr�r�)	rr�r5rr�r�rr
r^)rbr�ZamaskrYrYrZ�test_minmax�szTestUfuncs.test_minmaxcCsBx<dD]4}tddddgddddgd�}ttt|�dg��qWdS)Nz?bhilqpBHILQPfdgFDGOr]rrw)r^)rrr[r:)rbr�rcrYrYrZ�test_nonzero�s
zTestUfuncs.test_nonzeroN)r�r�r�rnrr	rrrYrYrYrZr��s
%
r�c@sLeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)�TestArrayMethodsc%Cstjddddddddd	d
ddd
ddddddddddddddddddd d!d"d#d$g$�}|jd%d%�}|jd&d'd'd&�}tjd(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(d(d(d(d)d(d)d(g$�}t||d*�}t||j|j�d*�}t||j|j�d*�}|||||||f|_dS)+Ng� @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-�����?rvrurwrr])�datar^)r_rr�rCra)rbrc�X�XXr��mx�mX�mXXrYrYrZrns$zTestArrayMethods.setupc	Csf|j\}}}}}}}|j�}t|j�|j�j�j��tt|j�|j�t|j|j�dd���dS)Nr)r�)	rar�r�trace�
compressedrJrr[r^)	rbrcrrr�rrrZmXdiagrYrYrZ�
test_traces
zTestArrayMethods.test_tracec	Csh|j\}}}}}}}|jdd�}tt|j|j��tt|j|jdd���tt|j|jjdd���dS)Nrwr�)raZcliprr[r^r�)	rbrcrrr�rrrZclippedrYrYrZ�	test_clips
zTestArrayMethods.test_clipcCs�|j\}}}}}}}|j\}}t|j�|j�j��tj|tj�}	tj|tj�}
x,t|�D] }|dd�|fj�j�|
|<qZWx$t|�D]}||j�j�|	|<q�Wt	t
|jd�|
��t	t
|jd�|	��dS)Nrr])rarCrZptprr_rPZfloat_r�rr[)rbrcrrr�rrrr�Zrows�cols�krYrYrZ�test_ptp%s
 zTestArrayMethods.test_ptpc
CsZ|j\}}}}}}}|jdd�}tt|d|dd�df��|jdd�}	t|	jd�dS)Nrr]rwrur�r�)rwrwruru)raZswapaxesrr[rrC)
rbrcrrr�rrrZ	mXswappedZ
mXXswappedrYrYrZ�
test_swapaxes2s
zTestArrayMethods.test_swapaxesc	Csd|j\}}}}}}}|jd�}tt|j|jd�jd���|jd�}tt|j|jd�jd���dS)Nrr])raZcumprodrr[r�r)	rbrcrrr�rrr�mXcprYrYrZ�test_cumprod9s


zTestArrayMethods.test_cumprodc	Csd|j\}}}}}}}|jd�}tt|j|jd�jd���|jd�}tt|j|jd�jd���dS)Nrr])raZcumsumrr[r�r)	rbrcrrr�rrrrrYrYrZ�test_cumsum@s


zTestArrayMethods.test_cumsumcCs,|j\}}}}}}}tt|jdd�|j�j���tt|jdd�|j�j���tt|jdd�j|jdd�j��tt|j�j|j�j��|jdd�|jdd�}}	x�td�D]t}
tt|	|
||
j�j���tt||
|dd�|
fj�j���tttj	||
�|dd�|
fj�j���q�WdS)N)r�rurr]rv)
rarr[�varrZstdrCr�r_rH)rbrcrrr�rrrZmXvar0ZmXvar1rrYrYrZ�test_varstdGs"&zTestArrayMethods.test_varstdN)r�r�r�rnrrrrrrr rYrYrYrZr
�s
r
cCs,|tkr|tkS|tkr |tkS||kj�S)N)r9r
)rfrgrYrYrZrUs
r)a�	functoolsrZnumpyr_Znumpy.core.umath�corerZnumpy.core.fromnumericrZ
numpy.testingrrrZnumpy.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@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPZnumpy.compatrQr`r[r\r�r
rrYrYrYrZ�<module>s$�5
 AW