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/lib/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

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

	�g�n���ddlZddlZddlZddlZddlmZmZmZmZm	Z	m
Z
mZmZm
Z
mZmZmZmZmZddlmZddlmZmZmZmZmZej6dkDZd�ZGd�d�ZGd	�d
�ZGd�d�Z Gd
�d�Z!Gd�d�Z"Gd�d�Z#Gd�d�Z$Gd�d�Z%Gd�d�Z&Gd�d�Z'Gd�d�Z(Gd�d�Z)Gd�d �Z*Gd!�d"�Z+Gd#�d$�Z,Gd%�d&�Z-d'�Z.y)(�N)�apply_along_axis�apply_over_axes�array_split�split�hsplit�dsplit�vsplit�dstack�column_stack�kron�tile�expand_dims�take_along_axis�put_along_axis)�	AxisError)�assert_�assert_equal�assert_array_equal�
assert_raises�assert_warns�c�B��tj���fd��}|S)z: hack in keepdims behavior into a function taking an axis c�P���|fd|i|��}|�d}tj||��S)N�axisr�r)�npr)�ar�kwargs�res�funcs    ��AC:\Python312\Lib\site-packages\numpy/lib/tests/test_shape_base.py�wrappedz_add_keepdims.<locals>.wrappeds2����1�*�4�*�6�*���<��D��~�~�c��-�-�)�	functools�wraps)r r"s` r!�
_add_keepdimsr&s%����_�_�T��.��.�
�Nr#c�$�eZdZd�Zd�Zd�Zd�Zy)�TestTakeAlongAxisc
� �ddlm}|ddd�}tjtjt�ft
tj�t
tj�t�ft
tj�t
tj�t�fg}|D][\}}}tt|j��dgzD]0}||fd|i|��}||fd|i|��}	t|t||	|����2�]y)	z- Test it translates from arg<func> to <func> r��rand���Nrr)�numpy.randomr+r�sort�argsort�dictr&�min�argmin�max�argmax�list�range�ndimrr)
�selfr+r�funcsr �argfuncrr�a_func�ai_funcs
          r!�test_argequivalentz$TestTakeAlongAxis.test_argequivalent s���%���A�q�M���W�W�b�j�j�$�&�)�
�2�6�6�
"�M�"�)�)�$<�d�f�E�
�2�6�6�
"�M�"�)�)�$<�d�f�E�
��&+�	M�!�D�'�6��U�1�6�6�]�+�t�f�4�
M���a�5�d�5�f�5��!�!�9�$�9�&�9���V�_�Q��d�%K�L�
M�	Mr#c��tjd�}tjdtj��}t||d��t	t
t|tjd�d��t	tt||jt�d��t	tt||jt�d��t	tt||d��t	t
t||d��y)z4 Test it errors when indices has too few dimensions )�
rA)rA���dtype�rrAN)r�ones�intprr�
ValueError�array�
IndexError�astype�bool�floatr�r:r�ais   r!�test_invalidzTestTakeAlongAxis.test_invalid2s����G�G�H���
�W�W�W�B�G�G�
,��	��2�A�&�	�j�/�1�b�h�h�q�k��J��j�/�1�b�i�i��o�A�N��j�/�1�b�i�i��6F�Q�O��i��!�R�b�A��j�/�1�b�t�Dr#c���tjd�}tjdtj��}t||d��}t	|j
|j
�y)zC Test everything is ok with empty results, even with inserted dims )r,r-r.)r,rr.rCrErN�rrFrGrr�shape�r:rrO�actuals    r!�
test_emptyzTestTakeAlongAxis.test_emptyEsD��
�W�W�Y�
��
�W�W�Y�b�g�g�
.�� ��B�Q�/���V�\�\�2�8�8�,r#c��tjd�}tjdtj��}t||d��}t	|j
d�y)�D Test that non-indexing dimensions are broadcast in both directions �r,r-rE�rErBr.rCrEr�r,rBr.NrRrTs    r!�test_broadcastz TestTakeAlongAxis.test_broadcastMs@��
�W�W�Y�
��
�W�W�Y�b�g�g�
.�� ��B�Q�/���V�\�\�9�-r#N)�__name__�
__module__�__qualname__r?rPrVr\�r#r!r(r(s��M�$E�&-�.r#r(c��eZdZd�Zd�Zd�Zy)�TestPutAlongAxisc�b�tjgd�gd�g�}tt|j��dgzD]o}|j�}t
tj�||��}t||d|��t
tj�||��}t||��qy)N�rA����<�(�2ri����)rrIr7r8r9�copyr&r6rr4r)r:�a_baserr�i_max�i_mins      r!�test_replace_maxz!TestPutAlongAxis.test_replace_maxVs������<��6�7����v�{�{�+�,��v�5�	'�D����
�A�-�M�"�)�)�,�Q�T�:�E��1�e�S�t�4�-�M�"�)�)�,�Q�T�:�E����&�	'r#c���tjd�}tjdtj��j	d�dz}t||dd��t
t||d��d�y	)
rXrYrArCrZr-rfrErN)rrF�arangerG�reshaperrrrNs   r!r\zTestPutAlongAxis.test_broadcastfsU��
�W�W�Y�
��
�Y�Y�r����
)�
1�
1�)�
<�q�
@���q�"�b�q�)��_�Q���3�R�8r#c���tjgd�gd�g�}tjdgdgg�}tjdgdgg�}|j�}t|||d��tj|gd�gd�gk(�sJ�|j�}tt�5}t|||d	��d	d	d	�d
tj�vsJ�y	#1swY�#xYw)z Test invalid inputs rdrgrrErBr�rBrBrB�rErErENzsingle dimension)	rrIrkr�allrrH�str�	exception)r:rl�indices�valuesr�excs      r!rPzTestPutAlongAxis.test_invalidms������<��6�7���(�(�Q�C�!��:�&�����A�3���*�%��
�K�K�M���q�'�6��2��v�v�a�I�y�1�1�2�2�2�
�K�K�M��
�:�
&�	:�#��1�g�v�D�9�	:�!�S����%7�7�7�7�	:�	:�s�,C�C'N)r]r^r_ror\rPr`r#r!rbrbUs��'� 9�8r#rbc��eZdZd�Zd�Zd�Zd�Zd�Zejfd�Z
ejfd�Zejfd�Zd	�Z
d
�Zd�Zd�Zd
�Zy)�TestApplyAlongAxisc��tjdd�}tttd|�t	|�tj|j
d�z�y)N�rfrA�drrE�rrFrr�lenrS�r:rs  r!�test_simplezTestApplyAlongAxis.test_simple�sC���G�G�H�c�"����S�!�Q�'��Q���������
�0C�)C�	Er#c��tjdd�}tttd|�t	|�tj|j
d�z�y)N)rA�er�rrEr�r�s  r!�test_simple101z!TestApplyAlongAxis.test_simple101�sC���G�G�I�s�#����S�!�Q�'��Q���������
�0C�)C�	Er#c��tjd�jd�}tt	tj
d|�gd�gd�gd�g�y)N��r,r,r,r)r�re�!)�$�'�*)�-�0�3)rrqrrrr�sumr�s  r!�test_3dzTestApplyAlongAxis.test_3d�s=���I�I�b�M�!�!�)�,���+�B�F�F�A�q�9�(�,��E�	Gr#c��d�}Gd�dtj�}tjddgddgg�j|�}tjddgdd	gg�j|�}t	|d|�}tt
||��t||�t	|d|�}tt
||��t||�y)
Nc��|dzS)NrBr`)�rows r!�doublez9TestApplyAlongAxis.test_preserve_subclass.<locals>.double�s����7�Nr#c��eZdZy)�<TestApplyAlongAxis.test_preserve_subclass.<locals>.MyNDArrayN�r]r^r_r`r#r!�	MyNDArrayr�����r#r�rrErBr,r-�)r�ndarrayrI�viewrr�
isinstancer)r:r�r��m�expected�results      r!�test_preserve_subclassz)TestApplyAlongAxis.test_preserve_subclass�s���	�	��
�
�	�
�H�H�q�!�f�q�!�f�%�&�+�+�I�6���8�8�a��V�a��V�,�-�2�2�9�=��!�&�!�Q�/���
�6�9�-�.��6�8�,�!�&�!�Q�/���
�6�9�-�.��6�8�,r#c���Gd�dtj�}d�}tjd�j|�}t	t|d|�tjgd���y)Nc��eZdZdZy)�9TestApplyAlongAxis.test_subclass.<locals>.MinimalSubclassrEN)r]r^r_�datar`r#r!�MinimalSubclassr��s���Dr#r�c��|jS�N)r�)rIs r!�minimal_functionz:TestApplyAlongAxis.test_subclass.<locals>.minimal_function�s���:�:�r#�r�r,rru)rr��zerosr�rrrI)r:r�r�rs    r!�
test_subclassz TestApplyAlongAxis.test_subclass�sR��	�b�j�j�	�	�
�H�H�V��!�!�/�2����-�q�!�4�b�h�h�y�6I�	
r#c��tjd�j|�}ttjd|�}tt
||��t|tjgd��j|��y)Nr�r�r�r�r�)	rrFr�rr�rr�rrI)r:�clsrrs    r!�test_scalar_arrayz$TestApplyAlongAxis.test_scalar_array�sY���G�G�F�O� � ��%���r�v�v�q�!�,���
�3��$�%��3����� 3� 8� 8�� =�>r#c��d�}tjd�j|�}t|d|�}t	t||��t
|tjgd��j|��t|d|�}t	t||��t
|tjgd��j|��y)Nc��t|jd�tjtj|d���S)z/ Sum x, returning a 0d array of the same class rET)�keepdims)rr9r�squeezer���xs r!�	sum_to_0dz3TestApplyAlongAxis.test_0d_array.<locals>.sum_to_0d�s*�������#��:�:�b�f�f�Q��6�7�7r#r�rr�rE)r,r,r,r,r,r,)rrFr�rrr�rrI)r:r�r�rrs     r!�
test_0d_arrayz TestApplyAlongAxis.test_0d_array�s���	8�
�G�G�F�O� � ��%���y�!�Q�/���
�3��$�%��3����� 3� 8� 8�� =�>��y�!�Q�/���
�3��$�%��3����);� <� A� A�#� F�Gr#c����fd�}tjd�jd�}t|d|�}tjt|jd�D�cgc]}||dd�|f���c}d��j��}tt|�t|��t||�t|d|�}tjt|jd�D�cgc]}|||dd�f���c}d��j��}tt|�t|��t||�tjd�jd	�}t|d|�}tjt|jd
�D��cgc]K}tjt|jd�D�cgc]}|||dd�|f���c}d����Mc}}d��j��}tt|�t|��t||�ycc}wcc}wcc}wcc}}w)Nc�r��t|jd�|ddd�|dd�dfzj��S)�/produces an asymmetric non-square matrix from xrEN���)rr9r�)r�r�s �r!�f1to2z5TestApplyAlongAxis.test_axis_insertion.<locals>.f1to2�s9��������#��d��d�G�a���4��j�(�.�.�s�3�3r#�r�rrEr�r�Z)r�r.r,rB)
rrqrrr�stackr8rSr�r�type)	r:r�r��a2drU�ir��a3d�js	 `       r!�test_axis_insertionz&TestApplyAlongAxis.test_axis_insertion�s����	4�
�i�i��n�$�$�V�,��"�%��C�0���8�8�%*�3�9�9�Q�<�%8�
� !�E�#�a��c�(�O�
����D��I�	�	�T�&�\�4��>�2��V�X�&�"�%��C�0���8�8�%*�3�9�9�Q�<�%8�
� !�E�#�a��c�(�O�
����4��9�	�	�T�&�\�4��>�2��V�X�&��i�i���&�&�y�1��!�%��C�0���8�8��3�9�9�Q�<�(�	
��
�H�H�+0����1��+>��&'��c�!�A�a�%�j�!���
�
�
��
�D��I�	�	�T�&�\�4��>�2��V�X�&��3
��
����
s$�!H6�/H;�".I
�I�'I
�I
c��Gd�dtj�}|j|�|j|�|j	|�y)Nc��eZdZy)�FTestApplyAlongAxis.test_subclass_preservation.<locals>.MinimalSubclassNr�r`r#r!r�r��r�r#r�)rr�r�r�r�)r:r�s  r!�test_subclass_preservationz-TestApplyAlongAxis.test_subclass_preservation�s<��	�b�j�j�	�����/����?�+�� � ��1r#c�<�d�}tjd�jd�}t|d|�}t	t|tjj��t|jd�t|dd�dd�dfj||dd�df�j�t|dd�dd�dfj||dd�df�j�t|dd�dd�dfj||dd�df�j�y)Nc��t|jd�|ddd�|dd�dfz}tjj	|dzdk(|�S)r�rENr�r.r)rr9r�ma�masked_where)r�rs  r!r�z8TestApplyAlongAxis.test_axis_insertion_ma.<locals>.f1to2�sK�������#��D�b�D�'�A�a�b��g�J�&�C��5�5�%�%�c�!�e�Q�h��4�4r#r�r�rr,rErB)rrqrrrrr�r��masked_arrayrr9r�mask)r:r�rrs    r!�test_axis_insertion_maz)TestApplyAlongAxis.test_axis_insertion_ma�s���	5�

�I�I�c�N�"�"�6�*���u�a��+���
�3���� 2� 2�3�4��S�X�X�q�!��3�q��1�u�:�?�?�E�!�A�a�C�&�M�,>�,>�?��3�q��1�u�:�?�?�E�!�A�a�C�&�M�,>�,>�?��3�q��1�u�:�?�?�E�!�A�a�C�&�M�,>�,>�?r#c	��d�}tj|dtjddgddgg��}t|tjddgddgg��y)Nc��|d|dfS)NrErr`r�s r!�	sample_1dz7TestApplyAlongAxis.test_tuple_func1d.<locals>.sample_1d�s���Q�4��1��:�r#rErBr,r-)rrrIr)r:r�rs   r!�test_tuple_func1dz$TestApplyAlongAxis.test_tuple_func1d�sQ��	��!�!�)�Q����1�a�&�1�a�&�9I�0J�K���3����1�a�&�1�a�&�)9� :�;r#c��d�}tjd�}tttj|d|�tttj|d|�d�}tjd�}tj|d|�}t|tjd��tttj|d|�y)Nc��td�y)NF)rr�s r!�
never_callz1TestApplyAlongAxis.test_empty.<locals>.never_calls	���E�Nr#)rrrrEc�2�tt|�dk(�y)NrrE)rr�r�s r!�
empty_to_1z1TestApplyAlongAxis.test_empty.<locals>.empty_to_1
s���C��F�a�K� �r#)rArrA)r�emptyrrHrrrF)r:r�rr�rUs     r!rVzTestApplyAlongAxis.test_emptys���	�
�H�H�V����j�"�"5�"5�z�1�a�H��j�"�"5�"5�z�1�a�H�	�
�H�H�W����$�$�Z��A�6���V�R�W�W�R�[�)��j�"�"5�"5�z�1�a�Hr#c�j�tjddhddhddhgddhd	d
hddhgg�}tjd
�d|�}tjhd�hd�hd�g�}t||�tj|j
�D]&}tt
||�t
||���(y)NrE�rB�r,r�r-�,r.�7r��Bc�&�tj|�Sr�)�set�union)rs r!�<lambda>z>TestApplyAlongAxis.test_with_iterable_object.<locals>.<lambda>s��s�y�y�!�}�r#r>rEr-r�r�>rBr.r�r�>r,r�r�r�)rrIrr�ndindexrSr�)r:r�rUr�r�s     r!�test_with_iterable_objectz,TestApplyAlongAxis.test_with_iterable_objects����H�H���W�q�"�g��2�w�'���W�q�"�g��2�w�'�
�
���$�$�%<�a��C���8�8�^�^�^�L�M���V�X�&����F�L�L�)�	=�A���f�Q�i��$�x��{�*;�<�	=r#N)r]r^r_r�r�r�r�r�rr�r�r�r�r�r�r�rVr�r`r#r!r}r}�se��E�
E�
G�
-�$
�%'�J�J�?�!#�
�
�H�')�j�j�#'�J2�@�<�I�&
=r#r}c��eZdZd�Zy)�TestApplyOverAxesc���tjd�jddd�}ttj|ddg�}t|tjdgdgdggg��y)	N�rBr,r-rrh�\�|)rrqrrrr�rrI)r:r�aoa_as   r!r�zTestApplyOverAxes.test_simple'sW���I�I�b�M�!�!�!�Q��*�������A�q�6�2���5�"�(�(�b�T�B�4�#��,?�+@�"A�Br#N)r]r^r_r�r`r#r!r�r�&s��Cr#r�c�*�eZdZd�Zd�Zd�Zd�Zd�Zy)�TestExpandDimsc��d}tj|�}tdd�D]T}t||�}t	|j
|dk(�t	tj|�j
|k(��Vy)N�rBr,r-r.���r-rE)rr�r8rrrSr�)r:�srr�bs     r!�test_functionalityz!TestExpandDims.test_functionality.se�����H�H�Q�K���"�a�L�	.�D��A�t�$�A��A�G�G�D�M�Q�&�'��B�J�J�q�M�'�'�1�,�-�	.r#c�^�tjd�}tj|d��jdk(sJ�tj|d��jdk(sJ�tj|d��jdk(sJ�tj|d	��jd
k(sJ�y)Nr��rrErBr)rErErEr,r,r,)rr����)rEr,r,r,rErE)rr,r.)rEr,r,rEr,rE)r���r�)rErEr,rEr,r,)rr�rrSr�s  r!�test_axis_tuplezTestExpandDims.test_axis_tuple6s����H�H�Y����~�~�a�i�0�6�6�:L�L�L�L��~�~�a�k�2�8�8�<N�N�N�N��~�~�a�i�0�6�6�:L�L�L�L��~�~�a�k�2�8�8�<N�N�N�Nr#c��d}tj|�}ttt|d�ttt|d�tjd�}ttt|d�ttt|d�y)Nr����r.r�)rr)rr.)rr�rrr)r:r�rs   r!�test_axis_out_of_rangez%TestExpandDims.test_axis_out_of_range=sZ�����H�H�Q�K���i��a��4��i��a��3��H�H�Y����i��a��9��i��a��8r#c�\�tjd�}ttt|d��y)Nr��rErEr)rr�rrHrr�s  r!�test_repeated_axisz!TestExpandDims.test_repeated_axisGs���H�H�Y����j�+�q�v�>r#c��tjd�jd�}tjj	||dzdk(��}tj
|d��}t
t|tjj��t|jd�t|jjd�y)	NrA)rBr.r,r�r�rEr)rBrEr.)rrqrrr�rIrrr��MaskedArrayrrSr�)r:r�expandeds   r!�test_subclasseszTestExpandDims.test_subclassesKs����I�I�b�M�!�!�&�)���E�E�K�K���!��q��K�)���>�>�!�!�,���
�8�R�U�U�%6�%6�7�8��X�^�^�Y�/��X�]�]�(�(�)�4r#N)r]r^r_r�r�rrr	r`r#r!r�r�-s��.�O�9�?�5r#r�c��eZdZd�Zd�Zd�Zd�Zd�Zejjed��d��Zd	�Z
d
�Zd�Zy)
�TestArraySplitc�Z�tjd�}ttt|d�y)NrAr)rrqrrHrr�s  r!�test_integer_0_splitz#TestArraySplit.test_integer_0_splitVs���I�I�b�M���j�+�q�!�4r#c
�&
�tjd�}t|d�}tjd�g}t||�t|d�}tjd�tjdd�g}t||�t|d�}tjd�tjdd�tjdd�g}t||�t|d�}tjd�tjdd�tjdd	�tjd	d�g}t||�t|d�}tjd�tjdd�tjdd�tjdd	�tjd	d�g}t||�t|d�}tjd�tjdd�tjdd�tjdd	�tjd	d
�tjd
d�g}t||�t|d�}tjd�tjdd�tjdd�tjdd�tjdd	�tjd	d
�tjd
d�g}t||�t|d	�}tjd�tjdd�tjdd�tjdd�tjdd�tjdd	�tjd	d
�tjd
d�g}t||�t|d
�}tjd�tjdd�tjdd�tjdd�tjdd�tjdd�tjdd	�tjd	d
�tjd
d�g	}t||�t|d�}tjd�tjdd�tjdd�tjdd�tjdd�tjdd�tjdd�tjdd	�tjd	d
�tjd
d�g
}t||�t|d�}tjd�tjdd�tjdd�tjdd�tjdd�tjdd�tjdd�tjdd	�tjd	d
�tjd
d�tjg�g}t||�y)NrArErBr.r,r-�r���	r�)rrqr�compare_resultsrI�r:rr�desireds    r!�test_integer_splitz!TestArraySplit.test_integer_splitZsw���I�I�b�M���!�Q����9�9�R�=�/����W�%��!�Q����9�9�Q�<����1�b�!1�2����W�%��!�Q����9�9�Q�<����1�a��"�)�)�A�r�2B�C����W�%��!�Q����9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��#�%����W�%��!�Q����9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��?�B�I�I�a��$4�6����W�%��!�Q����9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�"�5E�G����W�%��!�Q����9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��#�%��	��W�%��!�Q����9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��?�B�I�I�a��$4�6��	��W�%��!�Q����9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�"�5E�G��	��W�%��!�R� ���9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��#�%��	��W�%��!�R� ���9�9�Q�<����1�a��"�)�)�A�q�/��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��#�R�X�X�b�\�3��	��W�%r#c�b�tjtjd�tjd�g�}t|dd��}tjtjd�g�tjtjd�g�tjd�g}t||�t
|jj|djju�t|ddgd��}tjd�tjtjd�g�tjtjd�g�g}t||�t
|jj|djju�y)NrAr,rr�rrAr�rE�	rrIrqrr�rrrDr��r:rr�tgts    r!�test_integer_split_2D_rowsz)TestArraySplit.test_integer_split_2D_rows�s���H�H�b�i�i��m�R�Y�Y�r�]�3�4���!�Q�Q�'���x�x����2���(�"�(�(�B�I�I�b�M�?�*C��8�8�G�$�&����S�!��������B��
�
� 2� 2�2�3��!�a��V�!�,���x�x�� �"�(�(�B�I�I�b�M�?�";��x�x����2���(�*����S�!��������B��
�
� 2� 2�2�3r#c	�&�tjtjd�tjd�g�}t|dd��}tjtjd�tjd�g�tjtjdd�tjdd�g�tjtjdd�tjdd�g�g}t	||�y)NrAr,r�rr-r)rrIrqrrrs    r!�test_integer_split_2D_colsz)TestArraySplit.test_integer_split_2D_cols�s����H�H�b�i�i��m�R�Y�Y�r�]�3�4���!�Q�R�(���8�8�R�Y�Y�q�\�2�9�9�Q�<�8�9��8�8�R�Y�Y�q�!�_�b�i�i��1�o�>�?��8�8�R�Y�Y�q�"�-�r�y�y��B�/?�@�A�C��	��W�%r#c���tjtjd�tjd�g�}t|d�}tjtjd�g�tjtjd�g�tjd�g}t||�t
|jj|djju�y)z2 This will fail if we change default axis
        rAr,rr�Nrrs    r!�test_integer_split_2D_defaultz,TestArraySplit.test_integer_split_2D_default�s���
�H�H�b�i�i��m�R�Y�Y�r�]�3�4���!�Q����x�x����2���(�"�(�(�B�I�I�b�M�?�*C��8�8�G�$�&����S�!��������B��
�
� 2� 2�2�3r#zNeeds 64bit platform)�reasonc��tjdgd�}t|d�}tjdgd�}|gdz}tt	|��D](}t||j||j��*y)Nr)rrBr-)i@rB)r�broadcast_torr8r�rrS)r:rr�chunkrr�s      r!�,test_integer_split_2D_rows_greater_max_int32z;TestArraySplit.test_integer_split_2D_rows_greater_max_int32�sn���O�O�Q�C��.���!�Q��������\�2���g��k���s�3�x��	5�A���Q����s�1�v�|�|�4�	5r#c��tjd�}gd�}t||d��}tjdd�tjdd�tjdd�tjdd�g}t||�y)	NrA)rEr.rr�rrrEr.r)rrqrr�r:rryrrs     r!�test_index_split_simplez&TestArraySplit.test_index_split_simple�sf���I�I�b�M�����!�W�2�.���9�9�Q��?�B�I�I�a��O�R�Y�Y�q�!�_��9�9�Q��#�%����W�%r#c��tjd�}gd�}t||d��}tjg�tjdd�tjdd�tjdd�g}t	||�y)NrA)rr.rr�rrr.r�rrqrrIrr&s     r!�test_index_split_low_boundz)TestArraySplit.test_index_split_low_bound�sd���I�I�b�M�����!�W�2�.���8�8�B�<����1�a��"�)�)�A�q�/��9�9�Q��#�%����W�%r#c�d�tjd�}gd�}t||d��}tjg�tjdd�tjdd�tjdd�tjg�tjg�g}t	||�y)NrA)rr.rrA�r�rrr.rr)r&s     r!�test_index_split_high_boundz*TestArraySplit.test_index_split_high_bound�sy���I�I�b�M��#���!�W�2�.���8�8�B�<����1�a��"�)�)�A�q�/��9�9�Q��#�R�X�X�b�\�2�8�8�B�<�A����W�%r#N)r]r^r_r
rrrr�pytest�mark�skipif�IS_64BITr$r'r*r-r`r#r!rrUsW��5�;&�z
4�&�4��[�[���H��-C��D�5�E�5�&�&�&r#rc��eZdZd�Zd�Zy)�	TestSplitc��tjd�}t|d�}tjd�tjdd�g}t||�y)NrArBr.)rrqrrrs    r!�test_equal_splitzTestSplit.test_equal_split�sA���I�I�b�M���A�q�k���9�9�Q�<����1�b�!1�2����W�%r#c�Z�tjd�}ttt|d�y)NrAr,)rrqrrHrr�s  r!�test_unequal_splitzTestSplit.test_unequal_split�s���I�I�b�M���j�%��A�.r#N)r]r^r_r5r7r`r#r!r3r3�s��
&�/r#r3c�$�eZdZd�Zd�Zd�Zd�Zy)�TestColumnStackc�.�tttd�y�NrE)r�	TypeErrorr�r:s r!�test_non_iterablez!TestColumnStack.test_non_iterable�s���i��q�1r#c���tjd�}tjd�}tjddgddgddgg�}tj||f�}t||�y)N�rErBr,)rBr,r-rErBr,r-�rrIrr�r:rr�r�rUs     r!�test_1D_arrayszTestColumnStack.test_1D_arrays�sc���H�H�Y����H�H�Y����8�8�a��V���V���V�%�&�����!�Q��(���V�X�&r#c���tjdgdgdgg�}tjdgdgdgg�}tjddgddgddgg�}tj||f�}t||�y)NrErBr,r-rArBs     r!�test_2D_arrayszTestColumnStack.test_2D_arrays�s{���H�H�q�c�A�3���_�%���H�H�q�c�A�3���_�%���8�8�a��V���V���V�%�&�����!�Q��(���V�X�&r#c��tjtd��5td�t	d�D��ddd�y#1swYyxYw)N�arrays to stack must be��matchc3�FK�|]}tjd����y�w�r,N�rrq��.0�_s  r!�	<genexpr>z1TestColumnStack.test_generator.<locals>.<genexpr>s����8�!����1��8���!rB)r.�raisesr<rr8r=s r!�test_generatorzTestColumnStack.test_generators8��
�]�]�9�,E�
F�	9��8�u�Q�x�8�8�	9�	9�	9���A�A
N)r]r^r_r>rCrErSr`r#r!r9r9�s��2�'�'�9r#r9c�0�eZdZd�Zd�Zd�Zd�Zd�Zd�Zy)�
TestDstackc�.�tttd�yr;)rr<r
r=s r!r>zTestDstack.test_non_iterable	s���i���+r#c��tjd�}tjd�}t||g�}tjddggg�}t||�y�NrErB�rrIr
r�r:rr�rrs     r!�
test_0D_arrayzTestDstack.test_0D_arraysI���H�H�Q�K���H�H�Q�K���a��V�n���(�(�a��V�H�:�&���3��(r#c��tjdg�}tjdg�}t||g�}tjddggg�}t||�yrYrZr[s     r!�
test_1D_arrayzTestDstack.test_1D_arraysM���H�H�a�S�M���H�H�a�S�M���a��V�n���(�(�a��V�H�:�&���3��(r#c���tjdgdgg�}tjdgdgg�}t||g�}tjddggddggg�}t||�yrYrZr[s     r!�
test_2D_arrayzTestDstack.test_2D_arraysd���H�H�q�c�A�3�Z� ���H�H�q�c�A�3�Z� ���a��V�n���(�(�a��V�H��1�x�j�1�2���3��(r#c���tjddg�}tjddg�}t||g�}tjddgddggg�}t||�yrYrZr[s     r!�test_2D_array2zTestDstack.test_2D_array2!s[���H�H�a��V����H�H�a��V����a��V�n���(�(�a��V�a��V�,�-�.���3��(r#c��tjtd��5td�t	d�D��ddd�y#1swYyxYw)NrGrHc3�FK�|]}tjd����y�wrKrLrMs  r!rPz,TestDstack.test_generator.<locals>.<genexpr>*s����2�A�2�9�9�Q�<�2�rQrB)r.rRr<r
r8r=s r!rSzTestDstack.test_generator(s8��
�]�]�9�,E�
F�	3��2��q��2�2�	3�	3�	3�rTN)	r]r^r_r>r\r^r`rbrSr`r#r!rVrVs ��,�)�)�)�)�3r#rVc�(�eZdZdZd�Zd�Zd�Zd�Zy)�
TestHsplit�&Only testing for integer splits.

    c�0�tttdd�yr;)rrHrr=s r!r>zTestHsplit.test_non_iterable3����j�&�!�Q�/r#c�|�tjd�}	t|d�td�y#t$rYywxYw)NrErBr)rrIrrrHr�s  r!r\zTestHsplit.test_0D_array6s6���H�H�Q�K��	��1�a�L��A�J���	��	�s�/�	;�;c��tjgd��}t|d�}tjddg�tjddg�g}t||�y�N�rErBr,r-rBrEr,r-�rrIrrrs    r!r^zTestHsplit.test_1D_array>sI���H�H�\�"���Q��l���8�8�Q��F�#�R�X�X�q�!�f�%5�6����W�%r#c���tjgd�gd�g�}t|d�}tjddgddgg�tjddgddgg�g}t||�yrlrnrs    r!r`zTestHsplit.test_2D_arrayDse���H�H�l�� �
!���Q��l���8�8�a��V�a��V�,�-�r�x�x�!�Q��!�Q��8H�/I�J����W�%r#N�r]r^r_�__doc__r>r\r^r`r`r#r!rfrf/s���0��&�&r#rfc�(�eZdZdZd�Zd�Zd�Zd�Zy)�
TestVsplitrgc�0�tttdd�yr;)rrHr	r=s r!r>zTestVsplit.test_non_iterablePrir#c�Z�tjd�}ttt|d�yrY)rrIrrHr	r�s  r!r\zTestVsplit.test_0D_arrayS����H�H�Q�K���j�&�!�Q�/r#c��tjgd��}	t|d�td�y#t$rYywxYw�NrmrBr)rrIr	rrHr�s  r!r^zTestVsplit.test_1D_arrayWs7���H�H�\�"��	��1�a�L��A�J���	��	�s�1�	=�=c���tjgd�gd�g�}t|d�}tjgd�g�tjgd�g�g}t||�y�NrmrB)rrIr	rrs    r!r`zTestVsplit.test_2D_array_sO���H�H�l�� �
!���Q��l���8�8�\�N�+�R�X�X�|�n�-E�F����W�%r#Nrpr`r#r!rsrsLs���0�0��&r#rsc�*�eZdZd�Zd�Zd�Zd�Zd�Zy)�
TestDsplitc�0�tttdd�yr;)rrHrr=s r!r>zTestDsplit.test_non_iterableirir#c�Z�tjd�}ttt|d�yrY�rrIrrHrr�s  r!r\zTestDsplit.test_0D_arraylrvr#c�^�tjgd��}ttt|d�yrzrr�s  r!r^zTestDsplit.test_1D_arrayps���H�H�\�"���j�&�!�Q�/r#c��tjgd�gd�g�}	t|d�td�y#t$rYywxYwrx)rrIrrrHr�s  r!r`zTestDsplit.test_2D_arraytsA���H�H�l�� �
!��	��1�a�L��A�J���	��	�s�5�	A�Ac��tjgd�gd�ggd�gd�gg�}t|d�}tjddgddggddgddggg�tjddgddggddgddggg�g}t||�yrl)rrIrrrs    r!�
test_3D_arrayzTestDsplit.test_3D_array}s����H�H�|��!���!�"�
#���Q��l���8�8�q�!�f�q�!�f�-��A���A��/?�@�A��8�8�q�!�f�q�!�f�-��A���A��/?�@�A�C����W�%r#N)r]r^r_r>r\r^r`r�r`r#r!r|r|gs��0�0�0��&r#r|c��eZdZd�Zy)�TestSqueezec�F�ddlm}|ddddd�}|ddddd�}|dddd�}ttj|�tj
|d��ttj|�tj
|d��ttj|�tj
|d��d	ggg}tj|�}t
|d	�t
|jd�t
t|�tj�y)
Nrr*rfrArE)rfrArA)rfrArfrg�?)
r/r+rrr�rrrr9r�r�)r:r+rr��crs      r!�
test_basiczTestSqueeze.test_basic�s���%���R��Q��"����Q��A�r�"����A�r�2����2�:�:�a�=�"�*�*�Q��*E�F��2�:�:�a�=�"�*�*�Q��*E�F��2�:�:�a�=�"�*�*�Q��*A�B��e�W�I���j�j��m���S�#���S�X�X�q�!��T�#�Y��
�
�+r#N�r]r^r_r�r`r#r!r�r��s��,r#r�c���eZdZd�Zd�Zejjdejejg�d��Zd�Zejjdgd��d��Z
y	)
�TestKronc���tjd�}tjddgddgg�}tjddgddgg�}ttj||�|�tjddgddgg�}tjd�}ttj||�|�tjdg�}tjddgddgg�}tjddgddgg�}ttj||�|�tjddgddgg�}tjdg�}ttj||�|�tjdggdggg�}tjddgddgg�}tjddgddggddgddggg�}ttj||�|�tjddgddgg�}tjdggdggg�}tjddgddggddgddggg�}ttj||�|�y)	NrErBr,r-r�rr,r)rrIrr)r:rr��ks    r!r�zTestKron.test_basic�s���H�H�Q�K���H�H�q�!�f�q�!�f�%�&���H�H�q�!�f�q�!�f�%�&���2�7�7�1�a�=�!�,��H�H�q�!�f�q�!�f�%�&���H�H�Q�K���2�7�7�1�a�=�!�,�
�H�H�a�S�M���H�H�q�!�f�q�!�f�%�&���H�H�q�!�f�q�"�g�&�'���2�7�7�1�a�=�!�,��H�H�q�!�f�q�!�f�%�&���H�H�a�S�M���2�7�7�1�a�=�!�,�
�H�H��s�e�q�c�U�^�$���H�H�q�!�f�q�!�f�%�&���H�H��1�v��1�v�&�!�Q��!�Q��(8�9�:���2�7�7�1�a�=�!�,��H�H�q�!�f�q�!�f�%�&���H�H��s�e�q�c�U�^�$���H�H��1�v��1�v�&�!�Q��!�Q��(8�9�:���2�7�7�1�a�=�!�,r#c���Gd�dtj�}tjddg�}||j|j|j
�}t
tt||��tj�t
tt||��|�t
tt||��|�t
tt||��|�y)Nc��eZdZdZy)�*TestKron.test_return_type.<locals>.myarrayg�?N)r]r^r_�__array_priority__r`r#r!�myarrayr��s��!$�r#r�rB)	rr�rFrSrDr�rr�r)r:r�rr�s    r!�test_return_typezTestKron.test_return_type�s���	%�b�j�j�	%�
�G�G�Q��F�O��
�Q�W�W�a�g�g�q�v�v�
.���T�$�q�!�*�%�r�z�z�2��T�$�r�2�,�'��1��T�$�q�"�+�&��0��T�$�r�1�+�&��0r#�array_classc���|tjddg��}|tjddg��}|tjddg��}ttj||�|�y)Nr,r)rrFrr)r:r�rr�r�s     r!�test_kron_smokezTestKron.test_kron_smoke�sZ��
�����A���(�������A���(�������A���(���2�7�7�1�a�=�!�,r#c�d�tjjddgddggddgddgg��}tjjtjgd��tjtjd�t
����}t
|tj||��y)	NrErBr,r-rr)rEr-r-�rC)rr�rI�diag�identityrLrr)r:r�r�s   r!�test_kron_mazTestKron.test_kron_ma�s����E�E�K�K�!�Q��!�Q��(��A���A��/?�K�@���E�E�K�K����
�.��h�h�r�{�{�1�~�T�:�:�
�
<��	�1�b�g�g�a��m�,r#zshape_a,shape_b))rr)r@)r-r.r�)�rBrBrt))rErr)�rBrrBr�))rBrrrBr�c��tj|�}tj|�}dtdt|�t|�z
�z|z}dtdt|�t|�z
�z|z}tj||�}tj
||�}tj|j|�sJd��y)N)rErzUnexpected shape from kron)rrFr5r��multiplyr�array_equalrS)	r:�shape_a�shape_brr��normalised_shape_a�normalised_shape_b�expected_shaper�s	         r!�test_kron_shapezTestKron.test_kron_shape�s���
�G�G�G����G�G�G���!�C��3�w�<��G��+D�$E�E��O��!�C��3�w�<��G��+D�$E�E��O�����%7�9K�L���G�G�A�q�M���~�~�����)�	G�*F�	G�)r#N)r]r^r_r�r�r.r/�parametrizer�asarray�asmatrixr�r�r�r`r#r!r�r��ss��-�:	1��[�[�����
�
�B�K�K�0��-��-�-��[�[����
��	G��	Gr#r�c�$�eZdZd�Zd�Zd�Zd�Zy)�TestTilec��tjgd��}ddgddgg}tt|d�gd��tt|d�gd�gd�g�tt|d�gd�g�tt|d�gd	�gd
�g�tt|d�ddgddgddgddgg�tt|d�gd	�gd
�gd	�gd
�g�y)Nr�rErBr,r-)rrErBrrErBr��rErB)rErBrErB)r,r-r,r-�rBrE)rrIrr
�r:rr�s   r!r�zTestTile.test_basic�s����H�H�Y�����V�a��V����T�!�Q�Z�!3�4��T�!�V�_�'9�;M�&N�O��T�!�V�_�'9�&:�;��T�!�Q�Z�,��!=�>��T�!�V�_��1�v��1�v��1�v��1�v�&F�G��T�!�V�_�|�\�'3�\�'C�	Dr#c��tjd�}t|d�}|dz
}t|tjd��y)Nr.rErB)rrqr
rr�s   r!�(test_tile_one_repetition_on_array_gh4679z1TestTile.test_tile_one_repetition_on_array_gh4679�s5���I�I�a�L����A�J��	�Q����Q��	�	�!��%r#c���tjggg�}tjggg�}t|d�j}t|d�j}t	|d�t	|d�y)NrBr[)rBr)r,rBr)rrIr
rSr)r:rr�r�r�s     r!rVzTestTile.test_empty�s]���H�H�r�d�V����H�H�b�"�X�����A�J������I��$�$���Q����Q�	�"r#c���ddlm}gd�}gd�}|D]X}|dd|��}|D]F}tj||j�}t||�}t
||�}	t||	��H�Zy)Nr)�randint))rBr�r�r�)rBr,rB)r,rB))r,)rBr,)r,r-r,)r,rBr,)r-r,rBr-r�rA)�size)r/r�rrFrDr
rr)
r:r��repsrSr�r��rr�large�klarges
          r!�test_kroncomparezTestTile.test_kroncompareso��(�@��J���	,�A���2�A�&�A��
,���G�G�A�q�w�w�'���Q��
���a�����U�F�+�	
,�	,r#N)r]r^r_r�r�rVr�r`r#r!r�r��s��	D�&�#�,r#r�c��eZdZd�Zy)�TestMayShareMemoryc
��tjd�}tjd�}ttj||��ttj||ddd���ttj||ddd���ttj||dd�ddd�f��ttj|ddd�|��ttj|ddd�|��ttj|dd�ddd�f|��ttj|dd�ddd�f|��y)N)rjrh)rerhr�r�rBrE)rrFr�may_share_memory)r:r��d2s   r!r�zTestMayShareMemory.test_basics���G�G�H���
�W�W�[�
!����#�#�A�q�)�*���#�#�A�q��2��w�/�0���#�#�A�q��1��v�.�/���#�#�A�q���T�r�T��{�3�4��B�'�'��$�B�$���4�4�5��B�'�'��#�A�#���3�3�4��B�'�'��!�"�d��d�(��R�8�8�9���#�#�B�q�r�4�R�4�x�L�"�5�6r#Nr�r`r#r!r�r�s��7r#r�c��t|�t|�k7rtd��t||�D]\}}t||��y)zCompare lists of arrays.z Iterables have different lengthsN)r�rH�zipr)rrr��ys    r!rr"sB��
�3�x�3�w�<���;�<�<��C��!�!���1��1�a� �!r#)/�numpyrr$�sysr.rrrrrrr	r
rrr
rrr�numpy.exceptionsr�
numpy.testingrrrrr�maxsizer1r&r(rbr}r�r�rr3r9rVrfrsr|r�r�r�r�rr`r#r!�<module>r�s����
�
�����
'���
�;�;�����3.�3.�l'8�'8�Vc=�c=�LC�C�%5�%5�PC&�C&�L
/�
/� 9�9�:"3�"3�N&�&�:&�&�6&�&�B,�,�&LG�LG�^%,�%,�P7�7� !r#

Youez - 2016 - github.com/yon3zu
LinuXploit