iddddlZddlZddlmZmZmZddlZddlmZGddZ dZ dZ dS)N)assert_assert_array_equal assert_raises)randomceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZejejdjdkddZdZdZdS)TestRegressionctjdddD]j}tj|dd}t tj|tj ko!tj|tjkkdS)Ngg@2)nplinspacervonmisesrallpi)selfmurs c/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/random/tests/test_randomstate_regression.pytest_VonMises_rangez"TestRegression.test_VonMises_rangesv+c2q)) ? ?BAr**A BF1v:&&=26!ru*+=+= > > > > ? ?c ttjtjdddddkttjtjdddddkdg}t jd k}|r%t jd kr|d |D]!}ttj|dk"dS) N sizer)r r win32)r#r#) rr rrhypergeometricsysmaxsizeplatformappend)rargs is_64bitsargs rtest_hypergeometric_rangez(TestRegression.test_hypergeometric_rangesv,QBR@@@1DEEFFFv,RBR@@@1DEEFFF . K%'  ;00 KK9 : : : 5 5C F)3/!3 4 4 4 4 5 5rc6d}tjdtjd|}tj|dk|z }d|dd}t |d k|tj|d k|z }d|dd }t |d k|dS) Nrg?rr zFrequency was fz, should be > 0.45g?z, should be < 0.23gq= ףp?)rseed logseriesr sumr)rNrvsnfreqmsgs rtest_logseries_convergencez)TestRegression.test_logseries_convergence&s  A!,,,vdai  1$9t9999t S!!!vdai  1$9t9999t S!!!!!rcVgdgdgdgdfD]}tjdt|}tj|t j|d|d|d|d gt }tt j|t |dS) N)r r0rN)r r r0r0rrN)r r;r<N)r:r0rN90rrr r0dtype)rr1listshuffler arrayobjectr)rtshuffledexpecteds rtest_shuffle_mixed_dimensionz+TestRegression.test_shuffle_mixed_dimension6s!//000+++&&&( K KA K   AwwH N8 $ $ $x1qtQqT1Q4 8GGGH rx??? J J J J K Krc Dtj}tjgd}t dD]f}tj||dt |ddtjddz |gdS)N) rr0r rrJrrrirg$@)rp) r RandomStater rBranger1rchoiceones)rmresis rtest_call_within_randomstatez+TestRegression.test_call_within_randomstateBs   h55566q N NA KNNN FF4LLL qxxrwr{{3xGG M M M M  N NrctjdgdggdtjdgdggtjdtjdgdggtjddS)Nrr r)rmultivariate_normalr int_int64rs r#test_multivariate_normal_size_typesz2TestRegression.test_multivariate_normal_size_typesLst "A3!A6666"A3!BGAJJ????"A3!BHQKK@@@@@@rctjdtjddd}tt jt j| ddS)NiIg-C6?drzNans in random.beta)rr1betarr anyisnan)rxs rtest_beta_small_parametersz)TestRegression.test_beta_small_parametersTsV  J KS 1 1 1BF28A;;''')>?????rcntjdgd}gd}tjtjtjfD]t}tj||t|z }tj||}t||vtttj||dzudS)N)r r0r)rrr0r>)rLg?) rr1r float16float32float64rBr3rOrr ValueError)racountsdtprobscs r"test_choice_sum_of_probs_tolerancez1TestRegression.test_choice_sum_of_probs_tolerance[s  D II*bj"*4 E EBHV2...Vr.r) rr1r rBarangerCrNrArprqrrs r test_shuffle_of_array_of_objectsz/TestRegression.test_shuffle_of_array_of_objectsws  D HbillBIaLL1 @ @ @t  A N1       rcGddtj}tjdtjd|}tj|}t|tjgdt|tjd|Gdd}tjd|}tj|}t|tjgdt| tjd dS) NceZdZdS)3TestRegression.test_permutation_subclass..NN)__name__ __module__ __qualname__rrr4rzs Drr4r r)rr0r c4eZdZejdZdZdS)3TestRegression.test_permutation_subclass..Mr c|jS)N)rhrYs r __array__z=TestRegression.test_permutation_subclass..M.__array__s v rN)r{r|r}r rvrhrr~rrMrs2 ! A     rr)r0r rrrr ) r ndarrayrr1rvview permutationrrBr)rr4origpermrrQs rtest_permutation_subclassz(TestRegression.test_permutation_subclasssB           Ay||  ##!$''4)))!4!455541!2!21!5!5666          A ACC!!$$4///!:!:;;;1;;==")A,,77777rctjdkrdnd}tjd5t jddd| ddddS#1swxYwYdS) Nbigzi4znon-native byteorder is notmatchrr)rr?)r% byteorderpytestdeprecated_callrrandint)rother_byteord_dts rtest_warns_byteorderz#TestRegression.test_warns_byteorders$'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 DsAAActjd}tjd}|dd|ddksJdS)Ni[)r1rr\)r rrMr)rrs1rs2s r"test_named_argument_initializationz1TestRegression.test_named_argument_initializationsbi##I..i###33{{1c""ckk!S&9&9999999rcFtjddgdzd}|jtjtksJtjddgdzdd}|jtjtksJtjdd}|jtjtksJtjddd}|jtjtksJdS) Nrg?r0)rLrF)rLreplacerr)rr)r rrOr?int)rrls rtest_choice_retun_dtypez&TestRegression.test_choice_retun_dtypes I  RB47  3 3w"(3--'''' I  RB47E  B Bw"(3--'''' I  Ra  ( (w"(3--'''' I  RQ  7 7w"(3--''''''rlr!zCannot test with 32-bit C long)reasonctjdtjgdd}tjdd}t ||dS)Nr) l l/*/lu&-nlLC0lC34lRD7lvlg}1i tll^C?rXr>r!rr)rr1r rBrr)rrFactuals rtest_randint_117zTestRegression.test_randint_117sh  A8555>6?QC";7KKK# % % % % %rN)r{r|r}rr,r8rGrTrZrarmrtrwrrrrrmarkskipifr iinfomaxrrrr~rrrr sf???555 """ K K KNNNAAA@@@ E E E      888,DDD ::: ( ( ( [ )E1?AA--AA-,,,%%%%%rrctjdgjdksJtjdgdjdksJdS)Nr)rr)rJr rr)rJr rr)r multinomialshaper~rrtest_multinomial_emptyrsQ  b" % % +t 3 3 3 3  a) 4 4 4 :l J J J J J Jrctjtd5tjddddddS#1swxYwYdS)Nzpvals must be a 1-drrg333333?)rraises TypeErrorrrr~rrtest_multinomial_1d_pvalrs y(= > > >$$2s###$$$$$$$$$$$$$$$$$$s?AA) r%r numpy.testingrrrnumpyr rrrrr~rrrs ~%~%~%~%~%~%~%~%BKKK$$$$$r