https://t.me/RX1948
Server : Apache
System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64
User : dh_edsupp ( 6597262)
PHP Version : 8.2.26
Disable Function : NONE
Directory :  /lib/python3/dist-packages/numpy/ma/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Current File : //lib/python3/dist-packages/numpy/ma/tests/__pycache__/test_extras.cpython-310.pyc
o

6��a��@s�dZddlZddlZddlZddlZddlmZmZddl	m
Z
mZmZm
Z
ddlmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZm 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;Gdd�d�Z<Gd	d
�d
�Z=Gdd�d�Z>Gd
d�d�Z?Gdd�d�Z@Gdd�d�ZAGdd�d�ZBGdd�d�ZCGdd�d�ZDGdd�d�ZEGdd�d�ZFGdd�d�ZGGdd �d �ZHGd!d"�d"�ZIdS)#z�Tests suite for MaskedArray.
Adapted from the original test_ma by Pierre Gerard-Marchant

:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_extras.py 3473 2007-10-29 15:18:13Z jarrod.millman $

�N)�assert_warns�suppress_warnings)�assert_�assert_array_equal�assert_equal�assert_almost_equal)�array�arange�masked�MaskedArray�masked_array�getmaskarray�shape�nomask�ones�zeros�count)!�
atleast_1d�
atleast_2d�
atleast_3d�mr_�dot�polyfit�cov�corrcoef�median�average�unique�setxor1d�	setdiff1d�union1d�intersect1d�in1d�ediff1d�apply_over_axes�apply_along_axis�compress_nd�compress_rowcols�mask_rowcols�clump_masked�clump_unmasked�flatnotmasked_contiguous�notmasked_contiguous�notmasked_edges�
masked_all�masked_all_like�isin�diagflat�stack�vstackc@sLeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)�TestGenericcCsJtdtd�}tddgddgtd�}t||�t�ddgddgd��}td|d�}td	d	gd
d
g|d�}t||�td|d�}td	d	gd	d	ggd
d
gd
d
gg|d�}t||�t�ddd
dgfg�}td|d�}tddgddg|d�}t||�td|d�}tddgddg|d�}t||�td
|d�}tdggdgg|d�}t||�dS)N����dtype���maskr8�a�b�f��names�formats�rr�r9r9�r6r6�r<r>��bar>��bbr>�r9rC)r.�floatrr�npr8)�self�test�control�dt�rQ�</usr/lib/python3/dist-packages/numpy/ma/tests/test_extras.py�test_masked_all%s6

�
�
�
zTestGeneric.test_masked_allcCs�t�ddtfgdffg�}tj�d|�}tt|d�tjjj�tt|dd�tjjj�tt	|dd�d�t|ddj
d�t|ddjj
d�dS)Nr=�c�r9r9rCrQ�rLr8�object�mar.r�type�corer�lenr�_fill_value�rM�my_dtype�
masked_arrrQrQrR�"test_masked_all_with_object_nestedCsz.TestGeneric.test_masked_all_with_object_nestedcCsrt�dtdffg�}tj�d|�}tt|d�tjjj�tt	|d�d�t|dj
d�t|djj
d�dS)Nr=rUr9rCrQrVr]rQrQrR�test_masked_all_with_objectOsz'TestGeneric.test_masked_all_with_objectcCs�tddgtd�}t|�}tddgddgtd�}t||�t�ddgddgd��}td	d	gd
d
g|d�}t|�}tddgd
d
g|d�}t||�t�ddd
dgfg�}tddgddg|d�}t|�}t||�dS)Nr9r6r7r:r<r=r>r?rBrC)�
rbrErFrHrJ)rrKr/rrLr8)rM�baserNrOrPrQrQrR�test_masked_all_likeYs

�z TestGeneric.test_masked_all_likec	Cs�tdd�D]W}td|�D]N}tj|td�}tj||td�}td|�}|d|@dk|_d}||�D]}||j|��7}q3|t	krMt
|����|�q
|j|_t
|����|�q
qdS)Nr9�r6r7r)�rangerLr	�int�fullrr;�data�sumr*r�
compressed)	rMr>�i�j�k�jar<�s�slrQrQrR�check_clumpms
��zTestGeneric.check_clumpcCsVtt�d��}t|gd�<t|�}tdd�tdd�tdd�g}t||�|�t�dS)Nrb�rr9r6���	r�rtreru)rrLr	r
r)�slicerrr�rMr<rNrOrQrQrR�test_clump_masked}s
zTestGeneric.test_clump_maskedcCsNtt�d��}t|gd�<t|�}tdd�tdd�g}t||�|�t�dS)Nrbrsrwrtreru)rrLr	r
r*rxrrrryrQrQrR�test_clump_unmasked�s
zTestGeneric.test_clump_unmaskedcCs�td�}t|�}t|td|j�g�tjdtd�|_t|td|j�g�t	||dk|dkB|dkB<t|�}t|tdd�tdd�g�t	|dd�<t|�}t|g�dS)	Nrbrr7rwru�rtrv)
r	r+rrx�sizerLr�boolr;r
)rMr<rNrQrQrR�test_flatnotmasked_contiguous�sz)TestGeneric.test_flatnotmasked_contiguousN)�__name__�
__module__�__qualname__rSr`rardrrrzr{rrQrQrQrRr4#s

r4c@s<eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�TestAveragecCs$tgd�gd�d�}tdt|dd��tdt|gd�d��t|gd�d	d
�\}}td|�t|dk�t|dd�<tt|dd�jd	g�tgd�gd�d�}|�dd�}t|dd�d
f<tt|dd�ddg�tt|d
d�jdd	g�tddgt|dd��t|dd	d�\}}t|ddg�dS)N)���?�@�@)TFFF�r;r�r��axis)r�r�r�r���weightsT�r��returned�@r6r9r��r�r�r�)rrrrr
r;�reshape)rM�ott�result�wtsrQrQrR�test_testAverage1�s 
zTestAverage.test_testAverage1cCsBgd�}gd�gd�g}tdtjd�}tt|dd�d�tt|d|d�d�ttdtjd�d	td�g�}tt|d�tj�t�d��d
d�tt|dd�t�d�d
d	�tt|dd�t|dd�t|dd�d	g�tt|d|d
�d�tt|d|d�gd��tt|dd�t|dd�t|dd�d	g�td�}gd�}gd�gd�g}t	d�}gd�}	ttt
||�dd�d�ttt
||�dd�d�ttt
||�dd�jdg�ttt
||	�dd�d�tttt
||�dd��d�t
||�}
tt|
d�d�tt|
dd�gd��tt|
dd�ddg�tt|
d|d�gd��dS)N�rr9r9r9r9r�r9rrrrr9rtr7rr��@)r�r�r�r��(@r9r�g������
@)r�r�r�r�r��$@)rrr9r9rr)rr9r9r9r9r9Tr�)r�r���X@r�r�g@�@)r�r�r�r�r�r�)
r	rL�float_rrr�add�reducerrrr;r)rM�w1�w2�x�y�m1�m2�m3�m4�m5�zrQrQrR�test_testAverage2�sF& ���
�zTestAverage.test_testAverage2cCsftd�}td�d}t||g||ggddd�\}}tt|�t|��t|j|j�ttd�dddgdd�\}}tt|�t|��ttd�dd	�\}}tt|�t|��ttd�td�dd
�\}}tt|�t|��tddgddggt�}t|d
d
gdd
gg�}t|dd�}	t|	ddg�t|dd�}
t|
ddg�t|dd�}
t|
d�t|dd�}
t|
ddg�dS)Nrtrwr9Tr�)r6r6rwr�r�r�r�)r�r�r6�Fr���?r�r�g������@��?r�)r	rrrrrrKr)rMr<r=�r1r��r2r��a2d�a2dm�a2da�a2dmarQrQrR�test_testAverage3�s*
zTestAverage.test_testAverage3cCs@ttddg��}t|d�ttgd�gd�d��}t|d�dS)Nr9r6r�)r9r6rwr�)FFTTr�)rrr�rMr<rQrQrR�test_onintegers_with_mask�s
z%TestAverage.test_onintegers_with_maskc
Cs�tjgd�gd�gtd�}tgd�gd�g|d�}t|�}t�|���}t|j|j�t|j|j�t|dd�}t|jdd�t|jdd�d	}t|j|j�t|j|j�t|d
d�}t|jd
d�t|jd
d�d	}t|j|j�t|j|j�t�gd�gd�g�}	t||	d
�}
tj|��|	|d
�}t|
j|j�t|
j|j�t||	dd�}t|j|	dd�t|j|	dd�d	}t|j|j�t|j|j�t||	d
d�}t|j|	d
d�t|j|	d
d�d	}t|j|j�t|j|j�dS)N)rrrr9r�rr9rrrr7)ry�?@y@@y@@y@ @)y"@��?y@@y@@y@@r�rr�r�r9)r�r�r�r�r�)r�r�r�r�r�r�)r�r�)	rLrr~rrrkr�real�imag)
rMr;r<�av�expected�av0�	expected0�av1�	expected1r��wav�wav0�wav1rQrQrR�test_complex�sR
����  
���zTestAverage.test_complexcCs�tjjt�d��dd�gd�gd�gd�gd�}tgd�dd�}tgd�gd�d�}t|d|dd	�}t�gd
��}t||�t|d|dd	�}t�gd��}t||�dS)Nrvrw�r9rr�rrrr�)r|��Frr�)�@g�@g�@)r�g0��N@g0��N@)rLrXrr	r�rrr)rMr<�weights_unmasked�weights_masked�avg_unmasked�expected_unmasked�
avg_masked�expected_maskedrQrQrR�test_masked_weights's��
zTestAverage.test_masked_weightsN)	r�r�r�r�r�r�r�r�r�rQrQrQrRr��s",r�c@�$eZdZdd�Zdd�Zdd�ZdS)�TestConcatenatorcCszttdtgd���td�}gd�}t||d�}t|dd|f}tt|t��t|gd��t|jt|dd|f�dS)N�r9r6rwr�r|rtr|)r9rrrrr�r)r9r9r9r9r9rrr9r9r9r9r9)	rrrrrr�
isinstancerr;)rMr=�m�drTrQrQrR�test_1d<szTestConcatenator.test_1dcCs0tj�dd�}tj�dd�}t�tj�dd�d�}t�tj�dd�d�}t||d�}t||d�}td||f}t|jdk�t|dd�dd�f|�t|dd�dd�f|�t|j	tj
d||f�t||f}t|jdk�t|dd�dd�f|�t|dd�dd�f|�t|j	tj
||f�dS)Nr|rr��1)r|rb)rbr|)rL�random�rand�round_rrrrrr;�r_)rM�a_1�a_2�m_1�m_2�b_1�b_2r�rQrQrR�test_2dGs zTestConcatenator.test_2dcCspttjjdf}t|jddg�t|jdd�tddgtjjf}t|jgd��t|jdd�ddg�dS)Nr9TFr6)FFT)rrLrXr
rr;ri)rM�actualrQrQrR�test_masked_constant[sz%TestConcatenator.test_masked_constantN)r�r�r�r�r�r�rQrQrQrRr�9sr�c@�eZdZdd�Zdd�ZdS)�
TestNotMaskedcCs�tt�d��dd�gd�gd�gd�gd�gd�gd�}t|d�}t|d	d
g�t|d	�}t|d	ddg�t|dd
dg�t|d�}t|d	ddg�t|dddg�t|jd�}t|d	d
g�t|jd	�}t|d	ddg�t|dddg�t|jd�}t|d	ddg�t|dddg�t|d<t|d	�}t|d	ddg�t|dddg�t|d�}t|d	ddg�t|dddg�dS)N�r|)rrr9rr)rrrr9r9)r9r9rrr)rrrrr)r9r9r9rrr�r��rr9r6rwr�r9)rwrwrwr�r�)rrr6rrw)r�r6r�r�r�)r�r�r�r�r�������)r9r9r6r�r�)rr9r6r�)rrr6rw)r�r6r�r�)rrLr	r�r-rrir
�rMrirNrQrQrR�
test_edgeshs>��




zTestNotMasked.test_edgesc
Cs,tt�d��dd�gd�gd�gd�gd�}t|d�}t|tdd	d�td
dd�tddd�g�t|d�}t|tdd
d�tddd�gtdd
d�tddd�gtdd
d�tddd�gtdd
d�tddd�gtddd�gtddd�ggtddd�gg�t|d
�}t|tdd	d�ggtddd�tddd�gg�dS)Nr�rwru)rrrrr9r9r9r9)r9r9r9r9r9r9r9r9)rrrrrrr9rr�rr����r9r6rtre)rrLr	r�r,rrx)rMr<�tmprQrQrR�test_contiguous�s:��



�
�

�zTestNotMasked.test_contiguousN)r�r�r�r�r�rQrQrQrRr�es"r�c@s|eZdZdd�Zdd�Zdd�Zej�dgd��ej�d	d
ge	j
jdfe	j
jdfg�d
d���Z
dd�Zdd�Zdd�ZdS)�TestCompressFunctionsc	Cs(t�ttd����ddd�}t�d��t�}d|d<t||d�}t|�}t	|gd	�gd
�gd�ggd�gd
�gd�gg�t|d�}t	|gd�gd�gd�gd�ggd�gd�gd�gd�gg�t|d�}t	|gd�gd�gd�ggd�gd�gd�ggd�gd�gd�gg�t|d�}t|d�}t|d�}t	||�t	||�t	||�t|d�}t	|gd	�gd �gd
�gd�ggd!�gd"�gd#�gd$�ggd�gd%�gd
�gd�gg�t|d&�}t|d'�}t|d(�}t	||�t	||�t	||�t|d)�}t	|gd�gd�gd�ggd�gd�gd�gg�t|d*�}t	||�t|d+�}t	|gd	�gd
�gd�ggd!�gd#�gd$�ggd�gd
�gd�gg�t|d,�}t|d-�}t|d.�}t	||�t	||�t	||�t|d/�}t	|gd	�gd �gd
�gd�ggd�gd%�gd
�gd�gg�t|d0�}t	||�dS)1N�<rwr�r|�rwr�r|T�r9r9r9r�)rr6rwr�)rb��
�)����)�(�*�+�,)�2�4�5�6)�7�9�:�;rr�)r|rtrerurv)rb�r�r�r�)r�r�r�r�r�)r�)rrr)�-�.�/�0�1)r�3rrr)r�8r	r
rr9)��r�r�r�)�r�� �!�")�#�$�%�&�'rUr�)r�r6)r|rerurv)rr�r�r�)r��r��)rrrr)rrrr)rrrrr5r�)r�)rr9)rr��r9r6)r�r6)r9r�)r�r��rr6)rr�)
rLr�listrfr�r�astyper~r&r)rMr�r�r<�a2�a3�a4rQrQrR�test_compress_nd�s�
���

���


����








����







���



����








���
	z&TestCompressFunctions.test_compress_ndcCs�tt�d��dd�gd�gd�gd�gd�}tt|�ddgdd	gg�tt|d
�gd�gd�g�tt|d
�d
dgddgdd	gg�t|jgd�gd�gd�gd�}tt|�d
dgdd	gg�tt|d
�gd�gd�g�tt|d
�d
dgddgdd	gg�t|jgd�gd�gd�gd�}tt|�d	gg�tt|d
�gd�g�tt|d
�dgdgd	gg�t|jgd�gd�gd�gd�}tt|�jd
�tt|d
�jd
�tt|d
�jd
�dS)Nrvrwr�r�r�r�r|rerurr�)rtrerur9r6�rr9rrt�rr9r6�rrr9)rrLr	r�rr'�_datar}�rMr�rQrQrR�test_compress_rowcolss$�" "  z+TestCompressFunctions.test_compress_rowcolscCsHtt�d��dd�gd�gd�gd�gd�}tt|�jgd�gd�gd�g�tt|d�jgd�gd�gd�g�tt|d�jgd�gd�gd�g�t|jgd�gd	�gd�gd�}tt|�jgd	�gd�gd	�g�tt|d�jgd�gd�gd�g�tt|d�jgd	�gd	�gd	�g�t|jgd�gd	�gd�gd�}tt|�jgd�gd�gd
�g�tt|d�jgd�gd�gd�g�tt|d�jgd
�gd
�gd
�g�t|jgd�gd	�gd�gd�}tt|��	�t
u�tt|d��	�t
u�tt|d��	�t
u�tt|�j�	��tt|d�j�	��tt|d�j�	��dS)Nrvrwr�r�r�r�rr9r*)r9r9rr,)rrLr	r�rr(r;r-r�allr
r.rQrQrR�test_mask_rowcols1sN�
��� 
��� 
��� z'TestCompressFunctions.test_mask_rowcolsr�)Nrr9�func�rowcols_axisrr9cCsvtt�d��dd�gd�gd�gd�gd�}tt��|||d�}t|t||��Wd�dS1s4wYdS)Nrvrwr�r�r�r�)rrLr	r�r�DeprecationWarningrr()rMr�r2r3r��resrQrQrR�#test_mask_row_cols_axis_deprecationQs�
"�z9TestCompressFunctions.test_mask_row_cols_axis_deprecationcCs�t�dd�}gd�}t||d��dd�}t||d��dd�}t||dd�}t|jddgdd	gg�t||dd�}t|jgd
�gd�gd�g�t||dd�}t|t�|�d	�|�d	���t||dd�}t|t�|�d	�|�d	���gd
�}t||d��dd�}t||d��dd�}t||dd�}t|jd	dgddgg�t||dd�}t|jgd�gd�gd
�g�t||dd�}t|t�|�d	�|�d	���t|t||��t||dd�}t|t�|�d	�|�d	���gd�}t||d��dd�}t||d��dd�}t||�}t|jt�t||�}t|jt�t|gd�d��dd�}t|gd�d��dd�}t||dd�}t|jddgd	d	gg�t||dd�}t|t�|�d	�|�d	���t||dd�}t|jgd�gd�gd�g�t||dd�}t|t�|�d	�|�d	���t|gd
�d��dd�}t|gd�d��dd�}t||dd�}t|jd	d	gddgg�t||�}t|t�|�d	�|�d	���t||dd�}t|jgd�gd�gd�g�t||dd�}t|t�|�d	�|�d	���t|gd
�d��dd�}t|gd�d��dd�}t||dd�}t|jdd	gddgg�t||dd�}t|t�|�d	�|�d	���t||dd�}t|jgd�gd
�gd�g�t||dd�}t|t�|�d	�|�d	���dS)Nr9re)r9rrrrrr�r6rwT)�strictrr�r�F�rrrrrr9r,)rrrrrr)rrr9rrr)	rLr	rr�rrr;�filledr)rM�nr�r<r=rTrQrQrR�test_dot]sz


"zTestCompressFunctions.test_dotcCsnt�d�}t|�}ttt||��tu�ttt||��tu�ttt||��tu�ttt||��tu�dS)Nrw)rL�eyerrrYrr�rMr<r=rQrQrR�test_dot_returns_maskedarray�s
z2TestCompressFunctions.test_dot_returns_maskedarraycCsDtt�d��}tt�d��}t|||d�}t||u�t||�dS)Nrw)rwrw��out)rrLr<rrrr)rMr<r@r5rQrQrR�test_dot_out�s
z"TestCompressFunctions.test_dot_outN)r�r�r�r)r/r1�pytest�mark�parametrizerLrX�	mask_rows�	mask_colsr6r;r>rArQrQrQrRr��sp �	F	r�c@r�)�TestApplyAlongAxiscCs@td��ddd�}dd�}t|d|�}t|ddgdd	gg�dS)
Nr�r6rwcSs|dS�Nr9rQ)r=rQrQrR�myfunc�sz*TestApplyAlongAxis.test_3d.<locals>.myfuncr9r�rerb�r	r�r%r�rMr<rI�xarQrQrR�test_3d�szTestApplyAlongAxis.test_3dcCsFtd��ddd�}ddd�}t|d|dd�}t|dd	gd
dgg�dS)
Nr�r6rwrcSs|d|SrHrQ)r=�offsetrQrQrRrI�sz1TestApplyAlongAxis.test_3d_kwargs.<locals>.myfuncr9)rNr|rur)rrJrKrQrQrR�test_3d_kwargs�s
z!TestApplyAlongAxis.test_3d_kwargsN)r�r�r�rMrOrQrQrQrRrG��
rGc@seZdZdd�ZdS)�TestApplyOverAxescCs�td��ddd�}ttj|ddg�}t�dgdgdggg�}t||�t||d�t	�<ttj|ddg�}t�d	gd
gdggg�}t||�dS)Nr�r6rwr�rr��\�|r�r)
r	r�r$rLrjrrr
r%r~)rMr<rN�ctrlrQrQrR�
test_basic�s
zTestApplyOverAxes.test_basicN)r�r�r�rUrQrQrQrRrQ�srQc@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-S).�
TestMediancCs4tjjtjtjgtjtjggdd�}t|tj�dS)Nr�r�)rLrXr�infr�rM�rrQrQrR�test_pytype�s$zTestMedian.test_pytypecCs�tjjtj�tjtjgtjtjgg�dd�}t|tj�tjjtj�tjtjgtjtjgg�dd�}t|tj�tjjtjjtjtjgtjtjggdd�dd�}t|jd�tjjtjjtjtjgtjtjggdd�dd�}t|jd�dS)Nr�r�Tr�)rLrXrrrWrr;rXrQrQrR�test_inf�s8
��
��
���
���zTestMedian.test_infcCslt�d�}ttj�|�d�tttj�|��tu�td�}ttj�|�d�tttj�|��tu�d}ttj�|�d�tttj�|��tu�t�d��	dd�}ttjj|dd	�tj|dd	��ttjj|d
d	�tj|d
d	��ttjj|d
d	�tu�t�d��	dd�}ttjj|dd	�tj|dd	��ttjj|d
d	�tj|d
d	��ttjj|d
d	�tu�dS)Nrvr�rug@r|r��Hrr�r9gR@)
rLr	rrXrrrYrrfr�r.rQrQrR�test_non_masked�s"
    zTestMedian.test_non_maskedcCstt�d�dgddgdd�}ttj�|�d�ttj�|�jdd�tttj�|��t	u�tt�d	��
d
d�dgddgdd�}ttj�|�d
�ttj�|�jdd�tttj�|��t	u�tjj|ddd�}t|ddg�t|jdd�tt|�t	u�dS)z5test the examples given in the docstring of ma.medianrurr�r9r�r�rQ�shape mismatchrbr6r|rtr�r�T�r��overwrite_inputr�r�r5N)rrLr	rrXrrrrYrr�)rMr��ma_xrQrQrR�test_docstring_exampless"*z"TestMedian.test_docstring_examplescCs�d}td�D]p}dD]k}td||d�}t�t||�ddg�}|D] \}}ztjj|||d�Wq!tyAt|||||f��wt�|d|gddg�}|D]#\}}ztjj|||d�Wn
tj	yjYqQwt|||||f��q
qdS)	Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sr|)FTr9)�ndminr;FTr_)
rfr�	itertools�productrLrXr�	Exception�AssertionError�	AxisError)rM�msgrcr;r��argsr��overrQrQrR�test_axis_argument_errorss,�����z$TestMedian.test_axis_argument_errorscCsDtddd�}ttj�|�d�tddd�}ttj�|�tjj�dS)Nr9Fr�T)rrrLrXrr
r.rQrQrR�test_masked_0d*szTestMedian.test_masked_0dcCs�tt�d�dd�}ttj�|�tjj�ttj�|�jdd�tt	tj�|��tjj
ju�tt�d�dd�}ttj�|�d�ttj�|�jdd�tt	tj�|��tu�tt�d�gd�d�}ttj�|�d	�ttj�|�jdd�tt	tj�|��tu�tt�d�gd
�d�}ttj�|�d�ttj�|�jdd�tt	tj�|��tu�tt�d�gd�d�}ttj�|�d
�ttj�|�jdd�tt	tj�|��tu�tt�d�gd�d�}ttj�|�d
�ttj�|�jdd�tt	tj�|��tu�tt�d�gd�d�}ttj�|�d	�ttj�|�jdd�tt	tj�|��tu�tt�d�gd�d�}ttj�|�d	�ttj�|�jdd�tt	tj�|��tu�dS)Nr|Tr�rQr^Fr�r�r�)rr9r9r9r9r�)rr9r9rrr�r�rtr�r�)
rrLr	rrXrr
rrrYrZ�MaskedConstantrr.rQrQrR�test_masked_1d1s@zTestMedian.test_masked_1dcCsBttj�tgd�gd�d��jtj�tgd�gd�d��j�dS)N�r9r6rwr�r�r*)rrLrXrrr�rMrQrQrR�test_1d_shape_consistencyWs�z$TestMedian.test_1d_shape_consistencycCs�d\}}tt�dd|��}t|dd�<|dd�<ttj||ftd��}|dd�|dd�df<t�t|��}td|�D]}tj	�
|�|||dd�|f<q<tt|dd�df�d�tt|�d�tt|dd	�t�
|��tt|jdd	�t�
|��dS)
N)�er��r�rb���r7rr9r�)rrL�linspacer
�emptyrKr	r[rfr��shufflerrr�T)rMr:�pr�r��idxrlrQrQrRr�[szTestMedian.test_2dcCs�tt�d��dd��}t|dd�<|dd�<tt|�d�tttj	�|��t
u�tt|dd�gd��tttj	j|dd��t
u�tt|d	d�gd
��tttj	j|d	d��t
u�tt|d	d�jgd��dS)Nrrbrw����-@rr�)�+@r}g/@r9)
rrrrbr�r�r�rrr)
r9r9r9rrrrr9r9r9)rrLr	r�r
rrrrYrXrr;r.rQrQrR�
test_2d_waxiskszTestMedian.test_2d_waxiscCs�tj�d��ddd�}t||ddk<tt|d�ddgdd	gddgd
d	gg�d|_tt|d�dd
gddgddgg�tj�d��ddd�}t||ddk<tt|d�dd
gddgddgg�dS)Nr�rwr�r6rr�rvrtr�r�)r�rwr6�crbrr�r�r|rur�r�)rLrXr	r�r
rrrr.rQrQrRrMws("&zTestMedian.test_3dcCsLtt�d��dd��}t|dd�<|dd�<tt|dd�t|dd��dS)Nrrbrwr|r�r�r9)rrLr	r�r
rrr.rQrQrR�
test_neg_axis�szTestMedian.test_neg_axiscCs�dD]?}tt�|��}t|dd�<|dd�<tt�d��}t||d�}|dkr.t|d�nt|d�t||u�tt|�t	u�qdS)	N)r�>@r�g?@rwr|rQr?rr}g.@)
rrLr	r
rrrrrYr)rM�vr�r@rYrQrQrR�test_out_1d�s
�zTestMedian.test_out_1dcCsdD]}}tt�|��dd��}t|dd�<|dd�<tt�d��}t|d|d�}|dkrMtd	gdgd
�d	gddgddgd
dgdd�}ntd	gdgd�d	gddgddgd
dgdd�}t||�t||u�tt	|�t
u�qdS)N)rgD@rr�rbr�rwr|r9)r�r@rr�)rbr�r�r�TFr�r�)r~g�1@g�5@g�9@)rrLr	r�r
rrrrrYr)rMr�r�r@rY�erQrQrR�test_out�s ��
�zTestMedian.test_outcCsFddgddgddgg}tj�|d�}ddg}ttjj|dd�|�dS)Nr�r�r�rr�)rLrX�masked_equalrr)rMrir_r�rQrQrR�$test_single_non_masked_value_on_axis�s��z/TestMedian.test_single_non_masked_value_on_axiscCs�dtjdtd�fD]�}tj�dtjdggd�g�}||_tjj|dd�}tt�	|��t
|tj�tjj|��dd�}tt�	|��t
|tj�tjj|dd�}tt
|�t�t
|dtjdg�tjj|dd�}tt
|�t�t
|tjd	g�tjj|d
d�}tt
|�t�t
|tjd	g�q	tj�dtjdggd�g�}tjj|dd�d	f<t
tjj|dd�tj�t
tjj|dd�dtjdg�t
tjj|dd�tjdg�dS)NFrtr7r9rwrpr�rr6r�r�)rLrr~rXr�nanr;rr�isscalarr�ravelrrYrr
)rMr;�dmrYrQrQrR�test_nan�s. zTestMedian.test_nancCs�tj�t�d��}tj�t�d��}tj|d<tjj|d<ttjj|d|d�|�tj�t�d��}ttjj|d|d�|�tj�t�d	��}ttjj||d�|�dS)
N)r�)rwr�)r6r9rDrr?�rwr9rQ)	rLrXrrrr�r
rr)rM�or�rQrQrR�test_out_nan�s
zTestMedian.test_out_nancCs�tj�tjdtd��}tjj|ddd�<tj|d<ttj�|�tj�ttjj|dd�tj�tj�tjdtd��	ddd��}t�|j
�ddk|_|��}tj|d	<tj|d
<ttj�|�tj�t
t�tj�|���tjj|dd�}tj|d<tj|d<ttj�|d�|�tjj|dd�}tj|d
<tj|d<ttj�|d�|�tjj|dd�}tj|d<tj|d<ttj�|d�|�dS)Nr�r7rwr6rr�r�r9rp)r9r9r6�r6rwr"�r9rwr#)rLrXrr	rKr
r�rrr�r}r;�copyrr�r)rMr<�aorigr=rQrQrR�test_nan_behavior�s0
 







zTestMedian.test_nan_behaviorcCs�tjgd�gd�gtjd�}tjj||dkd�}ttjj|dd�d�ttjj|dd�jd�ttjj|d	d�|d	�ttj�|�d�dS)
N)rwrw�r7rwr�r9r�r�Fr)rLr�uint8rXrrrr;r�rQrQrR�test_ambigous_fill�szTestMedian.test_ambigous_fillcCstjtjfD�]v}t�|tjgtjtjgg�}tjj|t�|�d�}ttjj|dd�|tjg�ttjj|dd�|tjg�ttj�|�|�t�tjtj|gtjtj|gg�}tjj|t�|�d�}t	tjj|dd�|�t	tjj|dd�j
d�t	tjj|dd�|d�t	tj�|�|�t�||g||gg�}ttj�|�|�ttjj|dd�|�ttjj|dd�|�tj|d|dgdtjtjd	gd
tjtj|ggtjd�}tjj|t�|�d�}|dk�rttjj|dd�dd
|dg�ttj�|�d�nttjj|dd�dd
|dg�ttj�|�d�ttjj|dd�dd|g�tdd�D]P}tdd�D]G}t�tjg||g|gd�}tjj|t�|�d�}ttj�|�|�ttjj|dd�|�ttjj|dd�tjg||g|��q4�q-qdS)Nr�rr�r9Frei����rur|r�r7r��@r�g@g$�g"�g�rtrbr6)
rLrWrr�rXr�isnanrrrr;�float32rf)rMrWr<rlrmrQrQrR�test_specialsR"��
  "����zTestMedian.test_specialcCs�tj�tjgtd��}t��!}|�t�ttj�	|�tj
�t|jdj
tu�Wd�n1s2wYtj�tjgtdd��}t��(}|�t�t�ddt�ttj�	|�tj
�t|jdj
tu�Wd�n1sqwYtj�tjgtdd��}ttjj	|dd�|�ttjj	|d	d�|�tj�tjtj
tdd��}tjd
d��$}t�ddt�ttjj	|dd�|�t|dj
tu�Wd�dS1s�wYdS)Nr7rrw)r8rc�always�r6r�r9T)�record)rLrXrrrKrr��RuntimeWarningrrr�r�log�category�warnings�filterwarningsr�catch_warnings)rMr<�wr=rQrQrR�
test_empty.s,
�
�"�zTestMedian.test_emptycCsXtj�t�d��}tttj�|�t���t	�tj
|d<tttj�|�t���t	�dS)Nr�r6)rLrXrr	rrYrr%rWrKr�)rMr�rQrQrR�test_objectKs
 zTestMedian.test_objectN)r�r�r�rZr[r]rbrlrmrorrr�rrMr�r�r�r�r�r�r�r�r�r�r�rQrQrQrRrV�s.&	#	+rVc@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�TestCovcCsttj�d��|_dS�Nr�)rrLr�r�rirqrQrQrR�setupTsz
TestCov.setupcCsZ|j}tt�|�t|��ttj|dd�t|dd��ttj|ddd�t|ddd��dS)NF��rowvarT�r��bias)rirrLrr.rQrQrR�test_1d_without_missingWs�zTestCov.test_1d_without_missingcCsb|j�dd�}tt�|�t|��ttj|dd�t|dd��ttj|ddd�t|ddd��dS)Nrwr�Fr�Tr�)rir�rrLrr.rQrQrR�test_2d_without_missing_s�zTestCov.test_2d_without_missingcCs>|j}t|d<||��8}|��}tt�|�t|��ttj|dd�t|dd��ttj|ddd�t|ddd��zt|dd�Wn	tyJYnw|dd�}tt�||ddd��t||ddd���ttj||ddd�dd�t||ddd�dd��ttj||ddd�ddd�t||ddd�ddd��dS)Nr�Fr�Tr���allow_maskedr9)rir
�meanrkrrLr�
ValueError)rMr��nxrQrQrR�test_1d_with_missinggs,��,��zTestCov.test_1d_with_missingcCs6|j}t|d<|�dd�}t�t|���t�}t�||j	�}||�
d�dd�df�d�}tt
|�t�
|�|jdd|d�tt
|dd�tj
|dd�|jd|�t�|j	|�}||�
d��d�}tt
|d	d
�tj
|d	d
�|jdd|d�tt
|d	dd�tj
|d	dd�|jd|�dS)Nr�rwr�r9rr�T�r�Fr�r�)rir
r�rL�logical_notr
r%rgrryr�r9rrr)rMr��valid�frac�xfrQrQrR�test_2d_with_missings8 ��������zTestCov.test_2d_with_missingN)r�r�r�r�r�r�r�r�rQrQrQrRr�Rsr�c@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�TestCorrcoefcCs(ttj�d��|_ttj�d��|_dSr�)rrLr�r�ri�data2rqrQrQrRr��szTestCorrcoef.setupcCs�|j|j}}t�|�}t�||�}t��P}t�d�ttt|dd�|�	td�t
tj|dd�t|dd��t
t|dd�|�t
t||dd�|�t
t|dd�|�t
t||dd�|�Wd�dS1shwYdS)Nr�r���ddof�bias and ddof have no effectrrw�rir�rLrrr��simplefilterrr4�filterr)rMr�r�r��	expected2�suprQrQrR�	test_ddof�s

"�zTestCorrcoef.test_ddofcCs�|j|j}}t�|�}t��7}t�d�ttt||dd�ttt||dd�ttt|dd�|�	td�t
t|dd�|�Wd�dS1sIwYdS)Nr�TFr�r�r9r�)rMr�r�r�r�rQrQrR�	test_bias�s

"�zTestCorrcoef.test_biascCs�|j}tt�|�t|��ttj|dd�t|dd��t��}|�td�ttj|ddd�t|ddd��Wd�dS1s@wYdS)NFr�r�Tr�)rirrLrrr�r4�rMr�r�rQrQrRr��s
��"�z$TestCorrcoef.test_1d_without_missingcCs�|j�dd�}tt�|�t|��ttj|dd�t|dd��t��}|�td�ttj|ddd�t|ddd��Wd�dS1sDwYdS)Nrwr�Fr�r�Tr�)rir�rrLrrr�r4r�rQrQrRr��s
��"�z$TestCorrcoef.test_2d_without_missingc	Cs�|j}t|d<||��8}|��}tt�|�t|��ttj|dd�t|dd��t��}|�t	d�ttj|ddd�t|ddd��Wd�n1sMwYzt|dd�Wn	t
ycYnw|dd�}tt�||ddd��t||ddd���ttj||ddd�dd�t||ddd�dd��t��?}|�t	d�tt�||ddd��t||ddd�dd	��tt�||ddd��t||ddd�d
d��Wd�dS1s�wYdS)Nr�Fr�r�Tr�r�r9r�r6r�)rir
r�rkrrLrrr�r4r�)rMr�r�r�rQrQrRr��sB
����,���"�z!TestCorrcoef.test_1d_with_missingcCs$|j}t|d<|�dd�}t|�}t�|�}t|dd�dd�f|dd�dd�f�t��Z}|�td�tt|dd�dd�dd�f|dd�dd�f�tt|dd�dd�dd�f|dd�dd�f�tt|dd�dd�dd�f|dd�dd�f�Wd�dS1s�wYdS)	Nr�rwr�r�r�r�r9r�)	rir
r�rrLrrr�r4)rMr�rNrOr�rQrQrRr��s$
*���"�z!TestCorrcoef.test_2d_with_missingN)
r�r�r�r�r�r�r�r�r�r�rQrQrQrRr��s
r�c@r�)�TestPolynomialcCs�tj�d�}tj�d��dd�}tt||d�t�||d��|�t�}t|d<|�t�}t|d<|d<t||dd�dfdd	d
�\}}}}}tj|dd�|dd�df�	�dd	d
�\}}	}
}}t
|||||f||	|
||f�D]	\}
}t|
|�qpt||dd�dfdd	d
�\}}}}}tj|dd�|dd�dfdd	d
�\}}	}
}}t
|||||f||	|
||f�D]	\}
}t|
|�q�t||dd	d
�\}}}}}tj|dd�|dd�dd�fdd	d
�\}}	}
}}t
|||||f||	|
||f�D]	\}
}t|
|�q�tj�d�d}|��}|dd�}|dd�}|dd�}t||dd	|d�\}}}}}tj||dd	|d�\}}	}
}}t||�t
|||||f||	|
||f�D]
\}
}t|
|��qMdS)
Nrbrr�r6rwrrB�r�r�T�rhr9)rhr�)
rLr�r�r�rr�viewrr
rk�zipr�r)rMr�r��C�R�K�S�DrTrYrnrpr�r<�a_r��wo�xs�ys�wsrQrQrR�test_polyfits@

&"�&&0&4&
&�zTestPolynomial.test_polyfitcCs�tj�d�}tj�d��dd�}tj|d<tj|d<|�t�}|�t�}t|d<t|d<t||ddd	�\}}}}}tj|d
d�|d
d�dd�fddd	�\}}	}
}}t	|||||f||	|
||f�D]	\}
}t
|
|�qbdS)Nrbrr�r6rr�rwTr�r9)rLr�r�r�r�r�rr
rr�r)rMr�r�r�r�r�r�r�rTrYrnrpr�r<r�rQrQrR�test_polyfit_with_masked_NaNs)s



4&�z,TestPolynomial.test_polyfit_with_masked_NaNsN)r�r�r�r�r�rQrQrQrRr�s'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!S)"�TestArraySetOpscCsngd�}t|ddd�}tt|dt��t|dtgd�gd�d��t|dgd	��t|d
gd��dS)N�r9r9r9r6r6rwT��return_index�return_inverserrpr�r�r9)rrwr|r6)rrrr9r9r6)rrr�rrrr�rQrQrR�test_unique_onlist<sz"TestArraySetOps.test_unique_onlistcCs�tgd�gd�d�}t|ddd�}t|dtgd�gd�d��t|d	gd
��t|dgd��d
|_tgd�gd�d
d�}t|ddd�}t|dtgd�gd�d��t|d	gd
��t|dgd��dS)Nr�)rrr9rr9rr�Tr�r)r9r6rwr��rrrr9r9)rrwr|r6r6)rrrwr9rwr6rw)rir;�
fill_value)rrrr�r�rQrQrR�test_unique_onmaskedarrayEs�z)TestArraySetOps.test_unique_onmaskedarraycCs�tgd�dd�}t|ddd�}t|dtdgdgd��t|ddg�t|dgd��t}t|ddd�}t|dtt��t|ddg�t|ddg�dS)	Nr�Tr�r�rr9r6r�)rrrr
r�rQrQrR�test_unique_allmaskedUsz%TestArraySetOps.test_unique_allmaskedcCsdtt�d�gd�d�}tgd�gd�d�}t|�}t||�t|�d�|�d��t|j|j�dS)Nr|�r9rrrr9r�)r9r9r9r�)r9rrr9r)rrLr	rr#rr9r;)rMr�rOrNrQrQrR�test_ediff1dds
zTestArraySetOps.test_ediff1dcC�tt�d�gd�d�}t|td�}tgd�gd�d�}t||�t|�d�|�d��t|j|j�t|gd�d�}tgd	�gd
�d�}t||�t|�d�|�d��t|j|j�dS)Nr|r�r�)�to_begin)rr9r9r9r�)r9r9rrr9rrp)r9r6rwr9r9r9r�)rrrr9rrr9�	rrLr	r#r
rrr9r;�rMr�rNrOrQrQrR�test_ediff1d_tobeginm�

z$TestArraySetOps.test_ediff1d_tobegincCr�)Nr|r�r�)�to_end)r9r9r9r�r)r9rrr9r9rrp)r9r9r9r�r9r6rw)r9rrr9rrrr�r�rQrQrR�test_ediff1d_toend|r�z"TestArraySetOps.test_ediff1d_toendcCs�tt�d�gd�d�}t|ttd�}tgd�gd�d�}t||�t|�d�|�d��t|j|j�t|gd�td�}tgd	�gd
�d�}t||�t|�d�|�d��t|j|j�dS)Nr|r�r��r�r�)rr9r9r9r�r)r9r9rrr9r9rrp)rr9r9r9r�r9r6rw)r9r9rrr9rrrr�r�rQrQrR�test_ediff1d_tobegin_toend�s
�
z*TestArraySetOps.test_ediff1d_tobegin_toendcCs�t�d�}t|�}tgd�gd�d�}t||�tt|t��t|�d�|�d��t|j	|j	�t|t
t
d�}tgd�gd�d�}tt|t��t|�d�|�d��t|j	|j	�dS)	Nr|)r9r9r9r9)rrrrr�rr�r�r�)rLr	r#rrrr�rr9r;r
r�rQrQrR�test_ediff1d_ndarray�s

z$TestArraySetOps.test_ediff1d_ndarraycCsTtgd�gd�d�}tgd�gd�d�}t||�}tgd�gd�d�}t||�dS)N)r9rwrwrwr�r�)rwr9r9r9)r9rwr�r,)rr!r)rMr�r�rNrOrQrQrR�test_intersect1d�s

z TestArraySetOps.test_intersect1dcCs$tgd�gd�d�}tgd�gd�d�}t||�}t|tgd���tgd�gd�d�}gd�}t||�}t|tgd�gd	�d��tgd
��}tgd��}t||�}tt|t��t|gd��tgd
�gd�d�}tgd�gd	�d�}t||�}tt|t��t|gd��tgtgg��dS)N�r9r6r|rer��rrrrr9r��r9r6rwr�r|r�r8)rwr�re)r9r6rwr�r|)rwr�rer�r�rp)rtr|r�r�)r9rur6rw)rr9rr)rtr|r�ru)rrrrr�rr�rMr<r=rNrQrQrR�
test_setxor1d�s&



zTestArraySetOps.test_setxor1dcCs�t�d��gd��}t�gd��}d|d<t||d�}tgd�gd�d�}tdtd	�}d
|d<d
|d<d
|d
<t||�}tt|t	��t
||�t�|||j�|j@}t
||�dS)Nr�)r6rwr�r9)r9r6rr�)	rrbrrr9rwrr�r)	rr9rr9rr9rr9r)rir;r7Tr�r,)rr6rw)rLr	r�rrr~r0rr�rrr;)rMr<r;r=�ecrTr�rQrQrR�	test_isin�s �

zTestArraySetOps.test_isincCs�tgd�gd�d�}tgd�gd�d�}t||�}t|gd��tgd�gd�d�}tgd�gd	�d�}t||�}t|gd
��tgtgg��dS)Nr�r�r�r�r8)TTTFT�r|r|r6r9r��r9r|r�r,)TTFTT)rr"rrr�rQrQrR�	test_in1d�s

zTestArraySetOps.test_in1dcCs�tgd�gd�d�}tgd�gd�d�}tt�t||��t||dd��tgd�gd�d�}tgd	�gd
�d�}tt�t||��t||dd��tgtggdd��dS)Nr�r�r�r�r8T)�invertr�r�r,)rrrLr�r"rr=rQrQrR�test_in1d_invert�s  z TestArraySetOps.test_in1d_invertc	Cs�tgd�gd�d�}tgd�gd�d�}t||�}tgd�gd�d�}t||�tgd�gd�ggd	�gd
�gd�}tgd�gd�d�}tgd
�gd�d�}t||�}t||�tgtgg��dS)N)r9r6r|rer|r�r8r�r�)r9r6rwr�r|rer�)rrrrrrr9r+r�r�r,r�r�)rr9r6rwr�r|)rr rr)	rMr<r=rNrOr�r��ezr�rQrQrR�test_union1d�s

$

zTestArraySetOps.test_union1dcCs�tgd�gd�d�}tgd��}t||�}t|tgd�gd�d��td�}td�}tt||�tdd	g��tgtjgd�}tt|g�jtj�dS)
N)rtr|r�rerer9r6r9)rrrrrrrr9r�)r6r�rwrwr6r9r|)rtrer�r,rbrurv)rrrr	rL�uint32r8r�rQrQrR�test_setdiff1ds
zTestArraySetOps.test_setdiff1dcCs8t�gd��}t�gd��}tt||�t�dg��dS)N)r<r=rT)r<r=rprT)rLrrrr=rQrQrR�test_setdiff1d_char_arraysz)TestArraySetOps.test_setdiff1d_char_arrayN)r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrQrQrQrRr�:s"		
r�c@r�)�
TestShapeBasecCsntgd�gd�d�}t|�}t|jd�t|jj|jj�t|jd�t|jj|jj�t|jj|jj�dS)Nr+r*r�r�r�)rrrrr;rir=rQrQrR�test_atleast_2d$szTestShapeBase.test_atleast_2dcCs�td�}t|jd�t|jj|j�t|jj|j�tdd�}|D]}t|jd�t|jj|j�t|jj|j�q!td�}t|jd�t|jj|j�t|jj|j�tdd�}|D]}t|jd�t|jj|j�t|jj|j�q[td�}t|jd�t|jj|j�t|jj|j�tdd�}|D]}t|jd�t|jj|j�t|jj|j�q�td�}t|jd�t|jj|jj�dS)Nr�rUr�rCr�)rrrr;rirrr1)rMr=r<rQrQrR�test_shape_scalar.s<


zTestShapeBase.test_shape_scalarN)r�r�r�rrrQrQrQrRr"rPrc@r�)�	TestStackcCs�tgd�gd�d�}tgd�gd�d�}t||gdd�}t|jd�t|j|dj�t|j|d	j�t||g�}t|j|j�t|j|j�t||gd	d�}t|jd
�t|j|dd�dfj�t|j|dd�d	fj�dS)Nr+r*r��rvrurer�rr�r�r9�rwr6�rr2rrrr;r3ri�rMr<r=rTr�rQrQrR�
test_stack_1d\szTestStack.test_stack_1dcCs�tgd�dd�}tgd�dd�}t||gdd�}t|jd�t|j|dj�t|j|d	j�t||g�}t|j|j�t|j|j�t||gd	d�}t|jd
�t|j|dd�dfj�t|j|dd�d	fj�dS)Nr+Tr�r	Frr�r�r9r
rrrQrQrR�test_stack_masksnszTestStack.test_stack_masksc
Cs�d}tj�dd|�}tj�dd|�}tj�dd|��t�}tj�dd|��t�}t||d�}t||d�}t||gdd�}d|}	t|j|	�t	|j
|dj
�t	|j
|dj
�t||gd	d�}|d}	t|j|	�t	|j
|d
j
�t	|j
|dj
�d}tj�dd|�}tj�dd|�}tj�dd|��t�}tj�dd|��t�}t||d�}t||d�}t||gdd�}d|}	t|j|	�t	|j
|dj
�t	|j
|dj
�t||gd	d�}|d}	t|j|	�t	|j
|d
j
�t	|j
|dj
�dS)
Nr
rrbr6r�r�r5r9r�).r).r9)rwr6r�r|)rLr��randintr%r~rr2rrrr;)
rM�shp�d1�d2r�r��a1r&rT�c_shprQrQrR�
test_stack_nd�sDzTestStack.test_stack_ndN)r�r�r�r
rrrQrQrQrRrZsr)J�__doc__r�rdrB�numpyrL�
numpy.testingrr�numpy.ma.testutilsrrrr�
numpy.ma.corerr	r
rrr
rrrrr�numpy.ma.extrasrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r�r�r�r�rGrQrVr�r�r�r�rrrQrQrQrR�<module>s<4�
,F
}Ck:i8

https://t.me/RX1948 - 2025