5Z+2 P d dl mZmZ d dlZd dlZd dlmZmZ G d d Z dS ) )assert_assert_array_equalN) GeneratorMT19937c b e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd ZdS )TestRegressionc H t t d | _ d S )Nl (vn4Ti )r r mt19937selfs /builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/random/tests/test_generator_mt19937_regressions.py setup_methodzTestRegression.setup_method s )>!?!?@@ c t j ddd D ]p}| j |dd }t t j |t j k o!t j |t j k qd S )Ng g @ 2 )nplinspacer
vonmisesr allpi)r murs r
test_vonmises_rangez"TestRegression.test_vonmises_range s| +c2q)) ? ?B %%b!R00A BF1v:&&=26!ru*+=+= > > > > ? ?r c
N t t j | j dddd dk t t j | j dddd dk d}t | j j | dk d S ) N
size r ) r$ r$ )r r r r
hypergeometric)r argss r
test_hypergeometric_rangez(TestRegression.test_hypergeometric_range s t|221b"22FFJKKLLLt|222q"22FFJKKLLL 1+ +T2Q677777r c d}| j d| }t j |dk |z }d|dd}t |dk | t j |d k |z }d|dd
}t |d k | d S ) N g?r! r zFrequency was fz, should be > 0.45g? z, should be < 0.23gq=
ףp?)r
logseriesr sumr )r Nrvsnfreqmsgs r
test_logseries_convergencez)TestRegression.test_logseries_convergence s |%%c%22 vdai 1$9t9999t S!!!vdai 1$9t9999t S!!!!!r c g dg dg dg dfD ]}t t d }t j |t }| | t j |d |d |d |d
gt }t
t j |t | d S ) N)r r+ r N)r r r+ r+ r r N)r r5 r6 N)r4 r+ r Ni90 dtyper+ r r r )r r r arrayobjectshuffler )r tr
shuffledexpecteds r
test_shuffle_mixed_dimensionz+TestRegression.test_shuffle_mixed_dimension+ s !//000+++&&&( K KA //Gx000H OOH % % %x1qtQqT1Q4 8GGGH rx ? ? ? J J J J K Kr c
< t j g d }t d D ]u}t t | }t t d }t | ddt j d dz | vd S )N)
r r r r r r rA r r# r i r g $@)r" p)r r9 ranger r r choiceones)r resir
ms r
test_call_within_randomstatez+TestRegression.test_call_within_randomstate7 s h55566q N NA
++G'$--((A qxxrwr{{3x G G M M M M N Nr c | j dgdggd | j dgdggt j d | 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_types@ s
((!se!(<<< ((!se"'!**(EEE ((!se"(1++(FFFFFr c | j ddd }t t j t j | d d S )Ng-C 6?d r! zNans in mt19937.beta)r
betar r anyisnan)r xs r
test_beta_small_parametersz)TestRegression.test_beta_small_parametersH sL
L ff3 7 7BF28A;;''')?@@@@@r c g d}g d}t j t j t j fD ]}t j || t | z }| j || }t ||v t j
t 5 | j ||dz d d d n # 1 swxY w Y d S )N)r r+ r )r# r# r+ r7 )rB g?) r float16float32float64r9 r- r
rD r pytestraises
ValueError)r acountsdtprobscs r
"test_choice_sum_of_probs_tolerancez1TestRegression.test_choice_sum_of_probs_toleranceN s
II*bj"*4 4 4BHV2...V.NN)__name__
__module__ __qualname__ r r
r. ro x s Dr r. r r )r+ r r c 4 e Zd Z ej d Zd ZdS )3TestRegression.test_permutation_subclass..Mr c | j S )N)r] r s r
__array__z=TestRegression.test_permutation_subclass..M.__array__ s v
r N)rp rq rr r rk r] rw rs r r
Mru s2 ! A
r rx )r# r r r r+ r )
r ndarrayr r rk view permutationr r9 rw )r r. r
origpermrx rH s r
test_permutation_subclassz(TestRegression.test_permutation_subclassv sT
GAJJ''y|| ##""4((4)))!4!455541 !2!21!5!5666 GAJJ''
ACC""1%%4///!:!:;;;1;;==")A,,77777r c 4 | j d dk sJ t | j dg d | j dgd }t j dgt j }t || d S )Ng floatr7 )r
standard_gammar r r9 rX )r actualr> s r
test_gamma_0zTestRegression.test_gamma_0 s |**3//366664<66u==sCCC ,,cU',BB8RD
33368,,,,,r c t | j dd t j t j j d S )NgKH9r )rB r" )r r
geometricr iinforM maxr s r
test_geometric_tiny_probz'TestRegression.test_geometric_tiny_prob sI 4<11E1BB 8BH--1 3 3 3 3 3r N)rp rq rr r r r' r2 r? rI rN rU rb ri rl r~ r r rs r r
r r s A A A? ? ?8 8 8
"
"
"
K
K
KN N NG G GA A A 4 4 4 8 8 8.- - -3 3 3 3 3r r )
numpy.testingr r numpyr rZ numpy.randomr r r rs r r
r s 7 7 7 7 7 7 7 7
+ + + + + + + +S3 S3 S3 S3 S3 S3 S3 S3 S3 S3r