B )rrrrtest_unsupported_typesz!TestRandint.test_unsupported_typecCsx|jD]}|tjkrdn t|j}|tjkr4dnt|jd}tt|j|d||dtt|j||d|dtt|j|||dtt|jdd|dqWdS)Nrr"r!)r*) ityperbool_iinfominmaxrrrS)rdtlbndubndrrrtest_bounds_checkings z TestRandint.test_bounds_checkingcCsx|jD]}|tjkrdn t|j}|tjkr4dnt|jd}|d}t|j||dd|d||}t|j||dd|d|||d}t|j||dd|d|qWdS)Nrr"r!i)rKr*)rUrrVrWrXrYr rS)rrZr[r\tgtrrrtest_rng_zero_and_extremess  z&TestRandint.test_rng_zero_and_extremesc Csx|jD]}|tjkrdn t|j}|tjkr4dnt|jd}y|j|||dWqtk r}ztdt |Wdd}~XYqXqWdS)Nrr"r!)r*zMNo error should have been raised, but one was with the following message: %s) rUrrVrWrXrYrS ExceptionAssertionErrorstr)rrZr[r\errrtest_full_ranges zTestRandint.test_full_rangecCstjxV|jddD]D}x>dD]6}|jd|d|d}t||kt|dkq$WqW|jdddtjd}t|dkt|dkdS)Nr!)rFr"i)rKr*r) rrrArUrSrrYrXrV)rrZr\Zvalsrrrtest_in_bounds_fuzzs  zTestRandint.test_in_bounds_fuzzc Csddl}dddddddddd }x|jddD]r}tjd tjd kr^|jdd d |d }n|jdd d |d }| | tj  }t |t|j|kq0Wtjd |jddd tjd  tj }| | }t |ttjj|kdS)NrZ 7dd3170d7aa461d201a65f8bcf3944b0Z 1b7741b80964bb190c50d541dca1cac1Z 4dc9fcc2b395577ebb51793e58ed1a05Z 17db902806f448331b5a758d7d2ee672Z 27dd30c4e08a797063dffac2490b0be6) boolint16int32int64int8uint16r;uint64uint8r!ilittlei)rKr*r")hashlibrUrrrAsys byteorderrSZbyteswapZmd5viewrlZ hexdigestrr*namerh)rrrr^rZvalZresrrrtest_repeatabilitys*   zTestRandint.test_repeatabilitycCs`tj}ttjj}tttjj}tttjjd}tjj|||d}t||dS)Nr!)r*)rrkrWrYrnrrr )rrZr^r[r\actualrrrtest_int64_uint64_corner_cases z)TestRandint.test_int64_uint64_corner_casecCsxj|jD]`}|tjkrdn t|j}|tjkr4dnt|jd}|j|||d}||jt|qWxtj tj tj fD]l}|tj krdn t|j}|tj krdnt|jd}|j|||d}| t |d|t||q~WdS)Nrr"r!)r*r*)rUrrVrWrXrYrS assertEqualr*rhr)ZlongZ assertFalsehasattrtype)rrZr[r\samplerrrtest_respect_dtype_singletons z(TestRandint.test_respect_dtype_singletonN)r%r&r'rrrrSrVrlrorirmrjr;rkrnrUrTr]r_rdrgrxrzrrrrrrRs !rRc@s$eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=d>Z!d?d@Z"dAdBZ#dCdDZ$dEdFZ%dGdHZ&dIdJZ'dKdLZ(dMdNZ)dOdPZ*dQdRZ+dSdTZ,dUdVZ-dWdXZ.dYdZZ/d[d\Z0d]d^Z1d_d`Z2dadbZ3dcddZ4dedfZ5dgdhZ6didjZ7dkdlZ8dmdnZ9dodpZ:dqdrZ;dsdtZdydzZ?d{d|Z@d}d~ZAddZBddZCddZDddZEdS)TestRandomDistcCs d|_dS)NiI)rA)rrrrrEszTestRandomDist.setUpcCsJtj|jtjdd}tddgddgddgg}t||d d dS) NrJr"g_\*?g?g<p?g,o?gH?gZ?)decimal)rrrAZrandr=r )rrydesiredrrr test_rands   zTestRandomDist.test_randcCsJtj|jtjdd}tddgddgddgg}t||d d dS) NrJr"g!}|t}tjtdjtdj}tt |dkWdQRXtdj}t ||dS)Nlr!) rrrrrrrWrYrrr )rrrryrrrrtest_random_integers_max_int<s  z+TestRandomDist.test_random_integers_max_intc Cs^tLtdttttjjtdj tttjjtdj tdj WdQRXdS)Nerrorr) warningscatch_warnings simplefilterrrrrrrWrY)rrrrtest_random_integers_deprecatedKs  z.TestRandomDist.test_random_integers_deprecatedcCsHtj|jtjd}tddgddgddgg}t||dd dS) N)rJr"g_\*?g?g<p?g,o?gH?gZ?r)r)rrrAZ random_sampler=r )rryrrrrtest_random_sampleYs    z!TestRandomDist.test_random_samplecCs<tj|jtjdd}tddddg}t||dS)Nrer"rJ)rrrAchoicer=r )rryrrrrtest_choice_uniform_replaceasz*TestRandomDist.test_choice_uniform_replacecCsHtj|jtjjddddddgd}tddddg}t||dS)Nreg?g?)r-r!r")rrrArr=r )rryrrrrtest_choice_nonuniform_replacegsz-TestRandomDist.test_choice_nonuniform_replacecCs>tj|jtjjdddd}tdddg}t||dS)NrerJF)replacerr!)rrrArr=r )rryrrrrtest_choice_uniform_noreplacemsz,TestRandomDist.test_choice_uniform_noreplacecCsHtj|jtjjdddddddgd}tddd g}t||dS) NrerJFg?g333333?g?)rr-r"r!)rrrArr=r )rryrrrr test_choice_nonuniform_noreplacess  z/TestRandomDist.test_choice_nonuniform_noreplacecCsDtj|jtjddddgd}tddddg}t||dS)Nabcdre)rrrArr=r )rryrrrrtest_choice_nonintegerzsz%TestRandomDist.test_choice_nonintegerc Cstjj}tt|ddtt|ddtt|ddgddggdtt|gdtt|ddddgdddgddggdtt|ddgdd d d gdtt|ddgdd d gdtt|ddgdd d gdtt|dddgdd dtt|dddgdd dddgddS)NrrJg@r!r"reg?)r-g?g?g?gF)rr)rr-)rrrrr)rr~rrrtest_choice_exceptionssz%TestRandomDist.test_choice_exceptionsc Csddg}tttjjdddtttjjdddtttjjdd|dtttjjdd|dtttjjddgddttjjdgdddktddg}tjdtd }||d <ttjj|dd|kt}tttjjd|dd tttjjd|dd tttjjd|d|d tttjjd|d|d tttjjddg|dd ttjjdg|ddj d ktddg}tjdtd }||d <ttjj||dd |kd }ddddd d g}t tjjd|ddj |t tjjd|ddj |t tjjd|d|dj |t tjjd|d|dj |t tjjt d|ddj |dS)Ng?g?r"T)rF)rr-r!)r*r)r"rJg?g?rq)rrZisscalarrrr=emptyobjecttuplendimitemr r<r)rr-rZarrrrrrtest_choice_return_shapes:  "z'TestRandomDist.test_choice_return_shapecCs,tj|jtjd}d}t||dS)Nrs Ui+Wf)rrrAbytesr )rryrrrr test_bytess zTestRandomDist.test_bytesc Csxddddddddddddddd dd dd dg D]d}tj|j|d d ddddddddg }tj||}|dd ddd dddddg }t||qBWdS)NcSs tgS)N)rr=)r9rrrz-TestRandomDist.test_shuffle..cSs|S)Nr)r9rrrrrcSst|tjS)N)rasarrayastyperl)r9rrrrrcSst|tjS)N)rrrZfloat32)r9rrrrrcSst|tjS)N)rrrZ complex64)r9rrrrrcSst|tS)N)rrrr)r9rrrrrcSsdd|DS)NcSsg|] }||fqSrr).0irrr szATestRandomDist.test_shuffle....r)r9rrrrrcSstdd|DS)NcSsg|] }||gqSrr)rrrrrrszATestRandomDist.test_shuffle....)rr)r9rrrrrcSst||gjS)N)rZvstackT)r9rrrrrcSs(tdd|DdtdfdtjdfgS)NcSsg|] }||fqSrr)rrrrrrszATestRandomDist.test_shuffle....rr!r)rrrrj)r9rrrrsr!r"rJrer7rqrf r)rrrAshuffler )rZconvZalistryrrrr test_shuffles  zTestRandomDist.test_shufflecCstjttddddd}tjtdddd}|}|}xttdD]h}tj|t t |j |j t |j |j tj|t t |j |j t |j |j qXWdS)N)r7rerJr!r2) rZmaZ masked_valuesZreshaperrcopyrrr sorteddatamask)rrrZa_origZb_origrrrrtest_shuffle_maskeds" $ z"TestRandomDist.test_shuffle_maskedcCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) Ng?g?)rJr")rKgAKč?ghA?g ҇o>gάXf+q?gS1$?g'8zX ?r)r)rrrAbetar=r )rryrrrr test_betas zTestRandomDist.test_betacCsJtj|jtjjdddd}tddgddgd d gg}t||dS) NgZd;Y@gv/?)rJr")rK%+*0.-)rrrAr,r=r )rryrrrr test_binomials   zTestRandomDist.test_binomialcCsLtj|jtjjddd}tddgddgdd gg}t||d d dS) Nr)rJr")rKg+Q]uO@g+Q@gY7`ZqP@ge(fG@gvtR@gEIȋR@ )r)rrrA chisquarer=r )rryrrrrtest_chisquares   zTestRandomDist.test_chisquarecCsttj|jtddg}tjjj|dd}tddgddggd d gd d ggd dgddggg}t||dddS)NgI)rr-rrrtest_dirichlet_sizesz"TestRandomDist.test_dirichlet_sizecCsLtj|jtjjddd}tddgddgdd gg}t||d d dS) Ng8EGr?)rJr")rKgNU?gbI0?gVU\@g:3@gU?g"qo?r)r)rrrA exponentialr=r )rryrrrrtest_exponentials   zTestRandomDist.test_exponentialcCs*ttjjdddtttjjdddS)Nr)scaleg)r rrrrr)rrrrtest_exponential_0sz!TestRandomDist.test_exponential_0cCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) N M)rJr")rKg7?g`j?g6"+?gܳ2>?gc>+Y?g_}S?r)r)rrrAfr=r )rryrrrrtest_fs   zTestRandomDist.test_fcCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) Nr7rJ)rJr")rKgԏGM8@gȌ<@g ":@gv#)@gZQ?@gr[@@)r)rrrAgammar=r )rryrrrr test_gamma#s   zTestRandomDist.test_gammacCs.ttjjddddtttjjddddS)Nr)r<rg)r rrrrr)rrrr test_gamma_0+szTestRandomDist.test_gamma_0cCsHtj|jtjjddd}tddgddgddgg}t||dS) Ng_c97ݚ?)rJr")rKrfrr7r)rrrA geometricr=r )rryrrrrtest_geometric/s   zTestRandomDist.test_geometriccCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) Ng_c97ݚ?g@)rJr")locrrKgؕ?gs0?g#0g(xg-KD?gůp`@r)r)rrrAgumbelr=r )rryrrrr test_gumbel7s   zTestRandomDist.test_gumbelcCs*ttjjdddtttjjdddS)Nr)rg)r rrrrr)rrrr test_gumbel_0?szTestRandomDist.test_gumbel_0cCs tj|jtjjddddd}tddgddgddgg}t||tjjdd d d d}td d d d g}t||tjjd d d d d}td d d d g}t||tjjd dd d d}td d d d g}t||tjjd d d d d}td d d d g}t||dS)Ng333333$@g@r)rJr")rKrrr7rrJrerr)rrrAhypergeometricr=r )rryrrrrtest_hypergeometricCs$      z"TestRandomDist.test_hypergeometriccCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) Ng_c97ݚ?g@)rJr")rrrKg bO?g¸?gpJm @g{K*t @g:^w-gY~o?r)r)rrrAlaplacer=r )rryrrrr test_laplace]s   zTestRandomDist.test_laplacecCs*ttjjdddtttjjdddS)Nr)rg)r rrrrr)rrrrtest_laplace_0eszTestRandomDist.test_laplace_0cCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) Ng_c97ݚ?g@)rJr")rrrKg#K-z?gJ?g92@gر[@g#+^˿g-@r)r)rrrAlogisticr=r )rryrrrr test_logisticis   zTestRandomDist.test_logisticcCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) Ng_c97ݚ?g@)rJr")meansigmarKgɁ0@go,+4FB@g)ʭ6@g5?gnxNnP@g)U@r)r)rrrA lognormalr=r )rryrrrrtest_lognormalqs   zTestRandomDist.test_lognormalcCs*ttjjdddtttjjdddS)Nr)rr!g)r rrrrr)rrrrtest_lognormal_0yszTestRandomDist.test_lognormal_0cCsHtj|jtjjddd}tddgddgddgg}t||dS)Ngƀ@?)rJr")r-rKr"rqrrJ)rrrA logseriesr=r )rryrrrrtest_logseries}s   zTestRandomDist.test_logseriesc Cstj|jtjjddgddd}tddddd d gdd d d d d ggddddd dgd d ddddggddd dd dgdddd ddggg}t||dS) NrgUUUUUU?rq)rJr")rKrerJr7r"rfr!r)rrrAr3r=r )rryrrrrtest_multinomialszTestRandomDist.test_multinomialcCstj|jd}ddgddgg}d}tj|||}tddgddggd d gd d ggd dgddggg}t||ddtj||}tddg}t||ddddg}ddgddgg}tttjj||ttjj||ddt t tjj||dddS)N)g_c97ݚ?rr!r)rJr"g4Vj?gy'@gI1?gSB#@g e<@g4W(@g6R¿?g9r v"@gс?g{#@g4xg"f"@r)rgN}O6?gYMW"@r"ignore)Z check_validraise) rrrAZmultivariate_normalr=r r RuntimeWarningr rr)rrZcovrKryrrrrtest_multivariate_normals,  z'TestRandomDist.test_multivariate_normalcCsJtj|jtjjdddd}tddgddgd d gg}t||dS) Nr2g|Pk?)rJr")nr-rKiPiIi|ici i)rrrArPr=r )rryrrrrrQs   z%TestRandomDist.test_negative_binomialcCstj|jtjjdddd}tddgddgdd gg}t||d d tjjd d dd}tddgddgddgg}t||d d tj|jtjjdddd}tddgddgddgg}t||d d dS)Nr7)rJr")dfnoncrKgG7@g!ܴ*@g1}z9?@g+驸0@g26E_r#@gI!1@r)rg?g?g?gTD?g?T?g+qnl?g~|D?g밬?rg{01#@gprs'@g~;$@g"dI @g*@g }',@)rrrAnoncentral_chisquarer=r )rryrrrrtest_noncentral_chisquares"      z(TestRandomDist.test_noncentral_chisquarecCsPtj|jtjjddddd}tddgdd gd d gg}t||d d dS)Nr7r"r!)rJr")dfnumdfdenrrKg^~?g/+c?g". @g@g7?gE?r)r)rrrA noncentral_fr=r )rryrrrrtest_noncentral_fs   z TestRandomDist.test_noncentral_fcCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) Ng_c97ݚ?g@)rJr")rrrKg&&n@gOG_ @g~w@gtw]տgm&5@g$>@r)r)rrrAnormalr=r )rryrrrr test_normals   zTestRandomDist.test_normalcCs*ttjjdddtttjjdddS)Nr)rg)r rrrrr)rrrr test_normal_0szTestRandomDist.test_normal_0cCsPtj|jtjjddd}tddgddgdd gg}tjj||d d dS) Ng_c97ݚ?)rJr")rrKgW I@g Ԩy@gy]0Ag1~\Aga@gY 7A)Znulp)rrrAparetor=testingZassert_array_almost_equal_nulp)rryrrrr test_paretos zTestRandomDist.test_paretocCsHtj|jtjjddd}tddgddgddgg}t||dS)Ng_c97ݚ?)rJr")lamrKrr!)rrrApoissonr=r )rryrrrr test_poissons   zTestRandomDist.test_poissoncCs`tdj}d}tttjj|tttjj|gdtttjj|tttjj|gddS)Nrrr)rrWrYrrrr)rZlambigZlamnegrrrtest_poisson_exceptionss  z&TestRandomDist.test_poisson_exceptionscCsLtj|jtjjddd}tddgddgdd gg}t||d d dS) Ng_c97ݚ?)rJr")rrKg;'?gܴw*?g5?gFG?gD]?g$`?r)r)rrrApowerr=r )rryrrrr test_powers   zTestRandomDist.test_powercCsLtj|jtjjddd}tddgddgdd gg}t||d d dS) Nr)rJr")rrKg"u+@gN$MB*@g7v8B4@g55@ge&@g~ӣZ1@r)r)rrrArayleighr=r )rryrrrr test_rayleigh s   zTestRandomDist.test_rayleighcCs*ttjjdddtttjjdddS)Nr)rg)r rrr rr)rrrrtest_rayleigh_0szTestRandomDist.test_rayleigh_0cCsJtj|jtjjdd}tddgddgddgg}t||d d dS) N)rJr")rKgMoDL?g6([9grF=?gM' -gVgVA:Qy?r)r)rrrAZstandard_cauchyr=r )rryrrrrtest_standard_cauchys   z#TestRandomDist.test_standard_cauchycCsJtj|jtjjdd}tddgddgddgg}t||d d dS) N)rJr")rKg܁?gc!z?g&N"@gNI@gF>?g~ME?r)r)rrrAZstandard_exponentialr=r )rryrrrrtest_standard_exponentials   z(TestRandomDist.test_standard_exponentialcCsLtj|jtjjddd}tddgddgdd gg}t||d d dS) NrJ)rJr")r<rKg)@gľ@gd],q@gPk{@g9B #1@gv @r)r)rrrAstandard_gammar=r )rryrrrrtest_standard_gamma's   z"TestRandomDist.test_standard_gammacCs*ttjjdddtttjjdddS)Nr)r<g)r rrrrr)rrrrtest_standard_gamma_0/sz$TestRandomDist.test_standard_gamma_0cCsJtj|jtjjdd}tddgddgddgg}t||d d dS) N)rJr")rKg!#H@gu@g#@g}d]#@gN@gw )* @r)r)rrrAuniformr=r )rryrrrr test_uniformLs   zTestRandomDist.test_uniformcCstdj}tdj}tjj}tt|tj dtt|dtjtt|||tt|tj gdgtt|dgtjgtjjt |d|dddS)Nr>rr!g؅W4vC)rr) rZfinforXrYrrr OverflowErrorinfZ nextafter)rZfminZfmaxfuncrrrtest_uniform_range_boundsTs  z(TestRandomDist.test_uniform_range_boundscCsnGdddtj}td|}tttjj||Gdddtj}td|}tttjj|dddS)Nc@seZdZddZdS)zGTestRandomDist.test_scalar_exception_propagation..ThrowingFloatcSstdS)N)r)rrrr __float__lszQTestRandomDist.test_scalar_exception_propagation..ThrowingFloat.__float__N)r%r&r'r#rrrr ThrowingFloatksr$g?c@seZdZddZdS)zITestRandomDist.test_scalar_exception_propagation..ThrowingIntegercSstdS)N)r)rrrr__int__sszQTestRandomDist.test_scalar_exception_propagation..ThrowingInteger.__int__N)r%r&r'r%rrrrThrowingIntegerrsr&r!) rZndarrayr=rurrrrr)rr$Zthrowing_floatr&Z throwing_intrrr!test_scalar_exception_propagationds z0TestRandomDist.test_scalar_exception_propagationcCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) NgGz?gp= ף?)rJr")mukapparKgLZI@g'N"@gkr?gk=m@gډ?g^=\?r)r)rrrAvonmisesr=r )rryrrrr test_vonmisesys   zTestRandomDist.test_vonmisescCs:tj|jtjjdddd}tjt|dS)Ngg g5_PG>i@B)r(r)rK)rrrAr*rrZisfiniter8)rrrrrtest_vonmises_smallsz"TestRandomDist.test_vonmises_smallcCsNtj|jtjjdddd}tddgddgd d gg}t||d d dS) NgGz?gp= ף?)rJr")rrrKgZ@g- g@g8 m?gVP"?g`V?gx^&3?r)r)rrrAwaldr=r )rryrrrr test_walds   zTestRandomDist.test_waldcCsLtj|jtjjddd}tddgddgdd gg}t||d d dS) NgGz?)rJr")rrKg@6?g]A?g(1ڥR?gU?g2_u?g^}-Q?r)r)rrrAweibullr=r )rryrrrr test_weibulls   zTestRandomDist.test_weibullcCs*ttjjdddtttjjdddS)Nr)rg)r rrr0rr)rrrrtest_weibull_0szTestRandomDist.test_weibull_0cCsHtj|jtjjddd}tddgddgddgg}t||dS) NgGz?)rJr")rrKBr!rJr)rrrAzipfr=r )rryrrrr test_zipfs   zTestRandomDist.test_zipfN)Fr%r&r'rErrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrQrrrrrrrr r r rrrrrrrrr"r'r+r-r/r1r2r6rrrrrs $    "  rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=d>Z!d?S)@ TestBroadcastcCs d|_dS)Ni[)rA)rrrrrEszTestBroadcast.setUpcCstj|jdS)N)rrrA)rrrrsetSeedszTestBroadcast.setSeedcCspdg}dg}tjj}tdddg}|||d|}t||dd||||d}t||dddS) Nrr!gl ?g_x ?g =*BN?rJr)r)rrrr=r8r )rrrrrryrrrrszTestBroadcast.test_uniformcCsdg}dg}dg}tjj}tdddg}|||d|}t||dd tt||d|||||d}t||dd tt|||ddS) Nrr!rg-r@gYl@g-`w?rJr)r)rrrr=r8r rr)rrr bad_scalerrryrrrrszTestBroadcast.test_normalcCsdg}dg}dg}dg}tjj}tdddg}|||d|}t||d d tt||d|tt||d|||||d}t||d d tt|||dtt|||ddS) Nr!r"rr#gnQVf?gVKB?g v`k?rJr)r)rrrr=r8r rr)rrrbad_aZbad_brrryrrrrs$zTestBroadcast.test_betacCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rg.UgZ?giPrq?gE K?rJr)r)rrrr=r8r rr)rrr9rrryrrrrs zTestBroadcast.test_exponentialcCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rg.UgZ?giPrq?gE K?rJr)r)rrrr=r8r rr)rr< bad_shapeZ std_gammarryrrrrs z!TestBroadcast.test_standard_gammacCsdg}dg}dg}dg}tjj}tdddg}|||d|}t||d d tt||d|tt||d|||||d}t||d d tt|||dtt|||ddS) Nr!r"rr#g.UgZ?giPrq?gE K?rJr)r)rrrr=r8r rr)rr<rr;r9rrryrrrrs$zTestBroadcast.test_gammacCsdg}dg}dg}dg}tjj}tdddg}|||d|}t||d d tt||d|tt||d|||||d}t||d d tt|||dtt|||ddS) Nr!r"rr#gbyʜ?go?gT7@rJr)r)rrrr=r8r rr)rrr bad_dfnum bad_dfdenrrryrrrrs$zTestBroadcast.test_fc Csfdg}dg}dg}dg}dg}dg}tjj}tddd g}|||d||} t| |d d tt||d||tt||d||tt||d||||||d|} t| |d d tt|||d|tt|||d|tt|||d||||||d} t| |d d tt||||dtt||||dtt||||ddS) Nr"rJrerrr#gr^G"@g/ *@gц!@r)r)rrrr=r8r rr) rrrrr<r=bad_noncZnonc_frryrrrr/s8zTestBroadcast.test_noncentral_fcCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rg~}N??g$F?g'rS?rJr)r)rrrr=r8r rr)rrbad_dfrrryrrrrPs zTestBroadcast.test_chisquarecCsdg}dg}dg}dg}tjj}tdddg}|||d|}t||d d tt||d|tt||d|||||d}t||d d tt|||dtt|||ddS) Nr!r"rr#g w"@g2 WR@gRY@rJr)r)rrrr=r8r rr)rrrr?r>Znonc_chirryrrrr]s$z'TestBroadcast.test_noncentral_chisquarecCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rgT @g~Fl@gƚtp?rJr)r)rrrr=r8r rr)rrr?trryrrrrss zTestBroadcast.test_standard_tcCsdg}dg}dg}tjj}tdddg}|||d|}t||dd tt||d|||||d}t||dd tt|||ddS) Nr"r!rgW!@gOBg}?rJr)r)rrr*r=r8r rr)rr(r)Z bad_kappar*rryrrrr+szTestBroadcast.test_vonmisescCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rg&8??geFX?gՋː?rJr)r)rrrr=r8r rr)rrr:rrryrrrrs zTestBroadcast.test_paretocCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rg.UgZ?giPrq?gE K?rJr)r)rrr0r=r8r rr)rrr:r0rryrrrr1s zTestBroadcast.test_weibullcCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rgl ?g_x ?g =*BN?rJr)r)rrr r=r8r rr)rrr:r rryrrrr s zTestBroadcast.test_powercCsdg}dg}dg}tjj}tdddg}|||d|}t||dd tt||d|||||d}t||dd tt|||ddS) Nrr!rg:BKc?g3k?g~q?rJr)r)rrrr=r8r rr)rrrr9rrryrrrrszTestBroadcast.test_laplacecCsdg}dg}dg}tjj}tdddg}|||d|}t||dd tt||d|||||d}t||dd tt|||ddS) Nrr!rg'Zy?gdSO=?g3L1?rJr)r)rrrr=r8r rr)rrrr9rrryrrrrszTestBroadcast.test_gumbelcCsdg}dg}dg}tjj}tdddg}|||d|}t||dd tt||d|||||d}t||dd tt|||ddS) Nrr!rg“y?gJR?g ~&?rJr)r)rrrr=r8r rr)rrrr9rrryrrrrszTestBroadcast.test_logisticcCsdg}dg}dg}tjj}tdddg}|||d|}t||dd tt||d|||||d}t||dd tt|||ddS) Nrr!rgH"@gܮ @gvv :@rJr)r)rrrr=r8r rr)rrrZ bad_sigmarrryrrrrszTestBroadcast.test_lognormalcCsZdg}dg}tjj}tdddg}|||d}t||ddtt||ddS) Nr!rgݱlo?g/ӂ?gz~R?rJr)r)rrr r=r8r rr)rrr9r rryrrrr s zTestBroadcast.test_rayleighcCsdg}dg}dg}dg}tjj}tdddg}|||d|}t||d d tt||d|tt||d|||||d}t||d d tt|||dtt|||ddS) Ng?r!rr#g@m e?g`I߿?g|2?rJr)r)rrr.r=r8r rr)rrrZbad_meanr9r.rryrrrr/s$zTestBroadcast.test_waldc Csldg}dg}dg}dg}dg}|d\}}tjj}tdddg} |||d||} t| | dd tt||d||tt||d||tt||d||||||d|} t| | dd tt|||d|tt|||d|tt|||d||||||d} t| | dd tt||||dtt||||dtt||||ddS) Nr!rJr"reg,P;bD@g%G@g,~S@r)r)rrrr=r8r rr) rrrrZ bad_left_oneZ bad_mode_oneZ bad_left_twoZ bad_mode_tworrryrrrr)s8 zTestBroadcast.test_triangularc Csdg}dg}dg}dg}dg}tjj}tdddg}|||d|}t||tt||d|tt||d|tt||d|||||d}t||tt|||dtt|||dtt|||ddS)Nr!g?rg?rJ)rrr,r=r8r rr) rrr-bad_n bad_p_one bad_p_twoZbinomrryrrrrJs&  zTestBroadcast.test_binomialc Csdg}dg}dg}dg}dg}tjj}tdddg}|||d|}t||tt||d|tt||d|tt||d|||||d}t||tt|||dtt|||dtt|||ddS)Nr!g?rg?rrJ)rrrPr=r8r rr) rrr-rArBrCZ neg_binomrryrrrrQas&  z$TestBroadcast.test_negative_binomialcCs|tjj}dg}dg}|dg}tjj}tdddg}|||d}t||tt ||dtt ||ddS)Nr!rr"rrJ) rrrZpoisson_lam_maxrr=r8r rr)rZmax_lamrZ bad_lam_oneZ bad_lam_tworrryrrrrxs    zTestBroadcast.test_poissoncCsVdg}dg}tjj}tdddg}|||d}t||tt||ddS)Nr"rr!rJ)rrr5r=r8r rr)rrr:r5rryrrrr6s  zTestBroadcast.test_zipfcCsldg}dg}dg}tjj}tdddg}|||d}t||tt||dtt||ddS)Ng?rg?r"rJ)rrrr=r8r rr)rr-rBrCZgeomrryrrrrs  zTestBroadcast.test_geometricc Csdg}dg}dg}dg}dg}dg}dg}tjj}tdddg} |||d||} t| | tt||d||tt||d||tt||d||tt||d||||||d|} t| | tt|||d|tt|||d|tt|||d|tt|||d||||||d} t| | tt||||dtt||||dtt||||dtt||||ddS)Nr!r"rr#rrerJ)rrrr=r8r rr) rZngoodZnbadZnsampleZ bad_ngoodZbad_nbadZbad_nsample_oneZbad_nsample_twoZ hypergeomrryrrrrs<   z!TestBroadcast.test_hypergeometriccCsldg}dg}dg}tjj}tdddg}|||d}t||tt||dtt||ddS)Ng?r"rr!rJ)rrrr=r8r rr)rr-rBrCrrryrrrrs  zTestBroadcast.test_logseriesN)"r%r&r'rEr8rrrrrrrrrrrr+rr1r rrrrr r/rrrQrr6rrrrrrrr7s>  !      !  #r7c@s4eZdZddZddZddZddZd d Zd S) TestThreadcCstd|_dS)Nre)rseeds)rrrrrEszTestThread.setUpcsddlmtt|jf|}tt|jf|}fddt|j|D}dd|Ddd|Dx*t|j|D]\}}tj||q~Wt j j dkrt j dkrt||n t||dS) Nr)Threadcs(g|] \}}tj||fdqS))targetargs)rrr)rro)rFfunctionrrrsz-TestThread.check_function..cSsg|] }|qSr)start)rr9rrrrscSsg|] }|qSr)join)rr9rrrrsreZwin32)Z threadingrFrrrrEziprrZintpr*itemsizersplatformr r )rrJszZout1Zout2r@rrIr)rFrJrcheck_functions   zTestThread.check_functioncCsdd}|j|dddS)NcSs|jdd|d<dS)Ni')rK.)r)rDoutrrr gen_randomsz*TestThread.test_normal..gen_random)i')rP)rQ)rrSrrrrszTestThread.test_normalcCsdd}|j|dddS)NcSs|jtdd|d<dS)N)r2i)r.)rrZones)rDrRrrrrSsz'TestThread.test_exp..gen_random)r2i)rP)rQ)rrSrrrtest_expszTestThread.test_expcCsdd}|j|dddS)NcSs|jddgddd|d<dS)NrgUUUUUU?rqi')rK.)r3)rDrRrrrrSsz/TestThread.test_multinomial..gen_random)i'rq)rP)rQ)rrSrrrrszTestThread.test_multinomialN)r%r&r'rErQrrTrrrrrrDs rDc@s,eZdZddZddZddZddZd S) TestSingleEltArrayInputcCs4tdg|_tdg|_tdg|_d|_dS)Nr"rJre)r!)rr=argOneargTwoargThreetgtShape)rrrrrEszTestSingleEltArrayInput.setUpc Cstjjtjjtjjtjjtjjtjjtjjtjj tjj tjj tjj tjj f }tjj tjj f}x@|D]8}||kr|tdg}n ||j}||j|jqbWdS)Ng?)rrrrrrrr0r r rr5rrr=rVr{r<rY)rfuncs probfuncsr!rRrrrtest_one_arg_funcss       z*TestSingleEltArrayInput.test_one_arg_funcscCstjjtjjtjjtjjtjjtjjtjjtjj tjj tjj tjj tjj tjjtjjf}tjjtjjf}x|D]|}||krtdg}n|j}||j|}||j|j||jd|}||j|j||j|d}||j|jqnWdS)Ng?r)rrrrrrrrr*rrrrr.r,rPr=rWrVr{r<rY)rrZr[r!rWrRrrrtest_two_arg_funcss$        z*TestSingleEltArrayInput.test_two_arg_funcscCstjjtjjtjjg}xz|D]r}||j|j|j}||j |j ||jd|j|j}||j |j ||j|jd|j}||j |j qWdS)Nr) rrrrrrVrWrXr{r<rY)rrZr!rRrrrtest_three_arg_funcsAs   z,TestSingleEltArrayInput.test_three_arg_funcsN)r%r&r'rEr\r]r^rrrrrUs-rU__main__)Z __future__rrrrZnumpyrZ numpy.testingrrrrr r r r r rrrsrr(r1r@rRrr7rDrUr%rrrrs60 0 /+U