a { d d dl Z d dlZd dlmZmZmZ d dlZd dlmZ G d d Z d Z
d Z dS ) N)assert_assert_array_equal
assert_raises)randomc e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zej ej d j dk d d Zd Zd ZdS )TestRegressionc t j ddd D ]j}t j |dd }t t j |t j k o!t j |t j k kd S )Ng g @ 2 )nplinspacer vonmisesr allpi)selfmurs /builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/random/tests/test_randomstate_regression.pytest_VonMises_rangez"TestRegression.test_VonMises_range sv +c2q)) ? ?BAr**A BF1v:&&=26!ru*+=+= > > > > ? ? c
t t j t j dddd dk t t j t j dddd dk dg}t
j d k }|r%t
j d
k r| d |D ]!}t t j | dk "d S ) N
size r ) r r win32) r# r# ) r r
r r hypergeometricsysmaxsizeplatformappend)r args is_64bitsargs r test_hypergeometric_rangez(TestRegression.test_hypergeometric_range s v,QBR@@@1DEEFFFv,RBR@@@1DEEFFF
.
K%' ; 00 KK9 : : : 5 5C F)3/!3 4 4 4 4 5 5r c 6 d}t j d t j d| }t j |dk |z }d|dd}t |d k | t j |d
k |z }d|dd }t |d k | d S )
N r g?r r zFrequency was fz, should be > 0.45g? z, should be < 0.23gq=
ףp?)r seed logseriesr
sumr )r Nrvsnfreqmsgs r test_logseries_convergencez)TestRegression.test_logseries_convergence& s A !,,, vdai 1$9t9999t S!!!vdai 1$9t9999t S!!!!!r c V g dg dg dg dfD ]}t j d t | }t j | t j |d |d |d |d gt
}t t j |t
| d S ) N)r r0 r N)r r r0 r0 r r N)r r; r< N)r: r0 r N90 r r r r0 dtype)r r1 listshuffler
arrayobjectr )r tshuffledexpecteds r test_shuffle_mixed_dimensionz+TestRegression.test_shuffle_mixed_dimension6 s !//000+++&&&( K KA
K AwwH N8 $ $ $x1qtQqT1Q4 8GGGH rx ? ? ? J J J J K Kr c
D t j }t j g d }t d D ]f}t j | | d t
| ddt j d dz | gd S )N)
r r0 r r rJ r r r i r g $@)r p) r RandomStater
rB ranger1 r choiceones)r mresis r test_call_within_randomstatez+TestRegression.test_call_within_randomstateB s h55566q N NA KNNN
FF4LLL qxxrwr{{3x G G M M M M N Nr c t j dgdggd t j dgdggt j d t j dgdggt j d d S )Nr r r )r multivariate_normalr
int_int64r s r #test_multivariate_normal_size_typesz2TestRegression.test_multivariate_normal_size_typesL st "A3!A6666"A3!BGAJJ????"A3!BHQKK@@@@@@r c t j d t j ddd }t t j t j | d d S )NiIg-C 6?d r zNans in random.beta)r r1 betar r
anyisnan)r xs r test_beta_small_parametersz)TestRegression.test_beta_small_parametersT sV J KS 1 1 1BF28A;;''')>?????r c n t j d g d}g d}t j t j t j fD ]t}t j || t | z }t j || }t ||v t t t j ||dz ud S )N )r r0 r )r r r0 r> )rL g?) r r1 r
float16float32float64rB r3 rO r r
ValueError)r acountsdtprobscs r "test_choice_sum_of_probs_tolerancez1TestRegression.test_choice_sum_of_probs_tolerance[ s D II*bj"*4 E EBHV2...V r. r )
r r1 r
rB arangerC rN rA rp rq rr s r test_shuffle_of_array_of_objectsz/TestRegression.test_shuffle_of_array_of_objectsw s D HbillBIaLL1 @ @ @t A N1
r c G d dt j }t j d t j d | }t j | }t |t j g d t |t j d | G d d }t j d | }t j | }t |t j g d t | t j d d S )
Nc e Zd ZdS )3TestRegression.test_permutation_subclass..NN)__name__
__module__ __qualname__ r r r4 rz s Dr r4 r r )r r0 r c 4 e Zd Z ej d Zd ZdS )3TestRegression.test_permutation_subclass..Mr
c | j S )N)rh rY s r __array__z=TestRegression.test_permutation_subclass..M.__array__ s v
r N)r{ r| r} r
rv rh r r~ r r Mr s2 ! A
r r )r0 r r r r r
)
r
ndarrayr r1 rv view permutationr rB r )r r4 origpermr rQ s r test_permutation_subclassz(TestRegression.test_permutation_subclass sB
Ay|| ##!$''4)))!4!455541 !2!21!5!5666 A
ACC!!$$4///!:!:;;;1;;==")A,,77777r c t j dk rdnd}t j d 5 t j ddd| d d d d S # 1 swxY w Y d S )
Nbigzi4znon-native byteorder is notmatchr r )r r? )r% byteorderpytestdeprecated_callr randint)r other_byteord_dts r test_warns_byteorderz#TestRegression.test_warns_byteorder s $'MU$:$:55
#*G
H
H
H D D N1c2B C C C C D D D D D D D D D D D D D D D D D Ds AAAc t j d }t j d }| dd | dd k sJ d S )Ni[)r1 r r\ )r
r rM r )r rs1rs2s r "test_named_argument_initializationz1TestRegression.test_named_argument_initialization sb i##I..i###33{{1c""ckk!S&9&9999999r c F t j ddgdz d }|j t j t k sJ t j ddgdz dd }|j t j t k sJ t j dd }|j t j t k sJ t j ddd }|j t j t k sJ d S ) Nr g?r0 )rL r F)rL replacer r )r r )r
r rO r? int)r rl s r test_choice_retun_dtypez&TestRegression.test_choice_retun_dtype s I RB47 3 3w"(3--'''' I RB47E B Bw"(3--'''' I Ra ( (w"(3--'''' I RQ 7 7w"(3--''''''r lr! z Cannot test with 32-bit C long)reasonc t j d t j g dd }t j dd }t || d S )Nr )
l
l /*/ l u&-n l LC0 l C34 l RD7 l v l g}1 i tll ^C? rX r> r! r r )r r1 r
rB r r )r rF actuals r test_randint_117z TestRegression.test_randint_117 sh A8 5 5 5 >6?QC";7KKK# % % % % %r N)r{ r| r} r r, r8 rG rT rZ ra rm rt rw r r r r r markskipifr
iinfomaxr r r r~ r r r r
sf ? ? ?5 5 5 " " "
K
K
KN N NA A A@ @ @ E E E
8 8 8,D D D : : : ( ( ( [
)E1 ? A A- -A A-, , ,% % % % %r r c t j dg j dk sJ t j dg d j dk sJ d S )Nr )r r )rJ r
r r )rJ r
r r )r multinomialshaper~ r r test_multinomial_emptyr sQ b" % % +t 3 3 3 3 a) 4 4 4 :l J J J J J Jr c t j t d 5 t j dd d d d d S # 1 swxY w Y d S )Nzpvals must be a 1-dr r g333333?)r raises TypeErrorr r r~ r r test_multinomial_1d_pvalr s y(= > > > $ $2s###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s ?AA) r% r
numpy.testingr r r numpyr
r r r r r~ r r r s
~% ~% ~% ~% ~% ~% ~% ~%BK K K$ $ $ $ $r