403Webshell
Server IP : 217.160.0.135  /  Your IP : 216.73.217.85
Web Server : Apache
System : Linux www 6.18.52-i1-ampere #1203 SMP Mon Sep 14 18:29:59 CEST 2026 aarch64
User : sws1074145052 ( 1074145052)
PHP Version : 8.3.32
Disable Function : NONE
MySQL : OFF  |  cURL : ON  |  WGET : ON  |  Perl : ON  |  Python : OFF  |  Sudo : OFF  |  Pkexec : OFF
Directory :  /lib/python3/dist-packages/numpy/ma/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

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

���c����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>Gd�d	��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%��ZMdS)&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�vstackc�>�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	S)
�TestGenericc�(�tdt���}tddgddgt���}t||��t	jddgddgd���}td|���}td	d	gd
d
g|���}t||��td|���}td	d	gd	d	ggd
d
gd
d
gg|���}t||��t	jddd
dgfg��}td|���}tddgddg|���}t||��td|���}tddgddg|���}t||��td
|���}tdggdgg|���}t||��dS)N����dtype���maskr=�a�b�f��names�formats�rr�r>r>�r;r;�rArC��barC��bbrC�r>rH)r0�floatr
r�npr=)�self�test�control�dts    �</usr/lib/python3/dist-packages/numpy/ma/tests/test_extras.py�test_masked_allzTestGeneric.test_masked_all&s����$�e�,�,�,����A��a��V�5�9�9�9���T�7�#�#�#�
�X��c�
��S�z�B�B�
C�
C���$�b�)�)�)�����(���/?�r�J�J�J���T�7�#�#�#��&��+�+�+���&�&�)�F�F�+;�<�%�v�.���0@�A� �"�"�"��	�T�7�#�#�#�
�X�z�C�+�{�)C�#D�E�
F�
F���$�b�)�)�)����k�2�)�;�7�r�C�C�C���T�7�#�#�#��$�b�)�)�)����k�2�)�;�7�r�C�C�C���T�7�#�#�#��&��+�+�+���+����}�o�R�H�H�H���T�7�#�#�#�#�#�c�P�tjddtfgdffg��}tj�d|��}tt
|d��tjjj��tt
|dd��tjjj��tt|dd��d��t|ddj
d��t|ddjj
d��dS)NrB�c�r>r>rH��rQr=�object�mar0r�type�corer
�lenr�_fill_value�rR�my_dtype�
masked_arrs   rV�"test_masked_all_with_object_nestedz.TestGeneric.test_masked_all_with_object_nestedDs����8�c�c�6�]�O�T�#:�;�<�=�=���U�%�%�d�H�5�5�
��T�*�S�/�*�*�B�E�J�,B�C�C�C��T�*�S�/�#�.�/�/����1G�H�H�H��S��C���-�.�.��2�2�2��Z��_�S�)�/��8�8�8��Z��_�S�)�5�;�R�@�@�@�@�@rXc��tjdtdffg��}tj�d|��}tt
|d��tjjj��tt|d��d��t|dj
d��t|djj
d��dS)NrBr[r>rHr\r]rds   rV�test_masked_all_with_objectz'TestGeneric.test_masked_all_with_objectPs����8�c�F�D�>�2�3�4�4���U�%�%�d�H�5�5�
��T�*�S�/�*�*�B�E�J�,B�C�C�C��S��C��)�)�1�-�-�-��Z��_�*�F�3�3�3��Z��_�0�6��;�;�;�;�;rXc��tddgt���}t|��}tddgddgt���}t||��t	jddgddgd���}td	d	gd
d
g|���}t|��}tddgd
d
g|���}t||��t	jddd
dgfg��}tddgddg|���}t|��}t||��dS)Nr>r;r<r?rArBrCrDrGrH)�
rkrJrKrMrO)r
rPr1rrQr=)rR�baserSrTrUs     rV�test_masked_all_likez TestGeneric.test_masked_all_likeZs4���a��V�5�)�)�)���t�$�$����A��a��V�5�9�9�9���T�7�#�#�#�
�X��c�
��S�z�B�B�
C�
C���f�f�%�V�V�,<�B�G�G�G���t�$�$����8�,�F�F�3C�2�N�N�N���T�7�#�#�#�
�X�z�C�+�{�)C�#D�E�
F�
F����k�2�)�;�7�r�C�C�C���w�'�'���T�7�#�#�#�#�#rXc�x�tdd��D�]'}td|z��D�]}tj|t���}tj||t���}td|z��}|d|zzdk|_d}||��D]$}||j|���z
}�%|tkr5t|������|����|j|_t|������|������)dS)Nr>�r;r<r)�rangerQr�int�fullrr@�data�sumr,r�
compressed)	rRrC�i�j�k�jarA�s�sls	         rV�check_clumpzTestGeneric.check_clumpns,���q�!���
	:�
	:�A��1�a�4�[�[�
:�
:���I�a�s�+�+�+���W�Q���-�-�-�� ��A��&�&����1��+�!�+������!�A�$�$�*�*�B��������)�)�)�A�A���&�&� ������!3�!3�!5�!5�q�9�9�9�9��f�W�A�F� ������!3�!3�!5�!5�q�9�9�9�9�
:�
	:�
	:rXc�0�ttjd����}t|gd�<t	|��}tdd��tdd��tdd��g}t
||��|�t��dS)Nrk�rr>r;���	r�rror�)rrQrrr+�slicerr|�rRrArSrTs    rV�test_clump_maskedzTestGeneric.test_clump_masked~s������2���'�'�� &��
�
�
���A������A�;�;��a����U�1�b�\�\�:���T�7�#�#�#�����&�&�&�&�&rXc��ttjd����}t|gd�<t	|��}tdd��tdd��g}t
||��|�t��dS)Nrkr~r�rror�)rrQrrr,r�rr|r�s    rV�test_clump_unmaskedzTestGeneric.test_clump_unmasked�sy�����2���'�'�� &��
�
�
���a� � ����A�;�;��a����.���T�7�#�#�#�����(�(�(�(�(rXc��td��}t|��}t|td|j��g��tjdt���|_t|td|j��g��t||dk|dkz|dkz<t|��}t|tdd��tdd��g��t|dd�<t|��}t|g��dS)	Nrkrr<r�r��rr�)
rr-rr��sizerQr�boolr@r)rRrArSs   rV�test_flatnotmasked_contiguousz)TestGeneric.test_flatnotmasked_contiguous�s����2�J�J��'��*�*���T�E�!�Q�V�,�,�-�.�.�.���"�D�)�)�)����T�E�!�Q�V�,�,�-�.�.�.�*0��1�q�5�Q��U�
�q�A�v�
&�'�'��*�*���T�E�!�Q�K�K��q�!���5�6�6�6���!�!�!��'��*�*���T�2�����rXN)�__name__�
__module__�__qualname__rWrgrirmr|r�r�r�r\rXrVr8r8$s�������$�$�$�<
A�
A�
A�<�<�<�$�$�$�(:�:�:� 	'�	'�	'�)�)�)�����rXr8c���eZdZd�Zd�Zd�Zd�Zd�Zd�Ze	j
�dgd�d	d
ggd�dgd
gfgd�gd�gdgd�gddggd�ggd�gfg��d���Zd�Z
d	S)�TestAveragec� �tgd�gd����}tdt|d�����tdt|gd������t|gd�d	�
��\}}td|��t|dk��t|dd�<tt|d���jd	g��tgd�gd����}|�dd��}t|dd�d
f<tt|d���ddg��tt|d
���jdd	g��tddgt|d�����t|dd	���\}}t|ddg��dS)N)���?�@�@)TFFF�r@r�r��axis)r�r�r�r���weightsT�r��returned�@r;r>r��r�r�r�)r
rrrrr@�reshape)rR�ott�result�wtss    rV�test_testAverage1zTestAverage.test_testAverage1�s����$�$�$�+F�+F�+F�G�G�G���S�'�#�A�.�.�.�/�/�/��S�'�#�/?�/?�/?�@�@�@�A�A�A��c�+;�+;�+;�d�K�K�K�����S�&�!�!�!���s�
������A�A�A���W�S�q�)�)�)�.���7�7�7��$�$�$�+F�+F�+F�G�G�G���k�k�!�Q������A�A�A�q�D�	��W�S�q�)�)�)�C��:�6�6�6��W�S�q�)�)�)�.�q�1�D�6�:�:�:��b�"�X�w�s��3�3�3�4�4�4��c��D�9�9�9�����S�2�r�(�#�#�#�#�#rXc
�N�gd�}gd�gd�g}tdtj���}tt	|d���d��tt	|d|���d��ttdtj���d	td��zg��}tt	|d��tj�tjd����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���dS)N�rr>r>r>r>r�r>rrrrr>rr<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�)
rrQ�float_rrr
�add�reducerrrr@r)rR�w1�w2�x�y�m1�m2�m3�m4�m5�zs           rV�test_testAverage2zTestAverage.test_testAverage2�s���
�
�
�� � � �"4�"4�"4�
5���1�B�I�&�&�&���W�Q�Q�'�'�'��-�-�-��W�Q�Q��3�3�3�S�9�9�9��6�!�2�9�-�-�-�s�V�A�Y�Y��?�@�@���W�Q��%�%�r�v�}�}�R�Y�q�\�\�'B�'B�R�'G�#�'M�N�N�N��W�Q�Q�'�'�'���1����):�R�)?�@�@�@��W�Q�Q�'�'�'��a�a�(�(�(�'�!�!�*<�*<�*<�s�*B�C�	E�	E�	E��W�Q��b�1�1�1�8�<�<�<��W�Q�Q��3�3�3�.�.�.�	0�	0�	0��W�Q�Q�'�'�'��a�a�(�(�(�'�!�!�*<�*<�*<�s�*B�C�	E�	E�	E�
�1�X�X��
�
�
�� � � �"4�"4�"4�
5��
�!�W�W��
�
�
���W�\�!�R�0�0�q�9�9�9�3�?�?�?��W�\�!�R�0�0�q�9�9�9�3�?�?�?��W�\�!�R�0�0�q�9�9�9�>���G�G�G��W�\�!�R�0�0�q�9�9�9�3�?�?�?��U�7�<��2�#6�#6�Q�?�?�?�@�@�!�D�D�D���B�����W�Q��%�%�x�0�0�0��W�Q�Q�'�'�'�)E�)E�)E�F�F�F��W�Q�Q�'�'�'�#�s��4�4�4��W�Q�Q��3�3�3�2�2�2�	4�	4�	4�	4�	4rXc�b�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��dS)Nrr�r>Tr�)r;r;r�r�r�r�r�)r�r�r;�Fr���?r�r�g������@��?r�)rrrrrr
rPr)rRrArB�r1r��r2r��a2d�a2dm�a2da�a2dmas           rV�test_testAverage3zTestAverage.test_testAverage3�s����1�I�I���1�I�I��M���1�a�&�1�a�&�)��D�A�A�A���B��U�2�Y�Y��b�	�	�*�*�*��R�X�r�x�(�(�(���i���q�1�a�&�4�P�P�P���B��U�2�Y�Y��b�	�	�*�*�*���i���4�8�8�8���B��U�2�Y�Y��b�	�	�*�*�*���i���$�y�/�/�D�Q�Q�Q���B��U�2�Y�Y��b�	�	�*�*�*��a��V�a��V�$�e�,�,���C�5�%�.�4��-�!@�A�A���s��#�#�#���T�C��:�&�&�&���1�%�%�%���U�S�#�J�'�'�'���4�(�(�(���U�G�$�$�$���1�%�%�%���U�S�#�J�'�'�'�'�'rXc�z�tjgd����dd��}tj�|dgdgdgg���}tjgd����dd��}t	||dd���}td	gd
gdggdgdgdgg��}t
||��dS)N�r;r�r�r�r>FTr�)r�r�r)r�r��keepdimsr�r�r�)rQr
r�r_rrr)rRr�rB�w�actual�desireds      rV�test_testAverage4zTestAverage.test_testAverage4�s����H�Y�Y�Y���'�'��1�-�-���E�K�K��%��5�'�D�6� :�K�;�;���H�Y�Y�Y���'�'��1�-�-����A�A��=�=�=�����r�d�R�D�1�U�G�e�W�t�f�3M�N�N���V�W�%�%�%�%�%rXc���ttddg����}t|d��ttgd�gd������}t|d��dS)Nr>r;r�)r>r;r�r��FFTTr�)rr
r�rRrAs  rV�test_onintegers_with_maskz%TestAverage.test_onintegers_with_mask�sd���E�1�a�&�M�M�"�"���Q������E�,�,�,�-G�-G�-G�H�H�H�I�I���Q������rXc��tjgd�gd�gt���}tgd�gd�g|���}t	|��}tj|�����}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|���|	|�
��}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��dS)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�)	rQr
r�rrrur	�real�imag)
rRr@rA�av�expected�av0�	expected0�av1�	expected1r��wav�wav0�wav1s
             rV�test_complexzTestAverage.test_complexs����x����(���*�15�7�7�7���5�5�5�6�6�6�8�"�
$�
$�
$���Q�Z�Z���:�a�l�l�n�n�-�-���B�G�X�]�3�3�3��B�G�X�]�3�3�3��a�a� � � ���A�F��+�+�+�g�a�f�1�.E�.E�.E�b�.H�H�	��C�H�i�n�5�5�5��C�H�i�n�5�5�5��a�a� � � ���A�F��+�+�+�g�a�f�1�.E�.E�.E�b�.H�H�	��C�H�i�n�5�5�5��C�H�i�n�5�5�5��h�1�1�1�1�1�1�3�4�4���a��%�%�%���:�a�l�l�n�n�c�4�%�j�A�A�A���C�H�h�m�4�4�4��C�H�h�m�4�4�4��q�#�A�.�.�.���Q�V�S�q�9�9�9��Q�V�S�q�9�9�9�"�<�=�	��D�I�y�~�6�6�6��D�I�y�~�6�6�6��q�#�A�.�.�.���Q�V�S�q�9�9�9��Q�V�S�q�9�9�9�"�<�=�	��D�I�y�~�6�6�6��D�I�y�~�6�6�6�6�6rXz<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�g#@)r�r�r�c�V�tj�||d���}|jtj|��ksJ�t	||��tj�|||d���}|jtj|��ksJ�t	||��tj�|||dd���\}}	|jtj|��ksJ�t	||��|	jtj|��ksJ�t	|	|��dS)NT)r�r�)r�r�r�)r�r�r�r�)rQr_rrr)
rRr�r��expected_avgr��
expected_wavg�
expected_wsum�avg�wavg�wsums
          rV�test_basic_keepdimszTestAverage.test_basic_keepdims1s���e�m�m�A�D�4�m�8�8���y�B�H�\�2�2�2�2�2�2��3��-�-�-��u�}�}�Q�T�7�T�}�J�J���z�R�X�m�4�4�4�4�4�4��4��/�/�/��U�]�]�1�4��,0�4�#�A�A�
��d��z�R�X�m�4�4�4�4�4�4��4��/�/�/��z�R�X�m�4�4�4�4�4�4��4��/�/�/�/�/rXc��tj�tjd���dd��gd�gd�gd�g���}tgd�d���}tgd�gd����}t
|d|d�	��}tjgd
���}t||��t
|d|d�	��}tjgd���}t||��tj�gd�gd
����}t
|dddtjg���}t|d��tj�gd�gd�gd�ggd�gd�gd�g���}tj�|dtjdgd���}tj�dtjtjdggd����}t||��t|j
|j
��dS)Nr�r��r>rr�rrrr�)r���Frr�)�@g�@g�@)r�g0��N@g0��N@)r�r�r�r�r�r>r�r�)r�r��@r�)g"@r�r�r�)FTTF)TFTT)TFTFr�r��@)rQr_r
rr�rrr	�nanrr@)	rRrA�weights_unmasked�weights_masked�avg_unmasked�expected_unmasked�
avg_masked�expected_masked�avg_expecteds	         rV�test_masked_weightszTestAverage.test_masked_weightsHs>��
�E�K�K��	�!���,�,�Q��2�2�'�i�i����I�I�I�>�
�
@�
@��'����%�@�@�@��%�k�k�k�	�	�	�B�B�B���q�q�'7�%�I�I�I���H�%<�%<�%<�=�=���L�*;�<�<�<��Q�Q���O�O�O�
��(�#N�#N�#N�O�O���J��8�8�8�

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

'�&�&�%�%�%�&�&�&�
�	
�
�
���U�]�]�1�q�"�&�!�n�1�]�E�E�
��u�{�{�C������#=�#=�#=�#=�#�?�?��	�J��5�5�5��Z�_�l�&7�8�8�8�8�8rX)r�r�r�r�r�r�r�r�r��pytest�mark�parametrizer�r�r\rXrVr�r��s
������$�$�$�& 4� 4� 4�D(�(�(�0&�&�&����*7�*7�*7�X�[���F�
�)�)�T�C�5�)�)�)�d�V�c�U�	;��9�9�j�j�j�
!�1����&7��a�&�?�?�?�#�i�i�i�[�
2�	3���0�0�
��0�")9�)9�)9�)9�)9rXr�c� �eZdZd�Zd�Zd�ZdS)�TestConcatenatorc�t�ttdtgd�����td��}gd�}t	||���}t|dd|f}tt
|t����t|gd���t|jt|dd|f��dS)N�r>r;r�r�r�rr�)r>rrrrr�r)r>r>r>r>r>rrr>r>r>r>r>)	rrr
rrr�
isinstancer
r@)rRrB�m�drZs     rV�test_1dzTestConcatenator.test_1dws����3�/�0�%�8J�8J�8J�2K�2K�L�L�L���G�G���O�O�����#�#�#����1�a��
�O���
�1�k�*�*�+�+�+��1�B�B�B�C�C�C��1�6�3�q�!�Q��z�?�3�3�3�3�3rXc��tj�dd��}tj�dd��}tjtj�dd��d��}tjtj�dd��d��}t	||���}t	||���}t
d||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�rk)rkr�)rQ�random�rand�round_rrrrrr@�r_)rR�a_1�a_2�m_1�m_2�b_1�b_2rs        rV�test_2dzTestConcatenator.test_2d�s����i�n�n�Q��"�"���i�n�n�Q��"�"���i��	���q�!�,�,�a�0�0���i��	���q�!�,�,�a�0�0���3�S�)�)�)���3�S�)�)�)����S�#�
������7�"�#�#�#��1�Q�Q�Q����U�8�S�)�)�)��1�Q�Q�Q����U�8�S�)�)�)��1�6�2�5��c�3��#7�8�8�8���S��M�����7�"�#�#�#��1�R�a�R����T�7�C�(�(�(��1�Q�R�R����T�7�C�(�(�(��1�6�2�5��c��?�3�3�3�3�3rXc�R�ttjjdf}t	|jddg��t	|jdd��tddgtjjf}t	|jgd���t	|jdd�ddg��dS)Nr>TFr;)FFT)rrQr_rrr@rs)rRr�s  rV�test_masked_constantz%TestConcatenator.test_masked_constant�s����R�U�\�1�_�%���V�[�4��-�0�0�0��V�[��^�Q�'�'�'��a��V�R�U�\�)�*���V�[�"6�"6�"6�7�7�7��V�[��!��_�q�!�f�-�-�-�-�-rXN)r�r�r�rrrr\rXrVrrtsA������	4�	4�	4�4�4�4�(.�.�.�.�.rXrc��eZdZd�Zd�ZdS)�
TestNotMaskedc	�L�ttjd���dd��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	|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�)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�)rrQrr�r/rrsr�rRrsrSs   rV�
test_edgeszTestNotMasked.test_edges�s;���B�I�b�M�M�1�1�!�Q�7�7�"1�/�/�"1�/�/�"1�/�/�"1�/�/�"1�/�/�	"3�5�5�5���t�T�*�*���T�A�r�7�#�#�#��t�Q�'�'���T�!�W���@�A�A�A��T�!�W���@�A�A�A��t�Q�'�'���T�!�W���@�A�A�A��T�!�W���@�A�A�A��t�y�$�/�/���T�A�r�7�#�#�#��t�y�!�,�,���T�!�W���@�A�A�A��T�!�W���@�A�A�A��t�y�"�-�-���T�!�W���@�A�A�A��T�!�W���@�A�A�A���R���t�Q�'�'���T�!�W���@�A�A�A��T�!�W���@�A�A�A��t�R�(�(���T�!�W�|�\�:�;�;�;��T�!�W�|�\�:�;�;�;�;�;rXc�|�ttjd���dd��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��dS)Nrr�r�)rrrrr>r>r>r>)r>r>r>r>r>r>r>r>)rrrrrrr>rr�rr����r>r;rro)rrQrr�r.rr�)rRrA�tmps   rV�test_contiguouszTestNotMasked.test_contiguous�s������2���.�.�q�!�4�4�7�7�7�7�7�7�7�7�7�9�
:�
:�
:��#�1�d�+�+���S��!�Q�����"�b�$����"�b�$���
�	�	�	�#�1�a�(�(���S�
�1�a��
�
��a��D� 1� 1�2�
�1�a��
�
��a��D� 1� 1�2�
�1�a��
�
��a��D� 1� 1�2�
�1�a��
�
��a��D� 1� 1�2�
�1�a��
�
��
�1�a��
�
���
�1�a��
�
��	
�		�		�		�#�1�a�(�(���S�
�1�a��
�
���
�1�a��
�
��a��D� 1� 1�2�
�	�	�	�	�	rXN)r�r�r�r!r'r\rXrVrr�s3������ <� <� <�D����rXrc��eZdZd�Zd�Zd�Zej�dgd���ej�ddge	j
jdfe	j
jd	fg��d
�����Z
d�Zd�Zd
�ZdS)�TestCompressFunctionsc
��tjttd�������ddd��}tjd���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||��dS)1N�<r�r�r��r�r�r�T�r>r>r>r�)rr;r�r�)rk��
�)����)�(�*�+�,)�2�4�5�6)�7�9�:�;rr)r�rror�r�)rkr�r.r/r0)r1r#r2r3r4)r5�)r6r7r8)�-�.�/�0�1)r9�3r:r;r<)r=�8r>r?r@r>)��r$r%r)�r�� �!�")�#�$�%�&�'r[r)rr;)r�ror�r�)rIr$r%r)r�r��)rKrLrMrN)rOrQrRrS)rBrDrErFr:r)r�rr>)rr�r>r;)rr;)r>r)rr�rr;)rr)
rQr
�listrpr�r�astyper�r(r)rRr�rrA�a2�a3�a4s       rV�test_compress_ndz&TestCompressFunctions.test_compress_nd�s^���H�T�%��,�,�'�'�(�(�0�0��A�q�9�9���H�W���$�$�T�*�*����%���!�!�����
��N�N���Q�*�*�*�*�*�*�*�*�*�,�+�*�*�*�*�*�*�*�*�,�-�	.�	.�	.�
��1�����Q�.�.�.�.�.�.�.�.�.�.�.�.�0�/�.�.�.�.�.�.�.�.�.�.�.�0�	1�	2�	2�	2�
��1�����Q�.�.�.�.�.�.�.�.�.�0�/�.�.�.�.�.�.�.�.�0�/�.�.�.�.�.�.�.�.�0�
1�	2�	2�	2���D�
!�
!��
��B�
�
��
��E�
"�
"���Q������Q������Q�����
��1�����Q�/�/�/�)�/�/�*�*�*�*�*�*�,�+�*�*�*�*�*�*�*�*�*�*�*�,�+�*�*�*�*�*�*�*�*�*�*�*�,�-�	.�	.�	.���D�
!�
!��
��B�
�
��
��E�
"�
"���Q������Q������Q�����
��6�"�"���Q�.�.�.�.�.�.�.�.�.�0�/�.�.�.�.�.�.�.�.�0�1�	2�	2�	2���G�
$�
$���Q�����
��6�"�"���Q�*�*�*�*�*�*�*�*�*�,�+�*�*�*�*�*�*�*�*�,�+�*�*�*�*�*�*�*�*�,�
-�	.�	.�	.���G�
$�
$��
��G�
$�
$��
��H�
%�
%���Q������Q������Q�����
��6�"�"���Q�*�*�*�*�*�*�*�*�*�*�*�*�,�+�*�*�*�*�*�*�*�*�*�*�*�,�	-�	.�	.�	.���G�
$�
$���Q������rXc��ttjd���dd��gd�gd�gd�g���}t	t|��ddgdd	gg��t	t|d
��gd�gd�g��t	t|d
��d
dgddgdd	gg��t|jgd�gd�gd�g���}t	t|��d
dgdd	gg��t	t|d
��gd�gd�g��t	t|d
��d
dgddgdd	gg��t|jgd�gd�gd�g���}t	t|��d	gg��t	t|d
��gd�g��t	t|d
��dgdgd	gg��t|jgd�gd�gd�g���}t	t|��jd
��t	t|d
��jd
��t	t|d
��jd
��dS)Nr�r�r�r�r�r�r�ror�rr,)rror�r>r;�rr>rr�rr>r;�rrr>)r
rQrr�rr)�_datar��rRr�s  rV�test_compress_rowcolsz+TestCompressFunctions.test_compress_rowcolsXsv���"�)�A�,�,�&�&�q�!�,�,�!�	�	�9�9�9�i�i�i�8�
:�
:�
:���%�a�(�(�A�q�6�A�q�6�*:�;�;�;��%�a��+�+�i�i�i����-C�D�D�D��%�a��+�+�q�!�f�q�!�f�q�!�f�-E�F�F�F��!�'����I�I�I�y�y�y� A�B�B�B���%�a�(�(�A�q�6�A�q�6�*:�;�;�;��%�a��+�+�i�i�i����-C�D�D�D��%�a��+�+�q�!�f�q�!�f�q�!�f�-E�F�F�F��!�'����I�I�I�y�y�y� A�B�B�B���%�a�(�(�A�3�%�0�0�0��%�a��+�+�i�i�i�[�9�9�9��%�a��,�,��s�Q�C�!��o�>�>�>��!�'����I�I�I�y�y�y� A�B�B�B���%�a�(�(�-�q�1�1�1��%�a��+�+�0�!�4�4�4��%�a��+�+�0�!�4�4�4�4�4rXc���ttjd���dd��gd�gd�gd�g���}t	t|��jgd�gd�gd�g��t	t|d��jgd�gd�gd�g��t	t|d��jgd�gd�gd�g��t|jgd�gd	�gd�g���}t	t|��jgd	�gd�gd	�g��t	t|d��jgd�gd�gd�g��t	t|d��jgd	�gd	�gd	�g��t|jgd�gd	�gd�g���}t	t|��jgd�gd�gd
�g��t	t|d��jgd�gd�gd�g��t	t|d��jgd
�gd
�gd
�g��t|jgd�gd	�gd�g���}tt|���	��tu��tt|d���	��tu��tt|d���	��tu��tt|��j�	����tt|d��j�	����tt|d��j�	����dS)Nr�r�r�r�r�r-rr>r`)r>r>rrb)r
rQrr�rr*r@rcr�allrrds  rV�test_mask_rowcolsz'TestCompressFunctions.test_mask_rowcolsls����"�)�A�,�,�&�&�q�!�,�,�!�	�	�9�9�9�i�i�i�8�
:�
:�
:���\�!�_�_�)��i�i����I�I�I�6�	8�	8�	8��\�!�Q�'�'�,��i�i����I�I�I�6�	8�	8�	8��\�!�Q�'�'�,��i�i����I�I�I�6�	8�	8�	8��!�'����I�I�I�y�y�y� A�B�B�B���\�!�_�_�)��i�i����I�I�I�6�	8�	8�	8��\�!�Q�'�'�,��i�i����I�I�I�6�	8�	8�	8��\�!�Q�'�'�,��i�i����I�I�I�6�	8�	8�	8��!�'����I�I�I�y�y�y� A�B�B�B���\�!�_�_�)��i�i����I�I�I�6�	8�	8�	8��\�!�Q�'�'�,��i�i����I�I�I�6�	8�	8�	8��\�!�Q�(�(�-��i�i����I�I�I�6�	8�	8�	8��!�'����I�I�I�y�y�y� A�B�B�B����Q���#�#�%�%��/�0�0�0���Q��"�"�&�&�(�(�F�2�3�3�3���Q��"�"�&�&�(�(�F�2�3�3�3���Q���$�(�(�*�*�+�+�+���Q��"�"�'�+�+�-�-�.�.�.���Q��"�"�'�+�+�-�-�.�.�.�.�.rXr�)Nrr>�func�rowcols_axisrr>c�6�ttjd���dd��gd�gd�gd�g���}t	t
��5|||���}t
|t||����ddd��dS#1swxYwYdS)Nr�r�r�r�r�r�)r
rQrr�r�DeprecationWarningrr*)rRr�rirjr��ress      rV�#test_mask_row_cols_axis_deprecationz9TestCompressFunctions.test_mask_row_cols_axis_deprecation�s���

�"�)�A�,�,�&�&�q�!�,�,�!�	�	�9�9�9�i�i�i�8�
:�
:�
:���,�
-�
-�	=�	=��$�q�t�$�$�$�C���l�1�l�;�;�<�<�<�	=�	=�	=�	=�	=�	=�	=�	=�	=�	=�	=�	=����	=�	=�	=�	=�	=�	=s�,B�B�Bc	�v�tjdd��}gd�}t||����dd��}t||����dd��}t	||d���}t|jddgdd	gg��t	||d���}t|jgd
�gd�gd�g��t	||d���}t|tj|�d	��|�d	������t	||d���}t|tj|�d	��|�d	������gd
�}t||����dd��}t||����dd��}t	||d���}t|jd	dgddgg��t	||d���}t|jgd�gd�gd
�g��t	||d���}t|tj|�d	��|�d	������t|t	||����t	||d���}t|tj|�d	��|�d	������gd�}t||����dd��}t||����dd��}t	||��}t|jt��t	||��}t|jt��t|gd�����dd��}t|gd�����dd��}t	||d���}t|jddgd	d	gg��t	||d���}t|tj|�d	��|�d	������t	||d���}t|jgd�gd�gd�g��t	||d���}t|tj|�d	��|�d	������t|gd
�����dd��}t|gd�����dd��}t	||d���}t|jd	d	gddgg��t	||��}t|tj|�d	��|�d	������t	||d���}t|jgd�gd�gd�g��t	||d���}t|tj|�d	��|�d	������t|gd
�����dd��}t|gd�����dd��}t	||d���}t|jdd	gddgg��t	||d���}t|tj|�d	��|�d	������t	||d���}t|jgd�gd
�gd�g��t	||d���}t|tj|�d	��|�d	������dS)Nr>ro)r>rrrrrr�r;r�T)�strictrr-r�F�rrrrrr>rb)rrrrrr)rrr>rrr)	rQrrr�rrr@�filledr)rR�nrrArBrZs      rV�test_dotzTestCompressFunctions.test_dot�s
���I�a��O�O���������#�#�#�+�+�A�q�1�1�����#�#�#�+�+�A�q�1�1����1�T�"�"�"���Q�V�q�!�f�q�!�f�-�.�.�.���1�T�"�"�"���Q�V�i�i�i����I�I�I�>�?�?�?���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9��������#�#�#�+�+�A�q�1�1�����#�#�#�+�+�A�q�1�1����1�T�"�"�"���Q�V�q�!�f�q�!�f�-�.�.�.���1�T�"�"�"���Q�V�i�i�i����I�I�I�>�?�?�?���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9��Q��A�q�	�	�"�"�"���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9��������#�#�#�+�+�A�q�1�1�����#�#�#�+�+�A�q�1�1����1�I�I���Q�V�V�$�$�$���1�I�I���Q�V�V�$�$�$���!3�!3�!3�4�4�4�<�<�Q��B�B����!3�!3�!3�4�4�4�<�<�Q��B�B����1�T�"�"�"���Q�V�q�!�f�q�!�f�-�.�.�.���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9���1�T�"�"�"���Q�V�i�i�i����I�I�I�>�?�?�?���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9���!3�!3�!3�4�4�4�<�<�Q��B�B����!3�!3�!3�4�4�4�<�<�Q��B�B����1�T�"�"�"���Q�V�q�!�f�q�!�f�-�.�.�.���1�I�I���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9���1�T�"�"�"���Q�V�i�i�i����I�I�I�>�?�?�?���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9���!3�!3�!3�4�4�4�<�<�Q��B�B����!3�!3�!3�4�4�4�<�<�Q��B�B����1�T�"�"�"���Q�V�q�!�f�q�!�f�-�.�.�.���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9���1�T�"�"�"���Q�V�i�i�i����I�I�I�>�?�?�?���1�U�#�#�#���Q���q�x�x��{�{�A�H�H�Q�K�K�8�8�9�9�9�9�9rXc���tjd��}t|��}tt	t||����tu��tt	t||����tu��tt	t||����tu��tt	t||����tu��dS)Nr�)rQ�eyer
rr`rr
�rRrArBs   rV�test_dot_returns_maskedarrayz2TestCompressFunctions.test_dot_returns_maskedarray�s����F�1�I�I���!�H�H����S��A�Y�Y���;�.�/�/�/���S��A�Y�Y���;�.�/�/�/���S��A�Y�Y���;�.�/�/�/���S��A�Y�Y���;�.�/�/�/�/�/rXc���ttjd����}ttjd����}t	|||���}t||u��t
||��dS)Nr�)r�r���out)r
rQrvrrrr)rRrAr{rms    rV�test_dot_outz"TestCompressFunctions.test_dot_out�sj���"�&��)�)�����B�H�V�$�$�%�%���!�Q�C� � � ����s�
�����Q������rXN)r�r�r�r^rerhr�r�rrQr_�	mask_rows�	mask_colsrnrtrxr|r\rXrVr)r)�s�������n�n�n�`5�5�5�(/�/�/�@�[���V�\�\�\�2�2��[���f�n�5�!�u���2�R�U�_�a�4H�I�K�K�=�=�K�K�3�2�=�D:�D:�D:�L0�0�0�����rXr)c��eZdZd�Zd�ZdS)�TestApplyAlongAxisc��td���ddd��}d�}t|d|��}t|ddgddgg��dS)	Nr�r;r�c��|dS�Nr>r\)rBs rV�myfuncz*TestApplyAlongAxis.test_3d.<locals>.myfunc�s���Q�4�KrXr>r�rork�rr�r'r�rRrAr��xas    rV�test_3dzTestApplyAlongAxis.test_3d�sd���3�K�K����1�a�(�(��	�	�	��f�a��
+�
+���R�1�a�&�1�b�'�*�+�+�+�+�+rXc��td���ddd��}dd�}t|d|d���}t|ddgd	d
gg��dS)Nr.r;r�rc��|d|zSr�r\)rB�offsets  rVr�z1TestApplyAlongAxis.test_3d_kwargs.<locals>.myfunc�s���Q�v�X�;�rXr>)r�r�r�r�)rr�r�s    rV�test_3d_kwargsz!TestApplyAlongAxis.test_3d_kwargs�sn���2�J�J���q�!�Q�'�'��	�	�	�	��f�a��1�
5�
5�
5���R�1�a�&�1�b�'�*�+�+�+�+�+rXN)r�r�r�r�r�r\rXrVr�r��s2������,�,�,�,�,�,�,�,rXr�c��eZdZd�ZdS)�TestApplyOverAxesc��td���ddd��}ttj|ddg��}tjdgdgdggg��}t
||��t||dz�t��<ttj|ddg��}tjd	gd
gdggg��}t
||��dS)Nrr;r�r�rr+�\�|r�r8)
rr�r&rQrtr
rrrZr�)rRrArS�ctrls    rV�
test_basiczTestApplyOverAxes.test_basics����2�J�J���q�!�Q�'�'���r�v�q�1�a�&�1�1���x�2�$���s�e�,�-�.�.���T�4� � � �"(��1�q�5�.�.��
�
���r�v�q�1�a�&�1�1���x�2�$���r�d�+�,�-�-���T�4� � � � � rXN)r�r�r�r�r\rXrVr�r�s#������!�!�!�!�!rXr�c���eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zej�dgd����d���Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�ZdS)�
TestMedianc���tj�tjtjgtjtjggd���}t	|tj��dS)Nrr�)rQr_r�infr�rR�rs  rV�test_pytypezTestMedian.test_pytypesH���E�L�L�2�6�2�6�*�R�V�R�V�,<�=�B�L�G�G���Q�������rXc�*�tj�tj�tjtjgtjtjgg��d���}t|tj��tj�tj�tjtjgtjtjgg��d���}t|tj��tj�tj�tjtjgtjtjggd���d���}t|jd��tj�tj�tjtjgtjtjggd���d���}t|jd��dS)Nrr�Tr�)rQr_rrr�rr@r�s  rV�test_infzTestMedian.test_infs����E�L�L���+�+�b�f�b�f�-=�.0�f�b�f�-=�-?�@�@�FH�
�
J�
J���Q�������E�L�L���+�+�b�f�b�f�-=�.0�f�b�f�-=�-?�@�@�FJ�
�
L�
L���Q�������E�L�L���+�+�b�f�b�f�-=�.0�f�b�f�-=�-?�EI�,�K�K� �
�
"�
"��	�Q�V�T�"�"�"��E�L�L���+�+�b�f�b�f�-=�.0�f�b�f�-=�-?�EI�,�K�K�"�
�
$�
$��	�Q�V�T�"�"�"�"�"rXc�j�tjd��}ttj�|��d��tt
tj�|����tu��td��}ttj�|��d��tt
tj�|����tu��d}ttj�|��d��tt
tj�|����tu��tjd���	dd��}ttj�|d�	��tj|d�	����ttj�|d
�	��tj|d
�	����ttj�|d
�	��tu��tjd���	dd��}ttj�|d�	��tj|d�	����ttj�|d
�	��tj|d
�	����ttj�|d
�	��tu��dS)Nr�r�r�r�r�r��Hrr�r>gR@)
rQrrr_rrr`r
rpr�rds  rV�test_non_maskedzTestMedian.test_non_masked)s:���I�a�L�L���R�U�\�\�!�_�_�b�)�)�)���R�U�\�\�!�_�_�%�%�[�8�9�9�9��!�H�H���R�U�\�\�!�_�_�c�*�*�*���R�U�\�\�!�_�_�%�%�[�8�9�9�9�
���R�U�\�\�!�_�_�b�)�)�)���R�U�\�\�!�_�_�%�%�[�8�9�9�9��I�e���$�$�Q��*�*���R�U�\�\�!�!�\�,�,�b�i���.B�.B�.B�C�C�C��R�U�\�\�!�!�\�,�,�b�i���.B�.B�.B�C�C�C������Q�Q��'�'�{�:�;�;�;��I�f���%�%�a��+�+���R�U�\�\�!�!�\�,�,�b�i���.B�.B�.B�C�C�C��R�U�\�\�!�!�\�,�,�b�i���.B�.B�.B�C�C�C������Q�Q��'�'�{�:�;�;�;�;�;rXc�@�ttjd��dgdzdgdzz���}ttj�|��d��ttj�|��jdd��tttj�|����tu��ttjd	���
d
d��dgdzdgdzz���}ttj�|��d
��ttj�|��jdd��tttj�|����tu��tj�|dd���}t|ddg��t|jdd��tt|��tu��dS)z5test the examples given in the docstring of ma.medianr�rr�r>r�r�r\�shape mismatchrkr;r�rr�rT�r��overwrite_inputr�r�r:N)r
rQrrr_rrrr`r
r�)rRr��ma_xs   rV�test_docstring_examplesz"TestMedian.test_docstring_examples>s����"�)�A�,�,�a�S��U�a�S��U�]�3�3�3���R�U�\�\�!�_�_�c�*�*�*��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��"�)�B�-�-�'�'��1�-�-�Q�C��E�Q�C��E�M�B�B�B���R�U�\�\�!�_�_�c�*�*�*��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��u�|�|�A�B��|�=�=���T�B��8�$�$�$��T�Z��'7�8�8�8���T�
�
�k�)�*�*�*�*�*rXc�.�d}td��D�]}dD]�}td||���}tjt||��ddg��}|D]L\}}	tj�|||����)#t$rt|||||fz���wxYwtj|dz|gddg��}|D]R\}}	tj�|||���t|||||fz���#tj	$rY�OwxYw����dS)	Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sr�)FTr>)�ndminr@FTr�)
rpr
�	itertools�productrQr_r�	Exception�AssertionError�	AxisError)rR�msgr�r@r��argsr��overs        rV�test_axis_argument_errorsz$TestMedian.test_axis_argument_errorsMs���E���1�X�X�	N�	N�E�%�
N�
N���!�5�t�4�4�4��!�(���v�u�)=�)=��t�}�M�M��"&�N�N�J�D�$�N�����Q�T�4��H�H�H�H��$�N�N�N�,�S�D�%��t�3L�-L�M�M�M�N����!�(�E�A�I�,��)>���
�N�N��"&�N�N�J�D�$�N�����Q�T�4��H�H�H�-�S�D�%��t�3L�-L�M�M�M���<����������N�
N�	N�	Ns�"A<�<!B	�"C=�=D	�D	c��tdd���}ttj�|��d��tdd���}ttj�|��tjj��dS)Nr>Fr�T)r
rrQr_rrrds  rV�test_masked_0dzTestMedian.test_masked_0desh���!�%� � � ���R�U�\�\�!�_�_�a�(�(�(��!�$������R�U�\�\�!�_�_�b�e�l�3�3�3�3�3rXc��ttjd��d���}ttj�|��tjj��ttj�|��jdd��tttj�|����tjj
ju��ttjd��d���}ttj�|��d��ttj�|��jdd��tttj�|����tu��ttjd��gd����}ttj�|��d	��ttj�|��jdd��tttj�|����tu��ttjd��gd
����}ttj�|��d��ttj�|��jdd��tttj�|����tu��ttjd��gd����}ttj�|��d
��ttj�|��jdd��tttj�|����tu��ttjd��gd����}ttj�|��d
��ttj�|��jdd��tttj�|����tu��ttjd��gd����}ttj�|��d	��ttj�|��jdd��tttj�|����tu��ttjd��gd����}ttj�|��d	��ttj�|��jdd��tttj�|����tu��dS)Nr�Tr�r\r�Fr�r�r�)rr>r>r>r>r�)rr>r>rrr�r�rr�r�)
r
rQrrr_rrrrr`ra�MaskedConstantr
rds  rV�test_masked_1dzTestMedian.test_masked_1dls#���"�)�A�,�,�T�*�*�*���R�U�\�\�!�_�_�b�e�l�3�3�3��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%����)B�B�C�C�C��"�)�A�,�,�U�+�+�+���R�U�\�\�!�_�_�b�)�)�)��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��"�)�A�,�,�[�[�[�1�1�1���R�U�\�\�!�_�_�c�*�*�*��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��"�)�A�,�,�[�[�[�1�1�1���R�U�\�\�!�_�_�b�)�)�)��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��"�)�A�,�,�[�[�[�1�1�1���R�U�\�\�!�_�_�b�)�)�)��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��"�)�B�-�-�k�k�k�2�2�2���R�U�\�\�!�_�_�b�)�)�)��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��"�)�A�,�,�]�]�]�3�3�3���R�U�\�\�!�_�_�c�*�*�*��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9��"�)�B�-�-�m�m�m�4�4�4���R�U�\�\�!�_�_�c�*�*�*��R�U�\�\�!�_�_�*�B�0@�A�A�A���R�U�\�\�!�_�_�%�%�[�8�9�9�9�9�9rXc
��ttj�t	gd�gd������jtj�t	gd�gd������j��dS)Nr�r�r�r`)rrQr_rr
r�rRs rV�test_1d_shape_consistencyz$TestMedian.test_1d_shape_consistency�st���R�U�\�\�%����W�W�W�"=�"=�"=�>�>�D��U�\�\�%����W�W�W�"=�"=�"=�>�>�D�	G�	G�	G�	G�	GrXc��d\}}ttjdd|����}tx|dd�<|dd�<ttj||ft
�����}|dd�|dd�df<tjt|����}td|��D]0}tj	�
|��|||dd�|f<�1tt|dd�df��d��tt|��d��tt|d�	��tj
|����tt|jd�	��tj
|����dS)
N)�erK��r�rk���r<rr>r�)rrQ�linspacer�emptyrPrrbrpr�shufflerrr�T)rRrs�pr�r��idxrvs       rVrzTestMedian.test_2d�s\�����A����S�"�a�0�0�2�2��!�!��#�2�#���3�4�4�����1�a�&��6�6�6�7�7���A�A�A�$��!�!�!�Q�$���i��A�������q�!���	�	�A��I���c�"�"�"���f�A�a�a�a��d�G�G��V�A�a�a�a��d�G�_�_�a�(�(�(��V�A�Y�Y��"�"�"��V�A�A�&�&�&������4�4�4��V�A�C�a�(�(�(�"�(�1�+�+�6�6�6�6�6rXc�(�ttjd���dd����}tx|dd�<|dd�<tt
|��d��tttj	�|����tu��tt
|d���gd���tttj	�|d�����tu��tt
|d	���gd
���tttj	�|d	�����tu��tt
|d	���jgd���dS)NrKrkr�����-@rr�)�+@r�g/@r>)
rrrrkr/r#r4rrr)
r>r>r>rrrrr>r>r>)rrQrr�rrrrr`r_r
r@rds  rV�
test_2d_waxiszTestMedian.test_2d_waxis�sM�����2���.�.�r�1�5�5�6�6�����"�1�"���"�#�#���V�A�Y�Y��%�%�%���R�U�\�\�!�_�_�%�%�[�8�9�9�9��V�A�A�&�&�&�(:�(:�(:�;�;�;���R�U�\�\�!�!�\�,�,�-�-��<�=�=�=��V�A�A�&�&�&�(J�(J�(J�K�K�K���R�U�\�\�!�!�\�,�,�-�-��<�=�=�=��V�A�A�&�&�&�+�-K�-K�-K�L�L�L�L�LrXc��tj�d���ddd��}t||dzdk<tt
|d��ddgdd	gddgd
d	gg��d|_tt
|d��dd
gddgddgg��tj�d���ddd��}t||dzdk<tt
|d��dd
gddgddgg��dS)Nrr�r�r;rr.r�rr1r3)r�r�r;�crkr�r/r0r�r�r#r2)rQr_rr�rrrrrds  rVr�zTestMedian.test_3d�s���E�L�L����$�$�Q��1�-�-����!�a�%�1�*�
��V�A�q�\�\�R��G�a��W�r�1�g��B�x�#H�I�I�I�����V�A�q�\�\�R��H�r�2�h��R��#A�B�B�B��E�L�L����$�$�Q��1�-�-����!�a�%�1�*�
��V�A�q�\�\�R��H�q�!�f�r�2�h�#?�@�@�@�@�@rXc���ttjd���dd����}tx|dd�<|dd�<tt
|d���t
|d�����dS)NrKrkr�r�rr�r>)rrQrr�rrrrds  rV�
test_neg_axiszTestMedian.test_neg_axis�sq�����2���.�.�r�1�5�5�6�6�����"�1�"���"�#�#���V�A�B�'�'�'����):�):�):�;�;�;�;�;rXc��dD]�}ttj|����}tx|dd�<|dd�<ttjd����}t||���}|dkrt
|d��nt
|d��t||u��tt|��tu����dS)	N)rK�>@r�g?@r�r�r\rzrKr�g.@)
rrQrrrrrrr`r
)rR�vr�r{r�s     rV�test_out_1dzTestMedian.test_out_1d�s���#�
	,�
	,�A��R�Y�q�\�\�*�*�A�#�#�A�b�q�b�E�A�b�c�c�F��r�w�r�{�{�+�+�C��q�c�"�"�"�A��B�w�w��S�$�'�'�'�'��S�#�&�&�&��A��H�����D��G�G�{�*�+�+�+�+�
	,�
	,rXc�l�dD�]/}ttj|���dd����}tx|dd�<|dd�<ttjd����}t
|d|���}|dkr4td	gdzgd
�zd	gdzzdgdzdgd
zzdgdzz���}n3td	gdzgd�zd	gdzzdgdzdgd
zzdgdzz���}t||��t||u��tt|��tu����1dS)N)r5gD@rKr�rkrr�r�r>)r�r{rKr�)rkr/r#r4TFr�r�)r�g�1@g�5@g�9@)rrQrr�rrrrrr`r
)rRr�r�r{r��es      rV�test_outzTestMedian.test_out�sx��#�
	,�
	,�A��R�Y�q�\�\�1�1�"�b�9�9�:�:�A�#�#�A�b�q�b�E�A�b�c�c�F��r�w�r�{�{�+�+�C��q�q�c�*�*�*�A��B�w�w� �"��a��*:�*:�*:�!:�b�T�!�V�!C�'+�f�q�j�E�7�Q�;�&>�$��!��&K�M�M�M���!�"��a��*B�*B�*B�!B�b�T�!�V�!K�'+�f�Q�h�%����&:�d�V�A�X�&E�G�G�G����A�����A��H�����D��G�G�{�*�+�+�+�+�
	,�
	,rXr�)Nr>r[rV)r�r)�argnames�	argvaluesc�����tjdt���}tj�d��tj|j��dd�dfz}|�tj��}tj|t|��<ttj|j��|����|�d�jz}nCt|�j���t��fd�t�j��D����}ttj|����}t!�|d|���}||usJ�t#|j|��dS)	N)r�r�ror�r<)r���r�r[c3�>�K�|]}|�vrdn�j|V��dS)r>N)r)�.0rv�	axis_normrs  ��rV�	<genexpr>z/TestMedian.test_keepdims_out.<locals>.<genexpr>�sN�����L�L�89�Q�)�^�^�������L�L�L�L�L�LrXT)r�r�r{)rQrr�rr
rrZ�intpr��tuplerr�ndimrrpr�rr)	rRr�r@r��	shape_outr{r�r�rs	       @@rV�test_keepdims_outzTestMedian.test_keepdims_out�sP�����x�
�T�2�2�2���I���X�&�&���$�*�)=�)=�a�a�a��g�)F�F��
�H�H�R�W�������U�1�X�X�������,�,�4�8�8�8���<��q�v�
�I�I�,�T�1�6�:�:�I��L�L�L�L�L�=B�1�6�]�]�L�L�L�L�L�I��2�8�I�.�.�/�/�����t��=�=�=����}�}�}�}��V�\�9�-�-�-�-�-rXc���ddgddgddgg}tj�|d��}ddg}ttj�|d���|��dS)Nr�r�r�rr�)rQr_�masked_equalrr)rRrsrfr�s    rV�$test_single_non_masked_value_on_axisz/TestMedian.test_single_non_masked_value_on_axis�ss���R���R���R�����U�'�'��a�0�0�
���8���2�5�<�<�
��<�;�;�#�	%�	%�	%�	%�	%rXc�x�dtjdt���fD�]"}tj�dtjdggd�g��}||_tj�|d���}ttj	|����t|tj��tj�|���d���}ttj	|����t|tj��tj�|d���}tt|��t��t|dtjdg��tj�|d���}tt|��t��t|tjd	g��tj�|d
���}tt|��t��t|tjd	g����$tj�dtjdggd�g��}tjj|dd�d	f<ttj�|d���tj��ttj�|d���dtjdg��ttj�|d���tjdg��dS)NFrr<r>r�r�r�rr;rr�)rQrr�r_r
r�r@rr�isscalarr�ravelrr`r
r)rRr@�dmr�s    rV�test_nanzTestMedian.test_nans[���B�H�Q�d�3�3�3�4�	/�	/�D�����q�"�&�!�n�i�i�i�8�9�9�B��B�G�����R�d��+�+�A��B�K��N�N�#�#�#��q�"�&�)�)�)�����R�X�X�Z�Z�a��0�0�A��B�K��N�N�#�#�#��q�"�&�)�)�)�����R�a��(�(�A���a���+�.�.�.��q�1�b�f�a�.�1�1�1�����R�a��(�(�A���a���+�.�.�.��q�2�6�1�+�.�.�.�����R�b��)�)�A���a���+�.�.�.��q�2�6�1�+�.�.�.�.�
�U�[�[�1�b�f�a�.�)�)�)�4�
5�
5���5�<��1�1�1�a�4���2�5�<�<���<�6�6���?�?�?��2�5�<�<���<�3�3�a����^�D�D�D��2�5�<�<���<�3�3�b�f�c�]�C�C�C�C�CrXc���tj�tjd����}tj�tjd����}tj|d<tjj|d<ttj�|d|���|��tj�tjd����}ttj�|d|���|��tj�tjd	����}ttj�||���|��dS)
N)r��r�r��r;r>rIrrz�r�r>r\)	rQr_rrrr�rrr)rR�ors   rV�test_out_nanzTestMedian.test_out_nan#s����E���r�x��~�~�.�.���E���r�w�v���/�/���&��$���%�,��$���R�U�\�\�!�Q�A�\�.�.��2�2�2��E���r�x��~�~�.�.���R�U�\�\�!�Q�A�\�.�.��2�2�2��E���r�x��|�|�,�,���R�U�\�\�!��\�+�+�Q�/�/�/�/�/rXc�b�tj�tjdt�����}tjj|ddd�<tj|d<ttj�|��tj��ttj�|d���tj��tj�tjdt����	ddd����}tj|j
��dzdk|_|���}tj|d	<tj|d
<ttj�|��tj��ttjtj�|������tj�|d���}tj|d<tj|d<ttj�|d��|��tj�|d���}tj|d
<tj|d<ttj�|d��|��tj�|d���}tj|d<tj|d<ttj�|d��|��dS)Nrr<r�r;rr�r�r>r�)r>r>r;�r;r�rW�r>r�rX)rQr_rrrPrr�rrr�r�r@�copyrr�r)rRrA�aorigrBs    rV�test_nan_behaviorzTestMedian.test_nan_behavior.s3���E���r�y��5�9�9�9�:�:������#�#�A�#���v��!���2�5�<�<��?�?�B�F�3�3�3��2�5�<�<���<�2�2�B�F�;�;�;��E���r�y��5�9�9�9�A�A�!�Q��J�J�K�K����1�6�"�"�Q�&�!�+����������V��'�
��V��'�
�	�2�5�<�<��?�?�B�F�3�3�3����B�E�L�L��O�O�,�,�-�-�-�
�E�L�L��Q�L�'�'���&��$���&��$���R�U�\�\�!�Q�'�'��+�+�+�
�E�L�L��Q�L�'�'���&��$���&��$���R�U�\�\�!�Q�'�'��+�+�+�
�E�L�L��V�L�,�,���v��!���v��!���R�U�\�\�!�V�,�,�a�0�0�0�0�0rXc�&�tjgd�gd�gtj���}tj�||dk���}ttj�|d���d��ttj�|d���jd��ttj�|d	���|d	��ttj�|��d��dS)
N)r�r��r<r�r�r>r�r�Fr)rQr
�uint8r_rrrr@r�s  rV�test_ambigous_fillzTestMedian.test_ambigous_fillQs����H�k�k�k�;�;�;�/�r�x�@�@�@���E���q�q�A�v��.�.���2�5�<�<���<�2�2�C�8�8�8��2�5�<�<���<�2�2�7��?�?�?��2�5�<�<���<�2�2�A�a�D�9�9�9��2�5�<�<��?�?�C�0�0�0�0�0rXc	��tjtjfD�]�}tj|tjgtjtjgg��}tj�|tj|�����}ttj�|d���|tjg��ttj�|d���|tjg��ttj�|��|��tjtjtj|gtjtj|gg��}tj�|tj|�����}ttj�|d���|��ttj�|d���j
d��ttj�|d���|d��ttj�|��|��tj||g||gg��}ttj�|��|��ttj�|d���|��ttj�|d���|��tj|d|dgdtjtjd	gd
tjtj|ggtj���}tj�|tj|�����}|dkrbttj�|d���dd
|dg��ttj�|��d��nattj�|d���dd
|dg��ttj�|��d��ttj�|d���dd|g��tdd��D�]}tdd��D�]}tjtjg|z|g|zzgdz��}tj�|tj|�����}ttj�|��|��ttj�|d���|��ttj�|d���tjg|z|g|zz���������dS)Nr�rr�r>Froi����r�r�r�r<r�r�r�g@g$�g"�g�r�rkr;)
rQr�r
r�r_r�isnanrrrr@�float32rp)rRr�rArvrws     rV�test_specialzTestMedian.test_specialZsh���F�R�V�G�$�(	=�(	=�C���3���.�2�6�2�6�*:�;�<�<�A���"�"�1�2�8�A�;�;�"�7�7�A������a�a��0�0�3���.�A�A�A������a�a��0�0�3���.�A�A�A������a���#�.�.�.���2�6�2�6�3�/�"�&�"�&�#�1F�G�H�H�A���"�"�1�2�8�A�;�;�"�7�7�A��r�u�|�|�A�A�|�6�6��<�<�<��r�u�|�|�A�A�|�6�6�;�U�C�C�C��r�u�|�|�A�A�|�6�6��!��=�=�=��r�u�|�|�A����4�4�4���3��*�s�C�j�1�2�2�A������a���#�.�.�.������a�a��0�0�#�6�6�6������a�a��0�0�#�6�6�6���3��C�4��,�������2��b�f�b�f�c�2�4�!#�
�,�,�,�A���"�"�1�2�8�A�;�;�"�7�7�A��Q�w�w��R�U�\�\�!�!�\�4�4�r�2��t�R�6H�I�I�I��R�U�\�\�!�_�_�c�2�2�2�2��R�U�\�\�!�!�\�4�4�t�R�#��s�6K�L�L�L��R�U�\�\�!�_�_�d�3�3�3������a�a��0�0�3��c�2B�C�C�C��1�b�\�\�
=�
=���q�"���=�=�A���B�F�8�a�<�S�E�A�I�">�!?�!�!C�D�D�A���*�*�1�2�8�A�;�;�*�?�?�A� �����a���#�6�6�6� �����a�a��!8�!8�#�>�>�>� �����a�a��!8�!8�#%�6�(�Q�,�3�%�!�)�!;�=�=�=�=�=�
=�C(	=�(	=rXc�x�tj�tjgt�����}t��5}|�t��ttj�	|��tj
��t|jdj
tu��ddd��n#1swxYwYtj�tjgtd�����}t��5}|�t��tjddt��ttj�	|��tj
��t|jdj
tu��ddd��n#1swxYwYtj�tjgtd�����}t!tj�	|d���|��t!tj�	|d	���|��tj�tjtj
td�����}tjd
���5}tjddt��t!tj�	|d���|��t|dj
tu��ddd��dS#1swxYwYdS)Nr<rr�)r=r��always�r;r�r>T)�record)rQr_rr
rPrr��RuntimeWarningrrr�r�log�category�warnings�filterwarningsr�catch_warnings)rRrAr�rBs    rV�
test_emptyzTestMedian.test_empty�s���E���r�x��%�8�8�8�9�9��
�
 �
 �	9�A�
�H�H�^�$�$�$��r�u�|�|�A�����7�7�7��A�E�!�H�%��7�8�8�8�	9�	9�	9�	9�	9�	9�	9�	9�	9�	9�	9����	9�	9�	9�	9�
�E���r�x��%�q�A�A�A�B�B��
�
 �
 �	9�A�
�H�H�^�$�$�$��#�H�b�.�A�A�A��r�u�|�|�A�����7�7�7��A�E�!�H�%��7�8�8�8�		9�	9�	9�	9�	9�	9�	9�	9�	9�	9�	9����	9�	9�	9�	9�
�E���r�x��%�q�A�A�A�B�B���R�U�\�\�!�!�\�,�,�a�0�0�0��R�U�\�\�!�!�\�,�,�a�0�0�0�
�E���r�x���e�1�E�E�E�F�F��
�
$�D�
1�
1�
1�	5�Q��#�H�b�.�A�A�A������a�a��0�0�!�4�4�4��A�a�D�M�^�3�4�4�4�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5����	5�	5�	5�	5�	5�	5s9�A8C�C�C�BF<�<G�G�6A,L/�/L3�6L3c	���tj�tjd����}t	ttj�|�t������t��tj
|d<t	ttj�|�t������t��dS)Nr�r;)rQr_rrrr`rrZr^rPr�)rRr�s  rV�test_objectzTestMedian.test_object�s����E���r�y��}�}�-�-����R�U�\�\�!�(�(�6�"2�"2�3�3�4�4�e�<�<�<��v��!����R�U�\�\�!�(�(�6�"2�"2�3�3�4�4�e�<�<�<�<�<rXN)r�r�r�r�r�r�r�r�r�r�r�rr�r�r�r�r�r�r�rr�r�r�r�r�r�r�rrr\rXrVr�r�s������� � � �#�#�#�$<�<�<�*
+�
+�
+�N�N�N�04�4�4�$:�$:�$:�LG�G�G�7�7�7� 
M�
M�
M�	A�	A�	A�<�<�<�
,�,�,�,�,�,�"�[����
�
�
��	�	�.�.�	�	�.�$%�%�%�D�D�D�:	0�	0�	0�!1�!1�!1�F1�1�1�)=�)=�)=�V5�5�5�:=�=�=�=�=rXr�c�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�TestCovc�h�ttj�d����|_dS�Nr.)r
rQrrrsr�s rV�setup_methodzTestCov.setup_method�s#���"�)�.�.��,�,�-�-��	�	�	rXc�B�|j}ttj|��t|����ttj|d���t|d�����ttj|dd���t|dd�����dS)NF��rowvarT�r�bias)rsr	rQrrds  rV�test_1d_without_missingzTestCov.test_1d_without_missing�s����I���B�F�1�I�I�s�1�v�v�.�.�.��B�F�1�U�3�3�3�S��5�5I�5I�5I�J�J�J��B�F�1�U��>�>�>���%�d�;�;�;�	=�	=�	=�	=�	=rXc�j�|j�dd��}ttj|��t	|����ttj|d���t	|d�����ttj|dd���t	|dd�����dS)Nr�r�FrTr)rsr�r	rQrrds  rV�test_2d_without_missingzTestCov.test_2d_without_missing�s����I���a��#�#���B�F�1�I�I�s�1�v�v�.�.�.��B�F�1�U�3�3�3�S��5�5I�5I�5I�J�J�J��B�F�1�U��>�>�>���%�d�;�;�;�	=�	=�	=�	=�	=rXc
��|j}t|d<||���z}|���}t	tj|��t
|����t	tj|d���t
|d�����t	tj|dd���t
|dd�����	t
|d���n#t$rYnwxYw|dd�}t	tj||ddd���t
||ddd�����t	tj||ddd�d���t
||ddd�d�����t	tj||ddd�dd���t
||ddd�dd�����dS)NrFrTr��allow_maskedr>)rsr�meanrur	rQr�
ValueError)rRr��nxs   rV�test_1d_with_missingzTestCov.test_1d_with_missing�s����I����"��	�Q�V�V�X�X�
��
�\�\�^�^���B�F�2�J�J��A���/�/�/��B�F�2�e�4�4�4�c�!�E�6J�6J�6J�K�K�K��B�F�2�e�$�?�?�?���%�d�;�;�;�	=�	=�	=�	����&�&�&�&�&���	�	�	��D�	�����q��t�W���B�F�2�r�$�$�B�$�x�0�0�#�a��4�4�R�4��/�/�B�B�B��B�F�2�r�$�$�B�$�x��>�>�>���1�T�T�r�T�7�5�9�9�9�	;�	;�	;��B�F�2�r�$�$�B�$�x��D�I�I�I���1�T�T�r�T�7�5�t�D�D�D�	F�	F�	F�	F�	Fs�C'�'
C4�3C4c�"�|j}t|d<|�dd��}tjt|�����t��}tj||j	��}||�
d��dd�dfz
�d��}tt|��tj
|��|jddz
z|dz
z��tt|d���tj
|d���|jdz|z��tj|j	|��}||�
d��z
�d��}tt|d	�
��tj
|d	�
��|jddz
z|dz
z��tt|d	d���tj
|d	d���|jdz|z��dS)Nrr�r�r>rr�T�rFrr)rsrr�rQ�logical_notrrZrqrr�rrrr	rr)rRr��valid�frac�xfs     rV�test_2d_with_missingzTestCov.test_2d_with_missing�s����I����"��
�I�I�a��O�O����|�A���/�/�6�6�s�;�;���v�e�U�W�%�%���!�&�&��)�)�A�A�A�t�G�$�$�
,�
,�Q�
/�
/���C��F�F��F�2�J�J�!�'�!�*�q�.�9�T�B�Y�G�	I�	I�	I��C���-�-�-��F�2�D�1�1�1�A�G�A�J�>��E�	G�	G�	G��v�e�g�u�%�%���!�&�&��)�)�m�
#�
#�A�
&�
&���C��%�0�0�0��V�B�u�5�5�5��g�a�j�1�n�.�15���<�	>�	>�	>�	�C��%�d�;�;�;��V�B�u�4�@�@�@��W�Q�Z�(�*.�/�	1�	1�	1�	1�	1rXN)r�r�r�r
rrrr#r\rXrVr
r
�sb������.�.�.�=�=�=�=�=�=�F�F�F�01�1�1�1�1rXr
c�8�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	dS)	�TestCorrcoefc���ttj�d����|_ttj�d����|_dSr)r
rQrrrs�data2r�s rVr
zTestCorrcoef.setup_method�s@���"�)�.�.��,�,�-�-��	��2�9�>�>�"�-�-�.�.��
�
�
rXc��|j|j}}tj|��}tj||��}t	��5}tjd��ttt|d���|�	td��ttj|d���t|d�����tt|d���|��tt||d���|��tt|d���|��tt||d���|��ddd��dS#1swxYwYdS)Nr�r��ddof�bias and ddof have no effectrr��rsr'rQrrr�simplefilterrrl�filterr	)rRr�r�r��	expected2�sups      rV�	test_ddofzTestCorrcoef.test_ddof�s����y�$�*�1���;�q�>�>���K��1�%�%�	�
�
 �
 �		C�C��!�(�+�+�+��+�X�q�r�B�B�B�B��J�J�)�+I�J�J�J����A�A� 6� 6� 6����8K�8K�8K�L�L�L����� 4� 4� 4�h�?�?�?����A�B� 7� 7� 7��C�C�C����� 3� 3� 3�X�>�>�>����A�A� 6� 6� 6�	�B�B�B�		C�		C�		C�		C�		C�		C�		C�		C�		C�		C�		C�		C����		C�		C�		C�		C�		C�		Cs�C>E�E�Ec	���|j|j}}tj|��}t	��5}tjd��ttt||dd��ttt||dd��ttt|d���|�	td��tt|d���|��ddd��dS#1swxYwYdS)Nr�TFrr+r>r,)rRr�r�r�r0s     rV�	test_biaszTestCorrcoef.test_biass���y�$�*�1���;�q�>�>��
�
 �
 �	?�C��!�(�+�+�+��+�X�q�!�T�5�I�I�I��+�X�q�!�T�4�H�H�H��+�X�q�u�E�E�E�E��J�J�)�+I�J�J�J����� 3� 3� 3�X�>�>�>�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?����	?�	?�	?�	?�	?�	?s�B(C&�&C*�-C*c	���|j}ttj|��t|����ttj|d���t|d�����t	��5}|�td��ttj|dd���t|dd�����ddd��dS#1swxYwYdS)NFrr+Tr)rsr	rQrrr.rl�rRr�r0s   rVrz$TestCorrcoef.test_1d_without_missings*���I���B�K��N�N�H�Q�K�K�8�8�8��B�K��%�8�8�8�$�Q�u�5�5�5�	7�	7�	7�
�
 �
 �	F�C��J�J�)�+I�J�J�J����A�e�$� G� G� G� (��5�t� D� D� D�
F�
F�
F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F����	F�	F�	F�	F�	F�	Fs�8AC�C�Cc	���|j�dd��}ttj|��t	|����ttj|d���t	|d�����t��5}|�td��ttj|dd���t	|dd�����ddd��dS#1swxYwYdS)Nr�r�Frr+Tr)rsr�r	rQrrr.rlr5s   rVrz$TestCorrcoef.test_2d_without_missings:���I���a��#�#���B�K��N�N�H�Q�K�K�8�8�8��B�K��%�8�8�8�$�Q�u�5�5�5�	7�	7�	7�
�
 �
 �	F�C��J�J�)�+I�J�J�J����A�e�$� G� G� G� (��5�t� D� D� D�
F�
F�
F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F����	F�	F�	F�	F�	F�	Fs�AC*�*C.�1C.c�4�|j}t|d<||���z}|���}t	tj|��t
|����t	tj|d���t
|d�����t��5}|�td��t	tj|dd���t
|dd�����ddd��n#1swxYwY	t
|d���n#t$rYnwxYw|dd�}t	tj||ddd���t
||ddd�����t	tj||ddd�d���t
||ddd�d�����t��5}|�td��t	tj||ddd���t
||ddd�d�	����t	tj||ddd���t
||ddd�d
�����ddd��dS#1swxYwYdS)NrFrr+Trrr>rr;r))rsrrrur	rQrrr.rlr)rRr�rr0s    rVrz!TestCorrcoef.test_1d_with_missing%s ���I����"��	�Q�V�V�X�X�
��
�\�\�^�^���B�K��O�O�X�a�[�[�9�9�9��B�K��5�9�9�9�$�Q�u�5�5�5�	7�	7�	7�
�
 �
 �	F�C��J�J�)�+I�J�J�J����B�u�4� H� H� H� (��5�t� D� D� D�
F�
F�
F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F�	F����	F�	F�	F�	F�	��Q�U�+�+�+�+�+���	�	�	��D�	�����q��t�W���B�K��B�t�t��t�H�5�5�x��1�T�T�r�T�7�7K�7K�L�L�L��B�K��B�t�t��t�H�U�C�C�C�$�Q��$�$�B�$���>�>�>�	@�	@�	@�
�
 �
 �	>�C��J�J�)�+I�J�J�J����B��4�4�R�4�� 9� 9� (��A�d�d��d�G�!� <� <� <�
>�
>�
>����B��4�4�R�4�� 9� 9� (��A�d�d��d�G�!� <� <� <�
>�
>�
>�	>�	>�	>�	>�	>�	>�	>�	>�	>�	>�	>�	>����	>�	>�	>�	>�	>�	>s8�-AD
�
D�D�D(�(
D5�4D5�B&J
�
J�Jc��|j}t|d<|�dd��}t|��}t	j|��}t|dd�dd�f|dd�dd�f��t
��5}|�td��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��dS#1swxYwYdS)	Nrr�r�r+rr)r>r)	rsrr�rrQr	rr.rl)rRr�rSrTr0s     rVr#z!TestCorrcoef.test_2d_with_missingCs����I����"��
�I�I�a��O�O����{�{���+�a�.�.���D��"��c�r�c��N�G�C�R�C��"��H�,=�>�>�>�
�
 �
 �	3�C��J�J�)�+I�J�J�J����� 4� 4� 4�S�b�S�#�2�#�X� >� '����S�b�S�� 1�
3�
3�
3����� 3� 3� 3�C�R�C��"��H� =� '����S�b�S�� 1�
3�
3�
3����� 3� 3� 3�C�R�C��"��H� =� '����S�b�S�� 1�
3�
3�
3�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3����	3�	3�	3�	3�	3�	3s�CE�E�EN)
r�r�r�r
r1r3rrrr#r\rXrVr%r%�s�������/�/�/�C�C�C� ?�?�?�	F�	F�	F�	F�	F�	F�>�>�>�<3�3�3�3�3rXr%c��eZdZd�Zd�ZdS)�TestPolynomialc�j�tj�d��}tj�d���dd��}t	t||d��tj||d����|�t��}t|d<|�t��}tx|d<|d<t||dd�dfdd	�
��\}}}}}tj|dd�|dd�df�	��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	|
|���tj�d��dz}|���}|dd�}|dd�}|dd�}t||dd	|���\}}}}}tj||dd	|���\}}	}
}}t||��t|||||f||	|
||f��D]\}
}t	|
|���dS)
NrkrIrr;r�rrG�rrT�rrr>)rrr�)
rQrrr�r	r�viewr
rru�zipr�r)rRr�r��C�R�K�S�DrZr�rxrzrrA�a_r��wo�xs�ys�wss                    rV�test_polyfitzTestPolynomial.test_polyfitYs���
�I�N�N�2�����I�N�N�2���&�&�r�1�-�-���G�A�q�!�,�,�b�j��A�q�.A�.A�B�B�B�
�F�F�;������!��
�F�F�;����$�$��$��!�F�)�!�!�Q�q�q�q�!�t�W�a�d�;�;�;���A�q�!�Q��*�Q�q�r�r�U�A�a�b�b�!�e�H�,?�,?�,A�,A�1�*.�0�0�0���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�<�	'�	'�G�Q����2�&�&�&�&�!�!�Q�q�q�q�"�u�X�q�t�<�<�<���A�q�!�Q��*�Q�q��t�W�a��"��b��k�1�4�H�H�H���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�<�	'�	'�G�Q����2�&�&�&�&�!�!�Q���5�5�5���A�q�!�Q��*�Q�q��t�W�a��"��Q�Q�Q��i���F�F�F���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�<�	'�	'�G�Q����2�&�&�&�&��I�N�N�2����"��
�V�V�X�X��
�q��t�W��
�q��t�W��
�q��t�W��!�!�Q����:�:�:���A�q�!�Q��*�R��Q�T�R�@�@�@���A�q�!�Q��Q������A�q�!�Q��?�Q��1�a��O�<�<�	'�	'�G�Q����2�&�&�&�&�	'�	'rXc�l�tj�d��}tj�d���dd��}tj|d<tj|d<|�t��}|�t��}t|d<t|d<t||dd�	��\}}}}}tj|d
d�|d
d�dd�fdd�	��\}}	}
}}t|||||f||	|
||f��D]\}
}t|
|���dS)NrkrIrr;rr<r�Tr=r>)rQrrr�r�r>r
rrr?r	)rRr�r�r@rArBrCrDrZr�rxrzrrArEs               rV�test_polyfit_with_masked_NaNsz,TestPolynomial.test_polyfit_with_masked_NaNs�s*���I�N�N�2�����I�N�N�2���&�&�r�1�-�-���v��!���6��%��
�F�F�;����
�F�F�;������!����%��!�!�Q���5�5�5���A�q�!�Q��*�Q�q��t�W�a��"��Q�Q�Q��i���F�F�F���A�q�!�Q��A�q�!�Q��?�Q��1�a��O�<�<�	'�	'�G�Q����2�&�&�&�&�	'�	'rXN)r�r�r�rJrLr\rXrVr:r:Ws3������%'�%'�%'�N'�'�'�'�'rXr:c�n�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zd�Zd�ZdS)�TestArraySetOpsc�4�gd�}t|dd���}tt|dt����t	|dtgd�gd������t	|dgd	���t	|d
gd���dS)N�r>r>r>r;r;r�T��return_index�return_inverserr�r�r�r>)rr�r�r;)rrrr>r>r;)rrrr
rrr s   rV�test_unique_onlistz"TestArraySetOps.test_unique_onlist�s���!�!�!���d��d�C�C�C���
�4��7�K�0�0�1�1�1��T�!�W�l�9�9�9�9�9�9�E�E�E�F�F�F��T�!�W�i�i�i�(�(�(��T�!�W�0�0�0�1�1�1�1�1rXc��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���dS)NrP)rrr>rr>rr�TrQr)r>r;r�r�rrrr>r>)rr�r�r;r;)rrr�r>r�r;r�)rsr@�
fill_value)rrrrWr s   rV�test_unique_onmaskedarrayz)TestArraySetOps.test_unique_onmaskedarray�s:���.�.�.�5G�5G�5G�H�H�H���d��d�C�C�C���T�!�W�l�=�=�=�|�|�|�L�L�L�M�M�M��T�!�W�l�l�l�+�+�+��T�!�W�0�0�0�1�1�1�����!3�!3�!3�!3�!3�!3��C�C�C���d��d�C�C�C���T�!�W�l�=�=�=�|�|�|�L�L�L�M�M�M��T�!�W�l�l�l�+�+�+��T�!�W�0�0�0�1�1�1�1�1rXc���tgd�d���}t|dd���}t|dtdgdg�����t|ddg��t|dgd���t}t|dd���}t|dtt����t|ddg��t|ddg��dS)	Nr-Tr�rQrr>r;r�)rrrrr s   rV�test_unique_allmaskedz%TestArraySetOps.test_unique_allmasked�s����I�I�I�D�1�1�1���d��d�C�C�C���T�!�W�l�A�5��v�>�>�>�?�?�?��T�!�W�q�c�"�"�"��T�!�W�i�i�i�(�(�(����d��d�C�C�C���T�!�W�l�6�2�2�3�3�3��T�!�W�q�c�"�"�"��T�!�W�q�c�"�"�"�"�"rXc�X�ttjd��gd����}tgd�gd����}t	|��}t||��t|�d��|�d����t|j|j��dS)Nr��r>rrrr>r�)r>r>r>r�)r>rrr>r)rrQrr
r%rrrr@)rRr�rTrSs    rV�test_ediff1dzTestArraySetOps.test_ediff1d�s������1���O�O�O�<�<�<������<�<�<�8�8�8���q�z�z���T�7�#�#�#��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-�-�-rXc�v�ttjd��gd����}t|t���}tgd�gd����}t
||��t
|�d��|�d����t
|j|j��t|gd����}tgd	�gd
����}t
||��t
|�d��|�d����t
|j|j��dS)Nr�r\r�)�to_begin)rr>r>r>r�)r>r>rrr>rr�)r>r;r�r>r>r>r�)rrrr>rrr>�	rrQrr%rr
rrrr@�rRr�rSrTs    rV�test_ediff1d_tobeginz$TestArraySetOps.test_ediff1d_tobegin�s�����1���O�O�O�<�<�<���q�6�*�*�*������o�o�o�>�>�>���T�7�#�#�#��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-��q�9�9�9�-�-�-���-�-�-�4I�4I�4I�J�J�J���T�7�#�#�#��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-�-�-rXc�v�ttjd��gd����}t|t���}tgd�gd����}t
||��t
|�d��|�d����t
|j|j��t|gd����}tgd	�gd
����}t
||��t
|�d��|�d����t
|j|j��dS)Nr�r\r�)�to_end)r>r>r>r�r)r>rrr>r>rr�)r>r>r>r�r>r;r�)r>rrr>rrrr`ras    rV�test_ediff1d_toendz"TestArraySetOps.test_ediff1d_toend�s�����1���O�O�O�<�<�<���q��(�(�(������o�o�o�>�>�>���T�7�#�#�#��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-��q����+�+�+���-�-�-�4I�4I�4I�J�J�J���T�7�#�#�#��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-�-�-rXc��ttjd��gd����}t|tt���}tgd�gd����}t
||��t
|�d��|�d����t
|j|j��t|gd�t���}tgd	�gd
����}t
||��t
|�d��|�d����t
|j|j��dS)Nr�r\r��rdr_)rr>r>r>r�r)r>r>rrr>r>rr�)rr>r>r>r�r>r;r�)r>r>rrr>rrrr`ras    rV�test_ediff1d_tobegin_toendz*TestArraySetOps.test_ediff1d_tobegin_toend�s,�����1���O�O�O�<�<�<���q��&�9�9�9���*�*�*�1C�1C�1C�D�D�D���T�7�#�#�#��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-��q����V�<�<�<���0�0�0�5�5�5�7�7�7���T�7�#�#�#��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-�-�-rXc��tjd��}t|��}tgd�gd����}t	||��tt
|t����t	|�d��|�d����t	|j	|j	��t|tt���}tgd�gd����}tt
|t����t	|�d��|�d����t	|j	|j	��dS)	Nr��r>r>r>r>�rrrrr�rrgr�r�)rQrr%r
rrrr
rrr@rras    rV�test_ediff1d_ndarrayz$TestArraySetOps.test_ediff1d_ndarray�s"���I�a�L�L���q�z�z������<�<�<�8�8�8���T�7�#�#�#��
�4��-�-�.�.�.��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-��q��&�9�9�9���*�*�*�1C�1C�1C�D�D�D���
�4��-�-�.�.�.��T�[�[��^�^�W�^�^�A�%6�%6�7�7�7��T�Y���-�-�-�-�-rXc���tgd�gd����}tgd�gd����}t||��}tgd�gd����}t||��dS)N)r>r�r�r�rVr�)r�r>r>r>)r>r�rrb)r
r#r)rRr�r�rSrTs     rV�test_intersect1dz TestArraySetOps.test_intersect1dsu���,�,�,�\�\�\�2�2�2���,�,�,�\�\�\�2�2�2���1�a� � ���
�
�
����3�3�3���T�7�#�#�#�#�#rXc�4�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����dS)N�r>r;r�ror�rrrrr>r��r>r;r�r�r�rrq)r�r�ro)r>r;r�r�r�)r�r�rorrVr�)rr�r�r)r>r�r;r�)rr>rr)rr�r�r�)r
r rrrr
r�rRrArBrSs    rV�
test_setxor1dzTestArraySetOps.test_setxor1d
s����"�"�"����9�9�9���%�%�%�,>�,>�,>�?�?�?����1�~�~���T�5����+�+�,�,�,��"�"�"����9�9�9���O�O����1�~�~���T�5����\�\�\�B�B�B�C�C�C��)�)�)�����)�)�)������1�~�~���
�4��-�-�.�.�.��T�-�-�-�.�.�.��,�,�,�\�\�\�2�2�2���,�,�,�\�\�\�2�2�2����1�~�~���
�4��-�-�.�.�.��T�-�-�-�.�.�.��2�x��B�/�/�0�0�0�0�0rXc��tjd���gd���}tjgd���}d|d<t	||���}t	gd�gd����}tdt
�	��}d
|d<d
|d<d
|d
<t
||��}tt|t����t||��tj|||j��|jz}t||��dS)Nrr�r>)r>r;rr�)	rrkrIrKr>r�r�r$rM)	rr>rr>rr>rr>r)rsr@r<Tr�rb)rr;r�)rQrr�rr
r�r2rrr
rr@)rRrAr@rB�ecrZrs       rV�	test_isinzTestArraySetOps.test_isin$s��
�I�b�M�M�!�!�)�)�)�,�,���x�	�	�	�"�"����W�
��!�$������:�:�:�:�:�:�
<�
<�
<��
�9�D�
)�
)�
)����7����7����7����A�J�J���
�1�k�*�*�+�+�+��1�b�!�!�!��G�A�q�!�&��z�"�"�a�f�W�,���1�a� � � � � rXc�r�tgd�gd����}tgd�gd����}t||��}t|gd���tgd�gd����}tgd�gd	����}t||��}t|gd
���tgtgg����dS)Nrprqr�rrrq)TTTFT�r�r�r;r>r�r>r�rrb)TTFTT)r
r$rrrss    rV�	test_in1dzTestArraySetOps.test_in1d9s����"�"�"����9�9�9���%�%�%�,>�,>�,>�?�?�?���A�q�z�z���T�:�:�:�;�;�;��"�"�"����9�9�9���*�*�*�9�9�9�-�-�-���A�q�z�z���T�:�:�:�;�;�;��2�t�B��|�|�,�,�,�,�,rXc���tgd�gd����}tgd�gd����}ttjt	||����t	||d�����tgd�gd����}tgd	�gd
����}ttjt	||����t	||d�����tgt	ggd�����dS)Nrprqr�rrrqT)�invertryrzrb)r
rrQr}r$rrws   rV�test_in1d_invertz TestArraySetOps.test_in1d_invertGs����"�"�"����9�9�9���%�%�%�,>�,>�,>�?�?�?���R�Y�t�A�q�z�z�*�*�D��A�d�,C�,C�,C�D�D�D��"�"�"����9�9�9���*�*�*�9�9�9�-�-�-���R�Y�t�A�q�z�z�*�*�D��A�d�,C�,C�,C�D�D�D��2�t�B��4�8�8�8�9�9�9�9�9rXc���tgd�gd����}tgd�gd����}t||��}tgd�gd����}t||��tgd�gd�ggd	�gd
�g���}tgd�gd����}tgd
�gd����}t||��}t||��tgtgg����dS)N)r>r;r�ror�rrqr�rr)r>r;r�r�r�ror)rrrrrrr>rar,r�rbrrq)rr>r;r�r�r�)r
r"rr)	rRrArBrSrTr�r��ezr�s	         rV�test_union1dzTestArraySetOps.test_union1dSs"���%�%�%�,>�,>�,>�?�?�?���%�%�%�,>�,>�,>�?�?�?���q�!�}�}���.�.�.�5J�5J�5J�K�K�K���T�7�#�#�#�
�9�9�9�i�i�i�(�	�	�	�9�9�9�/E�F�F�F���/�/�/����8�8�8��
�%�%�%�,>�,>�,>�
?�
?�
?���A�q�M�M���Q������2�w�r�2���/�/�/�/�/rXc���tgd�gd����}tgd���}t||��}t|tgd�gd������td��}td��}tt||��tdd	g����tgtjg���}tt|g��jtj��dS)
N)rr�r�roror>r;r>)rrrrrrrr>r�)r;r�r�r�r;r>r�)rrorrbrkr�r�)r
r!rrrQ�uint32r=rss    rV�test_setdiff1dzTestArraySetOps.test_setdiff1des����*�*�*�1I�1I�1I�J�J�J���'�'�'�(�(����A�����T�5����)�)�)�<�<�<�=�=�=��2�J�J���1�I�I���Y�q�!�_�_�e�Q��F�m�m�4�4�4��"�b�i�b�)�)�)���Y�q�"�%�%�+�R�Y�7�7�7�7�7rXc��tjgd���}tjgd���}tt||��tjdg����dS)N)rArBrZ)rArBrzrZ)rQr
rr!rws   rV�test_setdiff1d_char_arrayz)TestArraySetOps.test_setdiff1d_char_arrayrsP���H�_�_�_�%�%���H�_�_�_�%�%���9�Q��?�?�B�H�c�U�O�O�<�<�<�<�<rXN)r�r�r�rTrXrZr]rbrerhrlrnrtrwr{r~r�r�r�r\rXrVrNrN�s������2�2�2�2�2�2� 
#�
#�
#�.�.�.�
.�
.�
.�
.�
.�
.�.�.�.� .�.�.� $�$�$�1�1�1�4!�!�!�*-�-�-�
:�
:�
:�0�0�0�$8�8�8�=�=�=�=�=rXrNc��eZdZd�Zd�ZdS)�
TestShapeBasec�z�tgd�gd����}t|��}t|jd��t|jj|jj��t|jd��t|jj|jj��t|jj|jj��dS)Nrar`r�r�r�)rrrrr@rsrws   rV�test_atleast_2dzTestShapeBase.test_atleast_2d{s����������3�3�3���q�M�M���Q�W�f�%�%�%��Q�V�\�1�6�<�0�0�0��Q�W�d�#�#�#��Q�V�\�1�6�<�0�0�0��Q�V�\�1�6�<�0�0�0�0�0rXc�R�td��}t|jd��t|jj|j��t|jj|j��tdd��}|D]U}t|jd��t|jj|j��t|jj|j���Vtd��}t|jd��t|jj|j��t|jj|j��tdd��}|D]U}t|jd��t|jj|j��t|jj|j���Vt
d��}t|jd��t|jj|j��t|jj|j��t
dd��}|D]U}t|jd��t|jj|j��t|jj|j���Vtd��}t|jd��t|jj|jj��dS)Nr�r[r�rHr-)rrrr@rsrrr3)rRrBrAs   rV�test_shape_scalarzTestShapeBase.test_shape_scalar�s7��

�s�O�O���Q�W�d�#�#�#��Q�V�\�1�7�+�+�+��Q�V�\�1�7�+�+�+��s�C� � ���	0�	0�A����$�'�'�'�����q�w�/�/�/�����q�w�/�/�/�/��s�O�O���Q�W�f�%�%�%��Q�V�\�1�7�+�+�+��Q�V�\�1�7�+�+�+��s�C� � ���	0�	0�A����&�)�)�)�����q�w�/�/�/�����q�w�/�/�/�/��s�O�O���Q�W�i�(�(�(��Q�V�\�1�7�+�+�+��Q�V�\�1�7�+�+�+��s�C� � ���	0�	0�A����)�,�,�,�����q�w�/�/�/�����q�w�/�/�/�/��S�M�M���Q�W�f�%�%�%��Q�V�\�1�6�<�0�0�0�0�0rXN)r�r�r�r�r�r\rXrVr�r�ys2������1�1�1�(1�(1�(1�(1�(1rXr�c� �eZdZd�Zd�Zd�ZdS)�TestNDEnumeratec	�R�tjd���d��}tj|t���}t||���}t
ttj|����tt|������t
tt|����tt|������t
tt|����tt|d�������dS)Nr�)r>r�r;r<r�F�ru)	rQrr��
zeros_liker�rrrYr4)rR�ordinary�
empty_mask�	with_masks    rV�test_ndenumerate_nomaskedz)TestNDEnumerate.test_ndenumerate_nomasked�s����9�R�=�=�(�(��3�3���]�8�4�8�8�8�
� ��
�;�;�;�	��T�"�.��2�2�3�3��+�h�/�/�0�0�	2�	2�	2��T�+�h�/�/�0�0��+�i�0�0�1�1�	3�	3�	3��T�+�i�0�0�1�1��+�i�E�B�B�B�C�C�	E�	E�	E�	E�	ErXc
���td��}td��}td��}ttt|����g��ttt|����g��ttt|d�����tt	tjd��dtgz������ttt|����g��ttt|d�����tt	tjd��dtgz������dS)Nr\)�dr�Fr�r�r)r0rrYr4r?rQ�ndindexr)rRrArBrZs    rV�test_ndenumerate_allmaskedz*TestNDEnumerate.test_ndenumerate_allmasked�s0���r�N�N���v�����y�!�!���T�+�a�.�.�)�)�2�.�.�.��T�+�a�.�.�)�)�2�.�.�.��T�+�a�E�:�:�:�;�;��#�b�j��0�0�#���.�A�A�B�B�	D�	D�	D��T�+�a�.�.�)�)�2�.�.�.��T�+�a�E�:�:�:�;�;��#�b�j��3�3�Y�&��5I�J�J�K�K�	M�	M�	M�	M�	MrXc
��ttjd���d��gd�gd�gd�g���}gd�}t	tt
|����|��t	ttt
|d�	������|j��t
|d�	��D]\}}t	|||���dS)
Nr.r�rj)r>r>rr>rkr�))rWr))r;rr�)r�r�)rIrk)r�r�Fr�)	rrQrr�rrYr4rbr�)rRrA�items�
coordinate�values     rV�test_ndenumerate_mixedmaskedz,TestNDEnumerate.test_ndenumerate_mixedmasked�s������2���.�.�v�6�6�+�|�|�+�|�|�+�|�|�-�
.�
.�
.��G�G�G���T�+�a�.�.�)�)�5�1�1�1��S��k�!��>�>�>�?�?�@�@�!�&�I�I�I�!,�Q�5�!A�!A�!A�	/�	/��J����:���.�.�.�.�	/�	/rXN)r�r�r�r�r�r�r\rXrVr�r��sG������	E�	E�	E�
M�
M�
M�
/�
/�
/�
/�
/rXr�c� �eZdZd�Zd�Zd�ZdS)�	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��dS)Nrar`r��r�r�ror�rr�r�r>�r�r;�rr5rrrr@r6rs�rRrArBrZrs     rV�
test_stack_1dzTestStack.test_stack_1d�s3���������3�3�3���������3�3�3���1�a�&�q�!�!�!���Q�W�f�%�%�%��1�6�1�Q�4�9�-�-�-��1�6�1�Q�4�9�-�-�-��A�q�6�N�N���1�6�1�6�*�*�*��1�6�1�6�*�*�*��1�a�&�q�!�!�!���Q�W�f�%�%�%��1�6�1�Q�Q�Q��T�7�<�0�0�0��1�6�1�Q�Q�Q��T�7�<�0�0�0�0�0rXc��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��dS)NraTr�r�Frr�r�r>r�r�r�s     rV�test_stack_maskszTestStack.test_stack_masks�s+�������.�.�.�������/�/�/���1�a�&�q�!�!�!���Q�W�f�%�%�%��1�6�1�Q�4�9�-�-�-��1�6�1�Q�4�9�-�-�-��A�q�6�N�N���1�6�1�6�*�*�*��1�6�1�6�*�*�*��1�a�&�q�!�!�!���Q�W�f�%�%�%��1�6�1�Q�Q�Q��T�7�<�0�0�0��1�6�1�Q�Q�Q��T�7�<�0�0�0�0�0rXc���d}tj�dd|��}tj�dd|��}tj�dd|���t��}tj�dd|���t��}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}tj�dd|��}tj�dd|��}tj�dd|���t��}tj�dd|���t��}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
��dS)
Nr�rrkr;r�r�r:r>r).r).r>)r�r;r�r�)rQr�randintrZr�rr5rrrr@)
rR�shp�d1�d2r�r��a1r[rZ�c_shps
          rV�
test_stack_ndzTestStack.test_stack_nd�s�����
�Y�
�
�q�"�c�
*�
*��
�Y�
�
�q�"�c�
*�
*��
�Y�
�
�q�!�S�
)�
)�
0�
0��
6�
6��
�Y�
�
�q�!�S�
)�
)�
0�
0��
6�
6��
�"�2�
&�
&�
&��
�"�2�
&�
&�
&���2�r�(��#�#�#���s�
���Q�W�e�$�$�$��2�7�A�a�D�I�.�.�.��2�7�A�a�D�I�.�.�.��2�r�(��$�$�$���d�
���Q�W�e�$�$�$��2�7�A�f�I�N�3�3�3��2�7�A�f�I�N�3�3�3���
�Y�
�
�q�"�c�
*�
*��
�Y�
�
�q�"�c�
*�
*��
�Y�
�
�q�!�S�
)�
)�
0�
0��
6�
6��
�Y�
�
�q�!�S�
)�
)�
0�
0��
6�
6��
�"�2�
&�
&�
&��
�"�2�
&�
&�
&���2�r�(��#�#�#���s�
���Q�W�e�$�$�$��2�7�A�a�D�I�.�.�.��2�7�A�a�D�I�.�.�.��2�r�(��$�$�$���d�
���Q�W�e�$�$�$��2�7�A�f�I�N�3�3�3��2�7�A�f�I�N�3�3�3�3�3rXN)r�r�r�r�r�r�r\rXrVr�r��sA������1�1�1�$1�1�1�$)4�)4�)4�)4�)4rXr�)N�__doc__rr�r��numpyrQ�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/r0r1r2r3r4r5r6r8r�rrr)r�r�r�r
r%r:rNr�r�r�r\rXrV�<module>r�sS������������
�
�
�
�����3�3�3�3�3�3�������������������������������������������������������������������������������������������������������������������������������DK9�K9�K9�K9�K9�K9�K9�K9�\).�).�).�).�).�).�).�).�XC�C�C�C�C�C�C�C�LF�F�F�F�F�F�F�F�R,�,�,�,�,�,�,�,�,
!�
!�
!�
!�
!�
!�
!�
!�T=�T=�T=�T=�T=�T=�T=�T=�n@1�@1�@1�@1�@1�@1�@1�@1�Fh3�h3�h3�h3�h3�h3�h3�h3�V7'�7'�7'�7'�7'�7'�7'�7'�te=�e=�e=�e=�e=�e=�e=�e=�P41�41�41�41�41�41�41�41�n#/�#/�#/�#/�#/�#/�#/�#/�LO4�O4�O4�O4�O4�O4�O4�O4�O4�O4rX

Youez - 2016 - github.com/yon3zu
LinuXploit