HEX
Server: Apache
System: Linux pdx1-shared-a1-38 6.6.104-grsec-jammy+ #3 SMP Tue Sep 16 00:28:11 UTC 2025 x86_64
User: mmickelson (3396398)
PHP: 8.1.31
Disabled: NONE
Upload Files
File: //lib/python3/dist-packages/numpy/lib/tests/__pycache__/test_twodim_base.cpython-310.pyc
o

6��a�G�@sLdZddlmZmZmZmZmZmZddlm	Z	m
Z
mZmZm
Z
mZmZmZmZmZmZmZmZmZmZmZmZddlZddlmZdd�ZGdd	�d	�ZGd
d�d�ZGdd
�d
�Z Gdd�d�Z!Gdd�d�Z"Gdd�d�Z#dd�Z$dd�Z%dd�Z&dd�Z'dd�Z(dd�Z)Gd d!�d!�Z*Gd"d#�d#�Z+Gd$d%�d%�Z,Gd&d'�d'�Z-dS)(z"Test functions for matrix module

�)�assert_equal�assert_array_equal�assert_array_max_ulp�assert_array_almost_equal�
assert_raises�assert_)�arange�add�fliplr�flipud�zeros�ones�eye�array�diag�histogram2d�tri�mask_indices�triu_indices�triu_indices_from�tril_indices�tril_indices_from�vanderN)�requires_array_functioncCst|�}t�||�}|S�N)rr	�outer)�n�data�r�B/usr/lib/python3/dist-packages/numpy/lib/tests/test_twodim_base.py�get_matsr c@sLeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)�TestEyecCsxttd�tgd�gd�gd�gd�g��ttddd�tgd�gd�gd�gd�gd��ttd�d	ktdtd��dS)
N���rrr�rr$rr�rrr$r�rrrr$�f��dtype�r$)rrr�bool��selfrrr�
test_basics&�����
�zTestEye.test_basiccCs`ttddd�tgd�gd�gd�gd�g��ttddd�tgd�gd	�gd�gd�g��dS)
Nr"r$��kr%r&r'�rrrr���r#�rrrr-rrr�	test_diag,s����zTestEye.test_diagcCsVttdd�tgd�gd�gd�gd�g��ttdd�tgd�gd�gd	�g��dS)
Nr"r+�r$rr�rr$r)rrr$�rrrr#r%r&r4r-rrr�test_2d9s
��
��zTestEye.test_2dcCs^ttdddd�tgd�gd�gd�g��ttdddd�tgd	�gd	�gd
�gd�g��dS)Nr+r"�r0r&r'r2���r8r6r7r4r-rrr�test_diag2dEs����zTestEye.test_diag2dcCs0ttddd�ddgddgg�ttddd�ddgddgg�ttddd�ddgddgg�ttddd�ddgddgg�ttddd�ddgddgddgg�ttddd�ddgddgddgg�ttddd�ddgddgddgg�ttddd�ddgddgddgg�ttddd�ddgddgddgg�dS)Nr:r$rr3r;r+����rrr-rrr�test_eye_boundsQs$$$$(zTestEye.test_eye_boundscCs$ttdddd�ddgddgg�dS)Nr:�S3r)�1�r>r-rrr�test_strings\s�zTestEye.test_stringscCs$ttddtd�ddgddgg�dS)Nr:r)TF)rrr,r-rrr�	test_bool`s$zTestEye.test_boolcCs\tdddd�}tddddd�}t||�|jjsJ�|jjr J�|jjr&J�|jjs,J�dS)Nr"r+r3r0�F)r1�order)rr�flags�c_contiguous�f_contiguous)r.�mat_c�mat_frrr�
test_ordercs
zTestEye.test_orderN)�__name__�
__module__�__qualname__r/r5r9r<r?rCrDrLrrrrr!s
r!c@s6eZdZdd�Zddd�Zdd�Zdd	�Zd
d�ZdS)
�TestDiagcCs�dtd��d�}td�}td�D]
}|||||f<qtt|�|�td�}|��}td�D]}|||||df<||||d|f<q/tt|dd�|�tt|dd�|�dS)	N�d��l)rRrR)�rTr:r0r;)r�astyper�rangerr�copy)r.�vals�br1�crrr�test_vectornszTestDiag.test_vectorNcCs�|durdtd�d�d�}td�}td�D]
}|||f||<qtt|�|�|d}td�D]}|||df||<q1tt|d�|dd��td�D]}||d|f||<qNtt|d	�|dd��dS)
NrQrRr$rS)rRrr+r:r;)r rUrrVrr)r.rXrYr1rrr�test_matrix|szTestDiag.test_matrixcCs(tdtd�dddd�}|�|�dS)NrQrRr$rErS)rFr*)rr r\)r.rXrrr�test_fortran_order�szTestDiag.test_fortran_ordercCs�ddgddgddgg}tt|dd�g�tt|dd�dg�tt|dd�ddg�tt|d	d�ddg�tt|d
d�dg�tt|dd�g�dS)Nr$r:r+r"rR�r0rr3r;r=)rr)r.�Arrr�test_diag_bounds�szTestDiag.test_diag_boundscCstttdggg�dS)Nr$)r�
ValueErrorrr-rrr�test_failure�szTestDiag.test_failurer)rMrNrOr[r\r]r`rbrrrrrPms
	rPc@�eZdZdd�ZdS)�
TestFliplrcCsnttttd��td�}|dd�ddd�f}tt|�|�gd�gd�g}gd�gd�g}tt|�|�dS)Nr"r3�rr$r:�r+r"rR)r:r$r)rRr"r+)rrar
r
r r�r.�arYrrrr/�s��zTestFliplr.test_basicN�rMrNrOr/rrrrrd��rdc@rc)�
TestFlipudcCs^td�}|ddd�dd�f}tt|�|�gd�gd�g}gd�gd�g}tt|�|�dS)Nr"r3rerf)r rrrgrrrr/�s��zTestFlipud.test_basicNrirrrrrk�rjrkc@sHeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Ze	d
d��Z
dS)�TestHistogram2dcCs
tgd��}tgd��}t�ddd�}t�ddd�}t||||f�d}tgd�gd�gd�gd	�gd
�gd�gd�gd�gd�g	�}t|j|�t|||�d}t|j|�tttd��ttd���\}}}t|tdd��t|t�ddd��t|t�ddd��dS)
N)g̙�
}��?g%�����?g�$F��?g�;c�jY�?gI�+����?)g�p5x���?g�?�_��?g�p����?g��b�d�?g�g<��=�?rr$�
)	rrrr$rrrrr)	rrrrrrr$rr)	rrrrrrrrr)	r$rr$rrrrrr)	rr$rrrrrrr�	�)	r�np�linspacerr�T�listrVr)r.�x�y�xedges�yedges�H�answerrrr�test_simple�s8����
 zTestHistogram2d.test_simplecCs�tgd��}tgd��}t||dddgddggdd�\}}}tgd	�gd
�gd�gd�gd
�gd�g�}t||dd�t|t�ddd��t|t�ddd��dS)N)r$r$r:r+r"r"r"rR)r$r+r:rr$r:r+r")r^rRrr^rRT)rV�density)�rrrr)rr$rr$r)rrr$rr)r$rrrr)rr$r$r$r�rrrrr$g @r+rT)rrrrrprq�r.rtrurx�xed�yedryrrr�	test_asym�s"���zTestHistogram2d.test_asymcCshtgd��}tgd��}t||gd�gd�gdd�\}}}tgd�gd�gd�g�d}t||d	�dS)
N)	r$r:r+r$r:r+r$r:r+)	r$r$r$r:r:r:r+r+r+)r$r:r+rRT)r{)r$r$��?)r�r���?g"@r+)rrrr~rrr�test_density�s���zTestHistogram2d.test_densitycCsDtj�d�dd}t||dddgddgfd�\}}}t|d�dS)NrQ��?g��.A)r"rRrr$)rV)rp�random�randrr)r.�rrxrr�rrr�test_all_outliers�s"z!TestHistogram2d.test_all_outlierscCsZtggddgddgfd�\}}}t|tdgg��tggdd�\}}}t|t�d��dS)Nrr$��binsr|r")r"r")rrrrpr)r.rh�edge1�edge2rrr�
test_empty�s zTestHistogram2d.test_emptycCs�tgd��}tgd��}d}t|||df�\}}}tgd�gd�gd�gd�gd�gd�gd�gd�gd�gd	�g
�}t||�t|tgd
���t||d|f�\}}}tgd�gd�gd
�gd�g�}t||�t|tgd
���dS)N)rg<D��瑷?g����?g!,��3{�?g�l�yT�?g�f��.�?r$)rg(@TT\�?g���IZ�?gY�N���?gK?~›�?g&W-��U�?r$)rg�������?g�������?g333333�?g�������?r�g333333�?gffffff�?g�������?g�������?r$r")g@r|r|r|)r|r�r|r|)r|r|r|r|)r�r|r|r|)r|r|r|r�)r|r�r�g�?r$)
r�r�r|r�r|r|r|r|r|r|)
r|r|r|r|r�r|r|r|r|r|)
r|r�r|r|r�r|r|r|r|r|)
r|r|r|r|r|r|r|r|r|r�)rrr)r.rtru�edgesrx�xe�yeryrrr�test_binparameter_combination�sB����
��
z-TestHistogram2d.test_binparameter_combinationcCs�Gdd�d�}ddg}|�}t||�}t||f||fifk�t||�}t||f||fifk�t|||d�}t||f||ft|d�f�t|||dgd�}t||f||ft|dgd�f�ttt|||gd�t|||d�}t||f||ft|d�f�dS)Nc@rc)z5TestHistogram2d.test_dispatch.<locals>.ShouldDispatchcSs
|||fSrr)r.�function�types�args�kwargsrrr�__array_function__s
zHTestHistogram2d.test_dispatch.<locals>.ShouldDispatch.__array_function__N)rMrNrOr�rrrr�ShouldDispatchrjr�r$r:r�rR)�weights)rr�dictr�	Exception)r.r��xy�s_dr�rrr�
test_dispatchs

  zTestHistogram2d.test_dispatchN)rMrNrOrzr�r�r�r�r�rr�rrrrrl�s
rlc@rc)�TestTricCsDtgd�gd�gd�g�}ttd�|�ttdtd�|�t��dS)Nr6�r$r$r�r$r$r$r+r))rrrr,rU)r.�outrrr�
test_dtype,s�zTestTri.test_dtypeN)rMrNrOr�rrrrr�+rjr�cCs~tjdtjdD]2}tjd|d�}t�|�}t�|�}t|ddgddgg�t||j�t|j|j�t|j|j�q
dS)N�AllFloat�
AllInteger)r:r:r)r$r)	rp�	typecodesr
�tril�triurrrrr*)r*rhrYrZrrr�test_tril_triu_ndim24s

�r�cCs
tjdtjdD]x}tjddgddggddgddggddgddggg|d�}tjddgddggddgddggddgddggg|d�}tjddgddggddgddggddgddggg|d�}t�|�}t�|�}t||�t||�t|j|j�t|j|j�q
dS)Nr�r�r$rr))rpr�rr�r�rrr*)r*rh�a_tril_desired�a_triu_desired�a_triu_observed�a_tril_observedrrr�test_tril_triu_ndim3@s:������



�r�cCsnt�ddtjggd�tjddgg�}t�gd�gd�tjddgg�}|j}tt�|�|�tt�|�|�dS)Nr$r�r6r�)rpr�infrrrr�r�)�arr�out_tril�out_triurrr�test_tril_triu_with_infYs
�

�r�cCs�tjdD]"}|dkrqtjd|d�}tt�|�j|j�tt�|�j|j�qtjddgddggd	d�}tt�|�j|j�tt�|�j|j�tjdd
d�}tt�|�j|j�tt�|�j|j�dS)N�All�V)r+r+r)z2001-01-01T12:00z2002-02-03T13:56z2004-01-01T12:00z2003-01-03T13:45�
datetime64zf4,f4)rpr�rrr�r*r�r)rZr�rrr�test_tril_triu_dtypefs 
��r�cCs\tdtj�}t�d��dd�}t||tgd���tdtjd�}t||tgd���dS)Nr+rn)rr$r:r"rR�r$)r$r:rR)rrpr�r�reshaperr)�iurh�iu1rrr�test_mask_indices|s
r�cCsLtd�}tddd�}tddd�}tdddd�}t�gd�gd�gd	�gd
�g�}t�dd��dd�}t||tgd
���t||tgd���d||<t|tgd�gd�gd�gd�g��d||<t|tgd�gd�gd�gd�g��d||<t|tgd�gd�gd�gd�g��d||<t|tgd�gd�gd�gd�g��dS)Nr"r:r0rR��m�r1r��r$r:r+r"�rRr^rTr��rnrmro���
���r$�)
r$rRr^rnrmror�r�r�r�)
r$r^rTror�r�r����r3)r3r:r+r")r3r3rTr�)r3r3r3r��r3r3r3r3)r3r:r+r"rR)r3r3r�rnrm)r3r3r3r�r�)r3r3r3r3����)r�r�r�r")r�r�r�r�)r�r�r�r"rR)r�r�r�r�rm)r�r�r�r�r�)rrprrr�r)�il1�il2�il3�il4rhrYrrr�test_tril_indices�s`
�
�
���������r�c@rc)�TestTriuIndicescCsLtd�}tddd�}tddd�}tdddd�}t�gd�gd�gd	�gd
�g�}t�dd��dd�}t||tgd
���t||tgd���d||<t|tgd�gd�gd�gd�g��d||<t|tgd�gd�gd�gd�g��d||<t|tgd�gd�gd�gd�g��d||<t|tgd�gd�gd�gd�g��dS)Nr"r:r0rRr�r�r�r�r�r�r$r�)
r$r:r+r"r^rTr�ror�r�)r$r:r+r"rRrTr�rnrmr�r�r�r�r�r3r�)rRr3r3r3)rnrmr3r3)r�r�r�r3)r3r3r3r3r3)r^r3r3r3r3)ror�r3r3r3)r�r�r�r3r3r�)r3r3r�r�)rRr3r3r�)r3r3r�r�r�)r^r3r3r�r�)ror�r3r3r�)rrprrr�r)r.r��iu2�iu3�iu4rhrYrrr�test_triu_indices�s`
�
�
���������z!TestTriuIndices.test_triu_indicesN)rMrNrOr�rrrrr��rjr�c@rc)�TestTrilIndicesFromcC�(tttt�d��tttt�d��dS�N)r:)r:r:r:)rrarrpr
r-rrr�test_exceptions��z#TestTrilIndicesFrom.test_exceptionsN�rMrNrOr�rrrrr��rjr�c@rc)�TestTriuIndicesFromcCr�r�)rrarrpr
r-rrrr��r�z#TestTriuIndicesFrom.test_exceptionsNr�rrrrr��rjr�c@seZdZdd�Zdd�ZdS)�
TestVandercCs�t�gd��}t|�}t�gd�gd�gd�gd�g�}t||dd�dd�f�|jd}td�D]}t||d�}t||dd�|||�f�q2dS)	N)rr$r;r+r})r$r$r$r$r$)r�i����r"r;r$)�Q�rnr+r$r$r^��N)rprrr�shaperV)r.rZ�v�powersr�rrrrr/�s
�
 �zTestVander.test_basiccCs|tgd�tjd�}t|�}t�gd�gd�gd�g�}t||�tddg�}t|dd	�}t�gd
�gd�g�}t||�dS)N)ro��r�r))�yror$)�r�r$)�r�r$��?�?��?�r+r�)y@r�r$)y��r�r$)rrp�int8rr)r.rZr��expectedrrr�test_dtypess
�

�zTestVander.test_dtypesN)rMrNrOr/r�rrrrr��sr�).�__doc__�
numpy.testingrrrrrr�numpyrr	r
rrr
rrrrrrrrrrrrp�numpy.core.tests.test_overridesrr r!rPrdrkrlr�r�r�r�r�r�r�r�r�r�r�rrrr�<module>s, LR/
v	

/2