gR&
@ d dl Z d dlZd dlZd dlZd dlZd dlmc mZ d dl
m Z m Z ej
ej j Z ej
ej j ZdZddz dz
Ze ZdZd ed D d
ed D z d ed d D z d
ed d D z d ed d D z e edd z Zd ed D d ed D z d ed d D z d ed d D z d ed d D z e edd z dgz Zd eD Zej d Z d Z d Z d Z d Z!d Z"d Z#d Z$d Z%d! Z&d" Z'ej( j) d# Z*d$ Z+d% Z,dS )& N)
assert_raises assert_equall l c " g | ] }t |z
S ) INT64_MIN.0js s/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/core/tests/test_extint128.py
r &&&qY]&&& c " g | ] }t |z
S r ) INT64_MAXr s r r r r r c " g | ] }t |z
S r INT64_MIDr s r r r s +++qY]+++r ic ( g | ]}d t z |z S r r r s r r r s ---Qy[1_---r c ( g | ]}t d z |z S r r r s r r r s! ...!Y\A...r iF c " g | ] }t |z
S r )
INT128_MINr s r r r '''Z!^'''r c " g | ] }t |z
S r )
INT128_MAXr s r r r r r c " g | ] }t |z
S r
INT128_MIDr s r r r ! s ,,,Z!^,,,r c ( g | ]}d t z |z S r r" r s r r r " s! ...!Qz\A...r c ( g | ]}t d z |z S r r" r s r r r # s! ///1Z]Q///r Fc g | ]
}|d k | S )r r )r xs r r r ( s 555!q1uuAuuur c ' K dg fd} | V dS # t $ rA ddl}dt d d| }t | w xY w)z}
Iterate over Cartesian product of *args, and if an exception is raised,
add information of the current iterate.
Nc 3 @ K t j D ] } | d< | V d S Nr ) itertoolsproduct)vargsvalues r iteratezexc_iter..iterate4 s< "D) AE!HGGGG r r zAt:
) Exception tracebackrepr
format_excAssertionError)r. r0 r3 msgr/ s ` @r exc_iterr8 + s FE
"gii " " " "58nnnn ' 2 2 4 4 46 S!!! "s
A A(c t j dft j dft j dfg} t | t
t
5 }|D ]\ }}}|\ }} ||| }t |cxk r t k s%n t t t j ||| St j ||| }||k rt || d d d d S # 1 swxY w Y d S )Nr r )
operatoraddsubmulr8 INT64_VALUESr
r r
OverflowErrormtextint_safe_binopr ) opsitxopabpyopopcds r test_safe_binoprL B s9
q q q C
#|\ 2 2 'b
'
'ICAHD"Q
A ////i//// mR-A1aLLLL(Ar2266 A&&&
' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's BCC Cc t t 5 } | D ].\ }t j | }||k rt || / d d d d S # 1 swxY w Y d S N)r8 r? rA
extint_to_128r rD rF rG s r test_to_128rQ Y s , #2 # #BA ##AAvv Q""" ## # # # # # # # # # # # # # # # # #s 2AA Ac 8 t t 5 } | D ]h\ }t |cxk r t k s#n t t
t j | >t
j | }||k rt || i d d d d S # 1 swxY w Y d S rN ) r8
INT128_VALUESr
r r r@ rA extint_to_64r rP s r
test_to_64rU a s - 'B ' 'BA ////i//// mR_a@@@@OA&&66 A&&&
'' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's A,BBBc t t t 5 } | D ]5\ }}||z }t j || }||k rt || 6 d d d d S # 1 swxY w Y d S rN )r8 r? rA extint_mul_64_64r rD rF rG rJ rK s r test_mul_64_64rY l s , - - # # #DAqAA#Aq))AAvv Q""" ## # # # # # # # # # # # # # # # # # 9A""A&)A&c T t t t 5 } | D ]p\ }}||z }t |cxk r t k s$n t t
t j || Et
j || }||k rt || q d d d d S # 1 swxY w Y d S rN ) r8 rS r r r r@ rA extint_add_128r rX s r test_add_128r] u - / / '2 ' 'DAqAA !1111z1111 mR->1EEEE%a++66 A&&& '' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' A4B B!$B!c T t t t 5 } | D ]p\ }}||z
}t |cxk r t k s$n t t
t j || Et
j || }||k rt || q d d d d S # 1 swxY w Y d S rN ) r8 rS r r r r@ rA extint_sub_128r rX s r test_sub_128rb r^ r_ c t t 5 } | D ]1\ }| }t j | }||k rt || 2 d d d d S # 1 swxY w Y d S rN )r8 rS rA extint_neg_128r rD rF rG rJ s r test_neg_128rf s - #B # #BAA!!$$AAvv Q""" ## # # # # # # # # # # # # # # # # #s 5AA A c t t 5 } | D ]S\ }|dk r| dz ddz dz
z }n|dz ddz dz
z }t j | }||k rt || T d d d d S # 1 swxY w Y d S )Nr r r r )r8 rS rA extint_shl_128r re s r test_shl_128ri s - #B # #BA1uuqyQVAX./!V3q)!!$$AAvv Q""" ## # # # # # # # # # # # # # # # # #s AA::A>A>c t t 5 } | D ]A\ }|dk r| dz }n|dz }t j | }||k rt || B d d d d S # 1 swxY w Y d S )Nr r )r8 rS rA extint_shr_128r re s r test_shr_128rl s - #B # #BA1uuraiLF!!$$AAvv Q""" ## # # # # # # # # # # # # # # # # #s AA((A,/A,c t t t 5 } | D ]6\ }}||k }t j || }||k rt || 7 d d d d S # 1 swxY w Y d S rN )r8 rS rA
extint_gt_128r rX s r test_gt_128ro s - / / #2 # #DAqAA A&&AAvv Q""" ## # # # # # # # # # # # # # # # # #s :A##A'*A'c t t t 5 } | D ]\ }}|dk rt || \ }}nt | | \ }}| }| }t j || \ }}||k s||k s ||z |z |k r6t
|| t
|| t
||z |z | d d d d S # 1 swxY w Y d S r* )r8 rS INT64_POS_VALUESdivmodrA extint_divmod_128_64r )rD rF rG rJ crrK drs r test_divmod_128_64rv s9 -!1 2 2 *b
*
*DAqAvv q! 22 r1
2BS+Aq11EArAvvbAaC"HMM Q""" R$$$ QqS2Xq)))
** * * * * * * * * * * * * * * * * *s B#C CCc t t t 5 } | D ]5\ }}||z }t j || }||k rt || 6 d d d d S # 1 swxY w Y d S rN )r8 rS rq rA extint_floordiv_128_64r rX s r test_floordiv_128_64ry s -!1 2 2 #b # #DAqQA)!Q//AAvv Q""" ## # # # # # # # # # # # # # # # # #rZ c t t t 5 } | D ];\ }}||z dz
|z }t j || }||k rt || < d d d d S # 1 swxY w Y d S )Nr )r8 rS rq rA extint_ceildiv_128_64r rX s r test_ceildiv_128_64r| s -!1 2 2 #b # #DAqQq A(A..AAvv Q""" ## # # # # # # # # # # # # # # # # #s ?A((A,/A,)-r+
contextlibr; pytestnumpynp numpy.core._multiarray_testscore_multiarray_testsrA
numpy.testingr r iinfoint64maxr minr
r r r r# rangelistr? rS rq contextmanagerr8 rL rQ rU rY r] rb rf ri rl ro markslowrv ry r| r r r r s\
) ) ) ) ) ) ) ) ) 5 5 5 5 5 5 5 5
BHRX " BHRX " VaZ
[
'&EE"II&&&&&EE"II&&&'++EE#rNN+++, .-eeCnn---. /.uuS"~~... /
DsB
('UU2YY'''''UU2YY'''(,,UU3^^,,,- /.uuS"~~.../ 0/c2/// 0
DsB G
65|555 " " ",' ' '.# # #' ' '# # # ' ' ' ' ' '# # # # # # # # ## # # * * *$# # ## # # # #r