1j< d dl Z d dlmZmZmZmZmZmZmZm Z m
Z
m Z m Z m
Z
mZ d dlmZmZmZmZmZ G d de Z G d de
Z G d d Z G d
d Z G d d
ZdS ) N)
logspacelinspace geomspacedtypearraysctypesarangeisnanndarraysqrt nextafterstackerrstate)assert_ assert_equal
assert_raisesassert_array_equalassert_allclosec @ e Zd Zd Zd ZeZd Zd Zd ZeZ d Z
d Z dS ) PhysicalQuantityc 8 t | | S N)float__new__)clsvalues w/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/core/tests/test_function_base.pyr zPhysicalQuantity.__new__ s }}S%((( c t t |t t t | t | z S r r
isinstancer r selfxs r __add__zPhysicalQuantity.__add__ :
1 .//000 a5;; 6777r c t t |t t t | t | z
S r r r" s r __sub__zPhysicalQuantity.__sub__ s:
1 .//000 d eAhh 6777r c t t |t t t | t | z
S r r r" s r __rsub__zPhysicalQuantity.__rsub__ r& r c Z t t | t | z S r r r r" s r __mul__zPhysicalQuantity.__mul__ ! a5;; 6777r c Z t t | t | z S r r, r" s r __div__zPhysicalQuantity.__div__ s! d eAhh 6777r c Z t t | t | z S r r, r" s r __rdiv__zPhysicalQuantity.__rdiv__# r. r N) __name__
__module__ __qualname__r r% __radd__r( r* r- __rmul__r0 r2 r r r r s ) ) )8 8 8 H8 8 88 8 88 8 8H8 8 88 8 8 8 8r r c e Zd ZdZdS )PhysicalQuantity2
N)r3 r4 r5 __array_priority__r8 r r r: r: ' s r r: c e Zd Zd Zd Zej dg d defd Z ej dg d defd Z
d Z d Z d Z
d
S ) TestLogspacec Z t dd }t t | dk t ddd }t |d dk t ddd }t |d dk t ddd
}t |g d d S ) Nr 2 d num@B Fendpoint r; rB @@ @g j@ .A)r r lenr r# ys r
test_basiczTestLogspace.test_basic- s QNNA" Qs # # #" !!! QE * * *" Qq ! ! !1 > > >?????r c t ddg }t ddg t |d }t d t | D d }t || t |d d }t fd
|D d }t || t |dd }t ||j d S ) N ?g @ @r@ c 6 g | ]\ }}t ||d S )r@ r .0_start_stops r
z6TestLogspace.test_start_stop_array..; 8 ; ; ;% VUA.. ; ; ;r rK axisr c > g | ]}t |d d S )r r@ rX rZ r[ stops r r] z6TestLogspace.test_start_stop_array..? 9 ) ) ) VT!Wa00 ) ) )r rE )r r r zipr Tr# startt1t2t3t4t5rc s @r test_start_stop_arrayz"TestLogspace.test_start_stop_array7 s r2hb"X
eT1
%
%
; ;),UD)9)9; ; ;ABD D DR
eT!Wa
(
(
) ) ) )"') ) )/02 2 2R
eT12
.
.
.R r r` )r rK rE c dddt ddg }t || }t fd|D |dz |j z }t || d S )NrK r@ rD baser` c 6 g | ]}t | S )rD rr rX )rZ _baserD rh rc s r r] z0TestLogspace.test_base_array..M s* J J JEXeTs
7
7
7 J J Jr r_ )r r r ndimr )r# r` rr ri rj rD rh rc s @@@r test_base_arrayz TestLogspace.test_base_arrayE s aV}}
eTsD
A
A
A
J J J J J JT J J J(bg%
R r c dt ddg }dt ddg }t ||| }t fdt || D |dz |j z }t || d S )NrK rp r@ rq c < g | ]\ }}t || S rt rX )rZ r\ ru rD rh s r r] z5TestLogspace.test_stop_base_array..Z s>
2
2
2 eU%
8
8
8
2
2
2r r_ )r r r re rv r )r# r` rc rr ri rj rD rh s @@r test_stop_base_arrayz!TestLogspace.test_stop_base_arrayR s aV}}aV}}
eTsD
A
A
A
2
2
2
2
2!$T4
2
2
2(bg%
R r c > t ddd }t |j t d t ddd }t |j t d t ddd }t |j t d d S Nr r@ float32r float64int32)r r r rP s r
test_dtypezTestLogspace.test_dtype` Q + + +QWeI . ./// Q + + +QWeI . ./// Q ) ) )QWeGnn-----r c t d }t d }t t || t dd d S NrU g @)r r r r# abs r test_physical_quantitiesz%TestLogspace.test_physical_quantitiesh D S ! ! S ! !Xa^^Xc3%7%788888r c t d t }t d t }t || }t | t u sJ t |t dd t ||d }t | t u sJ t |t ddd d S NrK rI rU rV )r viewr: r typer r# r r lss r
test_subclasszTestLogspace.test_subclassm !HHMM+ , , !HHMM+ , ,
a^^Bxx,,,,,R#s++,,,
aA
Bxx,,,,,R#sA../////r N)r3 r4 r5 rR rn pytestmark parametrizeintrw r{ r r r r8 r r r> r> + s @ @ @ [ VZZZ00
C
10
[ VZZZ00 10 . . .9 9 9
0 0 0 0 0r r> c J e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d S )
TestGeomspacec t dd }t t | dk t ddd }t |d dk t ddd }t |d dk t ddd
}t |g d t d d
d }t |g d t |j d t ddd }t |g d t |j d t ddd }t |g d t |j d d S )NrK rN rA rB rC rE rF FrG rI rJ rp ry )r rp r )rE r )r r rE )r r rO r r imagrP s r rR zTestGeomspace.test_basicz s] a A" a# & & &" !!! au - - -" a! $ $ $1 > > >??? a " " "999%%%161%%% b$A & & &1ooo...161%%% dBA & & &1ooo...161%%%%%r c ~ d}d}t ||d }t |d | t ||dd }t |d | t ||d }t |d | t |d | t ||dd }t |d | d S )
Ng333333?gL4@rK rC r FrD rH ry rE )r r r# rh rc rQ s r ,test_boundaries_match_start_and_stop_exactlyz:TestGeomspace.test_boundaries_match_start_and_stop_exactly s eTq ) ) )QqT5!!! eTq5 9 9 9QqT5!!! eTq ) ) )QqT5!!!QrUD!!! eTq5 9 9 9QqT5!!!!!r c L t d 5 t ddd }d d d n # 1 swxY w Y t |d d t t |d d
t |d d t d 5 t dddd
}d d d n # 1 swxY w Y t |d d t t |d d d S )Nignore)invalidry r rC r g rK rE g @Fr )r r r r r
allrP s r test_nan_interiorz TestGeomspace.test_nan_interior s
h
'
'
' ( ("aQ'''A ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( QqT4 a"g""$$%%%QqT3
h
'
'
' 8 8"aQ777A 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 QqT4 ae ""#####s 044-C
CCc t ddd }t |g d t |j d t ddd }t |g d t |j d t d
d d }t |g d
t ddd }t |g d t dddt }t |g d t ddd }t |ddt d z dt d z z dg t ddd }t |ddt d z dt d z z dg t ddd }t |ddt d z dt d z z
dg t ddd }t |ddt d z dt d z z dg t d d d }t |g d t dd d }t |dd g t d dd }t |d dg d S )!N ? 0@ rC )r y @y @y @r r @t)r y (y By [r ? ? @@ @@r )r y $@ $@y Y@ Y@r ? @ @@)r y $ $@y Y Y@r rE rK ry rD r )rE r rK y @y r rp y @ y @ @rI )r y9ҿŒ~y{xC @V]yMuU@PrN y0@5Sy
m^>@w@@r )r r r realcomplexr rP s r test_complexzTestGeomspace.test_complex s b#1 % % %000111161%%% c5a ( ( (:::;;;161%%% dJA . . .???@@@ e[a 0 0 0CCCDDD
b!' 2 2 2<<<((( dE1 % %D"T!WW*RQZ"7?@@@ dD! $ $D!DGG)BtAwwJ"6=>>> eT1 % %E2d1gg:baj#8$?@@@ dE1 % %D"T!WW*RQZ"7?@@@ eT1 % % " " " # # #
b"a B8$$$ b"a B8$$$$$r c " t ddd }t |j t d t ddd }t |j t d t ddd }t |j t d t ddt }t |j t d t ddt }t |j t d d S ) NrK rN r~ r r r float_r )r r r r r rP s r r zTestGeomspace.test_dtype s aI . . .QWeI . ./// aI . . .QWeI . ./// aG , , ,QWeGnn---
aE * * *QWeHoo... aG , , ,QWeI . ./////r c t ddgd }t ddgd }t ddgd }t |d
|d d }t |d
|d d }t |d
|d d }t d
dd }t ddd }t ddd } t ||d t ||d t || d d S )Nx rB int8r r uint16r rK r ^@ Y@ ^ Y @rL g{Gz?)rtolgh㈵>)r r r
r# lim1lim2lim3ri rj rk rl rm t6s
r test_start_stop_array_scalarz*TestGeomspace.test_start_stop_array_scalar s c3Zv...dD\000dD\222
tAwQ
+
+
tAwQ
+
+
tAwQ
+
+
ueQ
'
'
vvq
)
)
vvq
)
) BT****BT****BT******r c t g d }t g d t |d }t d t | D d }t || t |d d }t fd|D d }t || t |dd }t ||j d S )
N)rU g @@r r r rE )rM @r r y @ @rK r c 6 g | ]\ }}t ||d S r r rY s r r] z7TestGeomspace.test_start_stop_array.. s8 ; ; ;% feQ// ; ; ;r rK r_ r c > g | ]}t |d d S r r r rb s r r] z7TestGeomspace.test_start_stop_array.. s9 ) ) ) fd1gq11 ) ) )r rE )r r r re r rf rg s @r rn z#TestGeomspace.test_start_stop_array s 44455<<<==
udA
&
&
; ;),UD)9)9; ; ;ABD D DR
ud1gq
)
)
) ) ) )"') ) )/02 2 2R
udAB
/
/
/R r c t d }t d }t t || t dd d S r )r r r r s r r z&TestGeomspace.test_physical_quantities sD S ! ! S ! !Yq!__iS&9&9:::::r c t d t }t d t }t || }t | t u sJ t |t dd t ||d }t | t u sJ t |t ddd d S r )r r r: r r r )r# r r gss r r zTestGeomspace.test_subclass s !HHMM+ , , !HHMM+ , ,
q!__Bxx,,,,,R3,,---
q!Q
Bxx,,,,,R3Q//00000r c t t t dd t t t dd t t t dd d S )Nr r; )r
ValueErrorr r# s r test_boundszTestGeomspace.test_bounds sD j)Q333j)R333j)Q22222r N)
r3 r4 r5 rR r r r r r rn r r r r8 r r r r x s & & &," " "( $ $ $ &% &% &%P 0 0 0 + + +
; ; ;
1 1 13 3 3 3 3r r c n e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd ZdS ) TestLinspacec N t dd }t t | dk t ddd }t |d dk t ddd }t |d dk t t t ddd d S )
Nr r; rA rp rB rC rE FrG )r r rO r r rP s r rR zTestLinspace.test_basic s QOOA" Q $ $ $" QU + + +"
j(Arr::::::r c t t ddd }t |dgk | t t t ddd d S )Nr rK rT g @rC )listr r r TypeErrorrP s r test_cornerzTestLinspace.test_corner' sP !Q"" # #cU
A i1aS999999r c t ddd j }t ddd j }t ddd j }t || t || d S )Nr rK rp )r r r )r# ri rj rk s r test_typezTestLinspace.test_type, sd
aA
$
aA
$
aA
$R R r c > t ddd }t |j t d t ddd }t |j t d t ddd }t |j t d d S r} )r r r rP s r r zTestLinspace.test_dtype3 r r c t ddgd }t ddgd }t ddgd }t |d
|d d }t |d
|d d }t |d
|d d }t d
dd }t ddd }t ddd } t || t || t || d S )Nr rB r r r r r r r r rK r r r r r r rL )r r r r s
r r z)TestLinspace.test_start_stop_array_scalar; s dC[///c4[///dD\222
d1gtAw
*
*
d1gtAw
*
*
d1gtAw
*
*
feQ
'
'
eVQ
'
'
ffa
(
(R R R r c t ddgd }t ddgd t |d }t d t | D d
}t || t |d d }t fd |D d
}t || t |dd
}t ||j d S )Nr r r r rB r r c 6 g | ]\ }}t ||d S r r rY s r r] z6TestLinspace.test_start_stop_array..M r^ r rK r_ r c > g | ]}t |d d S r r rb s r r] z6TestLinspace.test_start_stop_array..Q rd r rE )r r r re r rf rg s @r rn z"TestLinspace.test_start_stop_arrayI s tSk000c4[///
eT1
%
%
; ;),UD)9)9; ; ;ABD D DR
eT!Wa
(
(
) ) ) )"') ) )/02 2 2R
eT12
.
.
.R r c t ddd }t g d }t ddd }t g d }t || t || d S )N ? @ @ @r )r y ? @y @ @y @ @r r r; )r y @ ?y @ ?y @ ?y $@ )r r r )r# r ri r rj s r r zTestLinspace.test_complexW sr **
AAA
B
BB""
EEE
F
FT2 T2 r c t d }t d }t t || t dd d S )NrT rU )r r r r s r r z%TestLinspace.test_physical_quantities_ r r c t d t }t d t }t || }t | t u sJ t |t dd t ||d }t | t u sJ t |t ddd d S )Nr rK rT rU )r r r: r r r r s r r zTestLinspace.test_subclassd r r c G d d } |t dd } |t dd }t t || t dd d S )Nc 4 e Zd ZdZd Zed Zd ZdS )3TestLinspace.test_array_interface..Arrayisha
A generic object that supports the __array_interface__ and hence
can in principle be converted to a numeric scalar, but is not
otherwise recognized as numeric, but also happens to support
multiplication by floats.
Data should be an object that implements the buffer interface,
and contains at least 4 bytes.
c || _ d S r _data)r# datas r __init__z.Arrayish.__init__~ s
!
r c dd| j ddS )Nr8 .Arrayish.__array_interface__ s !#tz#$& & &r c | S r r8 )r# others r r- z;TestLinspace.test_array_interface..Arrayish.__mul__ s r N)r3 r4 r5 __doc__r propertyr r- r8 r r Arrayishr s sW
"
"
"
&
& X
&
r r rK r r r )r r r )r# r onefives r test_array_interfacez!TestLinspace.test_array_interfacen s
0 huQe,,,--xau---..Xc4(((1a..99999r c t d D ]U}t |d |d dz }t t t d|dd| Vd S )Nr r rK r r; F)rH r )r r
r anyr )r# ftyperc s r test_denormal_numbersz"TestLinspace.test_denormal_numbers sv W% M ME UU1XXuuQxx0014D CD"uEJJJKK L L L L M Mr c t d D ]B}t t d||dz t t |dz t Cd S )Nr r rK r )ranger r r r )r# js r test_equivalent_to_arangez&TestLinspace.test_equivalent_to_arange sc t 1 1A !Q!3777 !3///
1
1
1
1 1 1r c dD ]}dD ]}t dd||d }t |t rt | dk sJ |dk r=|rdd gndd
g}t |d | t |d |d |dk r0|s.t |d dg t |d d t |d dgd | t
|d sJ d S ) N)r rK rp )FTr rK T)rH retsteprp rT rU g ?)r r! tuplerO r r r
)r# rD eptrQ y0_expects r test_retstepzTestLinspace.test_retstep s(
'
'C$
'
' Q3dCCC!!U++;A! ;!88.1 Ac
SzI&qtY777 1y|4444AXXcX&qtcU333 1s++++&qtcU4C4[999 1;;&&;&
'
'
'r c t dd }t dd }t ||d }t |t g d d S )NrK Or rp ry )rU g ?r r r r r s r test_objectzTestLinspace.test_object sY as###Qc""" UD! $ $1eMMM 2 233333r c t dddt }t g dt }t || d S )NrE ry r r )rE rE r r rK rK rp ry r )r r r r )r# rQ ts r test_round_negativez TestLinspace.test_round_negative sG R - - - ,,,C 8 8 81a r c t ddg }t ddg }t ||d }t |t ddgddgddgg d S )NrT rU r ry r r s r 'test_any_step_zero_and_not_mult_inplacez4TestLinspace.test_any_step_zero_and_not_mult_inplace sf sCj!!c3Z UD! $ $1ec3Z#sc3Z$H I IJJJJJr N)r3 r4 r5 rR r r r r rn r r r r r
r r r r r r8 r r r r s
; ; ;: : :
. . . 9 9 9
0 0 0 : : :DM M M1 1 1
' ' ' 4 4 4 ! ! !
K K K K Kr r )r numpyr r r r r r r r
r r r
r r
numpy.testingr r r r r r r r: r> r r r8 r r r s
8 8 8 8 8u 8 8 88 J0 J0 J0 J0 J0 J0 J0 J0Za3 a3 a3 a3 a3 a3 a3 a3H`K `K `K `K `K `K `K `K `K `Kr