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_stride_tricks.cpython-310.pyc
o

6��aAY�@sBddlZddlmZddlmZmZmZmZm	Z	m
Z
ddlmZm
Z
mZmZmZmZddlZdd�Zdd�Zd8d
d�Zdd
�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zd d!�Z d"d#�Z!d$d%�Z"d&d'�Z#Gd(d)�d)�Z$d*d+�Z%Gd,d-�d-ej&�Z'Gd.d/�d/e'�Z(d0d1�Z)d2d3�Z*d4d5�Z+d6d7�Z,dS)9�N��rational)�assert_equal�assert_array_equal�
assert_raises�assert_�assert_raises_regex�assert_warns)�
as_strided�broadcast_arrays�_broadcast_shape�broadcast_to�broadcast_shapes�sliding_window_viewcCs@dd�|D�}t|�}dd�|D�}|gt|�}t||�dS)NcS�g|]}t�|��qS���np�zeros��.0�srr�D/usr/lib/python3/dist-packages/numpy/lib/tests/test_stride_tricks.py�
<listcomp>�z)assert_shapes_correct.<locals>.<listcomp>cSsg|]}|j�qSr)�shape)r�arrrrs)r�lenr)�input_shapes�expected_shape�inarrays�	outarrays�	outshapes�expectedrrr�assert_shapes_corrects
r$cCs$dd�|D�}tttg|�R�dS)NcSrrrrrrrrrz4assert_incompatible_shapes_raise.<locals>.<listcomp>)r�
ValueErrorr)rr rrr� assert_incompatible_shapes_raisesr&Fc
Cs�tj|td�}ttj�|��}t�|��|�}|r|j}|j}|r/|ddd�}|ddd�}||}t||�\}}	t	||	�dS)N��dtype���)
rr�int�multiply�reduce�arange�reshape�Trr)
�shape0�shape1�
transposed�flipped�x0�n�x1�y�b0�b1rrr�assert_same_as_ufunc!sr:cCs:t�d�}t�d�}t||�\}}t||�t||�dS)N�
)rr-rr)�xr7�bx�byrrr�	test_same7s



r?cCsRt�d�}t�d�}ttd��t||dd�Wd�dS1s"wYdS)Nr;zgot an unexpected keyword�float64r')rr-r�	TypeErrorr)r<r7rrr�test_broadcast_kwargs>s


"�rBcCsnt�gd�g�}t�dgdgdgg�}t||�\}}t�gd�gd�gd�g�}|j}t||�t||�dS)N����rDrErF)r�arrayrr/r)r<r7r=r>�bx0�by0rrr�test_one_offIs
rJcCsLgd�}|D]}|g}t||�||g}t||�|||g}t||�qdS)N)
r�rD�rF�rrD)rrF�rDr)rFr�rDrF�rFrD�rFrF�r$)�datarr�
input_shapes2�
input_shapes3rrr�test_same_input_shapesSs


�rVcCs�ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd	dgdgdd	gdgddgdgddgdgdd	gd	gg
}|D]\}}t||�t|ddd
�|�qEdS)NrKrLrOrQrP�rDrDrN�rrrMr)rR�rSrrrrr�(test_two_compatible_by_ones_input_shapesns$












�
�rZcCs�ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgdd	gd	gdd
gd	gdd	gd	gdd	gd	gddgdgdd
gd
gddgdgddgdgdd
gd
gg}|D]\}}t||�t|ddd�|�qhdS)
NrrLrQrPrKrWrO�rrXrMrNr)rRrYrrr�3test_two_compatible_by_prepending_ones_input_shapes�s2



















�
�r\cCsDddgddggd�ddgg}|D]}t|�t|ddd��qdS)	NrL���rErF�rE�rLrLr]�rDrFr^�rErFrFr))r&)rSrrrr�)test_incompatible_shapes_raise_valueerror�s��rdcCs"gddgdg�ddgdg�ddgdg�ddgdg�ddgdg�ddgdg�ddgdg�ddgdg�d	dgdg�dd	gdg�ddgdg�ddgdg�dd	gd	g�d
dgdg�ddgdg�ddgdg�ddgdg�d
dgdg�ddgdg�ddgdg�ddgdg�d
dgdg�d
dgdg�d
dgdg�ddgdg�dd	gdg�ddgdg�d
dgdg�ddgdg�dd	gd	g�ddgdg�d
dgdg�d
d	gd	g�}|D]D\}}t|d|d
d|d|d
f�t|d
|d�t|d|d
d�d
|v�rt|d|d
dd�t|d|d
dd�q�dS)NrKrLrOrQrPrWrNrXrMrr[rrDz
Shapes: %s %sTF)r:rYrrr�test_same_as_ufunc�s�
�
�
�
�
�
�
�
�
	�

�
�
�

�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
�
 �
!�#�
��recCslt�d�dt�d�gt�d�dt�d�gt�d�dt�d�gt�d�dt�d�gt�d�dt�d�gt�d�d	t�d	�gt�d�dt�d�gt�d�d
t�d��dd�gt�d�dt�gd
�gd
�g�gt�d�dt�d�gt�d�dt�d�gt�d�dt�d�gt�d�dt�d�gt�d�dt�d�gt�d�dt�d�gg}|D]\}}}t||�}t||�q�dS)Nrr[rKrDrLrFr`rErCrOr)r_�rrDrE�rDrE)rrE�rErD)rEr)rrGr�onesr-r.r
r)rS�input_arrayrr#�actualrrr�test_broadcast_to_succeeds�s( �
�rlcsvddgddgddgddgddgddgddgd	dgdd
gddgddgg}|D]\}�t�|��tt��fd
d��q%dS)Nr[rrKrLr`r]rgrhrWr))r))r)rEcs
t���S�N)r
r��arr�target_shaperr�<lambda>s
z*test_broadcast_to_raises.<locals>.<lambda>)rrrr%)rS�
orig_shaperrnr�test_broadcast_to_raisess �
�rscs�tt�d�ttddg�d�ttt�d��d�ttt�d�t�d��d�ttt�d�gd�d�ttt�d�gd	�d�ttt�d�gddg�d�t�d�gdt�d
�gd�tt�fdd��dS)
NrrDrEr`rW�rFr^rg� �drFc�t��Srm)rr��bad_argsrrrq%�z&test_broadcast_shape.<locals>.<lambda>)rrrrirr%rrrxr�test_broadcast_shapes  r{cCsbgdgdgdgdgdgddgdgdgdgddgdggd�dgd	gd	gd
dgdgd
dgdgddgdgd
dgdgddgdgdd
gd
gddgdgddgdgddgdgddgdgddgdgddgdgddgd
gddgdgdd
gd
gdd
gd
gddgdgddgdgddgdgg}|D]\}}tt|�|�q�ttdgd�d�ttdgd�d�ttdgd�d�dS)Nr��rgr`rWrt))�r}��r~rDr|)r�rDr})r�r~r}rrOrPrQrNrXrMr[rKrLrE�rFrErurv)rr)rSrrprrr�test_broadcast_shapes_succeeds(sB





















�r�csnddgddggd�ddggd�d	dgg}|D]�tt�fd
d��qdgddgd�tt�fd
d��dS)NrLr]r_r`rarbrc)rgrPr�)r;r�rEcrwrm�rr)rrrrq\rzz.test_broadcast_shapes_raises.<locals>.<lambda>rucrwrmr�rrxrrrq_rz)rr%)rSr)ryrr�test_broadcast_shapes_raisesQs�r�cCs
t�dg�}t|�}t�dg�}t|t�dg��t�gd��}t|dd|jfd�}t�ddg�}t||�t�gd��}t|ddd|jfd�}t�gd�gd�gd�g�}t||�t�d	d
g�}tjd|d�}t�dd
�|d<t|dd|jfd�}gd�gd}dgdgd}t|j|�t||d�t||d�tjddd�}t|dd|jfd�}t|j|j�t�dgdgd��}tjd|d�}t|dd|jfd�}t|j|j�dd�t	d�D�}tj|t
d�}t|dd|jfd�}t|j|j�t|gd|�dS)N�rDrErFr^r`rE�r�stridesrDrFrtr)�num�i4)�obj�Or]r'r�r�r^r��V4�)�names�formatscSsg|]}t|��qSrr)r�irrrr�sz#test_as_strided.<locals>.<listcomp>)rrGr
r�itemsizer(�emptyr-r�ranger)r�a_viewr#�dt�expected_num�expected_obj�rrrr�test_as_stridedbsB

r�c@sLeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)�TestSlidingWindowViewcCsDt�d�}t|d�}t�ddgddgddgddgg�}t||�dS)Nr�rErrDrFr^)rr-rrGr)�selfro�arr_viewr#rrr�test_1d�s


�zTestSlidingWindowView.test_1dcCs�tjdd�dd�f\}}d||}d}t||�}t�ddgddggddgdd	ggddgd	d
gggddgddggdd	gdd
ggd	d
gd
dgggg�}t||�dS)NrFr^r;)rErErrD�rE��
�����r�ogridrrGr�r�r��jrorr�r#rrr�test_2d�s
���zTestSlidingWindowView.test_2dcCsdtjdd�dd�f\}}d||}t|dd�}t�gd�gd�gd�gd�gg�}t||�dS)	NrFr^r;r)rr;r�)rDr�r�)rEr�r�)rFr�r�r��r�r�r�ror�r#rrr�test_2d_with_axis�s
�z'TestSlidingWindowView.test_2d_with_axiscCsztjdd�dd�f\}}d||}t|dd�}t�gd�gd�gggd�gd	�gggd
�gd�ggg�}t||�dS)NrFr^r;r_rWrfrC�r;r�r��r�r�r��r�r�r��r�r�r�r�r�rrr�test_2d_repeated_axis�s
����z+TestSlidingWindowView.test_2d_repeated_axiscCs�tjdd�dd�f\}}d||}d}t||�}t�gd�gd�ggd�gd�gggd�gd�ggd�gd	�gggd�gd
�ggd	�gd�ggg�}t||�dS)Nr^r;r_rfr�rCr�r�r�)��ru)r�ru�!r�r�rrr�test_2d_without_axis�s
����z*TestSlidingWindowView.test_2d_without_axiscCstjdd�dd�f\}}d||}tjtdd��
t|d�Wd�n1s*wYtjtdd��
t|d�Wd�n1sFwYtjtdd��t|d	d
d�Wd�n1sdwYtjtdd��t|d
�Wd�dS1s�wYdS)Nr^r;zcannot contain negative values��match)r)rFz3must provide window_shape for all dimensions of `x`rKz2Must provide matching length window_shape and axisrbrM)�axisz.window shape cannot be larger than input array)r�r�)rr��pytest�raisesr%r)r�r�r�rorrr�test_errors�s.������"�z!TestSlidingWindowView.test_errorscCs�t�d�}t|ddd�}t|jj�tjtdd��d|d<Wd�n1s)wYt|dd	d�}t|jj�d|d
<t	|t�
gd���dS)Nr�rEF��	writeablez#assignment destination is read-onlyr�rFrXTrM)rrFrErFr^)rr-rr�flagsr�r�r�r%rrG)r�ro�viewrrr�test_writeable�s
�
�z$TestSlidingWindowView.test_writeablecCsnGdd�dtj�}t�d��|�}ttt|ddd�|��ttt|ddd�|��ttt|d�|��dS)Nc@seZdZdS)z1TestSlidingWindowView.test_subok.<locals>.MyArrayN)�__name__�
__module__�__qualname__rrrr�MyArray�sr�r�rEF��subokT)r�ndarrayr-r�r�
isinstancer)r�r�rorrr�
test_subok�s
��z TestSlidingWindowView.test_subokN)r�r�r�r�r�r�r�r�r�r�r�rrrrr��s	



r�cCs�t�d�}t|dd�}t|jj�t|dd�}t|jj�d|d<t|t�|d��d|j_t|dd�}t|dd�}t|jj�dS)Nr;Fr�TrF.)rrir
rr�r�r�	full_like)ror�rrr�as_strided_writeable�s
r�c@seZdZdd�ZdS)�VerySimpleSubClasscOstj|ddi|���|�S)Nr�T)rrGr�)�cls�args�kwargsrrr�__new__	szVerySimpleSubClass.__new__N)r�r�r�r�rrrrr�sr�c@seZdZdd�Zdd�ZdS)�SimpleSubClasscOs$tj|ddi|���|�}d|_|S)Nr�T�simple)rrGr��info)r�r�r�r�rrrr�szSimpleSubClass.__new__cCst|dd�d|_dS)Nr�r�z
 finalized)�getattrr�)r�r�rrr�__array_finalize__sz!SimpleSubClass.__array_finalize__N)r�r�r�r�r�rrrrr�
sr�cCs�tgd��}tt|�tu�t|dd|jfd�}tt|�tju�t|dd|jfdd�}tt|�tu�tgd��}t|dd|jfdd�}tt|�tu�t|jdk�t�	t
|���dd	�}t||�\}}tt|�tju�tt|�tju�t|j
|j
k�t||dd
�\}}tt|�tu�t|jdk�tt|�tju�t|j
|j
k�d}t||�}tt|�tju�t|j
|k�t||dd
�}tt|�tu�t|jdk�t|j
|k�dS)Nr�r`rEr�T)rr�r�zsimple finalizedr)rDr�)rEr^)r�r�typer
r�rr�r�r�r-rr.rrr
)rr��b�b_viewrrrr�test_subclassess8
r�c
Cs�t�gd��}t|d�}t|jjd�tt|jt	d�d�dt
|�fdt
d|�ffD]O\}}|D]H}|rqtt��t|jjd�Wd�n1sJwYtt
��d|dd�<Wd�n1sdwYt|jjd�q0t|jjd�q0q*t
|�t
d|�fD]}|D]}d|j_t|jjd�d|dd�<q�q�d|j_t
d|�\}}t|jjd�d}dg}tt�d�||�}t�d�}t
||�\}	}
t|	j|
jk�dS)NrCr_FrTr`rK)rrGr
rr�r�rr%�__setitem__�slicerr	�
FutureWarning�DeprecationWarningr
rrr)�original�result�is_broadcast�results�_rr��tricky_array�other�first�secondrrrr�<sF

�
�
��
��
r�cCsbt�gd��}dt|�fdtd|�ffD]\}}|D]}|r&t|�js%J�qt|�jr-J�qqdS)NrCFTr)rrGr�
memoryview�readonly)r�r�r�r�rrr�test_writeable_memoryviewks
���r�cCsXtjdtd�}tjdgdtd�}t|d�}t||�t|t�d��\}}t||�dS)Nrr'rFrL)rrG�objectr
rrri)rjr#rkr�rrr�test_reference_types~s

r�)FF)-�numpyr�numpy.core._rational_testsr�
numpy.testingrrrrrr	�numpy.lib.stride_tricksr
rrr
rrr�r$r&r:r?rBrJrVrZr\rdrerlrsr{r�r�r�r�r�r�r�r�r�r�r�r�rrrr�<module>s<  

 4).f
%/