b
D d dl Zd dlmZ d dlmZmZmZmZm Z m
Z
d dl m Z m
Z
mZmZmZmZ d dlZd Zd Zd"dZd Zd
Zd Zd Zd
Zd Zd Z d Z d Z d Z d Z d Z!d Z"d Z# G d d Z$d Z% G d dej& Z' G d d e' Z(d Z)d Z*d Z+d! Z,dS )# Nrational) assert_equalassert_array_equal
assert_raisesassert_assert_raises_regex assert_warns)
as_stridedbroadcast_arrays_broadcast_shape broadcast_tobroadcast_shapessliding_window_viewc d | D }t | }d |D }|gt | z }t || d S )Nc 6 g | ]}t j | S npzeros.0ss v/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_stride_tricks.py
z)assert_shapes_correct.. 222 222 c g | ] }|j
S r )shape)r as r r z)assert_shapes_correct.. s ,,,Q,,,r )r lenr ) input_shapesexpected_shapeinarrays outarrays outshapesexpecteds r assert_shapes_correctr( s] 32\222H (+I,,),,,I #h--/HH%%%%%r c N d | D }t t t g|R d S )Nc 6 g | ]}t j | S r r r s r r z4assert_incompatible_shapes_raise.. r r )r
ValueErrorr )r" r$ s r assert_incompatible_shapes_raiser, s6 32\222H* .:::::::r Fc t j | t }t t j | }t j | | }|r|j }|j }|r|d d d }|d d d }||z }t || \ }} t || d S )Ndtype)
r r intmultiplyreducearangereshapeTr r )
shape0shape1
transposedflippedx0nx1yb0b1s
r assert_same_as_ufuncrA ! s
& $ $ $B BK v & &''A 1 f % %B
T
T
"X
"X RA
b"
%
%FBq" r c t j d } t j d }t | | \ }}t | | t || d S )N
)r r4 r r )xr> bxbys r test_samerG 7 sU
"
A
"
A
a
#
#FBq" q" r c t j d } t j d }t t d 5 t | |d d d d d S # 1 swxY w Y d S )NrC zgot an unexpected keywordfloat64r. )r r4 r TypeErrorr )rD r> s r test_broadcast_kwargsrK > s "
A
"
A Y(C D D 0 0AY////0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s A A"%A"c t j g dg } t j dgdgdgg }t | | \ }}t j g dg dg dg }|j }t || t || d S )N rN rO rP )r arrayr r6 r )rD r> rE rF bx0by0s r test_one_offrT I s
))) A
1#sQC !!A
a
#
#FB
(IIIyyy)))4
5
5C
%CsB sB r c g d} | D ]>}|g}t || ||g}t || |||g}t || ?d S )N)
r rN rP r rN )r rP rN r )rP r rN rP rP rN rP rP r( )datar r"
input_shapes2
input_shapes3s r test_same_input_shapesra S s} D 4 4 w lE222
mU333 u-
mU3333 4 4r c ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gg
} | D ].\ }}t || t |d d d
| /d S ) NrV rW rZ r\ r[ rN rN rY r r rX r0 r] r^ r" r# s r (test_two_compatible_by_ones_input_shapesrf n s
t
& 6"
& 6"
& 6"
& 6"
& 6"
& 6"
& 6"
& 6"
& 6"
& 6"
& 6"
& 6" D )- B B$ n lN;;; l44R4 0.AAAAB Br c 4 ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgdd gd gdd
gd gdd gd gdd gd gddgd gdd
gd
gdd gd gdd gd gdd
gd
gg} | D ].\ }}t || t |d d d | /d S )
Nr rW r\ r[ rV rc rZ r rd rX rY r0 r] re s r 3test_two_compatible_by_prepending_ones_input_shapesri sM
dT
f v
$
f v
f v
dT
f v
$
f v
f v ) D, )- B B$ n lN;;; l44R4 0.AAAAB Br c z ddgddgg dddgg} | D ])}t | t |d d d *d S ) NrW rO rP rO rW rW rk rN rP rl rO rP rP r0 )r, )r^ r" s r )test_incompatible_shapes_raise_valueerrorrr so
t I D = = ( 666( ddd);<<<<= =r c
g ddgdgddgdgddgdgddgdgddgdgddgdgddgdgddgdgd dgdgdd gdgddgdgddgdgdd gd gd
dgdgddgdgddgdgddgdgd
dgdgddgdgddgdgddgdgd
dgdgd
dgdgd
d gd gd dgdgd d gdgddgdgd
dgdgdd gdgdd gd gddgdgd
dgdgd
d gd g} | D ]\ }}t |d |d
d|d d|d
t |d
|d t |d |d
d d
|vr>> \!_l1otDDD \ ! ! a ,q/5$ O O O a ,q/4 N N N O Or c @ t j d dt j d gt j d dt j d gt j d dt j d gt j d dt j d gt j d dt j d gt j d d t j d gt j d dt j d gt j d d
t j d dd gt j d d t j g d
g d
g gt j d dt j d gt j d dt j d gt j d dt j d gt j d dt j d gt j d dt j d gt j d dt j d gg} | D ]&\ }}}t
|| }t || 'd S )Nr rh rV rN rW rP rn rO rM rZ r0 rm r rN rO rN rO )r rO rO rN )rO r )r rQ r onesr4 r5 r r )r^ input_arrayr r' actuals r test_broadcast_to_succeedsr} s !dBHQKK( !dBHQKK( !dBHQKK( T271::& T271::& Y 2 23 1 tRYq\\* 1 vry|| 3 3Ar : :; 1 vrxIII(> ? ?@ Q
# Q
# Q
#
T271::& &"'&//2 &"'&//2% D( )- - -$ UH k5118V,,,,- -r c ddgddgddgddgddgddgddgd dgdd
gdd gdd gg } | D ]2\ }t j | t t fd
3d S )Nrh r rV rW rn rk rx ry rc r0 )r0 )r0 rO c $ t S N)r )arr target_shapes r z*test_broadcast_to_raises.. s ,sL*I*I r )r r r r+ )r^
orig_shaper r s @@r test_broadcast_to_raisesr s
r
r
r
t
t
t
r
u
D %) K K
Lhz""j"I"I"I"I"IJJJJK Kr c t t d t t ddg d t t t j d d t t t j d t j d d t t t j d gdz d t t t j d gd z d t t t j d gdz dgz d t j d gdz t j d
gdz z t t
fd d S ) Nr rN rO rn rc rP rl rx d rP c t S r )r
bad_argss r r z&test_broadcast_shape..%
&6&A r )r r
r rz r r+ r s @r test_broadcast_shaper sG !##R(((!1a&))4000!"'&//22F;;;!"'&//276??CCVLLL!RWV__$5$:.\ s
*:L*I r r c t S r r r s r r z.test_broadcast_shapes_raises.._ r r )r r+ )r^ r r" s @@r test_broadcast_shapes_raisesr Q s
t I '''
F
D K K j"I"I"I"IJJJJv{dVb[(H* A A A ABBBBBr c t j d g } t | }t j d g }t |t j d g t j g d } t | dd| j z f }t j ddg }t || t j g d } t | ddd| j z f }t j g dg dg dg }t || t j d d
g }t j d | } t j dd
| d< t | dd| j f }g dgdz }d gdz gdz }t |j | t ||d t ||d t j d d } t | dd| j f }t | j |j t j dgdgd }t j d | } t | dd| j f }t | j |j d t d D }t j |t } t | dd| j f }t | j |j t |gdz | d S )NrN rO rP rl rn rO r stridesrN rP r r )numi4)objOrk r. r r rl r V4 )namesformatsc , g | ]}t | S r r )r is r r z#test_as_strided.. s '''!'''r ) r rQ r r itemsizer/ emptyr4 r ranger )r a_viewr' dt expected_num expected_objrs r test_as_stridedr b s
$A
]]Fx Hvrx / /000
A
AJ/@
A
A
AFxA Hvx(((
A
!Q^1D
E
E
EFx |||\\\BCCHvx(((
=,/ 0 0B
R AyAAeH
!QZ
A
A
AF LL>A%LF1H:a e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestSlidingWindowViewc t j d }t |d }t j ddgddgddgddgg }t || d S )Nr rO r rN rP rl )r r4 r rQ r )selfr arr_viewr' s r test_1dz TestSlidingWindowView.test_1d sg ill&sA..8aV V V V % & & 8X.....r c t j d dd df \ }}d|z |z }d}t || }t j ddgddggddgdd ggddgd d
gggddgd d ggdd gd d
ggd d
gd
dgggg }t || d S )NrP rl rC )rO rO r rN rO
r ogridr rQ r r r jr r r r' s r test_2dz TestSlidingWindowView.test_2d s xBQB 1dQh&sE2281vBx 0 !1vBx 0 !1vBx 0 2 !#Bx"b 2 "Bx"b 2 "Bx"b 2 4 5 6 6 8X.....r c t j d dd df \ }}d|z |z }t |dd }t j g dg dg dg dgg }t || d S ) NrP rl rC r )r rC r )rN r r )rO r r )rP r r r r r r r r r' s r test_2d_with_axisz'TestSlidingWindowView.test_2d_with_axis s xBQB 1dQh&sAq118kkk )kk )kk )kk + , - - 8X.....r c t j d dd df \ }}d|z |z }t |dd }t j g dg dggg dg d ggg d
g d ggg }t || d S ) NrP rl rC rm rc rw rM rC r r r r r r r r r r r r r s r test_2d_repeated_axisz+TestSlidingWindowView.test_2d_repeated_axis s xBQB 1dQh&sFF;;8yyy (yy * + +|| +|| - . +|| +|| - . / 0 0 8X.....r c t j d dd df \ }}d|z |z }d}t || }t j g dg dgg dg dggg dg dgg dg d ggg dg d
gg d g d ggg }t || d S ) Nrl rC rm rw r rM r r r ) r )r r ! r r s r test_2d_without_axisz*TestSlidingWindowView.test_2d_without_axis s xBQB 1dQh&sE228yyy,,, 7 (yy,,, 7 9 +||\\\ : +||\\\ : < +||\\\ : +||\\\ : < = > > 8X.....r c j t j d dd df \ }}d|z |z }t j t d 5 t |d d d d n # 1 swxY w Y t j t d 5 t |d d d d n # 1 swxY w Y t j t d 5 t |d d
d d d n # 1 swxY w Y t j t d 5 t |d
d d d d S # 1 swxY w Y d S )Nrl rC z cannot contain negative valuesmatch)r0 rP z3must provide window_shape for all dimensions of `x`rV z2Must provide matching length window_shape and axisrp rX )axisz.window shape cannot be larger than input array)r r )r r pytestraisesr+ r )r r r r s r test_errorsz!TestSlidingWindowView.test_errors sS xBQB 1dQh
]:-M
N
N
N . . W - - - . . . . . . . . . . . . . . .
]KM M M + +
T * * * + + + + + + + + + + + + + + + ]JL L L = =
YV < < < < = = = = = = = = = = = = = = = ]FH H H - -
V , , , - - - - - - - - - - - - - - - - - -sG A A #A B B#&B#C$$C(+C(
D((D,/D,c t j d }t |dd }t |j j t
j t d 5 d|d< d d d n # 1 swxY w Y t |dd }t |j j d|d
< t |t j
g d d S ) Nr rO F writeablez#assignment destination is read-onlyr rP rd TrX )r rP rO rP rl ) r r4 r r flagsr r r r+ r rQ )r r views r test_writeablez$TestSlidingWindowView.test_writeable s ill"3U;;;DJ(()))
];= = = DJ #3T:::
$%%%T
3 9 9:::::s A..A25A2c G d dt j }t j d | }t t t
|dd | t t t
|dd | t t t
|d | d S )Nc e Zd ZdS )1TestSlidingWindowView.test_subok..MyArrayN)__name__
__module__ __qualname__r r r MyArrayr s Dr r r rO FsubokT)r ndarrayr4 r r
isinstancer )r r r s r
test_subokz TestSlidingWindowView.test_subok s bj ill ((J 239> @ @ @ &( ( ( ) ) )
.sATBBBGLLMMMJ 23 : :GDDDEEEEEr N) r r r r r r r r r r r r r r r r s / / / / / // / /
/
/
/ / / /- - -$ ; ; ;
F
F
F
F
Fr r c t j d } t | d }t |j j t | d }t |j j d|d< t
| t j | d d| j _ t | d }t | d }t |j j d S )NrC Fr TrP .)r rz r r r r r full_like)r r s r as_strided_writeabler s
'"++C cU + + +D
$ $%%% cT * * *D DJ !!!DIsBLa 0 0111 CI cU + + +D cT * * *D
$ $%%%%%r c e Zd Zd ZdS )VerySimpleSubClassc N t j |ddi| | S )Nr T)r rQ r )clsargskwargss r __new__zVerySimpleSubClass.__new__ s* x4T4V4499#>>>r N)r r r r r r r r r s# ? ? ? ? ?r r c e Zd Zd Zd ZdS )SimpleSubClassc ` t j |ddi| | }d|_ |S )Nr Tsimple)r rQ r info)r r r r s r r zSimpleSubClass.__new__ s5 x4T4V4499#>> r c 8 t |dd dz | _ d S )Nr r z
finalized)getattrr )r r s r __array_finalize__z!SimpleSubClass.__array_finalize__ s C,,|; r N)r r r r r r r r r r
s2
< < < < >>? ? ? ? ? ? ? ? ? ? ? ? ? ? ?!"455 " " !F111I" " " " " " " " " " " " " " " V\ 3T:::: V\ 3T:::: ; %X..$Q113 F%)FL " / 6 6 6F111II %HN H--IAv '///
EcG bhqkk5'::L HTNNE$\599ME6 EK6< '(((((s$ ,CC C 2DD
D
c t j g d } dt | fdt d| ffD ]9\ }}|D ]1}|rt | j sJ t | j rJ 2:d S )NrM FTr )r rQ r
memoryviewreadonly)r r r r
s r test_writeable_memoryviewr k s
x ""H#(*:8*E*E"F#')9!X)F)F"G"I 7 7 g 7 7F
7 "&))22222%f--66666 7 7 7r c $ t j dt } t j dgdz t }t | d }t || t | t j d \ }}t || d S )Nr r. rP rW )r rQ objectr r r rz )r{ r' r| r
s r test_reference_typesr ~ s (3f---Kx 000H
+t
,
,Fx((( bgajj99IFAx(((((r )FF)-numpyr numpy.core._rational_testsr
numpy.testingr r r r r r
numpy.lib.stride_tricksr r r
r r r r r( r, rA rG rK rT ra rf ri rr ru r} r r r r r r r r r r r r r r r r r r s / / / / / /
& & &; ; ; , 0 0 0 4 4 46B B B2 B B B@ = = = 1O 1O 1Oh- - -4K K K&C C C"&9 &9 &9RC C C"+( +( +(\cF cF cF cF cF cF cF cFL& & &$? ? ? ? ? ? ? ?
< < < < <' < < <"# "# "#J,) ,) ,)^7 7 7&) ) ) ) )r