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_shape_base.cpython-311.pyc
�

���c�h��D�ddlZddlZddlZddlZddlmZmZmZm	Z	m
Z
mZmZm
Z
mZmZmZmZmZmZddlmZmZmZmZmZejdkZd�ZGd�d��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,d&�Z-dS)'�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)�assert_�assert_equal�assert_array_equal�
assert_raises�assert_warns�c�F��tj����fd���}|S)z: hack in keepdims behavior into a function taking an axis c�N���|fd|i|��}|�d}tj||���S)N�axisr�r)�npr)�ar�kwargs�res�funcs    ��A/usr/lib/python3/dist-packages/numpy/lib/tests/test_shape_base.py�wrappedz_add_keepdims.<locals>.wrappeds>����d�1�*�*�4�*�6�*�*���<��D��~�c��-�-�-�-�)�	functools�wraps)rr!s` r �
_add_keepdimsr%s8����_�T���.�.�.�.���.�
�Nr"c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestTakeAlongAxisc
��ddlm}|ddd��}tjtjt��ft
tj��t
tj��t��ft
tj	��t
tj
��t��ftjtjtd���fg}|D]d\}}}tt|j����dgzD]8}||fd	|i|��}||fd	|i|��}	t!|t#||	|�
�����9�edS)z- Test it translates from arg<func> to <func> r��rand����)�kthNrr)�numpy.randomr*r�sort�argsort�dictr%�min�argmin�max�argmax�	partition�argpartition�list�range�ndimrr)
�selfr*r�funcsr�argfuncrr�a_func�ai_funcs
          r �test_argequivalentz$TestTakeAlongAxis.test_argequivalentsH��%�%�%�%�%�%��D��A�q�M�M���W�b�j�$�&�&�)�
�2�6�
"�
"�M�"�)�$<�$<�d�f�f�E�
�2�6�
"�
"�M�"�)�$<�$<�d�f�f�E�
�\�2�?�D�Q�K�K�K�8�	
��&+�	M�	M�!�D�'�6��U�1�6�]�]�+�+�t�f�4�
M�
M����a�5�5�d�5�f�5�5��!�'�!�9�9�$�9�&�9�9���V�_�Q��d�%K�%K�%K�L�L�L�L�
M�	M�	Mr"c��tjd��}tjdtj���}t||d���t	t
t|tjd��d���t	tt||�t��d���t	tt||�t��d���t	tjt||d���dS)z4 Test it errors when indices has too few dimensions )�
rD)rDr.��dtype�rrDN)r�ones�intprr�
ValueError�array�
IndexError�astype�bool�float�	AxisError�r=r�ais   r �test_invalidzTestTakeAlongAxis.test_invalid1s����G�H����
�W�W�B�G�
,�
,�
,��	��2�A�&�&�&�&�	�j�/�1�b�h�q�k�k��J�J�J�J��j�/�1�b�i�i��o�o�A�N�N�N�N��j�/�1�b�i�i��6F�6F�Q�O�O�O�O��b�l�O�Q���D�D�D�D�D�Dr"c���tjd��}tjdtj���}t||d���}t	|j|j��dS)zC Test everything is ok with empty results, even with inserted dims )r+r,r-)r+rr-rErGrN�rrHrIrr�shape�r=rrR�actuals    r �
test_emptyzTestTakeAlongAxis.test_emptyBsU��
�W�Y�
�
��
�W�Y�b�g�
.�
.�
.�� ��B�Q�/�/�/���V�\�2�8�,�,�,�,�,r"c��tjd��}tjdtj���}t||d���}t	|jd��dS)�D Test that non-indexing dimensions are broadcast in both directions �r+r,rG�rGr.r-rErGr�r+r.r-NrUrWs    r �test_broadcastz TestTakeAlongAxis.test_broadcastJsS��
�W�Y�
�
��
�W�Y�b�g�
.�
.�
.�� ��B�Q�/�/�/���V�\�9�-�-�-�-�-r"N)�__name__�
__module__�__qualname__rBrSrYr_�r"r r'r'sV������M�M�M�$E�E�E�"-�-�-�.�.�.�.�.r"r'c��eZdZd�Zd�ZdS)�TestPutAlongAxisc��tjgd�gd�g��}tt|j����dgzD]�}|���}t
tj��||���}t||d|���t
tj	��||���}t||����dS)N)rD��)�<�(�2ri����)rrKr:r;r<�copyr%r7rr5r)r=�a_baserr�i_max�i_mins      r �test_replace_maxz!TestPutAlongAxis.test_replace_maxSs�����<�<�<����6�7�7����v�{�+�+�,�,��v�5�	'�	'�D����
�
�A�-�M�"�)�,�,�Q�T�:�:�:�E��1�e�S�t�4�4�4�4�-�M�"�)�,�,�Q�T�:�:�:�E����&�&�&�&�	'�	'r"c��tjd��}tjdtj����d��dz}t||dd���t
t||d���d��d	S)
r[r\rDrEr]r,rhrGrN)rrH�arangerI�reshaperrrrQs   r r_zTestPutAlongAxis.test_broadcastcsv��
�W�Y�
�
��
�Y�r���
)�
)�
)�
1�
1�)�
<�
<�q�
@���q�"�b�q�)�)�)�)��_�Q���3�3�3�R�8�8�8�8�8r"N)r`rarbrpr_rcr"r rereRs2������'�'�'� 9�9�9�9�9r"rec��eZdZd�Zd�Zd�Zd�Zd�Zej	fd�Z
ej	fd�Zej	fd�Zd	�Z
d
�Zd�Zd�Zd
�ZdS)�TestApplyAlongAxisc���tjdd��}tttd|��t	|��tj|jd��z��dS)N�rhrD�drrG�rrHrr�lenrV�r=rs  r �test_simplezTestApplyAlongAxis.test_simplels`���G�H�c�"�"����S�!�Q�'�'��Q��������
�0C�0C�)C�	E�	E�	E�	E�	Er"c���tjdd��}tttd|��t	|��tj|jd��z��dS)N)rD�erxrrGryr{s  r �test_simple101z!TestApplyAlongAxis.test_simple101qs`���G�I�s�#�#����S�!�Q�'�'��Q��������
�0C�0C�)C�	E�	E�	E�	E�	Er"c��tjd���d��}tt	tjd|��gd�gd�gd�g��dS)N��r+r+r+r)r�rg�!)�$�'�*)�-�0�3)rrrrsrr�sumr{s  r �test_3dzTestApplyAlongAxis.test_3dvse���I�b�M�M�!�!�)�,�,���+�B�F�A�q�9�9�(�L�L�,�,�,����E�	G�	G�	G�	G�	Gr"c���d�}Gd�dtj��}tjddgddgg���|��}tjddgdd	gg���|��}t	|d|��}tt
||����t||��t	|d|��}tt
||����t||��dS)
Nc��|dzS)Nr.rc)�rows r �doublez9TestApplyAlongAxis.test_preserve_subclass.<locals>.double|s����7�Nr"c��eZdZdS)�<TestApplyAlongAxis.test_preserve_subclass.<locals>.MyNDArrayN�r`rarbrcr"r �	MyNDArrayr���������Dr"r�rrGr.r+r,�)r�ndarrayrK�viewrr�
isinstancer)r=r�r��m�expected�results      r �test_preserve_subclassz)TestApplyAlongAxis.test_preserve_subclass{s��	�	�	�	�	�	�	�	��
�	�	�	�
�H�q�!�f�q�!�f�%�&�&�+�+�I�6�6���8�a��V�a��V�,�-�-�2�2�9�=�=��!�&�!�Q�/�/���
�6�9�-�-�.�.�.��6�8�,�,�,�!�&�!�Q�/�/���
�6�9�-�-�.�.�.��6�8�,�,�,�,�,r"c���Gd�dtj��}d�}tjd���|��}t	t|d|��tjgd�����dS)Nc��eZdZdZdS)�9TestApplyAlongAxis.test_subclass.<locals>.MinimalSubclassrGN)r`rarb�datarcr"r �MinimalSubclassr��s�������D�D�Dr"r�c��|jS�N)r�)rKs r �minimal_functionz:TestApplyAlongAxis.test_subclass.<locals>.minimal_function�s
���:�r"�r�r+r)rGrGrG)rr��zerosr�rrrK)r=r�r�rs    r �
test_subclassz TestApplyAlongAxis.test_subclass�s���	�	�	�	�	�b�j�	�	�	�	�	�	�
�H�V���!�!�/�2�2����-�q�!�4�4�b�h�y�y�y�6I�6I�	
�	
�	
�	
�	
r"c�2�tjd���|��}ttjd|��}tt
||����t|tjgd����|����dS)Nr�r�r�r�r�)	rrHr�rr�rr�rrK)r=�clsrrs    r �test_scalar_arrayz$TestApplyAlongAxis.test_scalar_array�sz���G�F�O�O� � ��%�%���r�v�q�!�,�,���
�3��$�$�%�%�%��3������ 3� 3� 8� 8�� =� =�>�>�>�>�>r"c���d�}tjd���|��}t|d|��}t	t||����t
|tjgd����|����t|d|��}t	t||����t
|tjgd����|����dS)Nc�|�t|jd��tjtj|d�����S)z/ Sum x, returning a 0d array of the same class rGT)�keepdims)rr<r�squeezer���xs r �	sum_to_0dz3TestApplyAlongAxis.test_0d_array.<locals>.sum_to_0d�s3������#�#�#��:�b�f�Q��6�6�6�7�7�7r"r�rr�rG)r+r+r+r+r+r+)rrHr�rrr�rrK)r=r�r�rrs     r �
test_0d_arrayz TestApplyAlongAxis.test_0d_array�s���	8�	8�	8�
�G�F�O�O� � ��%�%���y�!�Q�/�/���
�3��$�$�%�%�%��3������ 3� 3� 8� 8�� =� =�>�>�>��y�!�Q�/�/���
�3��$�$�%�%�%��3���);�);�);� <� <� A� A�#� F� F�G�G�G�G�Gr"c�J������fd��tjd���d���t�d���}tj��fd�t�jd��D��d�������}tt|��t|����t||��t�d���}tj��fd	�t�jd��D��d�������}tt|��t|����t||��tjd
���d���t�d���}tj��fd�t�jd
��D��d�������}tt|��t|����t||��dS)Nc���t|jd��|ddd�|dd�dfz����S)�/produces an asymmetric non-square matrix from xrGN���)rr<r�)r�r�s �r �f1to2z5TestApplyAlongAxis.test_axis_insertion.<locals>.f1to2�sD�������#�#�#��d�d��d�G�a����4��j�(�.�.�s�3�3�3r"�r�rc�:��g|]}��dd�|f����Sr�rc��.0�i�a2dr�s  ��r �
<listcomp>z:TestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>�s:���
�
�
� !�E�E�#�a�a�a��c�(�O�O�
�
�
r"rGr�rc�:��g|]}��|dd�f����Sr�rcr�s  ��r r�z:TestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>�s:���
�
�
� !�E�E�#�a����c�(�O�O�
�
�
r"�Z)r�r-r+c	����g|]>�tj���fd�t�jd��D��d�����?S)c�<��g|]}��|dd��f����Sr�rc)r�r��a3dr��js  ���r r�zETestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>.<listcomp>�s>������&'���c�!�A�A�A�a�%�j�!�!���r"rr)r�stackr;rV)r�r�r�r�s @��r r�z:TestApplyAlongAxis.test_axis_insertion.<locals>.<listcomp>�s~����
�
�
��
�H�������+0���1��+>�+>�����
�
�
�
�
�
r"r.)
rrrrsrr�r;rVr�r�type)r=r�rXr�r�r�r�s `  @@@r �test_axis_insertionz&TestApplyAlongAxis.test_axis_insertion�sH������	4�	4�	4�	4�	4�
�i��n�n�$�$�V�,�,��"�%��C�0�0���8�
�
�
�
�
�%*�3�9�Q�<�%8�%8�
�
�
������D��I�I�	�	�T�&�\�\�4��>�>�2�2�2��V�X�&�&�&�"�%��C�0�0���8�
�
�
�
�
�%*�3�9�Q�<�%8�%8�
�
�
������4��9�9�	�	�T�&�\�\�4��>�>�2�2�2��V�X�&�&�&��i����&�&�y�1�1��!�%��C�0�0���8�
�
�
�
�
��3�9�Q�<�(�(�	
�
�
�
����
�D��I�I�	�	�T�&�\�\�4��>�>�2�2�2��V�X�&�&�&�&�&r"c��Gd�dtj��}|�|��|�|��|�|��dS)Nc��eZdZdS)�FTestApplyAlongAxis.test_subclass_preservation.<locals>.MinimalSubclassNr�rcr"r r�r��r�r"r�)rr�r�r�r�)r=r�s  r �test_subclass_preservationz-TestApplyAlongAxis.test_subclass_preservation�sp��	�	�	�	�	�b�j�	�	�	�����/�/�/����?�+�+�+�� � ��1�1�1�1�1r"c�Z�d�}tjd���d��}t|d|��}t	t|tjj����t|j	d��t|dd�dd�dfj||dd�df��j��t|dd�dd�dfj||dd�df��j��t|dd�dd�dfj||dd�df��j��dS)Nc��t|jd��|ddd�|dd�dfz}tj�|dzdk|��S)r�rGNr�r-r)rr<r�ma�masked_where)r�rs  r r�z8TestApplyAlongAxis.test_axis_insertion_ma.<locals>.f1to2�sT������#�#�#��D�D�b�D�'�A�a�b�b��g�J�&�C��5�%�%�c�!�e�Q�h��4�4�4r"r�r�rr+rGr.)rrrrsrrr�r��masked_arrayrr<r�mask)r=r�rrs    r �test_axis_insertion_maz)TestApplyAlongAxis.test_axis_insertion_ma�s%��	5�	5�	5�

�I�c�N�N�"�"�6�*�*���u�a��+�+���
�3��� 2�3�3�4�4�4��S�X�q�!�!�!��3�q�q�q����1�u�:�?�E�E�!�A�A�A�a�C�&�M�M�,>�?�?�?��3�q�q�q����1�u�:�?�E�E�!�A�A�A�a�C�&�M�M�,>�?�?�?��3�q�q�q����1�u�:�?�E�E�!�A�A�A�a�C�&�M�M�,>�?�?�?�?�?r"c	��d�}tj|dtjddgddgg����}t|tjddgddgg����dS)Nc�"�|d|dfS)NrGrrcr�s r �	sample_1dz7TestApplyAlongAxis.test_tuple_func1d.<locals>.sample_1d�s���Q�4��1��:�r"rGr.r+r,)rrrKr)r=r�rs   r �test_tuple_func1dz$TestApplyAlongAxis.test_tuple_func1d�si��	�	�	��!�)�Q���1�a�&�1�a�&�9I�0J�0J�K�K���3���1�a�&�1�a�&�)9� :� :�;�;�;�;�;r"c��d�}tjd��}tttj|d|��tttj|d|��d�}tjd��}tj|d|��}t|tjd����tttj|d|��dS)Nc�$�td��dS)NF)rr�s r �
never_callz1TestApplyAlongAxis.test_empty.<locals>.never_call�s���E�N�N�N�N�Nr")rrrrGc�F�tt|��dk��dS)NrrG)rrzr�s r �
empty_to_1z1TestApplyAlongAxis.test_empty.<locals>.empty_to_1�s���C��F�F�a�K� � � ��1r")rDrrD)r�emptyrrJrrrH)r=r�rr�rXs     r rYzTestApplyAlongAxis.test_empty�s���	�	�	�
�H�V�����j�"�"5�z�1�a�H�H�H��j�"�"5�z�1�a�H�H�H�	�	�	�
�H�W�����$�Z��A�6�6���V�R�W�R�[�[�)�)�)��j�"�"5�z�1�a�H�H�H�H�Hr"c��tjddhddhddhgddhd	d
hddhgg��}tjd
�d|��}tjhd�hd�hd�g��}t||��tj|j��D]8}tt
||��t
||�����9dS)NrG�r.�r+r�r,�,r-�7r��Bc��tj|�Sr�)�set�union)rs r �<lambda>z>TestApplyAlongAxis.test_with_iterable_object.<locals>.<lambda>s��s�y�!�}�r"r>rGr,r�r�>r.r-r�r�>r+r�r�r�)rrKrr�ndindexrVr�)r=rxrXr�r�s     r �test_with_iterable_objectz,TestApplyAlongAxis.test_with_iterable_objects����H���W�q�"�g��2�w�'���W�q�"�g��2�w�'�
�
�
���$�%<�%<�a��C�C���8�^�^�^�^�^�^�^�^�^�L�M�M���V�X�&�&�&���F�L�)�)�	=�	=�A���f�Q�i���$�x��{�*;�*;�<�<�<�<�	=�	=r"N)r`rarbr|rr�r�r�rr�r�r�r�r�r�r�rYr�rcr"r ruruks������E�E�E�
E�E�E�
G�G�G�
-�-�-�$
�
�
�%'�J�?�?�?�?�!#�
�H�H�H�H�')�j�#'�#'�#'�#'�J2�2�2�@�@�@�<�<�<�I�I�I�&
=�
=�
=�
=�
=r"ruc��eZdZd�ZdS)�TestApplyOverAxesc���tjd���ddd��}ttj|ddg��}t|tjdgdgdggg����dS)	N�r.r+r,rri�\�|)rrrrsrr�rrK)r=r�aoa_as   r r|zTestApplyOverAxes.test_simplesh���I�b�M�M�!�!�!�Q��*�*������A�q�6�2�2���5�"�(�b�T�B�4�#��,?�+@�"A�"A�B�B�B�B�Br"N)r`rarbr|rcr"r r�r�s(������C�C�C�C�Cr"r�c�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�TestExpandDimsc��d}tj|��}tdd��D]Z}t||��}t	|j|dk��t	tj|��j|k���[dS)N�r.r+r,r-���r,rG)rr�r;rrrVr�)r=�srr�bs     r �test_functionalityz!TestExpandDims.test_functionalitys~�����H�Q�K�K���"�a�L�L�	.�	.�D��A�t�$�$�A��A�G�D�M�Q�&�'�'�'��B�J�q�M�M�'�1�,�-�-�-�-�	.�	.r"c�6�tjd��}tj|d���jdksJ�tj|d���jdksJ�tj|d���jdksJ�tj|d	���jd
ksJ�dS)Nr��rrGr.r)rGrGrGr+r+r+)rr����)rGr+r+r+rGrG)rr+r-)rGr+r+rGr+rG)r���r�)rGrGr+rGr+r+)rr�rrVr{s  r �test_axis_tuplezTestExpandDims.test_axis_tuple!s����H�Y�����~�a�i�0�0�0�6�:L�L�L�L�L��~�a�k�2�2�2�8�<N�N�N�N�N��~�a�i�0�0�0�6�:L�L�L�L�L��~�a�k�2�2�2�8�<N�N�N�N�N�N�Nr"c�b�d}tj|��}ttjt|d��ttjt|d��tjd��}ttjt|d��ttjt|d��dS)Nr����r-r�)rr�)rr-)rr�rrPr)r=r�rs   r �test_axis_out_of_rangez%TestExpandDims.test_axis_out_of_range(s������H�Q�K�K���b�l�K��B�7�7�7��b�l�K��A�6�6�6��H�Y�����b�l�K��G�<�<�<��b�l�K��F�;�;�;�;�;r"c�h�tjd��}ttt|d���dS)Nr��rGrGr)rr�rrJrr{s  r �test_repeated_axisz!TestExpandDims.test_repeated_axis2s.���H�Y�����j�+�q�v�>�>�>�>�>�>r"c��tjd���d��}tj�||dzdk���}tj|d���}t
t|tjj����t|j
d��t|jj
d��dS)	NrD)r.r-r+r�r�rGr)r.rGr-)rrrrsr�rKrrr��MaskedArrayrrVr�)r=r�expandeds   r �test_subclasseszTestExpandDims.test_subclasses6s����I�b�M�M�!�!�&�)�)���E�K�K���!��q��K�)�)���>�!�!�,�,�,���
�8�R�U�%6�7�7�8�8�8��X�^�Y�/�/�/��X�]�(�)�4�4�4�4�4r"N)r`rarbr�r�rrrrcr"r r�r�sb������.�.�.�O�O�O�<�<�<�?�?�?�5�5�5�5�5r"r�c��eZdZd�Zd�Zd�Zd�Zd�Zej	�
ed���d���Zd	�Z
d
�Zd�ZdS)
�TestArraySplitc�f�tjd��}ttt|d��dS)NrDr)rrrrrJrr{s  r �test_integer_0_splitz#TestArraySplit.test_integer_0_splitAs)���I�b�M�M���j�+�q�!�4�4�4�4�4r"c
�R
�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||��dS)NrDrGr.r-r+r,�r���	r�)rrrr�compare_resultsrK�r=rr�desireds    r �test_integer_splitz!TestArraySplit.test_integer_splitEs����I�b�M�M���!�Q�����9�R�=�=�/����W�%�%�%��!�Q�����9�Q�<�<���1�b�!1�!1�2����W�%�%�%��!�Q�����9�Q�<�<���1�a���"�)�A�r�2B�2B�C����W�%�%�%��!�Q�����9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��#�#�%����W�%�%�%��!�Q�����9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��?�?�B�I�a��$4�$4�6����W�%�%�%��!�Q�����9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��?�?�B�I�a��O�O�R�Y�q�"�5E�5E�G����W�%�%�%��!�Q�����9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��#�#�%��	��W�%�%�%��!�Q�����9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��?�?�B�I�a��$4�$4�6��	��W�%�%�%��!�Q�����9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��?�?�B�I�a��O�O�R�Y�q�"�5E�5E�G��	��W�%�%�%��!�R� � ���9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��#�#�%��	��W�%�%�%��!�R� � ���9�Q�<�<���1�a���"�)�A�q�/�/��9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��#�#�R�X�b�\�\�3��	��W�%�%�%�%�%r"c�6�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��dS)NrDr+rr�rrDr�rG�	rrKrrrr�rrrFr��r=rr�tgts    r �test_integer_split_2D_rowsz)TestArraySplit.test_integer_split_2D_rows�s<���H�b�i��m�m�R�Y�r�]�]�3�4�4���!�Q�Q�'�'�'���x���2����(�(�"�(�B�I�b�M�M�?�*C�*C��8�G�$�$�&����S�!�!�!������B��
� 2�2�3�3�3��!�a��V�!�,�,�,���x�� � �"�(�B�I�b�M�M�?�";�";��x���2����(�(�*����S�!�!�!������B��
� 2�2�3�3�3�3�3r"c	� �tjtjd��tjd��g��}t|dd���}tjtjd��tjd��g��tjtjdd��tjdd��g��tjtjdd��tjdd��g��g}t	||��dS)NrDr+r�rr,r)rrKrrrrrs    r �test_integer_split_2D_colsz)TestArraySplit.test_integer_split_2D_cols�s����H�b�i��m�m�R�Y�r�]�]�3�4�4���!�Q�R�(�(�(���8�R�Y�q�\�\�2�9�Q�<�<�8�9�9��8�R�Y�q�!�_�_�b�i��1�o�o�>�?�?��8�R�Y�q�"�-�-�r�y��B�/?�/?�@�A�A�C��	��W�%�%�%�%�%r"c���tjtjd��tjd��g��}t|d��}tjtjd��g��tjtjd��g��tjd��g}t||��t
|jj|djju��dS)z2 This will fail if we change default axis
        rDr+rr�Nrrs    r �test_integer_split_2D_defaultz,TestArraySplit.test_integer_split_2D_default�s���
�H�b�i��m�m�R�Y�r�]�]�3�4�4���!�Q�����x���2����(�(�"�(�B�I�b�M�M�?�*C�*C��8�G�$�$�&����S�!�!�!������B��
� 2�2�3�3�3�3�3r"zNeeds 64bit platform)�reasonc��tjdgd��}t|d��}tjdgd��}|gdz}tt	|����D](}t||j||j���)dS)Nr)rr.r,)i@r.)r�broadcast_torr;rzrrV)r=rr�chunkrr�s      r �,test_integer_split_2D_rows_greater_max_int32z;TestArraySplit.test_integer_split_2D_rows_greater_max_int32�s����O�Q�C��.�.���!�Q��������\�2�2���g��k���s�3�x�x���	5�	5�A���Q���s�1�v�|�4�4�4�4�	5�	5r"c��tjd��}gd�}t||d���}tjdd��tjdd��tjdd��tjdd��g}t||��dS)	NrD)rGr-rr�rrrGr-r)rrrrr�r=r�indicesrrs     r �test_index_split_simplez&TestArraySplit.test_index_split_simple�s}���I�b�M�M���)�)���!�W�2�.�.�.���9�Q��?�?�B�I�a��O�O�R�Y�q�!�_�_��9�Q��#�#�%����W�%�%�%�%�%r"c��tjd��}gd�}t||d���}tjg��tjdd��tjdd��tjdd��g}t	||��dS)NrD)rr-rr�rrr-r�rrrrrKrr%s     r �test_index_split_low_boundz)TestArraySplit.test_index_split_low_bound�s{���I�b�M�M���)�)���!�W�2�.�.�.���8�B�<�<���1�a���"�)�A�q�/�/��9�Q��#�#�%����W�%�%�%�%�%r"c�h�tjd��}gd�}t||d���}tjg��tjdd��tjdd��tjdd��tjg��tjg��g}t	||��dS)NrD)rr-rrD�r�rrr-rr)r%s     r �test_index_split_high_boundz*TestArraySplit.test_index_split_high_bound�s����I�b�M�M��#�#�#���!�W�2�.�.�.���8�B�<�<���1�a���"�)�A�q�/�/��9�Q��#�#�R�X�b�\�\�2�8�B�<�<�A����W�%�%�%�%�%r"N)r`rarbrrrrr�pytest�mark�skipif�IS_64BITr#r'r*r-rcr"r r
r
@s�������5�5�5�;&�;&�;&�z
4�
4�
4�&�&�&�4�4�4��[���H��-C��D�D�5�5�E�D�5�&�&�&�&�&�&�&�&�&�&�&r"r
c��eZdZd�Zd�ZdS)�	TestSplitc��tjd��}t|d��}tjd��tjdd��g}t||��dS)NrDr.r-)rrrrrrs    r �test_equal_splitzTestSplit.test_equal_split�sP���I�b�M�M���A�q�k�k���9�Q�<�<���1�b�!1�!1�2����W�%�%�%�%�%r"c�f�tjd��}ttt|d��dS)NrDr+)rrrrrJrr{s  r �test_unequal_splitzTestSplit.test_unequal_split�s)���I�b�M�M���j�%��A�.�.�.�.�.r"N)r`rarbr5r7rcr"r r3r3�s2������
&�&�&�/�/�/�/�/r"r3c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestColumnStackc�<�tttd��dS�NrG)r�	TypeErrorr�r=s r �test_non_iterablez!TestColumnStack.test_non_iterable�s���i��q�1�1�1�1�1r"c���tjd��}tjd��}tjddgddgddgg��}tj||f��}t||��dS)N�rGr.r+)r.r+r,rGr.r+r,�rrKrr�r=rr�r�rXs     r �test_1D_arrayszTestColumnStack.test_1D_arrays�su���H�Y�����H�Y�����8�a��V���V���V�%�&�&����!�Q��(�(���V�X�&�&�&�&�&r"c���tjdgdgdgg��}tjdgdgdgg��}tjddgddgddgg��}tj||f��}t||��dS)NrGr.r+r,rArBs     r �test_2D_arrayszTestColumnStack.test_2D_arrays�s����H�q�c�A�3���_�%�%���H�q�c�A�3���_�%�%���8�a��V���V���V�%�&�&����!�Q��(�(���V�X�&�&�&�&�&r"c��tt��5td�td��D����ddd��dS#1swxYwYdS)Nc3�>K�|]}tjd��V��dS�r+N�rrr�r��_s  r �	<genexpr>z1TestColumnStack.test_generator.<locals>.<genexpr>�s*����9�9�1�"�)�A�,�,�9�9�9�9�9�9r"r.)r�
FutureWarningrr;r=s r �test_generatorzTestColumnStack.test_generator�s���
�-�
(�
(�	;�	;��9�9��a���9�9�9�:�:�:�	;�	;�	;�	;�	;�	;�	;�	;�	;�	;�	;�	;����	;�	;�	;�	;�	;�	;��'A	�	A
�A
N)r`rarbr>rCrErNrcr"r r9r9�sP������2�2�2�'�'�'�'�'�'�;�;�;�;�;r"r9c�2�eZdZd�Zd�Zd�Zd�Zd�Zd�ZdS)�
TestDstackc�<�tttd��dSr;)rr<r
r=s r r>zTestDstack.test_non_iterable�s���i���+�+�+�+�+r"c���tjd��}tjd��}t||g��}tjddggg��}t||��dS�NrGr.�rrKr
r�r=rr�rrs     r �
test_0D_arrayzTestDstack.test_0D_array�sX���H�Q�K�K���H�Q�K�K���a��V�n�n���(�a��V�H�:�&�&���3��(�(�(�(�(r"c���tjdg��}tjdg��}t||g��}tjddggg��}t||��dSrTrUrVs     r �
test_1D_arrayzTestDstack.test_1D_array�s\���H�a�S�M�M���H�a�S�M�M���a��V�n�n���(�a��V�H�:�&�&���3��(�(�(�(�(r"c���tjdgdgg��}tjdgdgg��}t||g��}tjddggddggg��}t||��dSrTrUrVs     r �
test_2D_arrayzTestDstack.test_2D_arraysu���H�q�c�A�3�Z� � ���H�q�c�A�3�Z� � ���a��V�n�n���(�a��V�H��1�x�j�1�2�2���3��(�(�(�(�(r"c���tjddg��}tjddg��}t||g��}tjddgddggg��}t||��dSrTrUrVs     r �test_2D_array2zTestDstack.test_2D_array2sl���H�a��V�����H�a��V�����a��V�n�n���(�a��V�a��V�,�-�.�.���3��(�(�(�(�(r"c��tt��5td�td��D����ddd��dS#1swxYwYdS)Nc3�>K�|]}tjd��V��dSrHrIrJs  r rLz,TestDstack.test_generator.<locals>.<genexpr>s*����3�3�Q�B�I�a�L�L�3�3�3�3�3�3r"r.)rrMr
r;r=s r rNzTestDstack.test_generators���
�-�
(�
(�	5�	5��3�3�%��(�(�3�3�3�4�4�4�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5����	5�	5�	5�	5�	5�	5rON)	r`rarbr>rWrYr[r]rNrcr"r rQrQ�sn������,�,�,�)�)�)�)�)�)�)�)�)�)�)�)�5�5�5�5�5r"rQc�*�eZdZdZd�Zd�Zd�Zd�ZdS)�
TestHsplit�&Only testing for integer splits.

    c�>�tttdd��dSr;)rrJrr=s r r>zTestHsplit.test_non_iterable����j�&�!�Q�/�/�/�/�/r"c��tjd��}	t|d��td��dS#t$rYdSwxYw)NrGr.r)rrKrrrJr{s  r rWzTestHsplit.test_0D_array!sR���H�Q�K�K��	��1�a�L�L�L��A�J�J�J�J�J���	�	�	��D�D�	���s�7�
A�Ac���tjgd���}t|d��}tjddg��tjddg��g}t||��dS�N�rGr.r+r,r.rGr+r,�rrKrrrs    r rYzTestHsplit.test_1D_array)s^���H�\�\�\�"�"���Q��l�l���8�Q��F�#�#�R�X�q�!�f�%5�%5�6����W�%�%�%�%�%r"c���tjgd�gd�g��}t|d��}tjddgddgg��tjddgddgg��g}t||��dSrgrirs    r r[zTestHsplit.test_2D_array/s����H�l�l�l��,�,� �
!�
!���Q��l�l���8�a��V�a��V�,�-�-�r�x�!�Q��!�Q��8H�/I�/I�J����W�%�%�%�%�%r"N�r`rarb�__doc__r>rWrYr[rcr"r rarasZ��������0�0�0����&�&�&�&�&�&�&�&r"rac�*�eZdZdZd�Zd�Zd�Zd�ZdS)�
TestVsplitrbc�>�tttdd��dSr;)rrJr	r=s r r>zTestVsplit.test_non_iterable;rdr"c�f�tjd��}ttt|d��dSrT)rrKrrJr	r{s  r rWzTestVsplit.test_0D_array>�)���H�Q�K�K���j�&�!�Q�/�/�/�/�/r"c��tjgd���}	t|d��td��dS#t$rYdSwxYw�Nrhr.r)rrKr	rrJr{s  r rYzTestVsplit.test_1D_arrayBsX���H�\�\�\�"�"��	��1�a�L�L�L��A�J�J�J�J�J���	�	�	��D�D�	���s�9�
A�Ac���tjgd�gd�g��}t|d��}tjgd�g��tjgd�g��g}t||��dS�Nrhr.)rrKr	rrs    r r[zTestVsplit.test_2D_arrayJsr���H�l�l�l��,�,� �
!�
!���Q��l�l���8�\�\�\�N�+�+�R�X�|�|�|�n�-E�-E�F����W�%�%�%�%�%r"Nrkrcr"r rnrn7sZ��������0�0�0�0�0�0����&�&�&�&�&r"rnc�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�
TestDsplitc�>�tttdd��dSr;)rrJrr=s r r>zTestDsplit.test_non_iterableTrdr"c�f�tjd��}ttt|d��dSrT�rrKrrJrr{s  r rWzTestDsplit.test_0D_arrayWrqr"c�j�tjgd���}ttt|d��dSrurzr{s  r rYzTestDsplit.test_1D_array[s/���H�\�\�\�"�"���j�&�!�Q�/�/�/�/�/r"c��tjgd�gd�g��}	t|d��td��dS#t$rYdSwxYwrs)rrKrrrJr{s  r r[zTestDsplit.test_2D_array_sh���H�l�l�l��,�,� �
!�
!��	��1�a�L�L�L��A�J�J�J�J�J���	�	�	��D�D�	���s�=�
A�
Ac��tjgd�gd�ggd�gd�gg��}t|d��}tjddgddggddgddggg��tjddgddggddgddggg��g}t||��dSrg)rrKrrrs    r �
test_3D_arrayzTestDsplit.test_3D_arrayhs����H�|�|�|��<�<�!��<�<��<�<�!�"�
#�
#���Q��l�l���8�q�!�f�q�!�f�-��A���A��/?�@�A�A��8�q�!�f�q�!�f�-��A���A��/?�@�A�A�C����W�%�%�%�%�%r"N)r`rarbr>rWrYr[r~rcr"r rwrwRs_������0�0�0�0�0�0�0�0�0����&�&�&�&�&r"rwc��eZdZd�ZdS)�TestSqueezec�r�ddlm}|ddddd��}|ddddd��}|dddd��}ttj|��tj|d����ttj|��tj|d����ttj|��tj|d����d	ggg}tj|��}t
|d	��t
|jd��t
t|��tj	��dS)
Nrr)rhrDrG)rhrDrD)rhrDrhrwg�?)
r0r*rrr�rsrr<r�r�)r=r*rr��crs      r �
test_basiczTestSqueeze.test_basicts��%�%�%�%�%�%��D��R��Q��"�"���D��Q��A�r�"�"���D��A�r�2�����2�:�a�=�=�"�*�Q��*E�*E�F�F�F��2�:�a�=�=�"�*�Q��*E�*E�F�F�F��2�:�a�=�=�"�*�Q��*A�*A�B�B�B��e�W�I���j��m�m���S�#�����S�X�q�!�!�!��T�#�Y�Y��
�+�+�+�+�+r"N�r`rarbr�rcr"r r�r�ss#������,�,�,�,�,r"r�c���eZdZd�Zd�Zej�dej	ej
g��d���Zd�Zej�dgd���d���Z
d	S)
�TestKronc���tjd��}tjddgddgg��}tjddgddgg��}ttj||��|��tjddgddgg��}tjd��}ttj||��|��tjdg��}tjddgddgg��}tjddgddgg��}ttj||��|��tjddgddgg��}tjdg��}ttj||��|��tjdggdggg��}tjddgddgg��}tjddgddggddgddggg��}ttj||��|��tjddgddgg��}tjdggdggg��}tjddgddggddgddggg��}ttj||��|��dS)	NrGr.r+r,r�rr,r)rrKrr)r=rr��ks    r r�zTestKron.test_basic�sR���H�Q�K�K���H�q�!�f�q�!�f�%�&�&���H�q�!�f�q�!�f�%�&�&���2�7�1�a�=�=�!�,�,�,��H�q�!�f�q�!�f�%�&�&���H�Q�K�K���2�7�1�a�=�=�!�,�,�,�
�H�a�S�M�M���H�q�!�f�q�!�f�%�&�&���H�q�!�f�q�"�g�&�'�'���2�7�1�a�=�=�!�,�,�,��H�q�!�f�q�!�f�%�&�&���H�a�S�M�M���2�7�1�a�=�=�!�,�,�,�
�H��s�e�q�c�U�^�$�$���H�q�!�f�q�!�f�%�&�&���H��1�v��1�v�&�!�Q��!�Q��(8�9�:�:���2�7�1�a�=�=�!�,�,�,��H�q�!�f�q�!�f�%�&�&���H��s�e�q�c�U�^�$�$���H��1�v��1�v�&�!�Q��!�Q��(8�9�:�:���2�7�1�a�=�=�!�,�,�,�,�,r"c��Gd�dtj��}tjddg��}||j|j|j��}t
tt||����tj��t
tt||����|��t
tt||����|��t
tt||����|��dS)Nc��eZdZdZdS)�*TestKron.test_return_type.<locals>.myarrayg�?N)r`rarb�__array_priority__rcr"r �myarrayr��s������!$���r"r�r.)	rr�rHrVrFr�rr�r)r=r�rr�s    r �test_return_typezTestKron.test_return_type�s���	%�	%�	%�	%�	%�b�j�	%�	%�	%�
�G�Q��F�O�O��
�W�Q�W�a�g�q�v�
.�
.���T�$�q�!�*�*�%�%�r�z�2�2�2��T�$�r�2�,�,�'�'��1�1�1��T�$�q�"�+�+�&�&��0�0�0��T�$�r�1�+�+�&�&��0�0�0�0�0r"�array_classc��|tjddg����}|tjddg����}|tjddg����}ttj||��|��dS)Nr+r)rrHrr)r=r�rr�r�s     r �test_kron_smokezTestKron.test_kron_smoke�su��
�K����A����(�(���K����A����(�(���K����A����(�(���2�7�1�a�=�=�!�,�,�,�,�,r"c�h�tj�ddgddggddgddgg���}tj�tjgd���tjtjd��t
������}t
|tj||����dS)	NrGr.r+r,rr)rGr,r,�rE)rr�rK�diag�identityrNrr)r=r�r�s   r �test_kron_mazTestKron.test_kron_ma�s����E�K�K�!�Q��!�Q��(��A���A��/?�K�@�@���E�K�K���
�
�
�.�.��h�r�{�1�~�~�T�:�:�:�:�
�
<�
<��	�1�b�g�a��m�m�,�,�,�,�,r"zshape_a,shape_b))rr)r@)r,r-r�)�r.r.)r.r.r.))rGrr)�r.rr.r�))r.rrr.r�c��tj|��}tj|��}dtdt|��t|��z
��z|z}dtdt|��t|��z
��z|z}tj||��}tj||��}tj|j|��s
Jd���dS)N)rGrzUnexpected shape from kron)rrHr6rz�multiplyr�array_equalrV)	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����!�C��3�w�<�<��G���+D�$E�$E�E��O��!�C��3�w�<�<��G���+D�$E�$E�E��O����%7�9K�L�L���G�A�q�M�M���~����)�)�	G�	G�*F�	G�	G�	G�	G�	Gr"N)r`rarbr�r�r.r/�parametrizer�asarray�matr�r�r�rcr"r r�r��s�������-�-�-�:	1�	1�	1��[�����
�B�F�+���-�-���-�-�-�-��[����
�
�
���	G�	G���	G�	G�	Gr"r�c�&�eZdZd�Zd�Zd�Zd�ZdS)�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��dS)Nr�rGr.r+r,)rrGr.rrGr.r��rGr.)rGr.rGr.)r+r,r+r,�r.rG)rrKrr
�r=rr�s   r r�zTestTile.test_basic�s3���H�Y�Y�Y������V�a��V����T�!�Q�Z�Z�!3�!3�!3�4�4�4��T�!�V�_�_�'9�'9�'9�;M�;M�;M�&N�O�O�O��T�!�V�_�_�'9�'9�'9�&:�;�;�;��T�!�Q�Z�Z�,�,�,����!=�>�>�>��T�!�V�_�_��1�v��1�v��1�v��1�v�&F�G�G�G��T�!�V�_�_�|�|�|�\�\�\�'3�|�|�\�\�\�'C�	D�	D�	D�	D�	Dr"c��tjd��}t|d��}|dz
}t|tjd����dS)Nr-rGr.)rrrr
rr�s   r �(test_tile_one_repetition_on_array_gh4679z1TestTile.test_tile_one_repetition_on_array_gh4679�sC���I�a�L�L����A�J�J��	�Q����Q��	�!���%�%�%�%�%r"c���tjggg��}tjggg��}t|d��j}t|d��j}t	|d��t	|d��dS)Nr.r^)r.r)r+r.r)rrKr
rVr)r=rr�r�rxs     r rYzTestTile.test_empty�sp���H�r�d�V�����H�b�"�X������A�J�J�����I���$���Q������Q�	�"�"�"�"�"r"c���ddlm}gd�}gd�}|D]_}|dd|���}|D]L}tj||j��}t||��}t
||��}	t||	���M�`dS)Nr)�randint))r.r�r�r�)r.r+r.)r+r.))r+)r.r+)r+r,r+)r+r.r+)r,r+r.r,r�rD)�size)r0r�rrHrFr
rr)
r=r��repsrVr�r��rr�large�klarges
          r �test_kroncomparezTestTile.test_kroncompare�s���(�(�(�(�(�(�@�@�@��J�J�J���	,�	,�A����2�A�&�&�&�A��
,�
,���G�A�q�w�'�'���Q��
�
���a������U�F�+�+�+�+�	
,�	,�	,r"N)r`rarbr�r�rYr�rcr"r r�r��sS������	D�	D�	D�&�&�&�#�#�#�,�,�,�,�,r"r�c��eZdZd�ZdS)�TestMayShareMemoryc
��tjd��}tjd��}ttj||����ttj||ddd�����ttj||ddd�����ttj||dd�ddd�f����ttj|ddd�|����ttj|ddd�|����ttj|dd�ddd�f|����ttj|dd�ddd�f|����dS)N)rkri)rgrir�r�r.rG)rrHr�may_share_memory)r=rx�d2s   r r�zTestMayShareMemory.test_basic�se���G�H����
�W�[�
!�
!����#�A�q�)�)�*�*�*���#�A�q���2��w�/�/�0�0�0���#�A�q���1��v�.�.�/�/�/���#�A�q����T�T�r�T��{�3�3�4�4�4��B�'��$�$�B�$���4�4�4�5�5�5��B�'��#�#�A�#���3�3�3�4�4�4��B�'��!�"�"�d�d��d�(��R�8�8�8�9�9�9���#�B�q�r�r�4�4�R�4�x�L�"�5�5�6�6�6�6�6r"Nr�rcr"r r�r��s#������7�7�7�7�7r"r�c��t|��t|��krtd���t||��D]\}}t||���dS)zCompare lists of arrays.z Iterables have different lengthsN)rzrJ�zipr)rrr��ys    r rr
sb��
�3�x�x�3�w�<�<����;�<�<�<��C��!�!�!�!���1��1�a� � � � �!�!r").�numpyrr#�sysr.�numpy.lib.shape_baserrrrrrr	r
rrr
rrr�
numpy.testingrrrrr�maxsizer1r%r'rerur�r�r
r3r9rQrarnrwr�r�r�r�rrcr"r �<module>r�s�����������
�
�
�
�
�
�
�
���������������������������������
��������������
�;�������1.�1.�1.�1.�1.�1.�1.�1.�h9�9�9�9�9�9�9�9�2c=�c=�c=�c=�c=�c=�c=�c=�LC�C�C�C�C�C�C�C�%5�%5�%5�%5�%5�%5�%5�%5�PC&�C&�C&�C&�C&�C&�C&�C&�L
/�
/�
/�
/�
/�
/�
/�
/� ;�;�;�;�;�;�;�;�:"5�"5�"5�"5�"5�"5�"5�"5�N&�&�&�&�&�&�&�&�:&�&�&�&�&�&�&�&�6&�&�&�&�&�&�&�&�B,�,�,�,�,�,�,�,�&LG�LG�LG�LG�LG�LG�LG�LG�^%,�%,�%,�%,�%,�%,�%,�%,�P7�7�7�7�7�7�7�7� !�!�!�!�!r"

Youez - 2016 - github.com/yon3zu
LinuXploit