1j d Z ddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z dZ ej d j ej d j k r*e dd d
d Zd eD Zd Zd
Z G d dej Zd Z e Z G d d Z G d d ZdS )z
The tests exercise the casting machinery in a more low-level manner.
The reason is mostly to test a new implementation of the casting machinery.
Unlike most tests in NumPy, these are closer to unit-tests rather
than integration tests.
N)
as_strided)assert_array_equal)_get_castingimplz?bhilqBHILQefdFDlq Lc P g | ]#}t t j | $S typenpdtype.0cs {/builddir/build/BUILD/cloudlinux-venv-1.0.11/venv/lib64/python3.11/site-packages/numpy/core/tests/test_casting_unittests.py
r s( :::qbhqkk""::: c # K t D ]u} | }t j |t | V |j dk r9| }t j |t | V vd S )N)id|)
simple_dtypespytestparamstr byteorder newbyteorder) dtype_classdts r simple_dtype_instancesr! s $ / /
[]]l2#b''******
<3 ""B,rc"gg... . . . / /r c | | t j k rdS | j dv r\| j dk rd}n<| j dk rd}n.| j dk rd}n | j dk rd }nt d
| | j d k r|dz
}|S | j d k rd
S | j dk rdS | j dk rdS | j dk rdS t d
| )zHReturns the string length when casting the basic dtypes to strings.
iu
z!did not find expected length for ig0 G` f r @ )r bool_kinditemsizeAssertionErrorchar)r lengths r get_expected_stringlengthr: ' s
q zT >Q FF
^q
FF
^q
FF
^q
FF !LU!L!LMM M : aKF
zSr s v s r s v
DUDD
E
EEr c " e Zd ZdZdZdZdZdZdS )Castingr r% r' r& r( N)__name__
__module__ __qualname__noequivsafe same_kindunsafer r r r< r< J s'
B
E DI
FFFr r< c t j d d } d | d dd d D }t j t j t j t j dd}i }t || dd D ]:\ }}i ||< t ||dd d D ]\ }}|| || |< ;|S ) Na
X ? b h i l q B H I L Q e f d g F D G S U V O M m
? # = = = = = = = = = = = = = = = = = = = = = . =
b . # = = = = . . . . . = = = = = = = = = = = . =
h . ~ # = = = . . . . . ~ = = = = = = = = = = . =
i . ~ ~ # = = . . . . . ~ ~ = = ~ = = = = = = . =
l . ~ ~ ~ # # . . . . . ~ ~ = = ~ = = = = = = . =
q . ~ ~ ~ # # . . . . . ~ ~ = = ~ = = = = = = . =
B . ~ = = = = # = = = = = = = = = = = = = = = . =
H . ~ ~ = = = ~ # = = = ~ = = = = = = = = = = . =
I . ~ ~ ~ = = ~ ~ # = = ~ ~ = = ~ = = = = = = . =
L . ~ ~ ~ ~ ~ ~ ~ ~ # # ~ ~ = = ~ = = = = = = . ~
Q . ~ ~ ~ ~ ~ ~ ~ ~ # # ~ ~ = = ~ = = = = = = . ~
e . . . . . . . . . . . # = = = = = = = = = = . .
f . . . . . . . . . . . ~ # = = = = = = = = = . .
d . . . . . . . . . . . ~ ~ # = ~ = = = = = = . .
g . . . . . . . . . . . ~ ~ ~ # ~ ~ = = = = = . .
F . . . . . . . . . . . . . . . # = = = = = = . .
D . . . . . . . . . . . . . . . ~ # = = = = = . .
G . . . . . . . . . . . . . . . ~ ~ # = = = = . .
S . . . . . . . . . . . . . . . . . . # = = = . .
U . . . . . . . . . . . . . . . . . . . # = = . .
V . . . . . . . . . . . . . . . . . . . . # = . .
O . . . . . . . . . . . . . . . . . . . . = # . .
M . . . . . . . . . . . . . . . . . . . . = = # .
m . . . . . . . . . . . . . . . . . . . . = = . #
c P g | ]#}t t j | $S r r r s r r z&_get_cancast_table..n s(
8
8
8Ad28A;;
8
8
8r r r' ).~=# r% )
textwrapdedentstripsplitr< rD rC rB rA zip)tabledtypes convert_castcancastfrom_dtrowto_dtr s r _get_cancast_tablerZ R s O
4 UWWUU4[[5
6 9
8q!$Q$
8
8
8F g.? GM L GFE!""I.. 6 6 FC1I.. 6 6HE1&21oGG U # # 6 Nr c e Zd ZdZej dddg ej dg d d Zd Zd S )
TestChangesz:
These test cases exercise some behaviour changes
stringSUfloating)er1 dr- c h t j || sJ t j || d sJ d S )N100r can_cast)selfr` r] s r test_float_to_stringz TestChanges.test_float_to_string s@ {8V,,,,,{8^^^4444444r c t j dd sJ t j dd sJ t j dd rJ t j dd rJ t j dd rJ t j ddd sJ t j d dd
sJ t j d dd
sJ d S ) Nrb VS20V1U1zd,irC castingV3r@ V0re rg s r test_to_voidzTestChanges.test_to_void s {3$$$$${5#&&&&& ;sD)))));ud+++++;tT*****{5#{;;;;;;{4d333333{4d33333333r N) r= r> r? __doc__r mark parametrizerh rs r r r r\ r\ s [ XSz22 [ Z)=)=)=>>5 5 ?> 325
4
4
4
4
4r r\ c e Zd ZdZd ZdZdZej de d Z
ej d ej d e d Z
ej de d
Zej g d d d
ej dddf e ej d d
ej d
ddfddej dddfddej d
ddfddej dddfd dej d
ddfdd ej d
ddfddej d
ddfddej d
d
g dfdd
ej dddf e ej d d
ej d
ddfddej dddfddej d
ddfd d ej dddfddej d
ddfddej d
ddfdd ej d
ddfd d ej d
d
g d fg d! Zd" Zej d#e ej d$d%d&g d' Zej d$d%d&g ej d( e d) Z ej d(g d* ej d$d%d&g d+ Z ej d,d-d.g ej d/d-d.g d0 Z d1 Z d2 Z ej d3 e d4 Z!ej d3 e d5 Z"ej d6d7d8g d9 Z#ej d:d;g ej dd?gdd@gdA dBf ej d=dd?gdd@gdA dBf ej d=dd?gddCgdA d
fg dD Z$ej dEdFdGdHdI e%d. s% EEE!&"4--EEEr r ur% ?c , g | ]}t | S r boolr vs r r z(TestCasting.get_data.. s 222a$q''222r c , g | ]}t | S r r r s r r z(TestCasting.get_data.. s ...!d1gg...r )
r6 sizer emptyflags c_contiguousalignedrange enumerater5 iinfomaxr8 ) rg dtype1dtype2r9 arr1valuesr, valuearr2s r get_datazTestCasting.get_data s >V_ ? ?Y&/1FFY&/1F xf---z&&&&z!!!!EEuV}}EEE!&)) HAuqyyV[C / / 0 0 44q8DGG >{c!!226222< ;# ..v...Fxf---z&&&&z!!!!!&)) HAuqyyV[C / / 0 0 44q8DGGT6!!r Tc |r|j j }|j j }n?|r d|j j z }d|j j z }n |j j dz }|j j dz }t | dz |j j z dz }t | dz |j j z dz }t j |t j } t j |t j }
|t | z | j k sJ |t | z |
j k sJ |rot | dd |j |j |f } t |
dd |j |j |f } nnt | dd |j |j |f } t |
dd |j |j |f } || d< |sU|r |j j nd}
| |
d|
}| d t | k sJ |r | j j
sJ | j j
sJ n | j j
rJ | j j
rJ |r | j j sJ | j j sJ n<| j j r| j j dk sJ | j j r| j j dk sJ | | fS )
z
Returns a copy of arr1 that may be non-contiguous or unaligned, and a
matching array for arr2 (although not a copy).
r' r% r& rz NrH .r )r r6 lenr zerosuint8nbytesr viewshapetobytescountr r r alignment)rg r r r contigstride1stride2 max_size1 max_size2
from_bytesto_bytesnew1new2offsetbufs r get_data_variationz TestCasting.get_data_variation s
.j)Gj)GG
.$*--G$*--GGj)A-Gj)A-GIIMDJ$77!; IIMDJ$77!; Xirx888
8IRX666 T"j&77777T"ho5555 6 j"o 2 24: > > "j7*6 6D hssm 0 0 < < "j7*6 6DD jn 1 1$* = = "j7*6 6D hqrrl / /
; ; "j7*6 6D S 0,3:TZ((F V_W_-5577C99U##s3xx//// /:* * **:* * ** *z. . ..z. . .. G:% % %%:% % %% %z) FTZ-AQ-F-F-FFz) FTZ-AQ-F-F-FFTzr from_Dtc t D ]n}t || } | | fD ]3}| |d f d d }| | k sJ ~ | | fD ]}| ||f \ }\ }} }
t | |k sJ t | |k sJ |
H|t
j k sJ t
j t | | k sJ |j | j k sJ n1||k r|j | j k sJ |t | | k sJ ||u r ||u sJ || u sJ 5pd S )Nr% )
r get_castingimplr _resolve_descriptorsr
r< r@ rA
CAST_TABLEisnative) rg r to_DtcastrW defaultrY ro from_resto_resview_offs r test_simple_cancastz TestCasting.test_simple_cancast s " 0 0E"7E22D #GIIwwyy'='='?'?@
0
0 33WdODDQGJ %%'')))) #egguuww';';'='= > 0 0E 5 5w6F G G :G /h >>W4444 <<50000 + ''* 4 4 4 4 &}
70CE0J J J J J ' 0FO C C C C C"e++#+#4#G#G#G#G &*W*=e*D D D D D %'' &( 2 2 2 2 $+0
0 0 0r z ignore::numpy.ComplexWarningrW c t D ].}|j d }t t | t | }| ||f \ }\ }}}||us||ur dS |t
j k }~~~| || \ } }
} | | |
f |
| k sJ |
| |
dd \ } }
| | |
f t |
|
|
|
k sJ |j
dk r |j
dk s|j s dS |
| |
dd \ } }
| | |
f t |
|
|
|
k sJ |
| |
dd \ } }
| | |
f t |
|
|
|
k sJ ~ ~
~0dS )a
This test checks numeric direct casts for dtypes supported also by the
struct module (plus complex). It tries to be test a wide range of
inputs, but skips over possibly undefined behaviour (e.g. int rollover).
Longdouble and CLongdouble are tested, but only using double precision.
If this test creates issues, it should possibly just be simplified
or even removed (checking whether unaligned/non-contiguous casts give
the same results is useful, though).
r NTFr% )r! r r r
r r< rB r _simple_strided_calltolistr r r r _supports_unaligned)rg rW rY r ro r r r rB r r r arr1_oarr2_os r test_simple_direct_castsz$TestCasting.test_simple_direct_casts# sh ,-- / % / %ELOE"4==$u++>>D484M4M% 5" 5" 1G'h w&&&*=*= gl*D&'!% w!>!> D$ % %tTl 3 3 3 ;;==F**** " 4 4T4u M MNFF % %vv&6 7 7 7 vt , , ,>>##t||~~5555 "a''EOq,@,@ 0 -A ! 4 4T4 M MNFF % %vv&6 7 7 7 vt , , ,>>##t||~~5555 ! 4 4T4 N NNFF % %vv&6 7 7 7 vt , , ,>>##t||~~5555_/ % / %r c " | }t j d t j d t j d t j d t j d t j d g}|D ] }t t | t | }| ||f \ }\ }}} ||u sJ ||u sJ ~~|t
| t | z sJ | J t j t j }
| ||
\ } } }
| | } t j d | d< |t j d k red | d
< |
| | f t j t 5 t | d
d d d n # 1 swxY w Y d S |
| | f d | D |
k sJ | | | d d
\ }}|
||f t# || | | k sJ d S )NM8M8[ms]M8[4D]m8m8[ms]m8[4D]NaT.r rH c , g | ]}t | S r )intr s r r z5TestCasting.test_numeric_to_times.. s 222qCFF222r TF)r r r r
r r int64r r
datetime64r r raises
ValueErrorr r r r r )rg r rW time_dtypestime_dtr ro r r r int64_dtr r r r r s r test_numeric_to_timesz!TestCasting.test_numeric_to_timesa s '))x~~rx'9'928H;M;Mx~~rx'9'928H;M;MO " % 6 % 6G"4==$w--@@D484M4M'"5$ 5$ 1G'h w&&&&W$$$$&Z 0g ?? ? ??###x))H!% w!A!A D$99W%%D
e,,DI"(4..(( R ))4,777]:.. " "RMMM" " " " " " " " " " " " " " " % %tTl 3 3 322DKKMM222f<<<< " 4 4T4u M MNFF % %vv&6 7 7 7 vt , , ,>>##t||~~55555K% 6 % 6s 4GG G )rW rY expected_castingexpected_view_offnomdenomzM8[ns]Nr r% r r zM8[5ms]i@B zM8[7ms] r zM8[1M]) r rH " !zm8[ns]r r zm8[5ms]zm8[7ms]r zm8[1M])r r r r r r c t j | }|t j | }t j g d }| t j d |j }|j j |j k sJ t j | | d sJ t | }t || } |
||f \ }
\ } } }
| |u sJ | |u s|J |
|k sJ |
|k sJ |!||z |z | }d|d< n.t j | }||d< | | }| | }t j | }|
t j
k rV|dk s|dk rJt j t 5 | ||f d d d n # 1 swxY w Y d S dD ]t}dD ]o}| |||| \ }}d|d< | ||f t' | d | d pud S ) N)r r% l i' il r r r .r r TT)r r arrayastyper r isnatr r
r r
empty_liker< rD r r r r r r )rg rW rY r r r r r DTyper ro r r r expected_outorig_arrorig_outr r arrouts r test_time_to_timez TestCasting.test_time_to_time s @ (7## HUOOE DDDEE rx 0 0 = =g>O P PQQ|%):::::x G,,Q/00000W
ue,,040I0I% 1" 1"-#(FX7""""%--/*****,,,,, ? "SLE 177??L #LOO =00L %L ',,U33L;;w''= .. gn $ $%4--5D== z**
@
@))8X*>???
@
@
@
@
@
@
@
@
@
@
@
@
@
@
@ F# R RG&
R
R 22 (GV = =S C))3*555"388G#4#4l6G6G6P6PQQQQ
R R Rs G44G8;G8c |j dk rdnd}|j |z |z }t j |j |j | S )Nr^ r% r( )r8 r6 r r r )rg r
change_lengthfactr9 s r string_with_modified_lengthz'TestCasting.string_with_modified_length sL J#%%qq1 4'-7x5?@EJ@@@AAAr other_DT string_charr^ r_ c t |dk rdnd}t t j | }t || } | }t | }t j | | }| |d f \ } \ }
} } | j ||z k sJ | t j k sJ | J t | | sJ dD ]g}
|
dk r
t j }n t j
}| ||
}| ||f \ } \ }} } | |u sJ | |k sJ | J ht || }| ||f \ } }} | t j k sJ | J t || }| |d f \ } \ }} } | t j k sJ | J || u sJ d S )Nr^ r% r( rH r r% r )
r
r r r r: r r6 r< rB
isinstancerC r rD )rg r r r string_DTr other_dtexpected_length string_dtsafety res_other_dtres_dtr r expected_safetyrY r s r test_string_cancastz TestCasting.test_string_cancast s8 3&&qqA+..// x338::3H==H >_ > >?? 373L3L4 4" 4"0& v/D"88888 %%%% &),,,,, ( $ $M !!"),")"344Y
NNE,0,E,E u%-' -' )FKQU????_,,,,#### y(33 " 7 7H8M N N8 '''' y(33(,(A(A
) ) % FX '''' 6!!!!!!r r c 4 t t j | }t t | | }t |t | }| |df \ }\ }}}||ur|j |j k sJ dS | |d \ } }
t j t | | } | |d } t j t | | }
| |d }t j t | | }|j
rJ |j
rJ dD ]B}| | | d| \ }} | | |
d| \ }}
| | |d| \ }}|
|| f |
||
f t | | |
|
||f t | | |j dk rd|d < |
| |f t | | d|d < |
||f t | | DdS )
ah
Tests casts from and to string by checking the roundtripping property.
The test also covers some string to string casts (but not all).
If this test creates issues, it should possibly just be simplified
or even removed (checking whether unaligned/non-contiguous casts give
the same results is useful, though).
Nrz rH r% )TFTbr .)r
r r r r r r r r r r r copyr r r r5 )rg r r r r cast_backr r r r r str_arrstring_dt_short
str_arr_shortstring_dt_long str_arr_longr other_arrs r "test_simple_string_casts_roundtripz.TestCasting.test_simple_string_casts_roundtrip s +..// tH~~y99#ItH~~>> *.*C*C4 +" +"' $L)a x ' '% )????? F ==488&(3x== ::: ::9bIIX oFFF
99)QGGxH
^DDD ++++ 0000# 4 4F!%!8!8'4"1 "1 Iw # 6 6-,,..f > > A} " 5 5,f 6 6OA|
% %y'&: ; ; ; % %y-&@ A A A w~~o > >
N N N % %y,&? @ @ @ w 5 5 5}## IcN * *GY+? @ @ @ x 3 3 3 IcN * *L)+D E E E x 3 3 3 35 4 4r )S8zU8c t j | }|dk rdnd}|j dk rdnd}t t j | }t t | | }|j |z }t j | | }| |d f \ } \ }
} } | j ||z k sJ t | | sJ d }
|j |k r#|j rt j
}d}
n,t j }n |dk r
t j }n t j
}| |
k sJ || k sJ dD ]}| || }| ||f \ } \ }} } | |u sJ |dk r | |
k sJ n| J |t j
k r | |k sJ h|dk r| t j k sJ |dk r | |k sJ |dk r| t j k sJ d S )Nr^ r% r( r r_ r )r r r8 r
r r6 r r r r< r@ rA rB rD r rC )rg r r r divr r r r r r r r r r r rY r s r test_string_to_string_cancastz)TestCasting.test_string_to_string_cancastP sl 8H%% 3&&qqA=C''aaQ+..// tH~~y99"+s2H >_ > >?? 373L3L4 4" 4"0& v/D"88888&),,,,, =K ' '
0")*$%!!")-
C
%lOO %nO,,,,, &(((( ' . .M44Y
NNE,0,E,E u%-' -' )FKQ U???? !! #444444 ''' '.00 00000 "" !222222 !## 00000 "" ----# . .r order1><order2c B t j | d }t j | d }t j dt j dd | }t j dt j dd | }|j dk r |j j rJ |j j rJ d}||d< || fD ]5}||d< |d |k sJ | d |k sJ 6d S )NU30y rz r% u this is a ünicode string‽r .) r r r r r r r r r ) rg r r r r data1data2elementdatas r test_unicode_byteswapped_castz)TestCasting.test_unicode_byteswapped_cast s+
V...))V...))28444QRR8==fEE28444QRR8==fEE q {* * ** {* * **0b EJJLL) / /D E#J9''''::<<#w..... / /r c t j g d d j j dk sJ t j g d d j j dk sJ d S )NV5rz r^ r# r_ r+ )r r r r r6 rr s r test_void_to_string_special_casez,TestCasting.test_void_to_string_special_case sr x$'''..s339BaGGGGx$'''..s339BeKKKKKKr c x t t j t }t t j t }t || }t
j t d 5 | t j d d f d d d d S # 1 swxY w Y d S )Nz+casting from object to the parametric DType)matchO)
r
r r objectr r r r TypeErrorr )rg object_dtype other_dtyper s r (test_object_to_parametric_internal_errorz4TestCasting.test_object_to_parametric_internal_error s BHV , ,-- 28C==)) |[99
]9GI I I = = % %rx}}d&; < < < = = = = = = = = = = = = = = = = = =s 8*B//B36B3r c t t j t }t |t | }| t j d |f \ }\ }}}|t j k sJ |J ||u sJ | t j d d f \ }\ }}}|t j k sJ |J || d k sJ d S )Nr rK ) r
r r r r r r< rD r rg r r r r r r r s r !test_object_and_simple_resolutionz-TestCasting.test_object_and_simple_resolution s BHV , ,-- |T%[[99(,(A(A# &)( )(% FX '''' (,(A(A# %)' )'% FX '''' ++C00000000r c $ t t j t }t t | | }| |d f \ }\ }}}|t j k sJ |J |t j d u sJ d S )Nr )r
r r r r r r< rB r$ s r test_simple_to_object_resolutionz,TestCasting.test_simple_to_object_resolution s BHV , ,-- tE{{L99(,(A(A
) ) % FX %%%% # &&&&&&r ro r@ rD c t j dg }|dk }t j d|| |k sJ t j |d| |k sJ d S )N)foozC; ?DD
AllIntegerAllFloatc t j g dt j t }| | }g d}t
|| d S )N)r r& irz )r r% r% )r r int8r r r r )rg r nonstandard_boolsresr+ s r test_nonstandard_bool_to_otherz*TestCasting.test_nonstandard_bool_to_other) s_ HZZZrw ? ? ? D DT J J &&u-- 993)))))r r )*r= r> r? r r r r ru rv r r filterwarningsr! r r r< r@ r r r r rA rB rD rC r r r r r r r r" r% r' r, r7 dictr= typecodesrD rK r r r rx rx sR D'" '" '"R9 9 9 9v [ Y
66 0 0 76 0B [ >?? [ Y(>(>(@(@AA:% :% BA @?:%x [ Y
66,6 ,6 76,6\ [
gj!Q
2c("(8$$113344d-q!-HglAq!
4gndAq
9GJ1a
8'"3T1e
D',eQ
? 7#4dAq
A'"3T4999; gj!Q
2c("(8$$113344d-q!-HglAq!
4gndAq
9GJ1a
8'"3T1e
D',eQ
? 7#4dAq
A'.$888:/
; < <:.R .R; < <:.R`B B B
[ Z 77 [ ]S#J77," ," 87 87,"\ [ ]S#J77 [ Z)?)?)A)ABB:4 :4 CB 87:4x [ Z)=)=)=>> [ ]S#J771. 1. 87 ?>1.f [ XSz22 [ XSz22/ / 32 32/(L L L= = = [ W&<&<&>&>??1 1 @?1" [ W&<&<&>&>?? ' ' @? ' [ Yx(899E E :9E [ g~ 6bh#st $%q6+ + , ,-.0 bh#st $%q6+ + , ,-.0 bh#st $%q6+ + , ,-13
4 5 5( ( 5 5( [ A # $$cUSEA3??? F
Tuqc
:
:
:C C $$c3Z#saVLLL
#
"= D
@ ( (A @ ( [ Wbl5&9::B B ;:B" [ W L & j)A AC C* *C C* * *r rx )rt r rN enumr} r@ numpyr numpy.lib.stride_tricksr
numpy.testingr numpy.core._multiarray_umathr r r r r6 replacer! r: IntEnumr< rZ r r\ rx r r r rV s
. . . . . . , , , , , , L L L L L L #
28C==XRXc]] 333!))#r22::3CCM::M:::
/ / / F F FF dl ( ( (T
!
!
4 4 4 4 4 4 4 46X
* X
* X
* X
* X
* X
* X
* X
* X
* X
*r