k-/I d dl Zd dlZd dlmZ d dlmZ d dlmZ d dl Z d dl
Z d dl m
Z
mZmZmZmZmZ d dlmZ G d d Z G d d
Z G d d Zd
ZdS ) N)BytesIO)path)Path)assert_ assert_equalassert_array_equalassert_array_almost_equal
assert_raisestemppath)picklec e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd ZdS )TestFromrecordsc t j g dg dgd }t |d d t |d j j d t |d j j d t |d j j d
t |d j j d d S )
N)i dbeg333333?) de?zcol1,col2,col3namesr col1icol2U col3f)nprec fromrecordsr itemdtypekinditemsizeselfrs q/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/core/tests/test_records.pytest_fromrecordsz TestFromrecords.test_fromrecords s F 1 1 1>>> B"2
4
4QqTYY[["3444QvY_)3///QvY_)3///QvY_-r222QvY_)3///// c dt fdt fg}t j g | }t |j d dS )z0 Verify fromrecords works with a 0-length input abr! r N)floatr r r r shape)r% r! r& s r' test_fromrecords_0lenz%TestFromrecords.test_fromrecords_0len sF u U|, F r / /QWd#####r) c g dg dg}g dg dg}g dg dg}t j |dt fdt fg }t |d | t |d | t j |ddg
}t |d | t |d | t || d S ) N)) r ) ) ))r7 r6 )r5 r4 )r r3 )r3 r4 r6 )r7 r5 r )r r5 r7 )r6 r4 r3 r+ r, r- r )r r r intr )r% data
expected_a
expected_br1r2s r' test_fromrecords_2dz#TestFromrecords.test_fromrecords_2d s $ $ $ $ $ $
ii+
ii+
V
c3Z#s,D
E
ERWj)))RWj))) V
S#J
7
7RWj)))RWj)))R r) c t j dddd }t |d d d S )Ns abcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgabcdefgzi2,a3,i4r4 bigformatsr0 byteorderr3 )idc s efgidcbar r arrayr r r$ s r' test_method_arrayz!TestFromrecords.test_method_array4 sB FLL):QRWL X XQqTYY[["=>>>>>r) c t j g dd }t |d d d S )N)r3 r+ )r r, )r4 ! c)r5 , d)r6 7 ex)r7 B r ) M gu1,f4,a1rB r3 )r g 6@ brD r$ s r' test_method_array2z"TestFromrecords.test_method_array28 sR FLL 1 1 1:D
F
FQqTYY[[/22222r) c t j g dd }t |dd d d d d S )NrH rU rV r3 r )r5 g F@ drD r$ s r' test_recarray_slicesz$TestFromrecords.test_recarray_slices= s^ FLL 1 1 1:D
F
FQqt!tWQZ__&&88888r) c ~ t j g d }t j g d }t j g d }t j |||gd }t |d d d|d< t |j t j g d d S ) N)r3 r r4 r5 )r+ ddxyz12)皙?r r4 r5 za,b,cr r3 )r r] @" )r rE r
fromarraysr r r+ )r% x1x2x3r& s r' test_recarray_fromarraysz(TestFromrecords.test_recarray_fromarraysB s
Xlll
#
#
X...
/
/
Xnnn
%
% F r2rl' : :QqTYY[[.1111QS"(<<<0011111r) c t j t j t d }t j |d }t |d }| d t j |ddd }| d t j |ddd }| d t }| | | d t j |ddd }|
t || t || d S )
Nr9 zrecarray_from_file.fitsrbi f8,i4,a5r4 r@ rA r )r joindirname__file__openseekr r fromfilerE r writereadcloser )r% data_dirfilenamefdr< r= bytes_arrayr3s r' test_recarray_fromfilez&TestFromrecords.test_recarray_fromfileK s2 9T\( 3 3V<<9X'@AA
(D
!
!
V__R1_
N
N
V\\"jU\
K
K
ii "''))$$$
V__[*AQV_
W
W
R R r) c
| d}t j |d }t j |d }t j |d }t t | D ]"}t t dd ||< #t j |||gd }t t | D ]w}t |j | t t dd k t |j | dk t |j
| dk xd S ) N
Or- f8r3 zdate,data1,data2r g ) r zerosrangelenlistr rc r datedata1data2)r% countr+ r, rL r mines r' test_recarray_from_objz&TestFromrecords.test_recarray_from_obj\ s HU# & & & HU$ ' ' ' HU$ ' ' 's1vv & &Aa %%AaDDv !Q2D EEs1vv , ,A TYq\T%2,,%7%77 9 9 9 TZ]c) + + + TZ]c) + + + + , ,r) c P t j ddgddg }t j | }t t | t j d t j t j dd }t t t j | d t j t j d
d }t t |
d d
t j t j d
d }t t |
d d
k d S )N)r3 g?)r g?)foo>>
FLL'222 3 3T!WW\\.112666 FLL$/// 0 0Q ]++r122222r) c t j dd }t t | t j d |d }t t | d t j d t t | d t j d
d S # t j d
w xY w) N)r3 ra 2003z.CN)__name__
__module__ __qualname__r r) r' Cr s Dr) r ))r+ f4,f4)r, V8)rL )f4r )rN i8r r3 r3 s 11111111r3 r r )r r r r )r rE r8 int64r r r! r r r r r r+ r, voidrL float32rN int32fieldsr ) r% r+ r, r& rvr rL
test_dtypearrsarrr arr2s r' test_recarray_viewsz#TestFromrecords.test_recarray_views s Hi, "CL-8
:
:
: H[[[ 1 1 1 RV\\!__*/;;;T"&,,q//**BK888RV\\!__*/:::T"&,,q//**BK888 QVVBK((.3RY???T!&&--.. <<<QVVBK((.3RX>>>T!&&--.. <<<
FLL'222 3 3
VVD\\
w
'
'T"XXr{+++RX]BI...
FLL +; +; +; < < <
=
=QsV\& 222T!C&\\2;///
FF1IIT!C&\\1%%%. . .
FLL5+!ua8 +!ua8:AK
M
MQSY^RY///QSY^RW---QSY^RZ000QSY^RX... FLL'222 3 3QVVG__*/;;;QVVH%%+0"(;;;QVVD\\',bg666QVVO,,27BBB )))271D+A+A+AB 0 0C&,,s##C88CI , 9 2:FFD #). 9 9 9 dT#YY / / / / 0 0r) c t j g dd }t j g dd }t |j |j k t |j |j k t t | D ]E}t || || k Fd S )N))r3 abcg
@r )r xyg @r3 )r g ?r zc1, c2, c3, c4r )
r r rE r r r! r0 r r r )r% rapaks r' test_recarray_from_namesz(TestFromrecords.test_recarray_from_names s
V\\ . . . . / /
V
!. !. !. . / /
BH$%%%BH$%%%s2ww 2 2A BqEJJLLBqEJJLL0 1 1 1 1 2 2r) c t j g dd }g d|_ t |d g d t t
|j t
|j u d|_ t |j dk g d|_ t |d g dg d |_
t |d
g d g t t |j
t j
j d S ) N))r3 r gffffff@)r r^ g@)r4 wrsr zfield, shape, meanr )r` g@gffffff
@meanr3 r4 )ABr r0 r6 field)r6 r6 r6 )r r rE r r r r varr0 r r
isinstance collectionsr Callable)r% r s r' test_recarray_conflict_fieldsz-TestFromrecords.test_recarray_conflict_fields s
V\\ ) ) ) 1 3 3 "//!"V*ooo>>>RW
bf -...F"###"??2g; (9:::2g; 444
28[_%=>>?????r) c t j ddgdt fdt fg }t |j ddg t |d j d t |j dd g t |d
j d t j dt fdt fg }t j ddg| }t |j ddg t |d j d t |j dd g t |d
j d d S ) N)r3 r+ )r bbbr+ r, r- r3 r r r r ) r r r r8 objectr r+ r, r! )r% r+ ndtypes r' $test_fromrecords_with_explicit_dtypez4TestFromrecords.test_fromrecords_with_explicit_dtype s( F * 5(+SzC=&A
C
CQS1a&!!!QqTVQ QS3,'''QrUWe$$$C:V}566 F * 5V D DQS1a&!!!QqTVQ QS3,'''QrUWe$$$$$r) c t j ddgddt fg }| t j }t |j d |j d k d d S ) N)abc r3 )r r r r r r- r r3 F)r rE r8 r r r r r s r' test_recarray_stringtypesz)TestFromrecords.test_recarray_stringtypes se Hk:. )E3<8
:
:
:
FF2; QU1Xq)511111r) c * t j d dft j d dfd}t j ddgddd t fd
t fgfd t fd |fg
}t t
|j t j t t
|d t j t t
|j t j
t t
|d t j
t |j j j t j t t
|d t j
t |j j j t j t t |j j j | t t
|j t j t t
|d t j t t
|d j t j t t
|d d t j t |d j j d t |d j d d t |d d j d t |d d d d t |d j j d t |d j d d t |d d j d t |d d d d d S )NS5r r7 )r D)r r r3 )abcdefgehi)r )r r4 r3 )r jklmnr r r r bazquxr- r r3 s fgehir )r r! r rE r8 r r r r r r r r dictr r r r )r%
qux_fieldsr+ s r' test_recarray_returntypesz)TestFromrecords.test_recarray_returntypes s HTNNA.bhtnna5HII
FLL@?A -!&#sc3Z(@ A!& uj.A C
D
D
T!%[["*---T!E(^^RZ000T!%[["+...T!E(^^R[111QU[%ry111T!E(^^R[111QU[%ry111T!%+,--z:::T!%[["*---T!E(^^RZ000T!A$(^^RY///T!A$u+&& 222QqTXZ###QqTXc]A&&&QqT%[]A&&&QqT%[%q)))QqTXZ***QqTXc]H---QqT%[]H---QqT%[%x00000r) c dgdz dgdz g}t j | }t |d g d t |d g d t j dd g }t j ||
}t |j d t |d g d t |d g d
d S )Ntestr4 f0)r r r f1)r r r )r z|S4)r z|Sr- r5 ) testr r )r) r) r) )r r rc r r! r# )r% colsr dts r' test_zero_width_stringsz'TestFromrecords.test_zero_width_strings4 s 1 rdQh'f %%SY 8 8 8999SY ---
X}l3
4
4f B //S\1%%%SY ; ; ;<<<SY00000r) N)r r r r( r1 r> rF rX r[ rg ry r r r r r r r r r r r r r) r' r r s5 0 0 0$ $ $ (? ? ?3 3 3
9 9 9
2 2 2 " , , , 3 3 3*
.
.
. / / /.<0 <0 <0|2 2 2
@
@
@
%
%
% 2 2 21 1 18 1 1 1 1 1r) r c e Zd Zd ZdS )
TestPathUsagec
t d 5 }t | }t j d t j d d }d|d< |j d 5 }| | d d d n # 1 swxY w Y t j j
|dd }t || d d d d S # 1 swxY w Y d S )
Nz.bin)suffix{ r{ rj )g ?r{ r r6 wb)rB r0 )
r r r randomseedrandastypern tofilecorerecordsrp r )r% r r+ rv xs r' test_tofile_fromfilez"TestPathUsage.test_tofile_fromfileE sf
V
$
$
$
%::D INN3 r""))*55A#AaD4
B
((1;/1 ) 3 3A
q! $ $ $
%
%
%
%
%
%
%
%
%
%
%
%
%
%
%
%
%
%s6 A6C8B) C8)B- -C80B- 1:C88C<?C<N)r r r r
r r) r' r r C s# % % % % %r) r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zej dg d d Zej dg d d Zd ZdS )
TestRecordc Z t j ddgg d | _ d S )Nr3 r r4 )r5 r6 r7 )r r r r )r r r- )r r r r9 )r% s r' setup_methodzTestRecord.setup_methodT s: F&& 9'=#4 #4 #4 ' 5 5 r) c | j }t |j d d d|d _ t |j d d d S Nr r3 r9 r r r s r' test_assignment1zTestRecord.test_assignment1Z sF IQVAY"""! QVAY"""""r) c | j }t |j d d d|j d< t |j d d d S r r r s r' test_assignment2zTestRecord.test_assignment2` sF IQVAY"""q QVAY"""""r) c F | j }d }t t || d S )Nc d| d _ d S )Nr3 r )col5)r s r' assign_invalid_columnzATestRecord.test_invalid_assignment..assign_invalid_columni s AaDIIIr) )r9 r
AttributeError)r% r+ r s r' test_invalid_assignmentz"TestRecord.test_invalid_assignmentf s3 I n&;Q?????r) c j t j ddgdg }d|j _ t t 5 ddg|_ d d d n # 1 swxY w Y t t 5 |j ddgg|j j
d R d d d d S # 1 swxY w Y d S ) Nr. )r3 )r r r- Fr r4 r ) r r rE flags writeabler
ValueErrorr setfieldr! r r$ s r' test_nonwriteable_setfieldz%TestRecord.test_nonwriteable_setfieldn s0 FLL$ k]L ; ; !
:
&
& a&AC
:
&
& 4 4 AJ!u 3qw~c 2 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s$
AA "A :!B((B,/B,c | j ddg }t |j j d t |j j g d | j ddg t j j d t t fd d S )Nr r )r r )r r )r z|V4)r r c j j S )N)r! descr)ys r' z5TestRecord.test_out_of_order_fields.. s !'- r) )r9 r r! r r& r
r! )r% r r' s @r' test_out_of_order_fieldsz#TestRecord.test_out_of_order_fieldsw s Ivv& 'QW]$4555QW]DDD F F F
Ivv& 'QW]$4555j"7"7"7"788888r) c
t j dg fgdt j fdt j dfg }t dt j dz D ]z}t
|t j t j || t
|d t j t j |d | {d S )Nr3 r+ r, r r- r protocol) r rE r r r HIGHEST_PROTOCOLr loadsdumpsr% r+ protos r'
test_pickle_1zTestRecord.test_pickle_1 s Hq"gYRXbh8J&K L L L1f 5 9:: K KE FL a%)H)H)H I I J J J 1v|FL1BG-I -I -I J J
K
K
K
K K Kr) c
D | j }t dt j dz D ]z}t |t j t j || t |d t j t j |d | {d S )Nr r3 r+ r )r9 r r r- r r. r/ r0 s r'
test_pickle_2zTestRecord.test_pickle_2 s I1f 5 9:: K KE FL a%)H)H)H I I J J J 1v|FL1BG-I -I -I J J
K
K
K
K K Kr) c x | j }t dt j dz D ]}t j t j |d | }t
|j j t
|j j t
|j j
t
|j j d S )Nr r3 r r+ ) r9 r r r- r. r/ r r c_contiguous f_contiguousr aligned)r% r+ r1 r s r'
test_pickle_3zTestRecord.test_pickle_3 s I1f 5 9:: & &E fl1Q4% @ @ @AAB BH) * * * BH) * * * BH& ' ' ' BH$ % % % % & &r) c t j ddg }t j d| }t d f}||d< d|d< |d \ }}|t j j j u sJ |\ }}t |t rJ t t ||d
t j
|d }t j | } |d | k sJ t j t" 5 |t j d | |u sJ d d d d S # 1 swxY w Y d S ) N)objr| )r8 r r3 r- s emanr; * r8 r
r| )r r! empty bytearray
__reduce__r
multiarrayscalarr bytesr
RuntimeErrorr r/ r. pytestwarnsDeprecationWarning)
r% r r+ r9 ctorargsr! r; dump unpickleds
r' test_pickle_voidzTestRecord.test_pickle_void s
X|\2
3
3 HQb ! ! !'""$%%qT__&&
drw)00000
s c5)))))lD%444 |AaD!!L&& ty \,
-
- 5 54
t,,44444 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s #D77D;>D;c > t j ddg }t j d| t j }t j g d |_ t j ddt fdt fd t fg }|ddg d S )
N)r r )r r| r r- r )r r r' z)
r r! r~ r r rE r r r/ r8 )r% r r& r s r' test_objview_recordz TestRecord.test_objview_record s
X}l3
4
4 HU" % % % * *2; 7 7## [sFmc5\C:%N
O
O
O
C9
r) c t j ddt dfg }d|d _ t |d j t j d d S )Nr3 r r6 r- r )r r r/ r r r )r% r s r' test_record_scalar_setitemz%TestRecord.test_record_scalar_setitem sN k!S%O#4555ASVXrwqzz*****r) c t j ddt fdt fg t t fd d S )N)r4 r r' r- c dg S )Nnofieldr )r s r' r( z/TestRecord.test_missing_field.. s YK(8 r) )r r~ r8 r
KeyError)r% r s @r' test_missing_fieldz TestRecord.test_missing_field sD htS#Jc
#;<<<h 8 8 8 899999r) c t j d t j dddg g}t j | }d S )Nr{ )r+ z Test that nested structured types are treated as records too r+ r, rL Ninnerr4 r )r r! r] r~ r r r r )r% r[ r dt_outerr9 data0s r' test_nested_fields_are_recordsz)TestRecord.test_nested_fields_are_records s XRXbh#rx I(7(S
T
T8gr]O,,x8$$))"+66$ ,,,,,$w-55555Q%+++++%.")4444444r) c t j dt j fdt j fdt j fg }|ddg }|j |j k sJ t j d|fg }t j d| t j }t |d j | |d }t |d j | dS )z) test that trailing padding is preserved r+ r, rL r` r4 r N)r r! r] r# r~ r r r )r% r
dt_padded_endra r9 rb s r' test_nested_dtype_paddingz$TestRecord.test_nested_dtype_padding s XRXbh#rx I
J
JC:
%44448g} 5 677x8$$))"+66T'](-888QU7^)=99999r) N)r r r r r r r r# r) r2 r4 r9 rL rO rQ rV rZ rE mark parametrizer^ rc rf r r) r' r r S sb 5 5 5 # # # # # # @ @ @4 4 4 9 9 9K K KK K K& & &5 5 52 + + + : : :
( ( ( [ Y 22
,
, 32
, [ Y 22 5 5 32 5
:
:
:
:
:r) r c g d} t t j | g k g d}t t j | dgk g d}t t j | ddgk g d}t t j | ddgk d S )N)r3 r r4 r5 r6 r7 )r3 r r3 r5 r6 r7 r3 )r3 r r3 r5 r3 r7 r r4 r )r r r3 r5 r3 r7 r r4 )r r r find_duplicate)l1l2l3s r' test_find_duplicatern s B BF ! !" % % +,,, B BF ! !" % %! ,--- ! ! !B BF ! !" % %!Q /000 ! ! !B BF ! !" % %!Q /00000r) )collections.abcr r ior osr pathlibr rE r r
numpy.testingr r r r r
r numpy.compatr r r r rn r r) r' ru sl
p1 p1 p1 p1 p1 p1 p1 p1f
%
%
%
%
%
%
%
% g: g: g: g: g: g: g: g:T 1 1 1 1 1r)