403Webshell
Server IP : 209.209.40.120  /  Your IP : 216.73.217.112
Web Server : Microsoft-IIS/10.0
System : Windows NT NEWWWW 10.0 build 17763 (Windows Server 2019) i586
User : NEWWWW$ ( 0)
PHP Version : 8.3.30
Disable Function : NONE
MySQL : OFF  |  cURL : ON  |  WGET : OFF  |  Perl : OFF  |  Python : OFF  |  Sudo : OFF  |  Pkexec : OFF
Directory :  C:/Python312/Lib/site-packages/numpy/ma/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : C:/Python312/Lib/site-packages/numpy/ma/tests/__pycache__/test_extras.cpython-312.pyc
�

	�g�9��P�dZddlZddlZddlZddlZddlmZddlm	Z	m
Z
ddlmZm
Z
mZmZddlmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZm 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?Gd�d	�Z@Gd
�d�ZAGd�d
�ZBGd�d�ZCGd�d�ZDGd�d�ZEGd�d�ZFGd�d�ZGGd�d�ZHGd�d�ZIGd�d�ZJGd�d�ZKGd �d!�ZLGd"�d#�ZMGd$�d%�ZNy)&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)�normalize_axis_tuple)�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�ndenumerate�stack�vstack�
_covhelperc�<�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
y	)
�TestGenericc��tdt��}tddgddgt��}t||�t	j
ddgddgd��}td|��}td	d	gd
d
g|��}t||�td|��}td	d	gd	d	ggd
d
gd
d
gg|��}t||�t	j
ddd
dgfg�}td|��}tddgddg|��}t||�td|��}tddgddg|��}t||�td
|��}tdggdgg|��}t||�y)N����dtype���maskr>�a�b�f��names�formats�rr�r?r?�r<r<�rBrD��barD��bbrD�r?rI)r0�floatr
r�npr>)�self�test�control�dts    �<C:\Python312\Lib\site-packages\numpy/ma/tests/test_extras.py�test_masked_allzTestGeneric.test_masked_all&sn���$�e�,����A��a��V�5�9���T�7�#�
�X�X��c�
��S�z�B�
C���$�b�)�����(���/?�r�J���T�7�#��&��+���&�&�)�F�F�+;�<�%�v�.���0@�A� �"��	�T�7�#�
�X�X�z�C�+�{�)C�#D�E�
F���$�b�)����k�2�)�;�7�r�C���T�7�#��$�b�)����k�2�)�;�7�r�C���T�7�#��&��+���+����}�o�R�H���T�7�#�c�.�tjddtfgdffg�}tjj	d|�}tt
|d�tjjj�tt
|dd�tjjj�tt|dd�d�t|ddjd�t|ddjjd�y)NrC�c�r?r?rI��rRr>�object�mar0r�type�corer
�lenr�_fill_value�rS�my_dtype�
masked_arrs   rW�"test_masked_all_with_object_nestedz.TestGeneric.test_masked_all_with_object_nestedDs����8�8�c�c�6�]�O�T�#:�;�<�=���U�U�%�%�d�H�5�
��T�*�S�/�*�B�E�E�J�J�,B�,B�C��T�*�S�/�#�.�/������1G�1G�H��S��C���-�.��2��Z��_�S�)�/�/��8��Z��_�S�)�5�5�;�;�R�@rYc��tjdtdffg�}tjj	d|�}tt
|d�tjjj�tt|d�d�t|djd�t|djjd�y)NrCr\r?rIr]r^res   rW�test_masked_all_with_objectz'TestGeneric.test_masked_all_with_objectPs����8�8�c�F�D�>�2�3�4���U�U�%�%�d�H�5�
��T�*�S�/�*�B�E�E�J�J�,B�,B�C��S��C��)�1�-��Z��_�*�*�F�3��Z��_�0�0�6�6��;rYc��tddgt��}t|�}tddgddgt��}t||�t	j
ddgddgd��}td	d	gd
d
g|��}t|�}tddgd
d
g|��}t||�t	j
ddd
dgfg�}tddgddg|��}t|�}t||�y)Nr?r<r=r@rBrCrDrErHrI)�
rlrKrLrNrP)r
rQr1rrRr>)rS�baserTrUrVs     rW�test_masked_all_likez TestGeneric.test_masked_all_likeZs����a��V�5�)���t�$����A��a��V�5�9���T�7�#�
�X�X��c�
��S�z�B�
C���f�f�%�V�V�,<�B�G���t�$����8�,�F�F�3C�2�N���T�7�#�
�X�X�z�C�+�{�)C�#D�E�
F����k�2�)�;�7�r�C���w�'���T�7�#rYc�0�tdd�D�]}td|z�D]�}tj|t��}tj||t��}td|z�}|d|zzdk7|_d}||�D]"}||j|j�z
}�$|tk(r)t|j�j�|���|j|_t|j�j�|�����	y)Nr?�r<r=r)�rangerRr�int�fullrrA�data�sumr,r�
compressed)	rSrD�i�j�k�jarB�s�sls	         rW�check_clumpzTestGeneric.check_clumpns����q�!��
	:�A��1�a�4�[�
:���I�I�a�s�+���W�W�Q���-�� ��A��&����1��+�!�+������A�$�*�B���������)�)�A�*���&� �����!3�!3�!5�q�9��f�f�W�A�F� �����!3�!3�!5�q�9�
:�
	:rYc���ttjd��}t|gd�<t	|�}tdd�tdd�tdd�g}t
||�|jt�y)Nrl�rr?r<���	r�r�rpr�)rrRrrr+�slicerr}�rSrBrTrUs    rW�test_clump_maskedzTestGeneric.test_clump_masked~s^������2��'�� &��
���A�����A�;��a���U�1�b�\�:���T�7�#�����&rYc���ttjd��}t|gd�<t	|�}tdd�tdd�g}t
||�|jt�y)Nrlrr�r�rpr�)rrRrrr,r�rr}r�s    rW�test_clump_unmaskedzTestGeneric.test_clump_unmasked�sW������2��'�� &��
���a� ����A�;��a���.���T�7�#�����(rYc��td�}t|�}t|td|j�g�tjdt��|_t|td|j�g�t||dk|dkDz|dk(z<t|�}t|tdd�tdd�g�t|ddt|�}t|g�y)	Nrlrr=r�r��r�r�)
rr-rr��sizerRr�boolrAr)rSrBrTs   rW�test_flatnotmasked_contiguousz)TestGeneric.test_flatnotmasked_contiguous�s����2�J��'��*���T�E�!�Q�V�V�,�-�.����"�D�)����T�E�!�Q�V�V�,�-�.�*0��1�q�5�Q��U�
�q�A�v�
&�'�'��*���T�E�!�Q�K��q�!��5�6���!��'��*���T�2�rYN)�__name__�
__module__�__qualname__rXrhrjrnr}r�r�r�r]rYrWr9r9$s+��$�<
A�<�$�(:� 	'�)�rYr9c���eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	e
jjdgd	�d
dggd�d
gdgfgd�gd�gdgd�gddggd�ggd�gfg�d��Z
d�Zy
)�TestAveragec��tgd�gd���}tdt|d���tdt|gd����t|gd�d	�
�\}}td|�t|dk(�t|ddtt|d��j
d	g�tgd�gd���}|j
dd�}t|dd�d
f<tt|d��ddg�tt|d
��j
dd	g�tddgt|d���t|dd	��\}}t|ddg�y)N)���?�@�@)TFFF�rAr�r��axis)r�r�r�r���weightsT�r��returned�@r<r?r��r�r�r�)r
rrrrrA�reshape)rS�ott�result�wtss    rW�test_testAverage1zTestAverage.test_testAverage1�s���$�+F�G���S�'�#�A�.�/��S�'�#�/?�@�A��c�+;�d�K�����S�&�!���s�
����A���W�S�q�)�.�.���7��$�+F�G���k�k�!�Q�����A�q�D�	��W�S�q�)�C��:�6��W�S�q�)�.�.�q�1�D�6�:��b�"�X�w�s��3�4��c��D�9�����S�2�r�(�#rYc
���gd�}gd�gd�g}tdtj��}tt	|d��d�tt	|d|��d�ttdtj��d	td�zg�}tt	|d�tjjtjd��d
zdz�tt	|d��tjd�d
zd	z�tt	|d��t	|d��t	|d��d	zg�tt	|d|�
�d�tt	|d|��gd��tt	|d��t	|d��t	|d��d	zg�td�}gd�}gd�gd�g}td�}gd�}	tt	t||�d��d�tt	t||�d��d�tt	t||�d��jdg�tt	t||	�d��d�ttt	t||�d���d�t||�}
tt	|
d�d�tt	|
d��gd��tt	|
d��ddg�tt	|
d|��gd��y)N�rr?r?r?r?r�r?rrrrr?r�r=rr��@�r�r�r�r��(@r?r�g������
@)r�r�r�r�r��$@)rrr?r?rr)rr?r?r?r?r?Tr�)r�r���X@r�r�g@�@)r�r�r�r�r�r�)
rrR�float64rrr
�add�reducerrrrAr)rS�w1�w2�x�y�m1�m2�m3�m4�m5�zs           rW�test_testAverage2zTestAverage.test_testAverage2�sD��
�� �"4�
5���1�B�J�J�'���W�Q�Q�'��-��W�Q�Q��3�S�9��6�!�2�:�:�.��f�Q�i��@�A���W�Q��%�r�v�v�}�}�R�Y�Y�q�\�'B�R�'G�#�'M�N��W�Q�Q�'����1���):�R�)?�@��W�Q�Q�'��a�a�(�'�!�!�*<�s�*B�C�	E��W�Q��b�1�8�<��W�Q�Q��3�.�	0��W�Q�Q�'��a�a�(�'�!�!�*<�s�*B�C�	E�
�1�X��
�� �"4�
5��
�!�W��
���W�\�!�R�0�q�9�3�?��W�\�!�R�0�q�9�3�?��W�\�!�R�0�q�9�>�>���G��W�\�!�R�0�q�9�3�?��U�7�<��2�#6�Q�?�@�!�D���B����W�Q��%�x�0��W�Q�Q�'�)E�F��W�Q�Q�'�#�s��4��W�Q�Q��3�2�	4rYc�j�td�}td�dz}t||g||ggdd��\}}tt|�t|��t|j|j�tt	d�dddgd��\}}tt|�t|��tt	d�d�	�\}}tt|�t|��tt	d�t	d�d�
�\}}tt|�t|��tddgddggt�}t|d
d
gdd
gg�}t|d��}	t|	ddg�t|d��}
t|
ddg�t|d��}
t|
d�t|d��}
t|
ddg�y)Nr�r�r?Tr��r<r<r�r�r�r�r�)r�r�r<�Fr���?r�r�g������@��?r�)rrrrrr
rQr)rSrBrC�r1r��r2r��a2d�a2dm�a2da�a2dmas           rW�test_testAverage3zTestAverage.test_testAverage3�sq���1�I���1�I��M���1�a�&�1�a�&�)��D�A���B��U�2�Y��b�	�*��R�X�X�r�x�x�(���i��q�1�a�&�4�P���B��U�2�Y��b�	�*���i��4�8���B��U�2�Y��b�	�*���i��$�y�/�D�Q���B��U�2�Y��b�	�*��a��V�a��V�$�e�,���C�5�%�.�4��-�!@�A���s��#���T�C��:�&���1�%���U�S�#�J�'���4�(���U�G�$���1�%���U�S�#�J�'rYc�T�tjgd��jdd�}tjj|dgdgdgg��}tjgd��jdd�}t	||dd��}td	gd
gdggdgdgdgg�}t
||�y)N�r<r�r�r�r?FTr�)r�r�r�)r�r��keepdimsr�r�r�)rRr
r�r`rrr)rSr�rC�w�actual�desireds      rW�test_testAverage4zTestAverage.test_testAverage4�s����H�H�Y��'�'��1�-���E�E�K�K��%��5�'�D�6� :�K�;���H�H�Y��'�'��1�-����A�A��=�����r�d�R�D�1�U�G�e�W�t�f�3M�N���V�W�%rYc��tjd�jddd�}tjgd��jddd�}tjdd�}tj
j||��}|dd�dd�df}t
|d	|�
�}tgd�gd���}t||�tjdd�}d
|dd�dd�df<d
|d<tj
j||��}t
|d	|�
�}ttjddggd���}t||�tjdd�}tj
j||��}|ddd�dd�f}t
|d|�
�}tddgddg��}t||�tjtd��5t
|d|�
�ddd�tjtd��5t
|d	|�
�ddd�t
|d|�
�}t
|d	|j�
�}t||�y#1swY�nxYw#1swY�HxYw)N�r<r�)r�r�r�r�r�r�r�r�r�r�r�r�r�Fr�r�rr?r�)�@� @�"@)FFFT�rrr?r�r�)TFFr?�r?r<g@g� @zHShape of weights must be consistent with shape of a along specified axis)�match�rr?r<)r?r)rRrr�r
rsr`rrr	�nan�pytest�raises�
ValueError�T)	rSr�r��m�yma�subw0r�r��subw1s	         rW�$test_weight_and_input_dims_differentz0TestAverage.test_weight_and_input_dims_different�s��
�I�I�b�M�!�!�!�Q��*���H�H�E�F�
�W�Q��1�
�	
�
�G�G�I�u�%���e�e�k�k�!�!�k�$���!�Q��'�
����6�5�9���|�2G�H���F�G�,��G�G�I�u�%����!�Q��'�
���'�
��e�e�k�k�!�!�k�$����6�5�9���
�V�V�R���%�'��	�F�G�,��G�G�I�u�%���e�e�k�k�!�!�k�$���!�Q��'�
����6�5�9����d�|�5�%�.�A���F�G�,��]�]��8�9�	8�
�C�i��7�		8��]�]��8�9�	5�
�C�f�e�4�		5���6�5�9���#�F�E�G�G�<���F�G�,�	8�	8��	5�	5�s�I�:I�I
�Ic��ttddg��}t|d�ttgd�gd����}t|d�y)Nr?r<r�)r?r<r�r��FFTTr�)rr
r�rSrBs  rW�test_onintegers_with_maskz%TestAverage.test_onintegers_with_mask1s:���E�1�a�&�M�"���Q����E�,�-G�H�I���Q��rYc�n�tjgd�gd�gt��}tgd�gd�g|��}t	|�}tj|j��}t
|j|j�t
|j|j�t	|d��}t	|jd��t	|jd��d	zz}t
|j|j�t
|j|j�t	|d
��}t	|jd
��t	|jd
��d	zz}t
|j|j�t
|j|j�tjgd�gd�g�}	t	||	�
�}
tj|j�|	|�
�}t
|
j|j�t
|
j|j�t	||	d��}t	|j|	d��t	|j|	d��d	zz}t
|j|j�t
|j|j�t	||	d
��}t	|j|	d
��t	|j|	d
��d	zz}t
|j|j�t
|j|j�y)N)rrrr?r�rr?rrrr=)ry�?@y@@y@@y@ @)y"@��?y@@y@@y@@r�rr�r�r?)r�r�r�r�r�)r�r�r�r�r�r��r�r�)	rRr
r�rrrvr	�real�imag)
rSrArB�av�expected�av0�	expected0�av1�	expected1r��wav�wav0�wav1s
             rW�test_complexzTestAverage.test_complex8sG���x�x��(�*�15�7���5�6�8�"�
$���Q�Z���:�:�a�l�l�n�-���B�G�G�X�]�]�3��B�G�G�X�]�]�3��a�a� ���A�F�F��+�g�a�f�f�1�.E�b�.H�H�	��C�H�H�i�n�n�5��C�H�H�i�n�n�5��a�a� ���A�F�F��+�g�a�f�f�1�.E�b�.H�H�	��C�H�H�i�n�n�5��C�H�H�i�n�n�5��h�h�1�1�3�4���a��%���:�:�a�l�l�n�c�4�%�j�A���C�H�H�h�m�m�4��C�H�H�h�m�m�4��q�#�A�.���Q�V�V�S�q�9��Q�V�V�S�q�9�"�<�=�	��D�I�I�y�~�~�6��D�I�I�y�~�~�6��q�#�A�.���Q�V�V�S�q�9��Q�V�V�S�q�9�"�<�=�	��D�I�I�y�~�~�6��D�I�I�y�~�~�6rYz<x, axis, expected_avg, weights, expected_wavg, expected_wsum�r?r<r�Nr�)r�r�r?g�?r�)r?r<r�)r?r��r)r�r�r�r?r�)r�r�g#@)r�r�r�c�\�tjj||d��}|jtj|�k(sJ�t	||�tjj|||d��}|jtj|�k(sJ�t	||�tjj|||dd��\}}	|jtj|�k(sJ�t	||�|	jtj|�k(sJ�t	|	|�y)NT)r�r�)r�r�r�)r�r�r�r�)rRr`rrr)
rSr�r��expected_avgr��
expected_wavg�
expected_wsum�avg�wavg�wsums
          rW�test_basic_keepdimszTestAverage.test_basic_keepdimsds����e�e�m�m�A�D�4�m�8���y�y�B�H�H�\�2�2�2�2��3��-��u�u�}�}�Q�T�7�T�}�J���z�z�R�X�X�m�4�4�4�4��4��/��U�U�]�]�1�4��,0�4�#�A�
��d��z�z�R�X�X�m�4�4�4�4��4��/��z�z�R�X�X�m�4�4�4�4��4��/rYc���tjjtjd�j	dd�gd�gd�gd�g��}tgd�d��}tgd�gd���}t
|d|d�	�}tjgd
��}t||�t
|d|d�	�}tjgd��}t||�tjjgd�gd
���}t
|dddtjg��}t|d�tjjgd�gd�gd�ggd�gd�gd�g��}tjj
|dtjdgd��}tjjdtjtjdggd���}t||�t|j|j�y)Nr�r��r?rr�rrrr�)r���Frr�)�@g�@g�@)rg0��N@g0��N@)r�r�r�r�r�r?r�r�)r�rr�r�)r�r�r�r�)FTTF)TFTT)TFTFr�r��@)rRr`r
rr�rrr	r�rrA)	rSrB�weights_unmasked�weights_masked�avg_unmasked�expected_unmasked�
avg_masked�expected_masked�avg_expecteds	         rW�test_masked_weightszTestAverage.test_masked_weights{s���
�E�E�K�K��	�	�!��,�,�Q��2�'��I�>�
�
@��'��%�@��%�k�	�B���q�q�'7�%�I���H�H�%<�=���L�*;�<��Q�Q���O�
��(�(�#N�O���J��8�

�E�E�K�K�1�2�
�
4���q�1�a��B�F�F�*;�<���L�#�.��E�E�K�K� � � �
�

'�%�&�
�	
�
���U�U�]�]�1�q�"�&�&�!�n�1�]�E�
��u�u�{�{�C��������#=�#=�#�?��	�J��5��Z�_�_�l�&7�&7�8rY)r�r�r�r�r�r�r�r�r�r�r��mark�parametrizerrr]rYrWr�r��s���$�& 4�D(�0&�1-�f�*7�X�[�[���F�
�T�C�5�)�d�V�c�U�	;��j�
!�1��&7��a�&�?�#�i�[�
2�	3��0�
�0�")9rYr�c��eZdZd�Zd�Zd�Zy)�TestConcatenatorc�"�ttdtgd���td�}gd�}t	||��}t|dd|f}tt
|t��t|gd��t|jt|dd|f�y)N�r?r<r�r�r�r�r�)r?rrrrr�r)r?r?r?r?r?rrr?r?r?r?r?)	rrr
rrr�
isinstancer
rA)rSrCr��dr[s     rW�test_1dzTestConcatenator.test_1d�sw���3�/�0�%�8J�2K�L���G�������#����1�a��
�O���
�1�k�*�+��1�B�C��1�6�6�3�q�!�Q��z�?�3rYc�d�tjjdd�}tjjdd�}tjtjjdd�d�}tjtjjdd�d�}t	||��}t	||��}t
d||f}t
|jdk(�t|dd�dd�f|�t|dd�dd�f|�t|jtjd||f�t
||f}t
|jdk(�t|dd�dd�f|�t|dd�dd�f|�t|jtj||f�y)Nr�rr��1)r�rl)rlr�)rR�random�rand�roundrrrrrrA�r_)rS�a_1�a_2�m_1�m_2�b_1�b_2rs        rW�test_2dzTestConcatenator.test_2d�sJ���i�i�n�n�Q��"���i�i�n�n�Q��"���h�h�r�y�y�~�~�a��+�Q�/���h�h�r�y�y�~�~�a��+�Q�/���3�S�)���3�S�)����S�#�
�������7�"�#��1�Q����U�8�S�)��1�Q���U�8�S�)��1�6�6�2�5�5��c�3��#7�8���S��M������7�"�#��1�R�a�R��T�7�C�(��1�Q�R��T�7�C�(��1�6�6�2�5�5��c��?�3rYc�\�ttjjdf}t	|j
ddg�t	|jdd�tddgtjjf}t	|j
gd��t	|jddddg�y)Nr?TFr<)FFT)rrRr`rrrArt)rSr�s  rW�test_masked_constantz%TestConcatenator.test_masked_constant�s~���R�U�U�\�\�1�_�%���V�[�[�4��-�0��V�[�[��^�Q�'��a��V�R�U�U�\�\�)�*���V�[�[�"6�7��V�[�[��!�_�q�!�f�-rYN)r�r�r�rr'r)r]rYrWrr�s��	4�4�(.rYrc��eZdZd�Zd�Zy)�
TestNotMaskedc	�`�ttjd�jdd�gd�gd�gd�gd�gd�g��}t	|d�}t|d	d
g�t	|d	�}t|d	ddg�t|dd
dg�t	|d�}t|d	ddg�t|dddg�t	|jd�}t|d	d
g�t	|jd	�}t|d	ddg�t|dddg�t	|jd�}t|d	ddg�t|dddg�t|d<t	|d	�}t|d	ddg�t|dddg�t	|d�}t|d	ddg�t|dddg�y)N�r�)rrr?rr)rrrr?r?)r?r?rrr)rrrrr)r?r?r?rrr�r��rr?r<r�r�r?)r�r�r�r�r�)rrr<rr�)r�r<r�r�r�)r�r�r�r�r�������)r?r?r<r�r�)rr?r<r�)rrr<r�)r�r<r�r�)rrRrr�r/rrtr�rSrtrTs   rW�
test_edgeszTestNotMasked.test_edges�s����B�I�I�b�M�1�1�!�Q�7�"1�"1�"1�"1�"1�	"3�5���t�T�*���T�A�r�7�#��t�Q�'���T�!�W���@�A��T�!�W���@�A��t�Q�'���T�!�W���@�A��T�!�W���@�A��t�y�y�$�/���T�A�r�7�#��t�y�y�!�,���T�!�W���@�A��T�!�W���@�A��t�y�y�"�-���T�!�W���@�A��T�!�W���@�A���R���t�Q�'���T�!�W���@�A��T�!�W���@�A��t�R�(���T�!�W�|�\�:�;��T�!�W�|�\�:�;rYc��ttjd�jdd�gd�gd�gd�g��}t	|d�}t|t
dd	d�t
d
dd�t
ddd�g�t	|d�}t|t
dd
d�t
ddd�gt
dd
d�t
ddd�gt
dd
d�t
ddd�gt
dd
d�t
ddd�gt
ddd�gt
ddd�ggt
ddd�gg�t	|d
�}t|t
dd	d�ggt
ddd�t
ddd�gg�y)Nr.r�r�)rrrrr?r?r?r?)r?r?r?r?r?r?r?r?)rrrrrrr?rr�rr����r?r<r�rp)rrRrr�r.rr�)rSrB�tmps   rW�test_contiguouszTestNotMasked.test_contiguous�sp������2��.�.�q�!�4�7�7�7�9�
:��#�1�d�+���S��!�Q����"�b�$���"�b�$��
�	�#�1�a�(���S�
�1�a��
��a��D� 1�2�
�1�a��
��a��D� 1�2�
�1�a��
��a��D� 1�2�
�1�a��
��a��D� 1�2�
�1�a��
��
�1�a��
���
�1�a��
��	
�		�#�1�a�(���S�
�1�a��
���
�1�a��
��a��D� 1�2�
�	rYN)r�r�r�r3r9r]rYrWr+r+�s
�� <�DrYr+c��eZdZd�Zd�Zd�Zejjdgd��ejjddge	jjdfe	jjd	fg�d
���Z
d�Zd�Zd
�Zy)�TestCompressFunctionsc
���tjttd���j	ddd�}tj
d�j
t�}d|d<t||��}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||�y)1N�<r�r�r��r�r�r�T�r?r?r?r�)rr<r�r�)rlr��
�)����)�(�*�+�,)�2�4�5�6)�7�9�:�;rr/)r�r�rpr�r�)rlr�r�r@rA)rBr5rCrDrE)rF�)rGrHrI)�-�.�/�0�1)rJ�3rKrLrM)rN�8rOrPrQr?)��r6r7r.)�r� �!�")�#�$�%�&�'r\r1)r1r<)r�rpr�r�)rZr6r7r.)r-�r�)r\r]r^r_)r`rbrcrd)rSrUrVrWr;r0)r0r�)rr1r�)r1r<)r?r0)r1r0�rr<)rr0)
rRr
�listrqr�r�astyper�r(r)rSr�r�rB�a2�a3�a4s       rW�test_compress_ndz&TestCompressFunctions.test_compress_nds0���H�H�T�%��,�'�(�0�0��A�q�9���H�H�W��$�$�T�*����%���!�!���
��N���Q�*�*�*�,�+�*�*�,�-�	.�
��1����Q�.�.�.�.�0�/�.�.�.�0�	1�	2�
��1����Q�.�.�.�0�/�.�.�0�/�.�.�0�
1�	2���D�
!��
��B�
��
��E�
"���Q����Q����Q���
��1����Q�/�)�*�*�,�+�*�*�*�,�+�*�*�*�,�-�	.���D�
!��
��B�
��
��E�
"���Q����Q����Q���
��6�"���Q�.�.�.�0�/�.�.�0�1�	2���G�
$���Q���
��6�"���Q�*�*�*�,�+�*�*�,�+�*�*�,�
-�	.���G�
$��
��G�
$��
��H�
%���Q����Q����Q���
��6�"���Q�*�*�*�*�,�+�*�*�*�,�	-�	.���G�
$���Q��rYc���ttjd�jdd�gd�gd�gd�g��}t	t|�ddgdd	gg�t	t|d
�gd�gd�g�t	t|d
�d
dgddgdd	gg�t|jgd�gd�gd�g��}t	t|�d
dgdd	gg�t	t|d
�gd�gd�g�t	t|d
�d
dgddgdd	gg�t|jgd�gd�gd�g��}t	t|�d	gg�t	t|d
�gd�g�t	t|d
�dgdgd	gg�t|jgd�gd�gd�g��}t	t|�jd
�t	t|d
�jd
�t	t|d
�jd
�y)Nr�r�rrr�r�r�rpr�rr>)r�rpr�r?r<�rr?rr�r�r�)r
rRrr�rr)�_datar��rSr�s  rW�test_compress_rowcolsz+TestCompressFunctions.test_compress_rowcols�s����"�)�)�A�,�&�&�q�!�,�!�9�i�8�
:���%�a�(�A�q�6�A�q�6�*:�;��%�a��+�i��-C�D��%�a��+�q�!�f�q�!�f�q�!�f�-E�F��!�'�'��I�y� A�B���%�a�(�A�q�6�A�q�6�*:�;��%�a��+�i��-C�D��%�a��+�q�!�f�q�!�f�q�!�f�-E�F��!�'�'��I�y� A�B���%�a�(�A�3�%�0��%�a��+�i�[�9��%�a��,��s�Q�C�!��o�>��!�'�'��I�y� A�B���%�a�(�-�-�q�1��%�a��+�0�0�!�4��%�a��+�0�0�!�4rYc��ttjd�jdd�gd�gd�gd�g��}t	t|�jgd�gd�gd�g�t	t|d�jgd�gd�gd�g�t	t|d�jgd�gd�gd�g�t|jgd�gd	�gd�g��}t	t|�jgd	�gd�gd	�g�t	t|d�jgd�gd�gd�g�t	t|d�jgd	�gd	�gd	�g�t|jgd�gd	�gd�g��}t	t|�jgd�gd�gd
�g�t	t|d�jgd�gd�gd�g�t	t|d�jgd
�gd
�gd
�g�t|jgd�gd	�gd�g��}tt|�j�tu�tt|d�j�tu�tt|d�j�tu�tt|�jj��tt|d�jj��tt|d�jj��y)Nr�r�rrr�r?rr?ro)r?r?rr�)r
rRrr�rr*rArpr�allrrqs  rW�test_mask_rowcolsz'TestCompressFunctions.test_mask_rowcols�s=���"�)�)�A�,�&�&�q�!�,�!�9�i�8�
:���\�!�_�)�)���I�6�	8��\�!�Q�'�,�,���I�6�	8��\�!�Q�'�,�,���I�6�	8��!�'�'��I�y� A�B���\�!�_�)�)���I�6�	8��\�!�Q�'�,�,���I�6�	8��\�!�Q�'�,�,���I�6�	8��!�'�'��I�y� A�B���\�!�_�)�)���I�6�	8��\�!�Q�'�,�,���I�6�	8��\�!�Q�(�-�-���I�6�	8��!�'�'��I�y� A�B����Q��#�#�%��/�0���Q��"�&�&�(�F�2�3���Q��"�&�&�(�F�2�3���Q��$�$�(�(�*�+���Q��"�'�'�+�+�-�.���Q��"�'�'�+�+�-�.rYr�)Nrr?�func�rowcols_axisrr?c��ttjd�jdd�gd�gd�gd�g��}t	t
�5|||��}t
|t||��ddd�y#1swYyxYw)Nr�r�rrr�r�)r
rRrr�r�DeprecationWarningrr*)rSr�rvrwr��ress      rW�#test_mask_row_cols_axis_deprecationz9TestCompressFunctions.test_mask_row_cols_axis_deprecation�sl��

�"�)�)�A�,�&�&�q�!�,�!�9�i�8�
:���,�
-�	=��q�t�$�C���l�1�l�;�<�	=�	=�	=�s�	!A3�3A<c	��tjdd�}gd�}t||��jdd�}t||��jdd�}t	||d��}t|jddgdd	gg�t	||d��}t|jgd
�gd�gd�g�t	||d��}t|tj|jd	�|jd	���t	||d��}t|tj|jd	�|jd	���gd
�}t||��jdd�}t||��jdd�}t	||d��}t|jd	dgddgg�t	||d��}t|jgd�gd�gd
�g�t	||d��}t|tj|jd	�|jd	���t|t	||��t	||d��}t|tj|jd	�|jd	���gd�}t||��jdd�}t||��jdd�}t	||�}t|jt�t	||�}t|jt�t|gd���jdd�}t|gd���jdd�}t	||d��}t|jddgd	d	gg�t	||d��}t|tj|jd	�|jd	���t	||d��}t|jgd�gd�gd�g�t	||d��}t|tj|jd	�|jd	���t|gd
���jdd�}t|gd���jdd�}t	||d��}t|jd	d	gddgg�t	||�}t|tj|jd	�|jd	���t	||d��}t|jgd�gd�gd�g�t	||d��}t|tj|jd	�|jd	���t|gd
���jdd�}t|gd���jdd�}t	||d��}t|jdd	gddgg�t	||d��}t|tj|jd	�|jd	���t	||d��}t|jgd�gd
�gd�g�t	||d��}t|tj|jd	�|jd	���ttjd�jddd�dd	gd	d	ggd	d	gd	d	ggg��}ttjd�jddd�d	d	gd	d	ggd	d	gd	dggg��}t	||d��}t|jddgddggd	d	gd	dgggd	d	gd	dggd	d	gd	dgggg�t	||d��}t|jd	d	gd	dggd	d	gd	d	gggd	d	gd	d	ggd	d	gd	d	gggg�t	||d��}t|jdd	gd	d	ggdd	gd	d	gggdd	gd	d	ggddgddgggg�t	||d��}t|jd	d	gd	d	ggd	d	gd	d	gggd	d	gd	d	ggdd	gd	d	gggg�ttjd�jddd�dd	gd	d	ggd	d	gd	d	ggg��}d}t	||d��}t|jdd	gd	d	ggd	d	gd	d	ggg�t	||d��}t|jdd	gd	d	ggd	d	gd	d	ggg�t	||d��}t|jdd	gd	d	ggd	d	gd	d	ggg�t	||d��}t|jdd	gd	d	ggd	d	gd	d	ggg�ttjd�jddd�dd	gd	d	ggd	d	gd	d	ggg��}ttjd�d	dg��}t	||d��}t|jddgddgg�t	||d��}t|jdd	gd	d	gg�y)Nr?rp)r?rrrrrr�r<r�T)�strictrr?rF�rrrrrr?r�)rrrrrr)rrr?rrrr�r�)	rRrrr�rrrA�filledr)rS�nr�rBrCr[s      rW�test_dotzTestCompressFunctions.test_dot�s	���I�I�a��O�������#�+�+�A�q�1�����#�+�+�A�q�1����1�T�"���Q�V�V�q�!�f�q�!�f�-�.���1�T�"���Q�V�V�i��I�>�?���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9������#�+�+�A�q�1�����#�+�+�A�q�1����1�T�"���Q�V�V�q�!�f�q�!�f�-�.���1�T�"���Q�V�V�i��I�>�?���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9��Q��A�q�	�"���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9������#�+�+�A�q�1�����#�+�+�A�q�1����1�I���Q�V�V�V�$���1�I���Q�V�V�V�$���!3�4�<�<�Q��B����!3�4�<�<�Q��B����1�T�"���Q�V�V�q�!�f�q�!�f�-�.���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9���1�T�"���Q�V�V�i��I�>�?���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9���!3�4�<�<�Q��B����!3�4�<�<�Q��B����1�T�"���Q�V�V�q�!�f�q�!�f�-�.���1�I���Q����q�x�x��{�A�H�H�Q�K�8�9���1�T�"���Q�V�V�i��I�>�?���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9���!3�4�<�<�Q��B����!3�4�<�<�Q��B����1�T�"���Q�V�V�q�!�f�q�!�f�-�.���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9���1�T�"���Q�V�V�i��I�>�?���1�U�#���Q����q�x�x��{�A�H�H�Q�K�8�9�����1��-�-�a��A�6�!"�A���A��/�1�a�&�1�a�&�1A�B�
D������1��-�-�a��A�6�!"�A���A��/�1�a�&�1�a�&�1A�B�
D����1�T�"���Q�V�V��A���A��'�1�a�&�1�a�&�)9�:��A���A��'�1�a�&�1�a�&�)9�:�<�	=�
��1�U�#���Q�V�V��A���A��'�1�a�&�1�a�&�)9�:��A���A��'�1�a�&�1�a�&�)9�:�<�	=�
��1�T�"���Q�V�V��A���A��'�1�a�&�1�a�&�)9�:��A���A��'�1�a�&�1�a�&�)9�:�<�	=�
��1�U�#���Q�V�V��A���A��'�1�a�&�1�a�&�)9�:��A���A��'�1�a�&�1�a�&�)9�:�<�	=�
����1��-�-�a��A�6�!"�A���A��/�1�a�&�1�a�&�1A�B�
D������1�T�"���Q�V�V��1�v��1�v�.�!�Q��!�Q��0@�A�B���1�U�#���Q�V�V��1�v��1�v�.�!�Q��!�Q��0@�A�B���1�T�"���Q�V�V��1�v��1�v�.�!�Q��!�Q��0@�A�B���1�U�#���Q�V�V��1�v��1�v�.�!�Q��!�Q��0@�A�B�����1��-�-�a��A�6�!"�A���A��/�1�a�&�1�a�&�1A�B�
D������1��Q��F�3����1�T�"���Q�V�V�q�!�f�q�!�f�-�.���1�U�#���Q�V�V�q�!�f�q�!�f�-�.rYc�d�tjd�}t|�}tt	t||��tu�tt	t||��tu�tt	t||��tu�tt	t||��tu�y)Nr�)rR�eyer
rrarr
�rSrBrCs   rW�test_dot_returns_maskedarrayz2TestCompressFunctions.test_dot_returns_maskedarray:st���F�F�1�I���!�H����S��A�Y��;�.�/���S��A�Y��;�.�/���S��A�Y��;�.�/���S��A�Y��;�.�/rYc���ttjd��}ttjd��}t	|||��}t||u�t
||�y)Nr�)r�r���out)r
rRr�rrrr)rSrBr�rzs    rW�test_dot_outz"TestCompressFunctions.test_dot_outCsK���"�&�&��)����B�H�H�V�$�%���!�Q�C� ����s�
���Q��rYN)r�r�r�rmrrrur�rrrRr`�	mask_rows�	mask_colsr{r�r�r�r]rYrWr;r;s���n�`5�(/�@�[�[���V�\�2��[�[���f�n�5�!�u�u����2�R�U�U�_�_�a�4H�I�K�=�K�3�=�m/�^0�rYr;c��eZdZd�Zd�Zy)�TestApplyAlongAxisc��td�jddd�}d�}t|d|�}t|ddgddgg�y)	Nr�r<r�c��|dS�Nr?r])rCs rW�myfuncz*TestApplyAlongAxis.test_3d.<locals>.myfuncPs���Q�4�KrYr?r�rprl�rr�r'r�rSrBr��xas    rW�test_3dzTestApplyAlongAxis.test_3dMsF���3�K����1�a�(��	��f�a��
+���R�1�a�&�1�b�'�*�+rYc��td�jddd�}d
d�}t|d|d��}t|ddgdd	gg�y)Nr�r<r�c��|d|zSr�r])rC�offsets  rWr�z1TestApplyAlongAxis.test_3d_kwargs.<locals>.myfuncZs���Q�v�X�;�rYr?)r�r�r�r�)rr�r�s    rW�test_3d_kwargsz!TestApplyAlongAxis.test_3d_kwargsWsH���2�J���q�!�Q�'��	��f�a��1�
5���R�1�a�&�1�b�'�*�+rYN)r�r�r�r�r�r]rYrWr�r�Ks��,�,rYr�c��eZdZd�Zy)�TestApplyOverAxesc��td�jddd�}ttj|ddg�}tj
dgdgdggg�}t
||�t||dzjt�<ttj|ddg�}tj
d	gd
gdggg�}t
||�y)Nr.r<r�r�rr=�\�|rrI)
rr�r&rRrur
rrrir�)rSrBrT�ctrls    rW�
test_basiczTestApplyOverAxes.test_basiccs����2�J���q�!�Q�'���r�v�v�q�1�a�&�1���x�x�2�$���s�e�,�-�.���T�4� �"(��1�q�5�.�.��
���r�v�v�q�1�a�&�1���x�x�2�$���r�d�+�,�-���T�4� rYN)r�r�r�r�r]rYrWr�r�as��!rYr�c���eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zej$j'dgd���d��Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zy)�
TestMedianc��tjjtjtjgtjtjggd��}t	|tj�y)Nr0r�)rRr`r�infr�rS�rs  rW�test_pytypezTestMedian.test_pytypeosE���E�E�L�L�2�6�6�2�6�6�*�R�V�V�R�V�V�,<�=�B�L�G���Q����rYc��tjjtjjtjtjgtjtjgg�d��}t|tj�tjjtjjtjtjgtjtjgg�d��}t|tj�tjjtjjtjtjgtjtjggd��d��}t|jd�tjjtjjtjtjgtjtjggd��d��}t|jd�y)Nr0r�Tr�)rRr`rrr�rrAr�s  rW�test_infzTestMedian.test_infss����E�E�L�L����+�+�b�f�f�b�f�f�-=�.0�f�f�b�f�f�-=�-?�@�FH�
�
J���Q������E�E�L�L����+�+�b�f�f�b�f�f�-=�.0�f�f�b�f�f�-=�-?�@�FJ�
�
L���Q������E�E�L�L����+�+�b�f�f�b�f�f�-=�.0�f�f�b�f�f�-=�-?�EI�,�K� �
�
"��	�Q�V�V�T�"��E�E�L�L����+�+�b�f�f�b�f�f�-=�.0�f�f�b�f�f�-=�-?�EI�,�K�"�
�
$��	�Q�V�V�T�"rYc���tjd�}ttjj	|�d�tt
tjj	|��tu�td�}ttjj	|�d�tt
tjj	|��tu�d}ttjj	|�d�tt
tjj	|��tu�tjd�jdd�}ttjj	|d�	�tj|d�	��ttjj	|d
�	�tj|d
�	��ttjj	|d
�	�tu�tjd�jdd�}ttjj	|d�	�tj|d�	��ttjj	|d
�	�tj|d
�	��ttjj	|d
�	�tu�y)Nr�r�r�r	r�r��Hrr�r?gR@)
rRrrr`rrrar
rqr�rqs  rW�test_non_maskedzTestMedian.test_non_masked�s����I�I�a�L���R�U�U�\�\�!�_�b�)���R�U�U�\�\�!�_�%�[�8�9��!�H���R�U�U�\�\�!�_�c�*���R�U�U�\�\�!�_�%�[�8�9�
���R�U�U�\�\�!�_�b�)���R�U�U�\�\�!�_�%�[�8�9��I�I�e��$�$�Q��*���R�U�U�\�\�!�!�\�,�b�i�i���.B�C��R�U�U�\�\�!�!�\�,�b�i�i���.B�C�������Q�Q��'�{�:�;��I�I�f��%�%�a��+���R�U�U�\�\�!�!�\�,�b�i�i���.B�C��R�U�U�\�\�!�!�\�,�b�i�i���.B�C�������Q�Q��'�{�:�;rYc���ttjd�dgdzdgdzz��}ttjj|�d�ttjj|�jdd�tttjj|��tu�ttjd	�jd
d�dgdzdgdzz��}ttjj|�d
�ttjj|�jdd�tttjj|��tu�tjj|dd��}t|ddg�t|jdd�tt|�tu�y)z5test the examples given in the docstring of ma.medianr�rr�r?r�r�r]�shape mismatchrlr<r�r�r�r0T�r��overwrite_inputr�r�r;N)r
rRrrr`rrrrar
r�)rSr��ma_xs   rW�test_docstring_examplesz"TestMedian.test_docstring_examples�sE���"�)�)�A�,�a�S��U�a�S��U�]�3���R�U�U�\�\�!�_�c�*��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��"�)�)�B�-�'�'��1�-�Q�C��E�Q�C��E�M�B���R�U�U�\�\�!�_�c�*��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��u�u�|�|�A�B��|�=���T�B��8�$��T�Z�Z��'7�8���T�
�k�)�*rYc�.�d}td�D]�}dD]�}td||��}tjt||�ddg�}|D](\}}	tj
j
|||���*tj|dz|gddg�}|D]9\}}	tj
j
|||��t|||||fz������y#t$rt|||||fz��wxYw#tjj$rY�wxYw)	Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sr�)FTr?)�ndminrAFTr�)rqr
�	itertools�productrRr`r�	Exception�AssertionError�
exceptions�	AxisError)rS�msgr�rAr��argsr��overs        rW�test_axis_argument_errorsz$TestMedian.test_axis_argument_errors�sE��E���1�X�	N�E�%�
N���!�5�t�4��!�(�(���v�u�)=��t�}�M��"&�N�J�D�$�N������Q�T�4��H�N�!�(�(�E�A�I�,��)>���
�N��"&�N�J�D�$�N������Q�T�4��H�-�S�D�%��t�3L�-L�M�M�
N�
N�	N��%�N�,�S�D�%��t�3L�-L�M�M�N���=�=�2�2����s�"C�"C4�C1	�4D	�D	c��tdd��}ttjj	|�d�tdd��}ttjj	|�tjj
�y)Nr?Fr�T)r
rrRr`rrrqs  rW�test_masked_0dzTestMedian.test_masked_0d�sN���!�%� ���R�U�U�\�\�!�_�a�(��!�$����R�U�U�\�\�!�_�b�e�e�l�l�3rYc���ttjd�d��}ttjj|�tjj�ttjj|�jdd�tttjj|��tjjju�ttjd�d��}ttjj|�d�ttjj|�jdd�tttjj|��tu�ttjd�gd���}ttjj|�d	�ttjj|�jdd�tttjj|��tu�ttjd�gd
���}ttjj|�d�ttjj|�jdd�tttjj|��tu�ttjd�gd���}ttjj|�d
�ttjj|�jdd�tttjj|��tu�ttjd�gd���}ttjj|�d
�ttjj|�jdd�tttjj|��tu�ttjd�gd���}ttjj|�d	�ttjj|�jdd�tttjj|��tu�ttjd�gd���}ttjj|�d	�ttjj|�jdd�tttjj|��tu�y)Nr�Tr�r]r�Fr�r�r�)rr?r?r?r?r�)rr?r?rrr�r�r�r�r)
r
rRrrr`rrrrrarb�MaskedConstantr
rqs  rW�test_masked_1dzTestMedian.test_masked_1d�sP���"�)�)�A�,�T�*���R�U�U�\�\�!�_�b�e�e�l�l�3��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%������)B�)B�B�C��"�)�)�A�,�U�+���R�U�U�\�\�!�_�b�)��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��"�)�)�A�,�[�1���R�U�U�\�\�!�_�c�*��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��"�)�)�A�,�[�1���R�U�U�\�\�!�_�b�)��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��"�)�)�A�,�[�1���R�U�U�\�\�!�_�b�)��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��"�)�)�B�-�k�2���R�U�U�\�\�!�_�b�)��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��"�)�)�A�,�]�3���R�U�U�\�\�!�_�c�*��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9��"�)�)�B�-�m�4���R�U�U�\�\�!�_�c�*��R�U�U�\�\�!�_�*�*�B�0@�A���R�U�U�\�\�!�_�%�[�8�9rYc
���ttjjt	gd�gd����j
tjjt	gd�gd����j
�y)Nr�rr�ro)rrRr`rr
r�rSs rW�test_1d_shape_consistencyz$TestMedian.test_1d_shape_consistency�sD���R�U�U�\�\�%��W�"=�>�D�D��U�U�\�\�%��W�"=�>�D�D�	GrYc��d\}}ttjdd|��}tx|dd|ddttj||ft
���}|dd|dd�df<tjt|��}td|�D]-}tjj|�|||dd�|f<�/tt|dd�df�d�tt|�d�tt|d�	�tj|��tt|jd�	�tj|��y)
N)�er\��r�rl���r=rr?r�)rrR�linspacer�emptyrQrrcrqr�shufflerrrr�)rSr��pr�r��idxrws       rWr'zTestMedian.test_2d�s�����A�����S�"�a�0�2��!�!��#�2���3�4������1�a�&��6�7���A�$��!�Q�$���i�i��A�����q�!��	�A��I�I���c�"���f�A�a��d�G�	�	�V�A�a��d�G�_�a�(��V�A�Y��"��V�A�A�&������4��V�A�C�C�a�(�"�(�(�1�+�6rYc��ttjd�jdd��}tx|dd|ddtt
|�d�tttjj
|��tu�tt
|d��gd��tttjj
|d���tu�tt
|d	��gd
��tttjj
|d	���tu�tt
|d	��jgd��y)Nr\rlr�����-@rr�)�+@r�g/@r?)
rrrrlr@r5rErrr)
r?r?r?rrrrr?r?r?)rrRrr�rrrrrar`r
rArqs  rW�
test_2d_waxiszTestMedian.test_2d_waxiss�������2��.�.�r�1�5�6�����"�1���"�#���V�A�Y��%���R�U�U�\�\�!�_�%�[�8�9��V�A�A�&�(:�;���R�U�U�\�\�!�!�\�,�-��<�=��V�A�A�&�(J�K���R�U�U�\�\�!�!�\�,�-��<�=��V�A�A�&�+�+�-K�LrYc���tjjd�jddd�}t||dzdk(<tt
|d�ddgdd	gddgd
d	gg�d|_tt
|d�dd
gddgddgg�tjjd�jddd�}t||dzdk(<tt
|d�dd
gddgddgg�y)Nr.r�r�r<rr�r�r�rBrD)r�r�r<�crlr�r@rAr�r�r5rC)rRr`rr�rrrrrqs  rWr�zTestMedian.test_3ds����E�E�L�L���$�$�Q��1�-����!�a�%�1�*�
��V�A�q�\�R��G�a��W�r�1�g��B�x�#H�I�����V�A�q�\�R��H�r�2�h��R��#A�B��E�E�L�L���$�$�Q��1�-����!�a�%�1�*�
��V�A�q�\�R��H�q�!�f�r�2�h�#?�@rYc��ttjd�jdd��}tx|dd|ddtt
|d��t
|d���y)Nr\rlr�r�r0r�r?)rrRrr�rrrrqs  rW�
test_neg_axiszTestMedian.test_neg_axissO������2��.�.�r�1�5�6�����"�1���"�#���V�A�B�'����):�;rYc�J�dD]�}ttj|��}tx|dd|ddttjd��}t||��}|dk(r
t
|d�nt
|d�t||u�tt|�tu���y)	N)r\�>@rg?@r�r�r]r�r\r�g.@)
rrRrrrrrrrar
)rS�vr�r�r�s     rW�test_out_1dzTestMedian.test_out_1ds���#�
	,�A��R�Y�Y�q�\�*�A�#�#�A�b�q�E�A�b�c�F��r�w�w�r�{�+�C��q�c�"�A��B�w��S�$�'��S�#�&��A��H���D��G�{�*�+�
	,rYc��dD�]}ttj|�jdd��}tx|dd|ddttj
d��}t
|d|��}|dk(r0td	gdzgd
�zd	gdzzdgdzdgd
zzdgdzz��}n/td	gdzgd�zd	gdzzdgdzdgd
zzdgdzz��}t||�t||u�tt|�tu���y)N)rFgD@r\r�rlr0r�r�r?)r�r�r\r�)rlr@r5rETFr�r�)r�g�1@g�5@g�9@)rrRrr�rrrrrrar
)rSr�r�r�r��es      rW�test_outzTestMedian.test_out,s��#�
	,�A��R�Y�Y�q�\�1�1�"�b�9�:�A�#�#�A�b�q�E�A�b�c�F��r�w�w�r�{�+�C��q�q�c�*�A��B�w� �"��a��*:�!:�b�T�!�V�!C�'+�f�q�j�E�7�Q�;�&>�$��!��&K�M��!�"��a��*B�!B�b�T�!�V�!K�'+�f�Q�h�%����&:�d�V�A�X�&E�G����A���A��H���D��G�{�*�+�
	,rYr�)Nr?r\r�)r�r0)�argnames�	argvaluesc�����tjdt��}tjjd�tj|j
�dd�dfz}|j
tj�}tj|t|�<ttj|j
�|���|�d�jz}n>t|�j��t��fd�t�j�D��}ttj|��}t!�|d|��}||usJ�t#|j
|�y)	N)r�r�rpr�r=)r���r�r\c3�H�K�|]}|�vrdn�j|���y�w)r?N)r)�.0rw�	axis_normrs  ��rW�	<genexpr>z/TestMedian.test_keepdims_out.<locals>.<genexpr>Rs+�����L�89�Q�)�^�������3�L�s�"T)r�r�r�)rRrr�rr
rri�intpr��tuplerr�ndimrrqr�rr)	rSr�rAr��	shape_outr�r�r�rs	       @@rW�test_keepdims_outzTestMedian.test_keepdims_out=s����x�x�
�T�2���I�I���X�&����$�*�*�)=�a��g�)F�F��
�H�H�R�W�W�������U�1�X���������,�4�8���<��q�v�v�
�I�,�T�1�6�6�:�I��L�=B�1�6�6�]�L�L�I��2�8�8�I�.�/�����t��=����}��}��V�\�\�9�-rYc��ddgddgddgg}tjj|d�}ddg}ttjj	|d��|�y)Nr�r�r�rr�)rRr`�masked_equalrr)rSrtrgr�s    rW�$test_single_non_masked_value_on_axisz/TestMedian.test_single_non_masked_value_on_axisYs]���R���R���R�����U�U�'�'��a�0�
���8���2�5�5�<�<�
��<�;�#�	%rYc�R�dtjdt��fD�]}tjj	dtj
dggd�g�}||_tjj|d��}ttj|��t|tj
�tjj|j�d��}ttj|��t|tj
�tjj|d��}tt|�t�t|dtj
dg�tjj|d��}tt|�t�t|tj
d	g�tjj|d
��}tt|�t�t|tj
d	g���tjj	dtj
dggd�g�}tjj|dd�d	f<ttjj|d��tj
�ttjj|d��dtj
dg�ttjj|d��tj
dg�y)NFr�r=r?r�r�r�rr<r0r�)rRrr�r`r
r�rArr�isscalarr�ravelrrar
r)rSrA�dmr�s    rW�test_nanzTestMedian.test_nanbs����B�H�H�Q�d�3�4�	/�D������q�"�&�&�!�n�i�8�9�B��B�G������R�d��+�A��B�K�K��N�#��q�"�&�&�)������R�X�X�Z�a��0�A��B�K�K��N�#��q�"�&�&�)������R�a��(�A���a��+�.��q�1�b�f�f�a�.�1������R�a��(�A���a��+�.��q�2�6�6�1�+�.������R�b��)�A���a��+�.��q�2�6�6�1�+�.�)	/�,�U�U�[�[�1�b�f�f�a�.�)�4�
5���5�5�<�<��1�a�4���2�5�5�<�<���<�6����?��2�5�5�<�<���<�3�a�����^�D��2�5�5�<�<���<�3�b�f�f�c�]�CrYc��tjjtjd��}tjjtjd��}tj
|d<tjj|d<ttjj|d|��|�tjjtjd��}ttjj|d|��|�tjjtjd	��}ttjj||��|�y)
N)r��r�r��r<r?rJrr��r�r?r])	rRr`rrrr�rrr)rS�ors   rW�test_out_nanzTestMedian.test_out_nans����E�E���r�x�x��~�.���E�E���r�w�w�v��/���&�&��$���%�%�,�,��$���R�U�U�\�\�!�Q�A�\�.��2��E�E���r�x�x��~�.���R�U�U�\�\�!�Q�A�\�.��2��E�E���r�x�x��|�,���R�U�U�\�\�!��\�+�Q�/rYc��tjjtjdt���}tjj
|ddd�<tj|d<ttjj|�tj�ttjj|d��tj�tjjtjdt��jddd��}tj|j�dzdk(|_|j�}tj|d	<tj|d
<ttjj|�tj�ttjtjj|���tjj|d��}tj|d<tj|d<ttjj|d�|�tjj|d��}tj|d
<tj|d<ttjj|d�|�tjj|d��}tj|d<tj|d<ttjj|d�|�y)Nr.r=r�r<rr�r�r?r�)r?r?r<�r<r�r��r?r�rg)rRr`rrrQrr�rrr�r�rA�copyrr�r)rSrB�aorigrCs    rW�test_nan_behaviorzTestMedian.test_nan_behavior�s���E�E���r�y�y��5�9�:��������#�A�#���v�v��!���2�5�5�<�<��?�B�F�F�3��2�5�5�<�<���<�2�B�F�F�;��E�E���r�y�y��5�9�A�A�!�Q��J�K�����1�6�6�"�Q�&�!�+���������V�V��'�
��V�V��'�
�	�2�5�5�<�<��?�B�F�F�3�����B�E�E�L�L��O�,�-�
�E�E�L�L��Q�L�'���&�&��$���&�&��$���R�U�U�\�\�!�Q�'��+�
�E�E�L�L��Q�L�'���&�&��$���&�&��$���R�U�U�\�\�!�Q�'��+�
�E�E�L�L��V�L�,���v�v��!���v�v��!���R�U�U�\�\�!�V�,�a�0rYc��tjgd�gd�gtj��}tjj	||dk(��}ttjj
|d��d�ttjj
|d��jd�ttjj
|d	��|d	�ttjj
|�d�y)
N)r�r��r=r�r�r?r�rFr)rRr
�uint8r`rrrrAr�s  rW�test_ambigous_fillzTestMedian.test_ambigous_fill�s����H�H�k�;�/�r�x�x�@���E�E���q�q�A�v��.���2�5�5�<�<���<�2�C�8��2�5�5�<�<���<�2�7�7��?��2�5�5�<�<���<�2�A�a�D�9��2�5�5�<�<��?�C�0rYc	��tjtjfD�]�}tj|tjgtjtjgg�}tjj|tj|���}ttjj|d��|tjg�ttjj|d��|tjg�ttjj|�|�tjtjtj|gtjtj|gg�}tjj|tj|���}ttjj|d��|�ttjj|d��jd�ttjj|d��|d�ttjj|�|�tj||g||gg�}ttjj|�|�ttjj|d��|�ttjj|d��|�tj|d|dgdtjtjd	gd
tjtj|ggtj��}tjj|tj|���}|dkDrZttjj|d��dd
|dg�ttjj|�d�nYttjj|d��dd
|dg�ttjj|�d�ttjj|d��dd|g�tdd�D�]}tdd�D�]}tjtjg|z|g|zzgdz�}tjj|tj|���}ttjj|�|�ttjj|d��|�ttjj|d��tjg|z|g|zz��������y)Nr�rr�r?Frpi����r�r�r�r=r�r�r�g@g$�g"�g�r�rlr<)
rRr�r
r�r`r�isnanrrrrA�float32rq)rSr�rBrwrxs     rW�test_specialzTestMedian.test_special�s����F�F�R�V�V�G�$�(	=�C����3����.�2�6�6�2�6�6�*:�;�<�A����"�"�1�2�8�8�A�;�"�7�A�������a�a��0�3����.�A�������a�a��0�3����.�A�������a��#�.����2�6�6�2�6�6�3�/�"�&�&�"�&�&�#�1F�G�H�A����"�"�1�2�8�8�A�;�"�7�A��r�u�u�|�|�A�A�|�6��<��r�u�u�|�|�A�A�|�6�;�;�U�C��r�u�u�|�|�A�A�|�6��!��=��r�u�u�|�|�A���4����3��*�s�C�j�1�2�A�������a��#�.�������a�a��0�#�6�������a�a��0�#�6����3��C�4��,���������2��b�f�f�b�f�f�c�2�4�!#�
�
�,�A����"�"�1�2�8�8�A�;�"�7�A��Q�w��R�U�U�\�\�!�!�\�4�r�2��t�R�6H�I��R�U�U�\�\�!�_�c�2��R�U�U�\�\�!�!�\�4�t�R�#��s�6K�L��R�U�U�\�\�!�_�d�3�������a�a��0�3��c�2B�C��1�b�\�
=���q�"��=�A����B�F�F�8�a�<�S�E�A�I�">�!?�!�!C�D�A����*�*�1�2�8�8�A�;�*�?�A� ������a��#�6� ������a�a��!8�#�>� ������a�a��!8�#%�6�6�(�Q�,�3�%�!�)�!;�=�=�
=�C(	=rYc�6�tjjtjgt���}t�5}|j
t�ttjj|�tj�t|jdjtu�ddd�tjjtjgtd���}t�5}|j
t�tjddt�ttjj|�tj�t|jdjtu�ddd�tjjtjgtd���}t!tjj|d��|�t!tjj|d	��|�tjjtjtjtd���}tj"d
��5}tjddt�t!tjj|d��|�t|djtu�ddd�y#1swY��?xYw#1swY��qxYw#1swYyxYw)Nr=rr�)r>r��always�r<r�r?T)�record)rRr`rr
rQrr�RuntimeWarningrrr�r�log�category�warnings�filterwarningsr�catch_warnings)rSrBr�rCs    rW�
test_emptyzTestMedian.test_empty�s����E�E���r�x�x��%�8�9��
�
 �	9�A�
�H�H�^�$��r�u�u�|�|�A�����7��A�E�E�!�H�%�%��7�8�	9�
�E�E���r�x�x��%�q�A�B��
�
 �	9�A�
�H�H�^�$��#�#�H�b�.�A��r�u�u�|�|�A�����7��A�E�E�!�H�%�%��7�8�		9�
�E�E���r�x�x��%�q�A�B���R�U�U�\�\�!�!�\�,�a�0��R�U�U�\�\�!�!�\�,�a�0�
�E�E���r�x�x����e�1�E�F��
�
$�
$�D�
1�	5�Q��#�#�H�b�.�A�������a�a��0�!�4��A�a�D�M�M�^�3�4�	5�	5�+	9�	9��	9�	9��	5�	5�s'�A5K5�BL�A%L�5K?�L�Lc	��tjjtjd��}t	ttjj
|jt���t�tj|d<t	ttjj
|jt���t�y)Nr�r<)rRr`rrrrarrir_rQr�)rSr�s  rW�test_objectzTestMedian.test_object�su���E�E���r�y�y��}�-����R�U�U�\�\�!�(�(�6�"2�3�4�e�<��v�v��!����R�U�U�\�\�!�(�(�6�"2�3�4�e�<rYN)r�r�r�r�r�r�r�r�r�r�r�r'r�r�r�r�r�r�rrr�r�r�r�rrrrrr]rYrWr�r�ns��� �#�$<�*
+�N�04�$:�LG�7� 
M�	A�<�
,�,�"�[�[����
��	�.�	�.�$%�D�:	0�!1�F1�)=�V5�:=rYr�c�0�eZdZd�Zd�Zd�Zd�Zd�Zd�Zy)�TestCovc�^�ttjjd��|_y�Nr�)r
rRrrrtr�s rW�setup_methodzTestCov.setup_methods���"�)�)�.�.��,�-��	rYc�n�|j}tt|d��djtj
�tt||d��djtj
�|dkD}t
ttjj||�d��djt�|jdd��t
ttjj||�|d��djt�t	j||f��y)NT��rowvarr?F)r�rr�r0)
rtrr7r>rRrrr`rrir�r�r6)rSr�rAs   rW�test_covhelperzTestCov.test_covhelper
s����I�I���
�1�T�*�1�-�3�3�R�Z�Z�@��
�1��%�0��3�9�9�2�:�:�F��3�w�������"�"�1�d�+�D�
��
�����
�\�\�!�R�
 � �		
�	�����"�"�1�d�+�q��
��
������I�I��u�t�e�n�%�		
rYc� �|j}ttj|�t|��ttj|d��t|d���ttj|dd��t|dd���y)NFrT�r�bias)rtr	rRrrqs  rW�test_1d_without_missingzTestCov.test_1d_without_missings`���I�I���B�F�F�1�I�s�1�v�.��B�F�F�1�U�3�S��5�5I�J��B�F�F�1�U��>���%�d�;�	=rYc�@�|jjdd�}ttj|�t	|��ttj|d��t	|d���ttj|dd��t	|dd���y)Nr�r�FrTr )rtr�r	rRrrqs  rW�test_2d_without_missingzTestCov.test_2d_without_missing&sm���I�I���a��#���B�F�F�1�I�s�1�v�.��B�F�F�1�U�3�S��5�5I�J��B�F�F�1�U��>���%�d�;�	=rYc
��|j}t|d<||j�z}|j�}t	tj|�t
|��t	tj|d��t
|d���t	tj|dd��t
|dd���	t
|d��|dd}t	tj||ddd��t
||ddd���t	tj||ddd�d��t
||ddd�d���t	tj||ddd�dd��t
||ddd�dd���y#t$rY��wxYw)Nr0FrTr ��allow_maskedr?)rtr�meanrvr	rRrr�)rSr��nxs   rW�test_1d_with_missingzTestCov.test_1d_with_missing.sI���I�I����"��	�Q�V�V�X�
��
�\�\�^���B�F�F�2�J��A��/��B�F�F�2�e�4�c�!�E�6J�K��B�F�F�2�e�$�?���%�d�;�	=�	����&�
�q��W���B�F�F�2�r�$�B�$�x�0�#�a��4�R�4��/�B��B�F�F�2�r�$�B�$�x��>���1�T�r�T�7�5�9�	;��B�F�F�2�r�$�B�$�x��D�I���1�T�r�T�7�5�t�D�	F���	��	�s�<
E;�;	F�Fc���|j}t|d<|jdd�}tjt|��j
t�}tj||j�}||jd�dd�dfz
jd�}tt|�tj|�|jddz
z|dz
z�tt|d��tj|d��|jdz|z�tj|j|�}||jd�z
jd�}tt|d	�
�tj|d	�
�|jddz
z|dz
z�tt|d	d��tj|d	d��|jdz|z�y)Nr0r�r�r?rr�T�r!Frr )rtrr�rR�logical_notrrirrrr�r(rr	rr)rSr��valid�frac�xfs     rW�test_2d_with_missingzTestCov.test_2d_with_missingFs����I�I����"��
�I�I�a��O�����|�A��/�6�6�s�;���v�v�e�U�W�W�%���!�&�&��)�A�t�G�$�$�
,�
,�Q�
/���C��F��F�F�2�J�!�'�'�!�*�q�.�9�T�B�Y�G�	I��C���-��F�F�2�D�1�A�G�G�A�J�>��E�	G��v�v�e�g�g�u�%���!�&�&��)�m�
#�
#�A�
&���C��%�0��V�V�B�u�5��g�g�a�j�1�n�.�15���<�	>�	�C��%�d�;��V�V�B�u�4�@��W�W�Q�Z�(�*.�/�	1rYN)	r�r�r�rrr"r$r*r1r]rYrWrrs!��.�
�(=�=�F�01rYrc�6�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	y)	�TestCorrcoefc��ttjjd��|_ttjjd��|_yr)r
rRrrrt�data2r�s rWrzTestCorrcoef.setup_method^s4���"�)�)�.�.��,�-��	��2�9�9�>�>�"�-�.��
rYc�`�|j|j}}tj|�}tj||�}t	�5}tjd�ttt|d��|jtd�ttj|d��t|d���tt|d��|�tt||d��|�tt|d��|�tt||d��|�ddd�y#1swYyxYw)Nr
r0��ddof�bias and ddof have no effectrr��rtr5rRrrr�simplefilterrry�filterr	)rSr�r�r��	expected2�sups      rW�	test_ddofzTestCorrcoef.test_ddofbs����y�y�$�*�*�1���;�;�q�>���K�K��1�%�	�
�
 �		C�C��!�!�(�+��+�X�q�r�B��J�J�)�+I�J�����A�A� 6����8K�L����� 4�h�?����A�B� 7��C����� 3�X�>����A�A� 6�	�B�		C�		C�		C�s
�C
D$�$D-c	��|j|j}}tj|�}t	�5}tjd�ttt||dd�ttt||dd�ttt|d��|jtd�tt|d��|�ddd�y#1swYyxYw)Nr
TFr,r9r?r:)rSr�r�r�r>s     rW�	test_biaszTestCorrcoef.test_biasrs����y�y�$�*�*�1���;�;�q�>��
�
 �	?�C��!�!�(�+��+�X�q�!�T�5�I��+�X�q�!�T�4�H��+�X�q�u�E��J�J�)�+I�J����� 3�X�>�	?�	?�	?�s�B
C�Cc	��|j}ttj|�t|��ttj|d��t|d���t	�5}|jtd�ttj|dd��t|dd���ddd�y#1swYyxYw)NFrr9Tr )rtr	rRrrr<ry�rSr�r>s   rWr"z$TestCorrcoef.test_1d_without_missings����I�I���B�K�K��N�H�Q�K�8��B�K�K��%�8�$�Q�u�5�	7�
�
 �	F�C��J�J�)�+I�J�����A�e�$� G� (��5�t� D�
F�	F�	F�	F�s
�+AB9�9Cc	��|jjdd�}ttj|�t	|��ttj|d��t	|d���t�5}|j
td�ttj|dd��t	|dd���ddd�y#1swYyxYw)Nr�r�Frr9Tr )rtr�r	rRrrr<ryrCs   rWr$z$TestCorrcoef.test_2d_without_missing�s����I�I���a��#���B�K�K��N�H�Q�K�8��B�K�K��%�8�$�Q�u�5�	7�
�
 �	F�C��J�J�)�+I�J�����A�e�$� G� (��5�t� D�
F�	F�	F�	F�s
�;AC	�	Cc�T�|j}t|d<||j�z}|j�}t	tj|�t
|��t	tj|d��t
|d���t�5}|jtd�t	tj|dd��t
|dd���ddd�	t
|d��|dd}t	tj||ddd��t
||ddd���t	tj||ddd�d��t
||ddd�d���t�5}|jtd�t	tj||ddd��t
||ddd�d�	��t	tj||ddd��t
||ddd�d
���ddd�y#1swY��'xYw#t$rY��(wxYw#1swYyxYw)Nr0Frr9Tr r&r?r,r<r7)rtrr(rvr	rRrrr<ryr�)rSr�r)r>s    rWr*z!TestCorrcoef.test_1d_with_missing�s����I�I����"��	�Q�V�V�X�
��
�\�\�^���B�K�K��O�X�a�[�9��B�K�K��5�9�$�Q�u�5�	7�
�
 �	F�C��J�J�)�+I�J�����B�u�4� H� (��5�t� D�
F�	F�	��Q�U�+��q��W���B�K�K��B�t��t�H�5�x��1�T�r�T�7�7K�L��B�K�K��B�t��t�H�U�C�$�Q��$�B�$���>�	@�
�
 �	>�C��J�J�)�+I�J�����B��4�R�4�� 9� (��A�d��d�G�!� <�
>�����B��4�R�4�� 9� (��A�d��d�G�!� <�
>�	>�	>�	F�	F���	��	��	>�	>�s,�AH�%
H�1BH�H�	H�H�H'c�.�|j}t|d<|jdd�}t|�}t	j|�}t|dd�dd�f|dd�dd�f�t
�5}|jtd�tt|d��dd�dd�f|dd�dd�f�tt|d��dd�dd�f|dd�dd�f�tt|d��dd�dd�f|dd�dd�f�ddd�y#1swYyxYw)	Nr0r�r�r9r1r7r?r,)	rtrr�rrRr	rr<ry)rSr�rTrUr>s     rWr1z!TestCorrcoef.test_2d_with_missing�s3���I�I����"��
�I�I�a��O����{���+�+�a�.���D��"��c�r�c��N�G�C�R�C��"��H�,=�>�
�
 �	3�C��J�J�)�+I�J����� 4�S�b�S�#�2�#�X� >� '����S�b�S�� 1�
3����� 3�C�R�C��"��H� =� '����S�b�S�� 1�
3����� 3�C�R�C��"��H� =� '����S�b�S�� 1�
3�	3�	3�	3�s
�0BD�DN)
r�r�r�rr?rAr"r$r*r1r]rYrWr3r3\s(��/�C� ?�	F�	F�>�<3rYr3c��eZdZd�Zd�Zy)�TestPolynomialc��tjjd�}tjjd�jdd�}t	t||d�tj
||d��|j
t�}t|d<|j
t�}tx|d<|d<t||dd�dfdd	�
�\}}}}}tj
|dd|dd�dfj�dd	�
�\}}	}
}}t|||||f||	|
||f�D]\}
}t	|
|��t||dd�dfdd	�
�\}}}}}tj
|dd|dd�dfdd	�
�\}}	}
}}t|||||f||	|
||f�D]\}
}t	|
|��t||dd	�
�\}}}}}tj
|dd|dd�dd�fdd	�
�\}}	}
}}t|||||f||	|
||f�D]\}
}t	|
|��tjjd�dz}|j�}|dd}|dd}|dd}t||dd	|��\}}}}}tj
||dd	|��\}}	}
}}t||�t|||||f||	|
||f�D]\}
}t	|
|��y)
NrlrZr0r<r�rrH�r0r0T�rsr?)rsr�)
rRrrr�r	r�viewr
rrv�zipr�r)rSr�r��C�R�K�S�Dr[r�ryr{rrB�a_r��wo�xs�ys�wss                    rW�test_polyfitzTestPolynomial.test_polyfit�s���
�I�I�N�N�2����I�I�N�N�2��&�&�r�1�-���G�A�q�!�,�b�j�j��A�q�.A�B�
�F�F�;�����!��
�F�F�;���$�$��$��!�F�)�!�!�Q�q�!�t�W�a�d�;���A�q�!�Q��*�*�Q�q�r�U�A�a�b�!�e�H�,?�,?�,A�1�*.�0���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�	'�G�Q����2�&�	'�"�!�Q�q�"�u�X�q�t�<���A�q�!�Q��*�*�Q�q��W�a��"��b��k�1�4�H���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�	'�G�Q����2�&�	'�"�!�Q���5���A�q�!�Q��*�*�Q�q��W�a��"��Q��i���F���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�	'�G�Q����2�&�	'�
�I�I�N�N�2���"��
�V�V�X��
�q��W��
�q��W��
�q��W��!�!�Q����:���A�q�!�Q��*�*�R��Q�T�R�@���A�q�!�Q��Q����A�q�!�Q��?�Q��1�a��O�<�	'�G�Q����2�&�	'rYc�8�tjjd�}tjjd�jdd�}tj|d<tj|d<|jt�}|jt�}t|d<t|d<t||dd�	�\}}}}}tj|d
d|d
d�dd�fdd�	�\}}	}
}}t|||||f||	|
||f�D]\}
}t|
|��y)NrlrZr0r<rrJr�TrKr?)rRrrr�r�rLr
rrrMr	)rSr�r�rNrOrPrQrRr[r�ryr{rrBrSs               rW�test_polyfit_with_masked_NaNsz,TestPolynomial.test_polyfit_with_masked_NaNs�s����I�I�N�N�2����I�I�N�N�2��&�&�r�1�-���v�v��!���6�6��%��
�F�F�;���
�F�F�;�����!����%��!�!�Q���5���A�q�!�Q��*�*�Q�q��W�a��"��Q��i���F���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�	'�G�Q����2�&�	'rYN)r�r�r�rXrZr]rYrWrHrH�s
��%'�N'rYrHc�r�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zd�Zd�Zd�Zy)�TestArraySetOpsc���gd�}t|dd��}tt|dt��t	|dtgd�gd����t	|dgd	��t	|d
gd��y)N�r?r?r?r<r<r�T��return_index�return_inverserr�rr�r?)rr�r�r<)rrrr?r?r<)rrrr
rrr2s   rW�test_unique_onlistz"TestArraySetOps.test_unique_onlistsZ��!���d��d�C���
�4��7�K�0�1��T�!�W�l�9�9�E�F��T�!�W�i�(��T�!�W�0�1rYc��tgd�gd���}t|dd��}t|dtgd�gd����t|d	gd
��t|dgd��d
|_tgd�gd�d
��}t|dd��}t|dtgd�gd����t|d	gd
��t|dgd��y)Nr^)rrr?rr?rr�Tr_r)r?r<r�r0�rrrr?r?)rr�r�r<r<)rrr�r?r�r<r�)rtrA�
fill_value)rrrrer2s   rW�test_unique_onmaskedarrayz)TestArraySetOps.test_unique_onmaskedarrays����.�5G�H���d��d�C���T�!�W�l�=�|�L�M��T�!�W�l�+��T�!�W�0�1�����!3�!3��C���d��d�C���T�!�W�l�=�|�L�M��T�!�W�l�+��T�!�W�0�1rYc�X�tgd�d��}t|dd��}t|dtdgdg���t|ddg�t|dgd��t}t|dd��}t|dtt��t|ddg�t|ddg�y)	Nr?Tr�r_rr?r<r)rrrrr2s   rW�test_unique_allmaskedz%TestArraySetOps.test_unique_allmaskeds����I�D�1���d��d�C���T�!�W�l�A�5��v�>�?��T�!�W�q�c�"��T�!�W�i�(����d��d�C���T�!�W�l�6�2�3��T�!�W�q�c�"��T�!�W�q�c�"rYc�,�ttjd�gd���}tgd�gd���}t	|�}t||�t|j
d�|j
d��t|j|j�y)Nr��r?rrrr?r�)r?r?r?r�)r?rrr?r)rrRrr
r%rrrA)rSr�rUrTs    rW�test_ediff1dzTestArraySetOps.test_ediff1d+sb������1��O�<����<�8���q�z���T�7�#��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-rYc�$�ttjd�gd���}t|t��}tgd�gd���}t
||�t
|jd�|jd��t
|j|j�t|gd���}tgd	�gd
���}t
||�t
|jd�|jd��t
|j|j�y)Nr�rjr�)�to_begin)rr?r?r?r�)r?r?rrr?rr�)r?r<r�r?r?r?r�)rrrr?rrr?�	rrRrr%rr
rrrA�rSr�rTrUs    rW�test_ediff1d_tobeginz$TestArraySetOps.test_ediff1d_tobegin4s�������1��O�<���q�6�*����o�>���T�7�#��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-��q�9�-���-�4I�J���T�7�#��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-rYc�$�ttjd�gd���}t|t��}tgd�gd���}t
||�t
|jd�|jd��t
|j|j�t|gd���}tgd	�gd
���}t
||�t
|jd�|jd��t
|j|j�y)Nr�rjr�)�to_end)r?r?r?r�r)r?rrr?r?rr�)r?r?r?r�r?r<r�)r?rrr?rrrrnros    rW�test_ediff1d_toendz"TestArraySetOps.test_ediff1d_toendCs�������1��O�<���q��(����o�>���T�7�#��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-��q��+���-�4I�J���T�7�#��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-rYc�8�ttjd�gd���}t|tt��}tgd�gd���}t
||�t
|jd�|jd��t
|j|j�t|gd�t��}tgd	�gd
���}t
||�t
|jd�|jd��t
|j|j�y)Nr�rjr��rrrm)rr?r?r?r�r)r?r?rrr?r?rr�)rr?r?r?r�r?r<r�)r?r?rrr?rrrrnros    rW�test_ediff1d_tobegin_toendz*TestArraySetOps.test_ediff1d_tobegin_toendRs�������1��O�<���q��&�9���*�1C�D���T�7�#��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-��q��V�<���0�5�7���T�7�#��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-rYc�X�tjd�}t|�}tgd�gd���}t	||�tt
|t��t	|jd�|jd��t	|j|j�t|tt��}tgd�gd���}tt
|t��t	|jd�|jd��t	|j|j�y)	Nr��r?r?r?r?�rrrrr�rrur�r�)rRrr%r
rrrr
rrArros    rW�test_ediff1d_ndarrayz$TestArraySetOps.test_ediff1d_ndarraybs����I�I�a�L���q�z����<�8���T�7�#��
�4��-�.��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-��q��&�9���*�1C�D���
�4��-�.��T�[�[��^�W�^�^�A�%6�7��T�Y�Y����-rYc��tgd�gd���}tgd�gd���}t||�}tgd�gd���}t||�y)N)r?r�r�r�rdr�)r�r?r?r?)r?r�r0r�)r
r#r)rSr�r�rTrUs     rW�test_intersect1dz TestArraySetOps.test_intersect1drs<���,�\�2���,�\�2���1�a� ���
��3���T�7�#rYc�v�tgd�gd���}tgd�gd���}t||�}t|tgd���tgd�gd���}gd�}t||�}t|tgd�gd	����tgd
��}tgd��}t||�}tt	|t
��t|gd��tgd
�gd���}tgd�gd	���}t||�}tt	|t
��t|gd��t
gtgg��y)N�r?r<r�rpr0�rrrrr?r��r?r<r�r�r�r0r~)r�r�rp�r?r<r�r�r��r�r�rpr0rdr�)r�r�r�r�r?r�r<r��rr?rr�r�r�r�r�)r
r rrrr
r�rSrBrCrTs    rW�
test_setxor1dzTestArraySetOps.test_setxor1dzs����"��9���%�,>�?����1�~���T�5��+�,��"��9������1�~���T�5��\�B�C��)����)�����1�~���
�4��-�.��T�-�.��,�\�2���,�\�2����1�~���
�4��-�.��T�-�.��2�x��B�/�0rYc���tgd�gd���}gd�}t||d��}t|tgd�gd����tgd	�gd
���}tgd�gd���}t||d��}tt	|t
��t|gd��td
gdgdgdgg�}tddgddgg�}t||d��}tt	|t
��t|gd��y)Nr~rr�r�T)�
assume_uniquer�rdr�r�r�rr?r�r<r�r�r�r�)r
r rrrr
r�s    rW�test_setxor1d_uniquez$TestArraySetOps.test_setxor1d_unique�s����"��9������1�D�1���T�5��\�B�C��,�\�2���,�\�2����1�D�1���
�4��-�.��T�-�.��A�3���a�S�1�#�&�'���A�q�6�A�q�6�"�#����1�D�1���
�4��-�.��T�-�.rYc���tjd�jgd��}tjgd��}d|d<t	||��}t	gd�gd���}tdt
�	�}d
|d<d
|d<d
|d
<t
||�}tt|t��t||�tj|||j�|jz}t||�y)Nr.r�r?)r?r<rr�)	rrlrZr\r?r�r�r6r^)	rr?rr?rr?rr?r)rtrAr=Trr�)rr<r�)rRrr�rr
r�r2rrr
rrA)rSrBrArC�ecr[rs       rW�	test_isinzTestArraySetOps.test_isin�s���
�I�I�b�M�!�!�)�,���x�x�	�"����W�
��!�$����:�:�
<��
�9�D�
)����7����7����7����A�J���
�1�k�*�+��1�b�!��G�G�A�q�!�&�&��z�"�a�f�f�W�,���1�a� rYc� �tgd�gd���}tgd�gd���}t||�}t|gd��tgd�gd���}tgd�gd	���}t||�}t|gd
��tgtgg��y)Nr~rr�r�r~)TTTFT�r�r�r<r?r0�r?r�r0r�)TTFTT)r
r$rrr�s    rW�	test_in1dzTestArraySetOps.test_in1d�sr���"��9���%�,>�?���A�q�z���T�:�;��"��9���*�9�-���A�q�z���T�:�;��2�t�B��|�,rYc��tgd�gd���}tgd�gd���}ttjt	||��t	||d���tgd�gd���}tgd	�gd
���}ttjt	||��t	||d���tgt	ggd���y)Nr~rr�r�r~T)�invertr�r�r�)r
rrRr�r$rr�s   rW�test_in1d_invertz TestArraySetOps.test_in1d_invert�s����"��9���%�,>�?���R�Y�Y�t�A�q�z�*�D��A�d�,C�D��"��9���*�9�-���R�Y�Y�t�A�q�z�*�D��A�d�,C�D��2�t�B��4�8�9rYc�l�tgd�gd���}tgd�gd���}t||�}tgd�gd���}t||�tgd�gd�ggd	�gd
�g��}tgd�gd���}tgd
�gd���}t||�}t||�tgtgg��y)N)r?r<r�rpr�r0r~r�r�)r?r<r�r�r�rpr0)rrrrrrr?r�r>rr�r/r)rr?r<r�r�r�)r
r"rr)	rSrBrCrTrUr�r��ezr�s	         rW�test_union1dzTestArraySetOps.test_union1d�s����%�,>�?���%�,>�?���q�!�}���.�5J�K���T�7�#�
�9�i�(�	�9�/E�F���/��8��
�%�,>�
?���A�q�M���Q����2�w�r�2��/rYc��tgd�gd���}tgd��}t||�}t|tgd�gd����td�}td�}tt||�tdd	g��tgtj
g��}tt|g�jtj
�y)
N)r�r�r�rprpr?r<r?)rrrrrrrr?r�)r<r�r�r�r<r?r�)r�rpr0r�rlr�r�)r
r!rrrR�uint32r>r�s    rW�test_setdiff1dzTestArraySetOps.test_setdiff1d�s����*�1I�J���'�(����A����T�5��)�<�=��2�J���1�I���Y�q�!�_�e�Q��F�m�4��"�b�i�i�b�)���Y�q�"�%�+�+�R�Y�Y�7rYc��tjgd��}tjgd��}tt||�tjdg��y)N)rBrCr[)rBrCr{r[)rRr
rr!r�s   rW�test_setdiff1d_char_arrayz)TestArraySetOps.test_setdiff1d_char_array�s8���H�H�_�%���H�H�_�%���9�Q��?�B�H�H�c�U�O�<rYN)r�r�r�rbrfrhrkrprsrvrzr|r�r�r�r�r�r�r�r�r]rYrWr\r\sW��2�2� 
#�.�
.�
.�.� .� $�1�4/�&!�*-�
:�0�$8�=rYr\c��eZdZd�Zd�Zy)�
TestShapeBasec���tgd�gd���}t|�}t|jd�t|jj|j
j�t|jd�t|jj|j
j�t|jj|j
j�y)Nr�ror�r�r�)rrrrrArtr�s   rW�test_atleast_2dzTestShapeBase.test_atleast_2d�s������3���q�M���Q�W�W�f�%��Q�V�V�\�\�1�6�6�<�<�0��Q�W�W�d�#��Q�V�V�\�\�1�6�6�<�<�0��Q�V�V�\�\�1�6�6�<�<�0rYc�Z�td�}t|jd�t|jj|j�t|jj|j�tdd�}|D]l}t|jd�t|jj|j�t|jj|j��ntd�}t|jd�t|jj|j�t|jj|j�tdd�}|D]l}t|jd�t|jj|j�t|jj|j��nt
d�}t|jd�t|jj|j�t|jj|j�t
dd�}|D]l}t|jd�t|jj|j�t|jj|j��ntd�}t|jd�t|jj|jj�y)Nr�r\r�rIr?)rrrrArtrrr3)rSrCrBs   rW�test_shape_scalarzTestShapeBase.test_shape_scalars��

�s�O���Q�W�W�d�#��Q�V�V�\�\�1�7�7�+��Q�V�V�\�\�1�7�7�+��s�C� ���	0�A�����$�'�������q�w�w�/�������q�w�w�/�	0�

�s�O���Q�W�W�f�%��Q�V�V�\�\�1�7�7�+��Q�V�V�\�\�1�7�7�+��s�C� ���	0�A�����&�)�������q�w�w�/�������q�w�w�/�	0�

�s�O���Q�W�W�i�(��Q�V�V�\�\�1�7�7�+��Q�V�V�\�\�1�7�7�+��s�C� ���	0�A�����)�,�������q�w�w�/�������q�w�w�/�	0�

�S�M���Q�W�W�f�%��Q�V�V�\�\�1�6�6�<�<�0rYN)r�r�r�r�r�r]rYrWr�r��s��1�(1rYr�c��eZdZd�Zd�Zd�Zy)�TestNDEnumeratec	���tjd�jd�}tj|t��}t||��}t
ttj|��tt|���t
tt|��tt|���t
tt|��tt|d����y)Nr)r?r�r<r=r�F�rv)	rRrr��
zeros_liker�rrrhr4)rS�ordinary�
empty_mask�	with_masks    rW�test_ndenumerate_nomaskedz)TestNDEnumerate.test_ndenumerate_nomasked5s����9�9�R�=�(�(��3���]�]�8�4�8�
� ��
�;�	��T�"�.�.��2�3��+�h�/�0�	2��T�+�h�/�0��+�i�0�1�	3��T�+�i�0�1��+�i�E�B�C�	ErYc
�2�td�}td�}td�}ttt|��g�ttt|��g�ttt|d���tt	tjd�dtgz���ttt|��g�ttt|d���tt	tjd�dtgz���y)Nr])�dr�Fr�r�r.)r0rrhr4rMrR�ndindexr)rSrBrCr[s    rW�test_ndenumerate_allmaskedz*TestNDEnumerate.test_ndenumerate_allmasked@s����r�N���v����y�!���T�+�a�.�)�2�.��T�+�a�.�)�2�.��T�+�a�E�:�;��#�b�j�j��0�#���.�A�B�	D��T�+�a�.�)�2�.��T�+�a�E�:�;��#�b�j�j��3�Y�&��5I�J�K�	MrYc
�f�ttjd�jd�gd�gd�gd�g��}gd�}t	tt
|��|�t	ttt
|d�	���|j�t
|d�	�D]\}}t	|||��y)
Nr�r�rx)r?r?rr?ryr�))r�r�))r<rr�)r�r�)rJrl)r�r�Fr�)	rrRrr�rrhr4rcr�)rSrB�items�
coordinate�values     rW�test_ndenumerate_mixedmaskedz,TestNDEnumerate.test_ndenumerate_mixedmaskedLs�������2��.�.�v�6�+�+�+�-�
.��G���T�+�a�.�)�5�1��S��k�!��>�?�@�!�&�&�I�!,�Q�5�!A�	/��J����:���.�	/rYN)r�r�r�r�r�r�r]rYrWr�r�3s��	E�
M�
/rYr�c��eZdZd�Zd�Zd�Zy)�	TestStackc��tgd�gd���}tgd�gd���}t||gd��}t|jd�t	|j
|dj
�t	|j
|d	j
�t
||g�}t	|j|j�t	|j
|j
�t||gd	��}t|jd
�t	|j
|dd�dfj
�t	|j
|dd�d	fj
�y)Nr�ror��r�r�rprrr�r�r?�r�r<�rr5rrrrAr6rt�rSrBrCr[rs     rW�
test_stack_1dzTestStack.test_stack_1d[s������3�����3���1�a�&�q�!���Q�W�W�f�%��1�6�6�1�Q�4�9�9�-��1�6�6�1�Q�4�9�9�-��A�q�6�N���1�6�6�1�6�6�*��1�6�6�1�6�6�*��1�a�&�q�!���Q�W�W�f�%��1�6�6�1�Q��T�7�<�<�0��1�6�6�1�Q��T�7�<�<�0rYc��tgd�d��}tgd�d��}t||gd��}t|jd�t	|j
|dj
�t	|j
|d	j
�t
||g�}t	|j|j�t	|j
|j
�t||gd	��}t|jd
�t	|j
|dd�dfj
�t	|j
|dd�d	fj
�y)Nr�Tr�r�Frr�r�r?r�r�r�s     rW�test_stack_maskszTestStack.test_stack_masksms������.�����/���1�a�&�q�!���Q�W�W�f�%��1�6�6�1�Q�4�9�9�-��1�6�6�1�Q�4�9�9�-��A�q�6�N���1�6�6�1�6�6�*��1�6�6�1�6�6�*��1�a�&�q�!���Q�W�W�f�%��1�6�6�1�Q��T�7�<�<�0��1�6�6�1�Q��T�7�<�<�0rYc��d}tjjdd|�}tjjdd|�}tjjdd|�jt�}tjjdd|�jt�}t||��}t||��}t
||gd��}d|z}	t|j|	�t|j|dj�t|j|dj�t
||gd	��}|dz}	t|j|	�t|j|d
j�t|j|dj�d}tjjdd|�}tjjdd|�}tjjdd|�jt�}tjjdd|�jt�}t||��}t||��}t
||gd��}d|z}	t|j|	�t|j|dj�t|j|dj�t
||gd	��}|dz}	t|j|	�t|j|d
j�t|j|dj�y)
Nr�rrlr<r�r�r;r?r0).r).r?)r�r<r�r�)rRr�randintrir�rr5rrrrA)
rS�shp�d1�d2r�r��a1rjr[�c_shps
          rW�
test_stack_ndzTestStack.test_stack_ndsv����
�Y�Y�
�
�q�"�c�
*��
�Y�Y�
�
�q�"�c�
*��
�Y�Y�
�
�q�!�S�
)�
0�
0��
6��
�Y�Y�
�
�q�!�S�
)�
0�
0��
6��
�"�2�
&��
�"�2�
&���2�r�(��#���s�
���Q�W�W�e�$��2�7�7�A�a�D�I�I�.��2�7�7�A�a�D�I�I�.��2�r�(��$���d�
���Q�W�W�e�$��2�7�7�A�f�I�N�N�3��2�7�7�A�f�I�N�N�3���
�Y�Y�
�
�q�"�c�
*��
�Y�Y�
�
�q�"�c�
*��
�Y�Y�
�
�q�!�S�
)�
0�
0��
6��
�Y�Y�
�
�q�!�S�
)�
0�
0��
6��
�"�2�
&��
�"�2�
&���2�r�(��#���s�
���Q�W�W�e�$��2�7�7�A�a�D�I�I�.��2�7�7�A�a�D�I�I�.��2�r�(��$���d�
���Q�W�W�e�$��2�7�7�A�f�I�N�N�3��2�7�7�A�f�I�N�N�3rYN)r�r�r�r�r�r�r]rYrWr�r�Ys��1�$1�$)4rYr�)O�__doc__rr�r��numpyrR�numpy._core.numericr�
numpy.testingrr�numpy.ma.testutilsrrrr	�
numpy.ma.corer
rrr
rrrrrrr�numpy.ma.extrasrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r9r�rr+r;r�r�r�rr3rHr\r�r�r�r]rYrW�<module>r�s$�����
��4��������������������D~9�~9�B).�).�XC�C�Lo�o�d	,�,�,
!�
!�T=�T=�nT1�T1�nh3�h3�V7'�7'�tx=�x=�v41�41�n#/�#/�LO4�O4rY

Youez - 2016 - github.com/yon3zu
LinuXploit