403Webshell
Server IP : 217.160.0.135  /  Your IP : 216.73.217.25
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/lib/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

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

���cH����dZddlZddlmZmZmZmZddlm	Z	m
Z
mZmZm
Z
mZmZmZddlZGd�d��ZGd�d��ZdS)	z-Test functions for 1D array set operations.

�N)�assert_array_equal�assert_equal�
assert_raises�assert_raises_regex)�ediff1d�intersect1d�setxor1d�union1d�	setdiff1d�unique�in1d�isinc
��eZdZd�Zd�Zd�Zd�Zd�Zej	�
dejgd�ej
���d	ejd
fejgd�ej
���ejgd�ej���d	dfejgd
�ej���ejejdfg��d���Zej	�
dejgd�ej���ddejgd�ej���fejgd�ej���ejdgej���d	ejgd�ej���fejgd�ej���ddejgd�ej���fejgd�ej
���ddejgd�ej
���fg��d���Zej	�
dgd���d���Zej	�
dgd���d���Zd�Zej	�
dgd���d ���Zej	�
dgd���d!���Zd"�Zej	�
dgd���d#���Zej	�
dd	d$g��d%���Zd&�Zej	�
d'ejejfejejfejej fej ejfejejfejejfg��ej	�
dgd���d(�����Z!ej	�
dgd���d)���Z"d*�Z#d+�Z$d,�Z%d-�Z&d.�Z'd/�Z(d0�Z)d1�Z*d2�Z+d3�Z,d4�Z-d	S)5�
TestSetOpsc���tjgd���}tjgd���}tjgd���}t||d���}t||��tjgd���}tjgd���}tjgd���}t||��}t||��tgtgg����dS)N������r��rr)rrrT��
assume_unique)rrrrr)rrrrrrr)�np�arrayrr)�self�a�b�ec�c�eds      �B/usr/lib/python3/dist-packages/numpy/lib/tests/test_arraysetops.py�test_intersect1dzTestSetOps.test_intersect1ds����H�\�\�\�"�"���H�_�_�_�%�%��
�X�i�i�i�
 �
 ����1�D�1�1�1���1�b�!�!�!�
�H�_�_�_�%�%���H�*�*�*�+�+��
�X�i�i�i�
 �
 ����1�����1�b�!�!�!��2�{�2�r�2�2�3�3�3�3�3�c��Gd�d��}|��}t||��}t||��tgd�gd���}t|gd���dS)Nc��eZdZd�ZdS)�4TestSetOps.test_intersect1d_array_like.<locals>.Testc�*�tjd��S)Nr)r�arange�rs r$�	__array__z>TestSetOps.test_intersect1d_array_like.<locals>.Test.__array__%s���y��|�|�#r&N)�__name__�
__module__�__qualname__r-�r&r$�Testr)$s#������
$�
$�
$�
$�
$r&r2�rrr)rr)rr2r�ress    r$�test_intersect1d_array_likez&TestSetOps.test_intersect1d_array_like"s���	$�	$�	$�	$�	$�	$�	$�	$�
�D�F�F���!�Q�����3��"�"�"��)�)�)�Y�Y�Y�/�/���3�	�	�	�*�*�*�*�*r&c�Z�tjgd���}tjgd���}t||dd���\}}}tjgd���}t||��t|||��t|||��tjgd���}tjgd���}t||d���\}}}tjgd���}t||��t|||��t|||��tjgd	�gd
�g��}tjgd�gd�g��}t||dd���\}}}tj||j��}tj||j��}	tjgd
���}
t|
||��t|
||	��tjgd�gd�g��}tjgd�gd�g��}t||d���\}}}tj||j��}tj||j��}	tjgd���}
t|
||��t|
||	��dS)N�rrrr)rrr�T)r�return_indices)rrr)rrrrrrr)r�rrrrrr)r9)rrrr8)rr:r�)rrrr8)�
�r:�	)rr8rr:)rrrr8r8)rrr:rr)rrrr)r<r=r:r)rrr:)rrrr�
unravel_index�shape)rrr r"�i1�i2�ee�ef�ui1�ui2�eas           r$�test_intersect1d_indicesz#TestSetOps.test_intersect1d_indices.s����H�\�\�\�"�"���H�\�\�\�"�"����1�D��N�N�N�	��2�r�
�X�i�i�i�
 �
 ���1�b�!�!�!��1�R�5�"�%�%�%��1�R�5�"�%�%�%�
�H�*�*�*�+�+���H�-�-�-�.�.����1�T�:�:�:�	��2�r�
�X�l�l�l�
#�
#���1�b�!�!�!��1�R�5�"�%�%�%��1�R�5�"�%�%�%�
�H�l�l�l�M�M�M�2�3�3���H�l�l�l�N�N�N�3�4�4����1�D��N�N�N�	��2�r���r�1�7�+�+����r�1�7�+�+��
�X�l�l�l�
#�
#���2�q��v�&�&�&��2�q��v�&�&�&�
�H�o�o�o����7�8�8���H�l�l�l�N�N�N�3�4�4����1�T�:�:�:�	��2�r���r�1�7�+�+����r�1�7�+�+��
�X�i�i�i�
 �
 ���2�q��v�&�&�&��2�q��v�&�&�&�&�&r&c��tjgd���}tjgd���}tjgd���}t||��}t||��tjgd���}tjgd���}tjgd���}t||��}t||��tjgd���}tjgd���}tjgd���}t||��}t||��tgtgg����dS)	Nrr)rrrr3)r8rr)rrrrrr8)rr:rr)r8rrr:)rrr	r�rrr r!r"s     r$�
test_setxor1dzTestSetOps.test_setxor1dUs*���H�\�\�\�"�"���H�_�_�_�%�%��
�X�i�i�i�
 �
 ���Q��N�N���1�b�!�!�!��H�Y�Y�Y�����H�Y�Y�Y����
�X�(�(�(�
)�
)���Q��N�N���1�b�!�!�!��H�\�\�\�"�"���H�\�\�\�"�"��
�X�(�(�(�
)�
)���Q��N�N���1�b�!�!�!��2�x��B�/�/�0�0�0�0�0r&c	��tjg��}tjdg��}tjddg��}tgt|����tdgt|d�����tdgt|d�����tddgt|dd�����tgt|����tdgt|����tgd�t|d	d
�����tgd�t|dd
gd	dg�����tdd
gt|d
�����tgd�t|d	dg�����td	dgt|d	�����tgd�t|dd
g�����dS)Nrrr)�to_begin)�to_end���)rMrN)rrr>rr>)rr8rrr:rr8r:)rrr:)rr8r)rrrr)r�	zero_elem�one_elem�two_elems    r$�test_ediff1dzTestSetOps.test_ediff1dms����H�R�L�L�	��8�Q�C�=�=���8�Q��F�#�#���2�w�y�1�1�2�2�2��A�3��	�A� >� >� >�?�?�?��A�3��	�!� <� <� <�=�=�=��B��7�G�I��1�$M�$M�$M�N�N�N��2�w�x�0�0�1�1�1��A�3��� 1� 1�2�2�2��9�9�9�g�h��1�&M�&M�&M�N�N�N��?�?�?�"�8�q�!�f�a��V�L�L�L�	N�	N�	N��A�q�6�7�8�A�#>�#>�#>�?�?�?��9�9�9�g�h��1�v�&F�&F�&F�G�G�G��A�q�6�7�8�a�#@�#@�#@�A�A�A��9�9�9�g�h�!�Q��&H�&H�&H�I�I�I�I�Ir&zary, prepend, append, expectedr3��dtypeNrN)rrrrM)��?g@g"@c��d�|��}tt|��5t|||���ddd��dS#1swxYwYdS)Nz dtype of `{}` must be compatible��aryrNrM)�formatr�	TypeErrorr)rrY�prepend�append�expected�msgs      r$�!test_ediff1d_forbidden_type_castsz,TestSetOps.test_ediff1d_forbidden_type_casts�s���:1�7�7��A�A��
 ��C�
0�
0�	&�	&���!�$�
&�
&�
&�
&�	&�	&�	&�	&�	&�	&�	&�	&�	&�	&�	&�	&����	&�	&�	&�	&�	&�	&s�A�A�Azary,prepend,append,expectedii)rrrrr)rrrr)rrrrr���)rrrrac�x�tj|||���}t||��|j|jksJ�dS)NrX)rrrrU)rrYr\r]r^�actuals      r$�test_ediff1d_scalar_handlingz'TestSetOps.test_ediff1d_scalar_handling�sO��:���#)�%,�.�.�.��	�V�X�&�&�&��|�x�~�-�-�-�-�-�-r&�kind)N�sort�tablec���
�d�}tj|tgdh����
�
�fd�}tjd���gd���}tjgd�gd�gd	�g��}|||��d
dg}d}|||��tjd
��}|||��|||��|||��|d|��||d��|dd���dkr&g}	||	|��|||	��||	|	��ttjtjfD]�}
�dkr|
tjkr�|
tjtjhvrtjgd�|
���}n |
thvrtjgd���}tjg|
���}|||��|||��|||����dS)Nc�z�tj|��������}||vS)N)r�asarray�flatten�tolist)rr s  r$�
_isin_slowz(TestSetOps.test_isin.<locals>._isin_slow�s0���
�1�
�
�%�%�'�'�.�.�0�0�A���6�Mr&r)�otypes�excludedc�d��t||����}�||��}t||��dS)N�re)rr)rr �x�y�	isin_slowres    ��r$�assert_isin_equalz/TestSetOps.test_isin.<locals>.assert_isin_equal�s<����Q���%�%�%�A��	�!�Q���A��q�!�$�$�$�$�$r&��rrr)r<��)rrr)���!)r>r:)rr8)r>rrrr8rgrT�TFF)r�	vectorize�boolr+�reshaper�int64�float64)rrermrurr r"�d�frrrU�ar�empty_arrayrts `           @r$�	test_isinzTestSetOps.test_isin�sr����
	�	�	��L��T�F�a�S�I�I�I�	�	%�	%�	%�	%�	%�	%�
�I�b�M�M�!�!�)�)�)�,�,���H�l�l�l�I�I�I�|�|�|�<�=�=����!�Q�����V�������!�Q����
�H�Q�K�K����!�Q������!�Q������!�Q����	��!�Q������!�Q������!�Q�����7�?�?��A���a��#�#�#���a��#�#�#���a��#�#�#��B�H�b�j�1�
	8�
	8�E��w���5�B�J�#6�#6�����2�:�.�.�.��X�l�l�l�%�8�8�8����4�&����X�2�2�2�3�3���(�2�U�3�3�3�K���k�2�.�.�.���b�+�.�.�.���k�;�7�7�7�7�
	8�
	8r&c��dD�]q}gd�}gd�|z}tjgd���}t||d|���}t||��d|d<tjgd	���}t||d|���}t||��d
\|d<|d<tjgd���}t||d|���}t||��tjgd
���}gd�|z}gd�}t|||���}t||��|gd�|zz}gd�}t|||���}t||��tjgd���}tjgd�|z��}tjgd���}t|||���}t||��tjgd���}tjgd�|z��}tjgd���}t|||���}t||��tjddg��}tjddg|z��}tjddg��}t|||���}t||����stjdg��}tjdg��}tjdg��}t|||���}t||��|dvr"ttgg|���g��dSdS)N�rr<rr�TFTTT�rrer:r)FFTT)rr:r)TFTF�rrrrrrrrrrrrrrrw)FTFTTTTTTFTFFFrq)rrr)TTTTTTTTTTTFTT)rrrrr)rrrrrr)TFTTTrrF>Nrf�rrr
r)rre�multrr r!r"s       r$�	test_in1dzTestSetOps.test_in1dsU���/	&�/	&�D����A����$�&�A���3�3�3�4�4�B��Q���D�9�9�9�A��q�"�%�%�%��A�a�D���4�4�4�5�5�B��Q���D�9�9�9�A��q�"�%�%�%��J�A�a�D�!�A�$���4�4�4�5�5�B��Q���D�9�9�9�A��q�"�%�%�%���C�C�C�D�D�A��	�	�D� �A�4�4�4�B��Q���%�%�%�A��q�"�%�%�%��I�I�I��$�$�A�+�+�+�B��Q���%�%�%�A��q�"�%�%�%������&�&�A������4�/�0�0�A���3�3�3�4�4�B��Q���%�%�%�A��q�"�%�%�%������)�)�A���+�+�+�d�2�3�3�A���9�9�9�:�:�B��Q���%�%�%�A��q�"�%�%�%���!�Q�� � �A���!�Q��$��'�'�A���5�%�.�)�)�B��Q���%�%�%�A��q�"�%�%�%�%��H�a�S�M�M���H�a�S�M�M��
�X�u�g�
�
����A�D�!�!�!���1�b�!�!�!��>�!�!��t�B���6�6�6��;�;�;�;�;�"�!r&c���tjgd���}tjddg��}tjgd���}t||��}t||��dS)N)rr r"r��er"r�r rr")TFTFFTFFr�rJs     r$�test_in1d_char_arrayzTestSetOps.test_in1d_char_arrayAsd���H�=�=�=�>�>���H�c�3�Z� � ��
�X�K�K�K�
L�
L����A�J�J���1�b�!�!�!�!�!r&c
��dD]b}tjgd���}gd�|z}ttjt	|||�����t	||d|������c|dvr�dD]�}tjgd�tj���}gd�|z}tj|tj���}ttjt	|||�����t	||d|�������d	Sd	S)
zTest in1d's invert parameterr�r�rwrqT��invertre>NrfrTN)rrrr�r
�float32)rrer�rr s     r$�test_in1d_invertzTestSetOps.test_in1d_invertJsG��
�	C�	C�D���C�C�C�D�D�A��	�	�D� �A��r�y��a���)>�)>�)>�?�?�#�A�q��D�A�A�A�
C�
C�
C�
C��>�!�!��
G�
G���H�G�G�G�"$�*�.�.�.���I�I��$���H�Q�b�j�1�1�1��"�2�9�T�!�Q�T�-B�-B�-B�#C�#C�#'��1�T��#E�#E�#E�G�G�G�G�
"�!�
G�
Gr&c�.�tjd���dd��}tjdd���dd��}tjdd���dd��}tjgd���}t	t||d|�	��|��t	t||d
|�	��|��t	t||d|�	��|��t	t||d
|�	��|��dS)Nr8rrr>�?ry)FFFTTTTr�F)rr+r�rrr
)rrerr �long_br!s      r$�test_in1d_ravelzTestSetOps.test_in1d_ravel_s:��
�I�a�L�L� � ��A�&�&���I�a��O�O�#�#�A�q�)�)����1�b�!�!�)�)�"�a�0�0��
�X�=�=�=�
>�
>���4��1�D�t�D�D�D��	�	�	��4��1�E�%)�+�+�+��	�	�	�	�4��6��%)�+�+�+��	�	�	�	�4��6��%)�+�+�+��	�	�	�	�	r&c�`�tjgd�tj���}tjgd�tj���}tjgd�t���}t	|t||����t	tj|��t||d�����tjgd�tj���}tjgd�tj���}tjgd	���}t||d�
��}t	||��dS)z(Hit the standard isin code with integers)rrrrrr�e��ArT)rrrr�)rrrrrrrT�r�r)rrrrrr�r�rN)rrr�rrr
r�)rrr r^r!r"s      r$�!test_in1d_hit_alternate_algorithmz,TestSetOps.test_in1d_hit_alternate_algorithmts��
�H�,�,�,�B�H�=�=�=���H�^�^�^�2�8�4�4�4���8�1�1�1��>�>�>���8�T�!�Q�Z�Z�0�0�0��2�9�X�.�.��Q��$�0G�0G�0G�H�H�H��H�\�\�\���2�2�2���H�)�)�)���:�:�:��
�X�/�/�/�
0�
0����A�T�*�*�*���1�b�!�!�!�!�!r&c	�0�tjddg��}tjgd���}tjddg��}t|t|||�����ttj|��t||d|�����dS)z&Test that in1d works for boolean inputTF)FFFrqr�N)rrrr
r�)rrerr r^s     r$�test_in1d_booleanzTestSetOps.test_in1d_boolean�s���
�H�d�E�]�#�#���H�*�*�*�+�+���8�U�D�M�*�*���8���1�4�0�0�0�	2�	2�	2��2�9�X�.�.���1�T��=�=�=�	?�	?�	?�	?�	?r&rfc�X�tj�d��}|�ddd���}|�ddd���}t	||��}|�d��}|�d��}t
|t	|||�����dS)z(Test that in1d works for timedelta inputr�dr<)�size�timedelta64[s]rqN)r�random�RandomState�randintr
�astyper)rre�rstaterr �truth�a_timedelta�b_timedeltas        r$�test_in1d_timedeltazTestSetOps.test_in1d_timedelta�s�����&�&�q�)�)���N�N�1�c��N�+�+���N�N�1�c��N�+�+���Q��
�
���h�h�/�0�0���h�h�/�0�0���5�$�{�K�d�"K�"K�"K�L�L�L�L�Lr&c���tjgd�d���}|}tjt��5t||d���ddd��dS#1swxYwYdS)N�rrrr�rTrgrq)rr�pytest�raises�
ValueErrorr
�rrr s   r$�test_in1d_table_timedelta_failsz*TestSetOps.test_in1d_table_timedelta_fails�s����H�Y�Y�Y�&6�7�7�7��
��
�]�:�
&�
&�	%�	%���A�G�$�$�$�$�	%�	%�	%�	%�	%�	%�	%�	%�	%�	%�	%�	%����	%�	%�	%�	%�	%�	%s�A�A�Az
dtype1,dtype2c��tj|tj��}tjgd�|���}|rtjgd�|���}ntjgd�|���}tjgd���}|dkoLt	|tjko|tjk|tjko|tjkf��}|rHtjtd���5t|||�	��d
d
d
��d
S#1swxYwYd
Stt|||�	��|��d
S)z7Test that in1d works as expected for mixed dtype input.)rrrrrT)i����r�)r�r�)TTFFrgzexceed the maximum)�matchrqN)r�
issubdtype�
signedintegerr�any�int8�int16r�r��RuntimeErrorr
r)	r�dtype1�dtype2re�is_dtype2_signed�ar1�ar2r^�expect_failures	         r$�test_in1d_mixed_dtypez TestSetOps.test_in1d_mixed_dtype�s����=���1A�B�B���h�|�|�|�6�2�2�2���	8��(�>�>�>��8�8�8�C�C��(�=�=�=��7�7�7�C��8�6�6�6�7�7������S��b�g��4�&�B�H�"4��b�h��4�6�R�W�#4�2
�.�.��
�	D���|�3G�H�H�H�
*�
*��S�#�D�)�)�)�)�
*�
*�
*�
*�
*�
*�
*�
*�
*�
*�
*�
*����
*�
*�
*�
*�
*�
*�
�t�C��4�8�8�8�(�C�C�C�C�Cs�1D�D�Dc��tjdD]�}tjgd�t���}tjgd�|���}tjgd�t���}t	t|||���|��||}}tjgd�t���}t	t|||���|����dS)	z4Test that in1d works as expected for bool/int input.�
AllIntegerr}rT�rrrr)FTTrq)TTTTN)r�	typecodesrrrr
)rrerUrr r^s      r$�test_in1d_mixed_booleanz"TestSetOps.test_in1d_mixed_boolean�s����\�,�/�	@�	@�E���-�-�-�T�:�:�:�A������U�3�3�3�A��x� 3� 3� 3�4�@�@�@�H��t�A�q�t�4�4�4�h�?�?�?��a�q�A��x� 8� 8� 8��E�E�E�H��t�A�q�t�4�4�4�h�?�?�?�?�	@�	@r&c��dg}tjdgdz��}tjdg��}tj||��}t||��dS�Nrr<Fr��rr�r�r^�results     r$�test_in1d_first_array_is_objectz*TestSetOps.test_in1d_first_array_is_object�sV���f���h��s�2�v�����8�U�G�$�$�����c�"�"���6�8�,�,�,�,�,r&c��d}tjdgdz��}tjdg��}tj||��}t||��dSr�r�r�s     r$� test_in1d_second_array_is_objectz+TestSetOps.test_in1d_second_array_is_object�sT�����h��v�b�y�!�!���8�U�G�$�$�����c�"�"���6�8�,�,�,�,�,r&c��dg}tjdgdz��}tjdg��}tj||��}t||��dS)Nr<Tr�r�s     r$� test_in1d_both_arrays_are_objectz+TestSetOps.test_in1d_both_arrays_are_object�sV���f���h��v�b�y�!�!���8�T�F�#�#�����c�"�"���6�8�,�,�,�,�,r&c�$�tjdtfdtfg��}tjdg|���}tjdgdz|���}tjdg��}tj||��}t
||��dS)N�field1�field2)rNrTr<T)rrU�int�objectrr
r)r�dtr�r�r^r�s      r$�+test_in1d_both_arrays_have_structured_dtypez6TestSetOps.test_in1d_both_arrays_have_structured_dtype�s����X��#���6�(:�;�
<�
<���h�	�{�"�-�-�-���h�	�{�2�~�R�0�0�0���8�T�F�#�#�����c�"�"���6�8�,�,�,�,�,r&c�V�tjddgt���}tjddgt���}tjddg��}tj||��}t	||��tj||d���}t	|tj|����tjgd�t���}|dd�}tjgd�t���}|dd�}tjddg��}tj||��}t	||��tj||d���}t	|tj|����tjgd�t���}|dd�}tjddgt���}tjdd	g��}tj||��}t	||��tj||d���}t	|tj|����dS)
N�rrrTTr�)r�)rrrrO)r��rrrF)rrr�r
rr�r�s     r$�'test_in1d_with_arrays_containing_tuplesz2TestSetOps.test_in1d_with_arrays_containing_tuples�s����h��a�y��/�/�/���h��a�y��/�/�/���8�T�4�L�)�)�����c�"�"���6�8�,�,�,����c�$�/�/�/���6�2�9�X�#6�#6�7�7�7��h�(�(�(��7�7�7���#�2�#�h���h�(�(�(��7�7�7���#�2�#�h���8�T�4�L�)�)�����c�"�"���6�8�,�,�,����c�$�/�/�/���6�2�9�X�#6�#6�7�7�7��h�(�(�(��7�7�7���#�2�#�h���h��a�y��/�/�/���8�T�5�M�*�*�����c�"�"���6�8�,�,�,����c�$�/�/�/���6�2�9�X�#6�#6�7�7�7�7�7r&c��tjgd���}tjgd���}ttt||d���tjgd�t
���}tjgd�t
���}ttt||d���tjtjfD]�}tjgd�|���}tjd	tj|��j	g|���}ttt||d���tj||d
���}t|dgdgd
zz��tj||d���}t|dgdgd
zz����d
S)z&Test that in1d raises expected errors.)rrrrr)rrr8r:r<�	quicksortrq)rrrr rrTrg)rOrrrrrONTFrrf)rrrr�r
r��int32r��iinfo�maxr�r)rr�r��obj_ar1�obj_ar2rU�overflow_ar2r�s        r$�test_in1d_errorszTestSetOps.test_in1d_errorss����h����'�'���h�'�'�'�(�(���j�$��S�{�C�C�C�C��(�.�.�.�f�=�=�=���(�.�.�.�f�=�=�=���j�$���w�G�G�G�G��h���)�	=�	=�E��(�+�+�+�5�9�9�9�C��8�R���%���)<�$=�U�K�K�K�L�

���c�<�g�
�
�
�
��W�S�,�T�:�:�:�F��v��v���!��';�<�<�<��W�S�,�V�<�<�<�F��v��v���!��';�<�<�<�<�'	=�	=r&c���tjgd���}tjgd���}tjgd���}t||��}t||��tjgd�gd�g��}tjgd���}tjgd���}t||��}t||��tgtgg����dS)N)rrrrr�rrrrrrr)rrrrrrr�)rrr)rrrrr)rrrrrr)rrr
r)	rrr r!r"rrrs�ez�zs	         r$�test_union1dzTestSetOps.test_union1d3s����H�_�_�_�%�%���H�*�*�*�+�+��
�X�(�(�(�
)�
)���A�q�M�M���1�b�!�!�!�
�H�i�i�i����+�,�,���H�_�_�_�%�%��
�X�(�(�(�
)�
)���A�q�M�M���1�b�!�!�!��2�w�r�2���/�/�/�/�/r&c�Z�tjgd���}tjgd���}tjddg��}t||��}t||��tjd��}tjd��}tjddg��}t||��}t||��tgtgg����tjdtj��}t
t|g��jtj��dS)	N)r8rrrrrrrr�r8r��rxr1)rrrrr+�uint32rrUrJs     r$�test_setdiff1dzTestSetOps.test_setdiff1dEs����H�-�-�-�.�.���H�*�*�*�+�+��
�X�q�!�f�
�
���a��O�O���1�b�!�!�!��I�b�M�M���I�b�M�M��
�X�r�2�h�
�
���a��O�O���1�b�!�!�!��2�y��R�0�0�1�1�1��H�R���#�#���Y�q�"�%�%�+�R�Y�7�7�7�7�7r&c���tjgd���}tjgd���}tjddg��}t||d���}t||��dS)N)rrr)rrrrrTr)rrrr)rrr r^rcs     r$�test_setdiff1d_uniquez TestSetOps.test_setdiff1d_uniqueWse���H�Y�Y�Y�����H�Y�Y�Y�����8�Q��F�#�#���1�a�t�4�4�4���V�X�&�&�&�&�&r&c��tjgd���}tjgd���}tt||��tjdg����dS)N)rr r")rr �sr")rrrrr�s   r$�test_setdiff1d_char_arrayz$TestSetOps.test_setdiff1d_char_array^sP���H�_�_�_�%�%���H�_�_�_�%�%���9�Q��?�?�B�H�c�U�O�O�<�<�<�<�<r&c��tjgd���}tjgd���}t||��}t||��}t	||��}t||��}t
||��dS)N)rrrrr:)r>r:rrrrr)rrr	rr
rr)rrr �c1�aux1�aux2�c2s       r$�
test_manywayszTestSetOps.test_manywayscsy���H�_�_�_�%�%���H�*�*�*�+�+��
�a��^�^���1�a� � ���q�!�}�}��
�t�T�
"�
"���2�r�"�"�"�"�"r&).r.r/r0r%r5rHrKrSr��mark�parametrizerrr��nanr�r�r`r�r�r�rdr�r�r�r�r�r�r�r�r��uint8�uint16r�r�r�r�r�r�r�r�r�r�r�r�r�r1r&r$rrs~������4�4�4�$
+�
+�
+�%'�%'�%'�N1�1�1�0J�J�J�&�[���=�
���)�)�)�2�8�	,�	,�	,�	
�	��	�	�
���)�)�)�2�8�	,�	,�	,�	���)�)�)�2�:�	.�	.�	.�	
�	�	�
���,�,�,�b�g�	.�	.�	.�	��	��	�	�%@���.
&�
&�/��.
&��[���%�
�"�(�9�9�9�B�H�
-�
-�
-�
�
�
�"�(�<�<�<�r�x�
0�
0�
0�
2��"�(�9�9�9�B�J�
/�
/�
/�
�"�(�A�3�b�j�
)�
)�
)�
�
�"�(�9�9�9�B�J�
/�
/�
/�
1��"�(�9�9�9�B�H�
-�
-�
-�
�
�
�"�(�<�<�<�r�x�
0�
0�
0�
2��"�(�9�9�9�B�H�
-�
-�
-�
�
�
�"�(�=�=�=���
1�
1�
1�
3�	
���*.�.�+��*.��[���V�%<�%<�%<�=�=�:8�:8�>�=�:8�x�[���V�%<�%<�%<�=�=�;<�;<�>�=�;<�z"�"�"��[���V�%<�%<�%<�=�=�G�G�>�=�G�(�[���V�%<�%<�%<�=�=���>�=��("�"�"� �[���V�%<�%<�%<�=�=�?�?�>�=�?��[���V�d�F�^�4�4�M�M�5�4�M�%�%�%��[����
�W�b�h��
�X�r�w��
�X�r�y�!�
�Y���!�
�X�r�x� �
�X�r�x� �
	
�
�
��[���V�%<�%<�%<�=�=�D�D�>�=�
�
�D�.�[���V�%<�%<�%<�=�=�
@�
@�>�=�
@�-�-�-�-�-�-�-�-�-�-�-�-�8�8�8�B =� =� =�D0�0�0�$8�8�8�$'�'�'�=�=�=�
#�#�#�#�#r&rc��eZdZd�Zd�Zd�Zd�Zej�	dddg��d���Z
d	�Zd
�Zd�Z
d�Zd
�ZdS)�
TestUniquec	��d�}gd�dz}gd�}gd�}gd�dz}tjgd�d��}g}|�tjd��|�tjd	��|�d
��|�d��|D]<}tj||��}	tj||��}
||	|
||||���=d}tjt|��|��}	||	dd�<tjt|��|��}
||
dd�<||	|
||||��d
d
g}tjtt||����|��}	tjtt||����|��}
||	|
||||��gd�}	ttj|	��ddg��gd�}gd�}gd�}t|��}
t|
|��t|d���\}}t||��t||��tjd��}d|d<tj|d���\}}t|tj
d����g}tj|d���d}tj|d���d}tj|dd���dd�\}}t|jtj��t|jtj��t|jtj��t|jtj��dtjdtjg}ddtjg}gd�}gd�}gd �}ttj|��|��ttj|d���||f��ttj|d���||f��ttj|d�!��||f��d"tjd#t%d$tj��t%dtj��g}d#d"t%d$tj��g}gd%�}gd&�}gd'�}ttj|��|��ttj|d���||f��ttj|d���||f��ttj|d�!��||f��tjd(��}tjd)��|tjd*��|g}tjd*��tjd)��|g}gd�}gd�}gd �}ttj|��|��ttj|d���||f��ttj|d���||f��ttj|d�!��||f��tjd(��}tjdd+��|tjdd,��|g}tjdd,��tjdd+��|g}gd�}gd�}gd �}ttj|��|��ttj|d���||f��ttj|d���||f��ttj|d�!��||f��tjgd-z}tjg}d.g}gd/�}d-g}ttj|��|��ttj|d���||f��ttj|d���||f��ttj|d�!��||f��dS)0Nc�`�d}|�d|��}t|��}t|||��|�d|��}t|ddd��\}}	t|||��t|	||��|�d|��}t|ddd��\}}	t|||��t|	||��|�d|��}t|ddd��\}}	t|||��t|	||��|�d|��}t|ddd��\}}
}t|||��t|
||��t|||��|�d	|��}t|ddd��\}}
}t|||��t|
||��t|||��|�d
|��}t|ddd��\}}
}t|||��t|
||��t|||��|�d|��}t|ddd��\}}
}}t|||��t|
||��t|||��t|||��dS)Nzcheck {0} failed for type {1}�values�return_indexTF�return_inverse�
return_countszreturn_index and return_inversezreturn_index and return_countsz return_inverse and return_countsz.return_index, return_inverse and return_counts)rZrr)
rr rArBr"r��base_msgr_�v�j�j1�j2�j3s
             r$�	check_allz,TestUnique.test_unique_1d.<locals>.check_allrs���6�H��/�/�(�B�/�/�C��q�	�	�A��q�!�S�)�)�)��/�/�.�"�5�5�C��!�T�5�%�0�0�D�A�q��q�!�S�)�)�)��q�"�c�*�*�*��/�/�"2�B�7�7�C��!�U�D�%�0�0�D�A�q��q�!�S�)�)�)��q�"�c�*�*�*��/�/�/�2�6�6�C��!�U�E�4�0�0�D�A�q��q�!�S�)�)�)��q�!�S�)�)�)��/�/�"C�R�H�H�C��q�$��e�4�4�I�A�r�2��q�!�S�)�)�)��r�2�s�+�+�+��r�2�s�+�+�+��/�/�"B�B�G�G�C��q�$��t�4�4�I�A�r�2��q�!�S�)�)�)��r�2�s�+�+�+��r�1�c�*�*�*��/�/�"D�b�I�I�C��q�%��t�4�4�I�A�r�2��q�!�S�)�)�)��r�2�s�+�+�+��r�1�c�*�*�*��/�/�$7�9;�=�=�C�"�1�d�D�$�7�7�M�A�r�2�r��q�!�S�)�)�)��r�2�s�+�+�+��r�2�s�+�+�+��r�1�c�*�*�*�*�*r&)rrrrrrrr<)rrrr)rrrr)rrrrrrr)rrrrr��AllFloat�
datetime64[D]�timedelta64[D]�O)��i)��?��?�rrr)�rrrr�r3)rrrrrrT)r	rr.)rr)rr	g@rV)rrr)rrrr)rrr)r
y@�y�?�?�)rrr)rrrrr)rrr�natz
2020-12-26z
2020-12-24�D�hrrr�)r�multiply�extendr�r]r�empty�len�list�ziprr�	chararray�zerosrrU�intpr�complex�
datetime64�timedelta64)rrrr rArBr"�typesr��aa�bb�unq�inv�a1�a2�a2_inv�a1_idx�a3_idx�a3_inv�ua�ua_idx�ua_inv�ua_cntr�all_nanss                         r$�test_unique_1dzTestUnique.test_unique_1dps���.	+�.	+�.	+�`
"�!�!�"�$���L�L��
�\�\��
"�
"�
"�2�
%���K����b�)�)����
���R�\�,�/�0�0�0�
���R�\�*�-�.�.�.�
���_�%�%�%�
���%�&�&�&��	-�	-�B���!�R���B���!�R���B��I�b�"�b�"�a��,�,�,�,���
�X�c�!�f�f�b�
!�
!����1�1�1��
�X�c�!�f�f�b�
!�
!����1�1�1���	�"�b�"�b�!�R�(�(�(���
#��
�X�d�3�q�!�9�9�o�o�r�
*�
*��
�X�d�3�q�!�9�9�o�o�r�
*�
*���	�"�b�"�b�!�R�(�(�(�$�
#�
#���2�9�R�=�=�8�X�*>�?�?�?�
%�$�$���i�i�� � � ��
�A�Y�Y���2�s�#�#�#��A�d�3�3�3�
��F��2�s�#�#�#��6�3�'�'�'�
�L��O�O����#���Y�q��6�6�6�
��F��6�2�8�A�;�;�/�/�/�
����1�4�0�0�0��3����1�T�2�2�2�1�5����1�4�26�8�8�8�89���<�����V�\�2�7�+�+�+��V�\�2�7�+�+�+��V�\�2�7�+�+�+��V�\�2�7�+�+�+��"�&�#�r�v�&���3���
���������������R�Y�q�\�\�2�&�&�&��R�Y�q�t�4�4�4�r�6�l�C�C�C��R�Y�q��6�6�6��V��E�E�E��R�Y�q��5�5�5��F�|�D�D�D��R�V�V�W�S�"�&�%9�%9�7�3���;O�;O�P���f�g�c�2�6�2�2�
3������ ���������R�Y�q�\�\�2�&�&�&��R�Y�q�t�4�4�4�r�6�l�C�C�C��R�Y�q��6�6�6��V��E�E�E��R�Y�q��5�5�5��F�|�D�D�D��m�E�"�"��
�]�<�
(�
(�#�r�}�\�/J�/J�C�P���m�L�)�)�2�=��+F�+F��
L���������������R�Y�q�\�\�2�&�&�&��R�Y�q�t�4�4�4�r�6�l�C�C�C��R�Y�q��6�6�6��V��E�E�E��R�Y�q��5�5�5��F�|�D�D�D��n�U�#�#��
�^�A�s�
#�
#�S�"�.��C�*@�*@�#�F���n�Q��$�$�b�n�Q��&<�&<�c�
B���������������R�Y�q�\�\�2�&�&�&��R�Y�q�t�4�4�4�r�6�l�C�C�C��R�Y�q��6�6�6��V��E�E�E��R�Y�q��5�5�5��F�|�D�D�D��F�8�a�<���f�X�������������R�Y�x�(�(�"�-�-�-��R�Y�x�d�;�;�;�b�&�\�J�J�J��R�Y�x��=�=�=��F�|�L�L�L��R�Y�x�t�<�<�<�r�6�l�K�K�K�K�Kr&c�l�tt|jt��tt|jdtfdtfg��tt
jttjd��d���tt
jttjd��d���dS)Nrr r<r��axis���)	rr[�_run_axis_testsr�r�r�	AxisErrorrr+r,s r$�test_unique_axis_errorsz"TestUnique.test_unique_axis_errorss����i��!5�v�>�>�>��i��!5��S�z�C��=�1�	3�	3�	3�	�b�l�F�B�I�b�M�M��B�B�B�B��b�l�F�B�I�b�M�M��C�C�C�C�C�Cr&c��d}gd�gd�g}tj|��}tt|d���t|d���|��tt|d���t|d���|��dS)NzUnique failed on list of lists�rrrrr=r)rrjrr)rr_�inp�inp_arrs    r$�test_unique_axis_listz TestUnique.test_unique_axis_list!s���.���y�y�)�)�)�$���*�S�/�/���6�#�A�.�.�.��w�Q�0G�0G�0G��M�M�M��6�#�A�.�.�.��w�Q�0G�0G�0G��M�M�M�M�Mr&c��g}|�tjd��|�tjd��|�d��|�d��|�dtfdtfg��|�dtfdt
fg��|D]}|�|���d}tjdtj�	���	d
d���
t��}tjdd
gd
d
ggt�	��}tt|d���||��d}tjddgddgddgddgg��}tjddgg��}tt|d���||��dS)Nr�rrrrr z&Non-bitwise-equal booleans test failedr<rTrOrFTrr=z"Negative zero equality test failedg�r)r rr�r]r��floatr@r+rr��viewrrrr)rr+rUr_�datar�s      r$�test_unique_axiszTestUnique.test_unique_axis(s�����
���R�\�,�/�0�0�0�
���R�\�*�-�.�.�.�
���_�%�%�%�
���%�&�&�&�
���s�C�j�3��*�-�.�.�.�
���s�C�j�3��,�/�0�0�0��	(�	(�E�� � ��'�'�'�'�6���y��2�8�,�,�,�4�4�R��;�;�@�@��F�F����E�4�=�4��,�7�t�D�D�D���6�$�Q�/�/�/���=�=�=�2���x�$���s�D�k�D�#�;��d��L�M�M����D�#�;�-�(�(���6�$�Q�/�/�/���=�=�=�=�=r&r>rrOc�x�tjgd���}t||���}t|gd���dS)N)rrrrrrrrr=r7�rrrr)rr>rr�uniqs    r$�test_unique_1d_with_axisz#TestUnique.test_unique_1d_with_axis>sD���H�-�-�-�.�.���a�d�#�#�#���4����.�.�.�.�.r&c�Z�tjdtj���}t|dddd���\}}}}t	|j|j��t
|tjd�����t
|tjdg����t
|tjddg����t
|tjdg����t|d	ddd���\}}}}t	|j|j��t
|tjd�����t
|tjg����t
|tjg����t
|tjg����d
}tj|���}tt|����D]Z}t|��}	||dkrd|	|<nd	|	|<t
t||���tj|	������[dS)N)rr)r@rUrT�r>rr	r
)rr)r@rr)rrrrrrrr=)rr!r�rrrUrr�ranger"r#)
r�single_zerorO�idxr/�cntr@�multiple_zerosr>�expected_shapes
          r$�test_unique_axis_zerosz!TestUnique.test_unique_axis_zerosDs���h�V�2�7�;�;�;��$�[�q�t�48��N�N�N���c�3��	�T�Z��!2�3�3�3��4����!7�!7�!7�8�8�8��3���!��
�
�.�.�.��3���!�Q�� 0� 0�1�1�1��3���!��
�
�.�.�.�%�[�q�t�48��N�N�N���c�3��	�T�Z��!2�3�3�3��4����!7�!7�!7�8�8�8��3������-�-�-��3������-�-�-��3������-�-�-�&�����.�.�.���#�e�*�*�%�%�	?�	?�D�!�%�[�[�N��T�{�a���'(��t�$�$�'(��t�$��v�n�4�@�@�@�!�x�n�=�=�=�
?�
?�
?�
?�	?�	?r&c�D�tjgd�d���}tj�|d��}tj|��}tj|dd���\}}}d}t|j|j|��t|j|j|��dS)N)�@rrrrr�r�rrrrrrr�rrrTrT)rr
z6Unique returned different results when asked for index)rr�ma�masked_equalrrrK�mask)rrrrsr�v2rr"r_s        r$�test_unique_maskedzTestUnique.test_unique_maskedhs����H�F�F�F�"�
$�
$�
$���E���q�!�$�$���I�a�L�L���9�Q�T��F�F�F���A�q�F���1�6�2�7�C�0�0�0��1�6�2�7�C�0�0�0�0�0r&c��d}dD]E}tjdgdgg|��}tj|d���}t||||z���FdS)Nz*sort order incorrect for integer type '%s'�bhilqrOrr=rN)r�fmtr�rr s     r$� test_unique_sort_order_with_axisz+TestUnique.test_unique_sort_order_with_axisush��;���	/�	/�B���2�$����b�)�)�A��	�!�!�$�$�$�A��q�!�S�2�X�.�.�.�.�	/�	/r&c��tjgd�gd�gd�gd�g���|��}d}tjddg��}tt	|��|�|��|��d}tjgd�gd�g��}tt	|d���|�|��|��d}tjgd	�gd
�gd	�gd
�g��}tt	|d���|�|��|��d}tjddgddggddgddggg���|��}tj|ddgd���}tt	|d���||��t	|dd
d
d
���\}}}}	d}t||||��d}t|||��d}t|	tjddg��|��t	|dd
d
d
���\}}}}	d}t|dd�|f|��d}t|dd�|f|��d}t|	tjgd���|��dS)N)rrrr)rrrrz&Unique with 1d array and axis=0 failedrrz&Unique with 2d array and axis=0 failedr=z&Unique with 2d array and axis=1 failed)rrrrDz&Unique with 3d array and axis=2 failedrTrRz-Unique's return_index=True failed with axis=0z/Unique's return_inverse=True failed with axis=0z.Unique's return_counts=True failed with axis=0z-Unique's return_index=True failed with axis=1z/Unique's return_inverse=True failed with axis=1z.Unique's return_counts=True failed with axis=1)rrr)rrr�rr�take)
rrUrKr_r��data3drOrUr/rVs
          r$r@zTestUnique._run_axis_tests~s����x����%���%���%���'�(�(�)/��u�
�
�	
�
7����1�a�&�!�!���6�$�<�<����u�)=�)=�s�C�C�C�6����<�<�<����6�7�7���6�$�Q�/�/�/����u�1E�1E�s�K�K�K�6����9�9�9�i�i�i����I�I�I�F�G�G���6�$�Q�/�/�/����u�1E�1E�s�K�K�K�6����Q��F���F�$���F���F�$�%�&�&�'-�f�U�m�m�	����!�Q��a�0�0�0���6�&�q�1�1�1�6�3�?�?�?�$�T���48��N�N�N���c�3��=���4��9�d�C�0�0�0�?���4��9�d�+�+�+�>���3���!�Q�� 0� 0�#�6�6�6�$�T���48��N�N�N���c�3��=���4����3��<��.�.�.�?���4����3��<��.�.�.�>���3������ 3� 3�S�9�9�9�9�9r&c	��tjddtjtjtjg��}tj|��}tj|d���}t	|tjdtjg����t	|tjdtjtjtjg����dS)NrF)�	equal_nan)rrrrr)rrr.�not_unqs    r$�test_unique_nanequalsz TestUnique.test_unique_nanequals�s����H�a��B�F�B�F�B�F�3�4�4���i��l�l���)�A��/�/�/���3���!�R�V�� 5� 5�6�6�6��7�B�H�a�������-H�$I�$I�J�J�J�J�Jr&N)r.r/r0r;rBrGrLr�r�r�rPrYr`rdr@rkr1r&r$rrns�������gL�gL�gL�RD�D�D�N�N�N�>�>�>�,�[���V�a��W�-�-�/�/�.�-�/�
"?�"?�"?�H1�1�1�/�/�/�*:�*:�*:�XK�K�K�K�Kr&r)�__doc__�numpyr�
numpy.testingrrrr�numpy.lib.arraysetopsrrr	r
rrr
rr�rrr1r&r$�<module>rps?��������?�?�?�?�?�?�?�?�?�?�?�?����������������������
�
�
�]	#�]	#�]	#�]	#�]	#�]	#�]	#�]	#�@BK�BK�BK�BK�BK�BK�BK�BK�BK�BKr&

Youez - 2016 - github.com/yon3zu
LinuXploit