bgY ,ddlZddlZddlmZddlZddlmZmZm Z m Z m Z m Z dZ ejdddgdZejd ejd ejgd ejd d d fdZdZejdddgdZejdgddZdZejdejdejdzejdzdZejddgejejfejejfejdejdfgd Zejddgejejfejejfejdejdfgd!Z ejd"e!e"ejej#gejd#ej$ej%ej&gd$Z'ejdej(d%d&Z)d'Z*d(Z+d)Z,d*Z-d+Z.d,Z/ejgd-d.d/d0gejd/d0gd/d0ggfd1d2d3gejd2d3gd2d3gd2d3ggfgd4Z0d5Z1dS)6N)rational)assert_ assert_equalassert_array_equal assert_raises assert_warns HAS_REFCOUNTctt}tjdtj}t|}t tj|tj|trJtj |}tj|t |tj |t tjdtjtjtj tjtrVtj |}tjdtjt |tj |t tjdtjtjdtjtrtj tj}tjtjdtjtjt |tj tjtj tdf}tj tdf}tj tdf}t tjdtjtjdtjt tjdj |t tjdtj |t tjd|tjd t tjd |tjd|tj tdf}tj tdf}tj tdf} t tjd tjtjdtjt tjd j |t tjd tj |t tjd |tjtd t tjd | tjd| ttdt} t!t#| dtjt%dtj} t | gdt!| j jtjutjt%d} t | gdt!| j jtjutjdtj} tdtft)| j} t tj| tj| tjddgdt,fdt.fdtfg} tdtft)| j} t ttj| jt| jtdtft)d} t tj| tjtjdtjd}t5tjD]}|g}tj|t9t:tj|gtjt tjdgdztjtjdtj tjt tjdggdztjtjd tj tjt tjdgdzgtjtjd!tj tjt tjdgdzgdztjtjd"tj tjt tjdgdztjtjdtj tjt tjdggdztjtjd tj tjt tjdgdzgtjtjd!tj tjt tjdgdzgdztjtjd"tj tjt tjdgdztjtjdtjt tjdggdztjtjd tjt tjdgdzgtjtjd!tjt tjdgdzgdztjtjd"tjt tjd#tjtjdtj tjt tjd$gdztjtjd tj tjt tjd#gtjtjd!tj tjt tjd#gdztjtjd"tj tjt tjd%tjtjdtjt tjd&gdztjtjd tjt tjd%gtjtjd!tjt tjd%gdztjtjd"tjdS)'N)r dtype?s1.0s1.1z1.0z1.1__dict__get)gH@gG@gH@)1.0Y@o)__array_interface__)r @Hello)r@Worldf0f1f2)__array_struct__cBtjdtjS)Nrr )nparrayfloat64)xs l/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/core/tests/test_api.pyz"test_array_array..ZsrxRZ'H'H'H) __array__? )r/)r/r )r r/)r/r/) NNNNNNNNNNN) rrrrrrrrrr)r) typeobjectr&onesr(rr'r sys getrefcountnanr bytesstrgetattr __builtins__rhasattr memoryviewuint8dictrintfloatr$datarangeMAXDIMSr ValueErrorfloat32full)tobjones11tndarray old_refcountS2S3S5U2U3U5builtinsdatarnestedis r*test_array_arrayrV sl <x00  \3?8#<#<===$bj111"& 333555:t,,  RZ((((\3?4#8#8999#RZ0002:...000@rz22  #RZ000 CCCC\3?2:#>#>??? 5!*  B 5!*  B 5!*  B& 3332:...000&!!',,,&...4b999&+++RXe__===$b)))272R+@+@+@AAA 3(  B 3(  B 3(  B%rz2222:...000%&+++%s+++12666%r***BHSYY,?,?@@@#R((("'"B*?*?*?@@@|Z>>H GHe $ $%%% (:f%%RZ 8 8 8C((())) CINbj ())) (:f%% & &Clll### CINbh &''' bj)))A S6) a&; < < < > >A!2:...222 #%67s dE]T3K@ B B BA S6) 1#5 6 6 6 8 8Arx{{'((%--888 KS6) HH I I I K K M MA!2:... 0K0KLLLF 2:  HV*bh CCCC4&2+RZ888bj999;;;D6(R-rz:::"& ;;;===D6B;-rz:::"& ;;;===D6B;-",BJ???26<<<>>>4&2+RZ888bj999;;;D6(R-rz:::"& ;;;===D6B;-rz:::"& ;;;===D6B;-",BJ???26<<<>>>3%"*BJ777bj111333C5'B,bj999 333555C52:,bj999 333555C52:,+2:>>>444666,bj999bj999;;;7)b. ;;;"& ;;;===<. ;;;"& ;;;===<.2-RZ@@@26<<<>>>+RZ888bj1113336(R-rz::: 333555;-rz::: 333555;-",BJ???44466666r,r'TFctdd}|rtj|}tt5tj|dddddS#1swxYwYdS)Nr rM8r )rr&r'r TypeError)r'rts r*test_array_impossible_castsr[s !QB  Xb\\ y ! !!! 4    !!!!!!!!!!!!!!!!!!sAA#&A#rSr)rrrrctj5tj|}t ||j|jjsJ ddddS#1swxYwYdSr0)pytestdeprecated_callr&fastCopyAndTransposerTflagsowndata)rSbs r*test_fastCopyAndTransposerfs   ! !  #A & &Qws8AA!AcJtjdddd}|d}t ||t |jtjdt |j|j|jd}t |j|t |jtjdt |jj|j|d}t ||t||u|dd}t||u|dd d }t ||t||ut|j j |dd d }t ||t||ut|j j |d d }t ||t |jtjd tt|jdd |ddd}t||uGddtj}tjgdgdgd|}|ddd}t||u|dd}t ||t t#|||ddd}t ||t||utt#||utjdgd}|d}t ||t |jtjdtjdgd}|d}t ||t |jtjdtjdgd}|d}t ||t |jtjdtjdgd}|d}t ||t |jtjd tjd!dd}t%|tjd"d#tjd!dd}t%|tjd$d%tjd!gdd}t%|tjd"d#tjd!gdd}t%|tjd$d%tjd!d}t%|tjd"d#tjd!d}t%|tjd$d%tjd&d}tj|d'd(}t|jdktjd)gd}tt|jd*d tjd)d}tt|jd+d tt|jdS),Nr]f4r rri4FcopyF)orderrkCc8safecastingr)subokrkceZdZdS)$test_array_astype..MyNDArrayN)__name__ __module__ __qualname__rr,r* MyNDArrayrus r,ry)rr r)rrTsdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaOSS100daaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaUU100s aaaaaaaaaaS10 aaaaaaaaaaU10l ~l!]cs123456789012345678901234567890S30123456789012345678901234567890U30uaŀuint32)bufferr shapeiS1U1)r&arangereshapeastyperr stridesrbrrc f_contiguous c_contiguousrrYndarrayr'viewr1rsize)rSrerys r*test_array_astypers !4   ((A..A AA"(4..)))AI&&&  4Aa"(4..)))ai((( AA a E""A AFOOO Su--AA a AG !!! Su--AA AFOOO AG !!! v&&AA"(4..))))QXtV<<<< QU++A AFOOO     BJ    )))YYY't44499)DDA T..A AFOOO E""AAa)$$$ U//AA a DGG9 $%%% (3'''A  AA"(6**+++ '#&&&A  AA"(6**+++ '#&&&A  AA"(5//*** &%%%A  AA"(5//*** /s;;;BB3GGAq"(#4EBBBCCC /s;;;BB3GGAq"(#35AAABBB 01===DDSIIAq"(#4EBBBCCC 01===DDSIIAq"(#35AAABBB /s;;;Aq"(#4EBBBCCC /s;;;Aq"(#35AAABBB #&&&A !81555A AFaK $t$$$A)QXtV<<<< T"""A)QXtV<<<<)QX&&&&&r,dtr|rcRtjdgt}||jjtj|djksJtj||dsJtj||drJtjd|dsJdS)Nr runsaferq same_kindr{)r&r'r2rr itemsizecan_castrarrs r*+test_array_astype_to_string_discovery_emptyr"s (B4v & & &C ::b>>  (BHXXX,>,>,G G G G G ;sB 1 1 111 1{3K88888 8 ;sB 1 1 111 111r,)dfS13U32ctj|}tjg|}|djj|jksJdS)Nr V)r&r r'rrrs r*test_array_astype_to_voidr0sL "B (2R C ::c?? )R[ 8 8 8 8 8 8r,crtjgdd}|jdksJdS)Nr{r rV8)r&r'rr )rs r* test_object_array_astype_to_voidr6s? (2S ! ! ! ( ( - -C 9      r,tuintr?r@ctjdtj}ttj|j|dS)Nr/r )r&r'complex_rComplexWarningr)rrSs r*test_array_astype_warningr<s7 2;'''A"AHa00000r,r out_dtypezS10,S9z?,?ctjgd|}tjgd|}t|||dS)z Currently, for `astype` strings are cast to booleans effectively by calling `bool(int(string)`. This is not consistent (see gh-9875) and will eventually be deprecated. )10z10z00r )TTFFN)r&r'rr)r rrexpecteds r*test_string_to_boolean_castrDs\ (3335 A A ACx222)DDDHszz),,h77777r,cdD]Z}tj|g|}tt5||dddn #1swxYwY[dS)zn These currently error out, since cast to integers fails, but should not error out in the future. )FalseTruerz non-emptyr N)r&r'rrDr)r rinvalidrs r*"test_string_to_boolean_cast_errorsrRs<""hy... : & & " " JJy ! ! ! " " " " " " " " " " " " " " """sAA A str_type scalar_typecl|d}||dksJtj|gt|ddksJtj||ddksJtjd|}||d<|ddksJdS)Ns1+3jy?@r rr r)r&r'r2rzeros)rrvaluers r*test_string_to_complex_castr`s K E ;u   % % % % 8UG6 * * * 1 1+ > >r Bd J J J J 8E?? ! !+ . .r 2d : : : : (1K ( ( (C CF q6T>>>>>>r,AllFloatctjd|}d|d<tj|dsJtjtjd|dsJtjtjdg|dsJtjtjd|dsJdS)Nr r rr)r&risnanr'r)r rs r*test_none_to_nan_castrls (1E " " "C CF 8C==   8BHT/// 0 0 444 4 8BHdV5111 2 21 555 5 8BHTNN))%00 1 1" 555 555r,c,tjdddd}tj|dt |dtj|jdt |dtjgdgd gd }tj|d | t |gd gdgtjddgddgddggd }tj|jd| t |gdgdgdS)Nr]rhr rrr.@)rr r)rrr ? @where)rrr)rrrrr @)rrr)rrr)r&rrcopytorrbr')rSmasks r*test_copyto_fromscalarrws( !4   ((A..AIaCIac3C 8YYY *# 6 6 6DIaD!!!!___ooo6777 8aVaVaV,C 8 8 8DIac3d####___ooo677777r,ctjdddd}tj|gdgdgt |gdgdgtj|ddddf|dddd ddft |gd gd gt t tj|d tj|d d t |d tj|dgdt |gdgdgt t tj|dgdtj|dd gdgdgt |gdgdgtj|ddddf|dddd ddfdd gd d ggt |gdgdgt t tjgdgddS)Nr]rir rr)rr r)r]rr r )rr]rr rr r.rrq)TFTr)rr rrr)rr r )r rr)rrr)rrzrz)rzr rr)rzrrr rr)rrrz)r&rrrrrrY)rSs r* test_copytors- !4   ((A..AIa)))YYY'(((YYY *+++Ia2A2h$$B$2+'''YYY *+++)RY3///Iah''''AIa---....YYY *+++)RY36I6I6IJJJJIahyyy))).DEEEEYYY *+++Ia2A2h$$B$2+1v1v.>????YYY *+++)RY 999=====r,c d d}dg|zgdz}tjt||z }tjt||z }tj||d}tj|||dddd}tj|}t d|zD];tj|}fdt |D}tj|}tj|||t|d k|t| t|tj|}tj|||dddt|d k|dddt| t|tj|}tj|ddd|ddd|dddt|dddd k|dddt|ddd t|dddtj|}tj|ddd|ddd|ddd t|dddd k|ddd t|ddd t|ddd d D]}tj|}tj|}tj| tj }|||d k<tj|||t|d k|t| t|=tj|}tj||dt| |j tj|}tj|}tj||d t| |j dS)NiT)TTTTrr rc"g|] }|zdk S)rr).0r)rUs r* z&test_copyto_permut..s! 0 0 0aa!e\ 0 0 0r,r )rrF) r&rlenr3r'rrBrrsumrr=r) padlrrrpowercimaskrUs @r*test_copyto_permutrs C  ////A QA As A 8A;;stt DIa$ttt*%%%% E A 1e8__** HUOO 0 0 0 05<< 0 0 0x{{ !Qd####161%%%QUUWWc!ff%%% HUOO !Qd44R4j))))161TTrT7+++QUUWWc!ff%%% HUOO !CCaC&!CCaC&SSqS 22221SSqS6Q;##A#///QsssVZZ\\3q1v;;/// HUOO !CCaC&!CCaC&TTrT 33331SSqS6Q;$$B$000QsssVZZ\\3q2w<<000) * *AA8A;;DHQKK$$RX..E E$!)  Ia$ ' ' ' ' qAvq ) ) ) #a&& ) ) ) ) * AIa$!&!!! A AIa%    !&!!!!!r,ctjddddd}|d}tjddddddd}dfd }t |jjt |jj t |jj t |jjt |jj t |jj |d ||d dd |d ||d dd |d ||d dd tj|d ||d dd tj|d ||d dd tj|d ||d dd |d||dd d |d||dd d |d||dd d tj|d||dd d tj|d||dd d tj|d||dd d |d ||d dd |d ||dd d |d ||ddd tj|d ||d dd tj|d ||dd d tj|d ||ddd dS)Nrr rrzrlrmFct||ut||tjj|tjj|dSr0)rrrcrr)r)yccontigfcontigrress r*check_copy_resultz*test_copy_order..check_copy_resultsUQ!VQSY+W555SY+W55555r,rnT)rrrK)F) r&rrrkswapaxesrrcrr)rSrerrrs @r*test_copy_orderrsi " aAq))A SA " aAq))221a88A666666 AG !!! $ $%%% $ $%%% AG !!! $ $%%% $ $%%% &&s&  Cc1dE4HHHH &&s&  Cc1dE5IIII &&s&  Cc1dE5IIII '!3   Cc1dE4HHHH '!3   Cc1dE5IIII '!3   Cc1dE5IIII &&s&  Cc1eT5IIII &&s&  Cc1eT4HHHH &&s&  Cc1eT5IIII '!3   Cc1eT5IIII '!3   Cc1eT4HHHH '!3   Cc1eT5IIII &&s&  Cc1dE4HHHH &&s&  Cc1eT4HHHH &&s&  Cc1eUDIIII '!3   Cc1dE4HHHH '!3   Cc1eT4HHHH '!3   Cc1eUDIIIIIIr,c @tjddddddddf}|jdddz|_tjddd}d}||dd||dd|tjd d d |tjd gdgggd d d |tjdd d|tjdd dd |tj|ddd|tj|ddd d|tj|ddd dd ||dd d ||ddddddfd d ||ddd ||ddddddddddfd d ||d d |tjdd d dS)N)rzrzr r)i)rrr rrrrzczt|jj|kt|jj|kdSr0)rrcrr)rSrrs r* check_contigz+test_contiguous_flags..check_contig#s8$/000$/00000r,F)rrrrrTr rlrrrrjrn)rkrm)ndminrkrmr.)rr.r)r&r3rremptyr'ravelsqueeze)rSrers r*test_contiguous_flagsrst  33Q3qqq7#A "1" 'AI   ))!Q//A111 LE5!!!LE5!!!L/**D$777LQC!:,c222D$???L&!!4///L&,,,eT:::L!%(((%777L!%s333T5AAAL!15<<tjdgd}tjgdd}tj||}t|dtjgddt|dtjgdddS)Nrzu4,u4,u4r )r)rzrr])r\rr)rrrr )r&r'broadcast_arraysr)rSreresults r*test_broadcast_arraysr>s )J///A 222*EEEA  A & &FBH%F%F%FjYYYZZZBH%F%F%FjYYYZZZZZr,)r fill_valueexpected_outputrg@r)rrrg@cPtj||}t||dSr0)r&rFr)rrroutputs r*test_full_from_listrFs+WUJ ' 'F)))))r,ctjdtj}|tjd}tj||rJ|tjtjj}tj||rJ|tjd}tj||sJ|tjtjj}tj||sJ|tjtjj}tj||sJ|tj d}t|||tj tjj}t||tt|jtj tjjdS)Nr/r TrjF) r&rintprmay_share_memory _CopyModeALWAYS IF_NEEDEDNEVERr(rrrD)rres_true res_always res_false res_if_needed res_nevers r*test_astype_copyflagrMs )Bbg & & &Czz"'z--H"31111 1BG",*=>>J"3 3333 3 27 //I sI . ... .JJrwR\-CJDDM sM 2 222 2 27); <r!s //////  J6J6J6X4-00!!10!  ! Q""n'n'n'`Sz** 2 2+* 27778899989  JvE**RZ-@@111 ';/ )RX  '28  "(8  hbhuoo . 01188 118';/ )RX  '28  "(8  hbhuoo . 011"" 11"c5"'2;%GHH r}bn57777IH",z":;;66<;6888"">">">H3"3"3"j;J;J;Jz;;;@[[[CCC 3+xrx#sc3Z(@AA B 3+xrx#sc3Z#(MNN O QRR**RR*+++++r,