Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_stride_tricks.cpython-39.pyc
a
z[ycAY � @ sB d dl Zd dlmZ d dlmZmZmZmZm Z m
Z
d dl m Z m
Z
mZmZmZmZ d dlZdd� Zdd� Zd8d
d �Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z dd� Z d d � Z d d � Z d d!� Z d"d#� Z!d$d%� Z"d&d'� Z#G d(d)� d)�Z$d*d+� Z%G d,d-� d-ej&�Z'G d.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_viewc C s@ dd� | D �}t |� }dd� |D �}|gt|� }t||� d S )Nc S s g | ]}t �|��qS � ��np�zeros��.0�sr r �H/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_stride_tricks.py�
<listcomp> � z)assert_shapes_correct.<locals>.<listcomp>c S s g | ]
}|j �qS r )�shape)r �ar r r r r )r �lenr )� input_shapes�expected_shape�inarraysZ outarraysZ outshapes�expectedr r r �assert_shapes_correct s
r! c C s$ dd� | D �}t ttg|�R � d S )Nc S s g | ]}t �|��qS r r r r r r r r z4assert_incompatible_shapes_raise.<locals>.<listcomp>)r �
ValueErrorr )r r r r r � assert_incompatible_shapes_raise s r# Fc
C s� t j| td�}tt j�|��}t �|��|�}|r>|j}|j}|r^|d d d� }|d d d� }|| }t||�\}} t || � d S )N��dtype���)
r r �intZmultiply�reduce�arange�reshape�Tr r )
Zshape0Zshape1Z
transposedZflippedZx0�nZx1�yZb0�b1r r r �assert_same_as_ufunc! s r/ c C s: t �d�} t �d�}t| |�\}}t| |� t||� d S )N�
)r r) r r )�xr- �bx�byr r r � test_same7 s
r4 c C sP t �d�} t �d�}ttd�� t| |dd� W d � n1 sB0 Y d S )Nr0 zgot an unexpected keywordZfloat64r$ )r r) r � TypeErrorr )r1 r- r r r �test_broadcast_kwargs> s
r6 c C sn t �g d�g�} t �dgdgdgg�}t| |�\}}t �g d�g d�g d�g�}|j}t||� t||� d S )N�� � � r8 r9 r: )r �arrayr r+ r )r1 r- r2 r3 Zbx0Zby0r r r � test_one_offI s
r< c C sL g d�} | D ]:}|g}t ||� ||g}t ||� |||g}t ||� q d S )N)
r �r8 �r: �r r8 )r r: �r8 r )r: r �r8 r: �r: r8 �r: r: �r! )�datar r Z
input_shapes2Z
input_shapes3r r r �test_same_input_shapesS s
rF c C s� ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gg
} | D ]&\}}t ||� t |d d d
� |� q�d S ) Nr= r> rA rC rB �r8 r8 r@ �r r r? r& rD �rE r r r r r �(test_two_compatible_by_ones_input_shapesn s"
�
rJ c C s� ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgdd gd gdd
gd gdd gd gdd gd gddgd gdd
gd
gdd gd gdd gd gdd
gd
gg} | D ]&\}}t ||� t |d d d � |� q�d S )
Nr r> rC rB r= rG rA �r rH r? r@ r& rD rI r r r �3test_two_compatible_by_prepending_ones_input_shapes� s0
�
rL c C sD ddgddgg d�ddgg} | D ] }t |� t |d d d� � q d S ) Nr> �� �r9 r: �r9 �r> r> rM �r8 r: rN �r9 r: r: r& )r# )rE r r r r �)test_incompatible_shapes_raise_valueerror� s �rT c " C s� ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gd
dgdgddgdgddgdgddgdgd
dgdgddgdgddgdgddgdgd
dgdgd
dgdgd
d gd gd dgdgd d gdgddgdgd
dgdgdd gdgdd gd gddgdgd
dgdgd
d gd gg!} | D ]�\}}t |d |d
d|d |d
f � t |d
|d � t |d |d
d� d
|v�rRt |d |d
dd� t |d |d
dd� �qRd S )Nr= r> rA rC rB rG r@ rH r? r rK r r8 z
Shapes: %s %sTF)r/ rI r r r �test_same_as_ufunc� sV
�# �
rU c C sn t �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 �g d
�g d
�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||� �qHd S )Nr rK r= r8 r> r: rP r9 r7 rA r& rO �r r8 r9 �r8 r9 )r r9 �r9 r8 )r9 r )r r; r �onesr) r* r
r )rE � input_arrayr r �actualr r r �test_broadcast_to_succeeds� s& �
r\ c sv ddgddgddgddgddgddgddgd dgdd
gdd gdd gg } | D ]&\}�t �|�� tt� �fd
d�� qJd S )NrK r r= r> rP rM rW rX rG r& )r& )r& r9 c s
t � ��S �N)r
r ��arr� target_shaper r �<lambda> r z*test_broadcast_to_raises.<locals>.<lambda>)r r r r" )rE Z
orig_shaper r^ r �test_broadcast_to_raises s �
rb c s� t t� d� t tddg�d� t tt�d��d� t tt�d�t�d��d� t tt�d�gd � d� t tt�d�gd � d� t tt�d�gd dg � d� t�d�gd t�d
�gd � tt� fd d �� d S )
Nr r8 r9 rP rG �r: rN rW � �d r: c s t � � S r] )r r ��bad_argsr r ra % r z&test_broadcast_shape.<locals>.<lambda>)r r r rY r r" r r rf r �test_broadcast_shape s rh c C sd g dgdgdgdgdgddgdgdgdgddgdgg d�dgd gd gd
d gd gd
dgdgddgdgd
dgdgddgdgdd
gd
gddgdgddgdgddgdgddgdgddgdgddgdgddgd
gddgdgdd
gd
gdd
gd
gddgdgddgdgddgdgg} | D ]\}}t t|� |� �q
t tdgd � d� t tdgd � d� t tdgd � d� d S )Nr �� rW rP rG rc ))� rj �� rk r8 ri )rm r8 rj )rm rk rj rl rA rB rC r@ rH r? rK r= r> r9 �r: r9 rd re )r r )rE r r` r r r � test_broadcast_shapes_succeeds( sB
� ro c sn ddgddgg d�ddgg d�d dgg} | D ]�t t�fd
d �� q,dgd dgd � t t� fd
d �� d S )Nr> rM rO rP rQ rR rS )rW rB rn )r0 rm r9 c s t � � S r] �r r )r r r ra \ r z.test_broadcast_shapes_raises.<locals>.<lambda>rd c s t � � S r] rp r rf r r ra _ r )r r" )rE r )rg r r � test_broadcast_shapes_raisesQ s �rq c C s
t �d g�} t| �}t �d g�}t|t �d g�� t �g d��} t| dd| j fd�}t �ddg�}t||� t �g d��} t| ddd| j fd�}t �g d�g d�g d�g�}t||� t �d d
g�}t jd |d �} t �dd
�| d< t| dd| jfd�}g d�gd }d gd gd }t|j|� t||d � t||d � t jd dd �} t| dd| jfd�}t| j|j� t �dgdgd��}t jd |d �} t| dd| jfd�}t| j|j� dd� t d�D �}t j|t
d �} t| dd| jfd�}t| j|j� t|gd |� d S )N�r8 r9 r: rN rP r9 �r �stridesr8 r: rc r )�numZi4)�obj�OrM r$ rm ru rN rv ZV4� )�names�formatsc S s g | ] }t |��qS r r )r �ir r r r � r z#test_as_strided.<locals>.<listcomp>) r r; r
r �itemsizer% �emptyr) r �ranger )r �a_viewr ZdtZ expected_numZ expected_obj�rr r r �test_as_stridedb sB
r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�TestSlidingWindowViewc C sD t �d�}t|d�}t �ddgddgddgddgg�}t||� d S )Nrm r9 r r8 r: rN )r r) r r; r )�selfr_ �arr_viewr r r r �test_1d� s
�z TestSlidingWindowView.test_1dc C s� t jd d�d d�f \}}d| | }d}t||�}t �ddgddggddgdd ggddgd d
gggddgd d ggdd gd d
ggd d
gd
dgggg�}t||� d S )Nr: rN r0 )r9 r9 r r8 � r9 � �
� � � � �r �ogridr r; r �r� r{ �jr_ r r� r r r r �test_2d� s
���z TestSlidingWindowView.test_2dc C sd t jd d�d d�f \}}d| | }t|dd�}t �g d�g d�g d�g d�gg�}t||� d S ) Nr: rN r0 r )r r0 r� )r8 r� r� )r9 r� r� )r: r� r� r� �r� r{ r� r_ r� r r r r �test_2d_with_axis� s
�z'TestSlidingWindowView.test_2d_with_axisc C sz t jd d�d d�f \}}d| | }t|dd�}t �g d�g d�ggg d�g d �ggg d
�g d �ggg�}t||� d S ) Nr: rN r0 rO rG rV r7 �r0 r� r� �r� r� r� �r� r� r� �r� r� r� r� r� r r r �test_2d_repeated_axis� s
����z+TestSlidingWindowView.test_2d_repeated_axisc C s� t jd d�d d�f \}}d| | }d}t||�}t �g d�g d�gg d�g d�ggg d�g d�gg d�g d �ggg d�g d
�gg d �g d �ggg�}t||� d S ) NrN r0 rO rV r� r7 r� r� r� )� � rd )r� rd �! r� r� r r r �test_2d_without_axis� s
����z*TestSlidingWindowView.test_2d_without_axisc C s t jd d�d d�f \}}d| | }tjtdd�� t|d� W d � n1 sT0 Y tjtdd�� t|d� W d � n1 s�0 Y tjtdd�� t|d d
d � W d � n1 s�0 Y tjtd d�� t|d
� W d � n1 �s0 Y d S )NrN r0 z cannot contain negative values��match)r& r: z3must provide window_shape for all dimensions of `x`r= z2Must provide matching length window_shape and axisrR r? )Zaxisz.window shape cannot be larger than input array)rm rm )r r� �pytest�raisesr" r )r� r{ r� r_ r r r � test_errors� s& (�(�,�z!TestSlidingWindowView.test_errorsc C s� t �d�}t|ddd�}t|jj � tjtdd�� d|d< W d � n1 sR0 Y t|dd d�}t|jj� d|d
< t |t �
g d ��� d S ) Nrm r9 F�� writeablez#assignment destination is read-onlyr� r: rH Tr? )r r: r9 r: rN ) r r) r r �flagsr� r� r� r" r r; )r� r_ �viewr r r �test_writeable� s
�& z$TestSlidingWindowView.test_writeablec C sn G dd� dt j�}t �d��|�}ttt|ddd�|� � ttt|ddd�|�� ttt|d�|� � d S )Nc @ s e Zd ZdS )z1TestSlidingWindowView.test_subok.<locals>.MyArrayN)�__name__�
__module__� __qualname__r r r r �MyArray� s r� rm r9 F��subokT)r �ndarrayr) r� r �
isinstancer )r� r� r_ r r r �
test_subok� s
��z TestSlidingWindowView.test_subokN) r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c C s� t �d�} t| dd�}t|jj � t| dd�}t|jj� d|d< t| t �| d�� d| j_t| dd�}t| dd�}t|jj � d S )Nr0 Fr� Tr: .)r rY r
r r� r� r Z full_like)r_ r� r r r �as_strided_writeable� s
r� c @ s e Zd Zdd� ZdS )�VerySimpleSubClassc O s t j|ddi|���| �S )Nr� T)r r; r� )�cls�args�kwargsr r r �__new__ s zVerySimpleSubClass.__new__N)r� r� r� r� r r r r r� s r� c @ s e Zd Zdd� Zdd� ZdS )�SimpleSubClassc O s$ t j|ddi|���| �}d|_|S )Nr� T�simple)r r; r� �info)r� r� r� r� r r r r� s zSimpleSubClass.__new__c C s t |dd�d | _d S )Nr� rx z
finalized)�getattrr� )r� rv r r r �__array_finalize__ s z!SimpleSubClass.__array_finalize__N)r� r� r� r� r� r r r r r�
s r� c C s� t g d��} tt| �t u � t| dd| j fd�}tt|�tju � t| dd| j fdd�}tt|�t u � tg d��} t| dd| j fdd�}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� d S ) Nrr rP r9 rs T)r rt r� zsimple finalizedr& r8 r� )r9 rN )r� r �typer
r| r r� r� r� r) r r* r r r
)r r �bZb_viewr r r r �test_subclasses s8
r� c
C s� t �g d��} t| d�}t|jjd� tt|jt d �d� dt
| �fdt
d| �ffD ]�\}}|D ]�}|r�t t �� t|jjd� W d � n1 s�0 Y t t
�� d|d d �< W d � n1 s�0 Y t|jjd� q`t|jjd� q`qTt
| �t
d| �fD ]4}|D ](}d|j_t|jjd� d|d d �< �q�qd| j_t
d| �\}}t|jjd� d}dg}tt �d�||�}t �d�}t
||�\} }
t| j|
jk� d S )Nr7 rO Fr TrP r= )r r; r
r r� r� r r" � __setitem__�slicer r �
FutureWarning�DeprecationWarningr
r r r ) �original�result� is_broadcast�results�_r rt Z tricky_array�other�first�secondr r r r� < s>
�
,
*�
r� c C sb t �g d��} dt| �fdtd| �ffD ]4\}}|D ]&}|rLt|�jsZJ �q4t|�jr4J �q4q(d S )Nr7 FTr )r r; r �
memoryview�readonly)r� r� r� r� r r r �test_writeable_memoryviewk s
� r� c C sX t jdtd�} t jdgd td�}t| d�}t||� t| t �d��\}}t||� d S )Nr r$ r: r> )r r; �objectr
r r rY )rZ r r[ r� r r r �test_reference_types~ s
r� )FF)-Znumpyr Znumpy.core._rational_testsr Z
numpy.testingr r r r r r Znumpy.lib.stride_tricksr
r r r
r r r� r! r# r/ r4 r6 r< rF rJ rL rT rU r\ rb rh ro rq r� r� r� r� r� r� r� r� r� r� r r r r �<module> s:
4).f
%/
Mini Shell