o gE @s<ddlZddlZddlZddlmZddlZddlZddlZddlZ ddl m Z m Z m Z ddlmZddlmZmZmZmZmZmZmZddlmZddlmZddlmZmZdd lmZm Z dd l!m"Z"Gd d d Z#d de#ifddgdifddgdifdde#ifdde#ifde#ddfgZ$ej%j&dkrej%j'dkrdZ(ndZ(ddedfdde)e(fdde*dfd dedfd!dedfd"de)d#fgZ+d$dedfd%dedfd&de)e(fd'de*dfd(de)d#fgZ,d)de)e(fd*de*dfd+dedfd,dedfd-de)d#fgZ-gd.fd/d0ggd1fd2d3e j.d4gd5fd2d3e j/d4gd5fd0d0gd6fgZ0d7e1d8fd9e1d8fgZ2d:ed;fdfe j.e j/fe1d>fgZ4de j.fe1d>fe j.e j.fe1d>fgZ5ej67d?e3e2e5ej67d@e$dAdBZ8ej67d@e$dCdDZ9dEdFZ:dGdHZ;ej67d@e$dIdJZdQdRZ?dSdTZ@dUdVZAej67dWgdXdYdZZBej67dWddgGd[d\d\ZCGd]d^d^ZDGd_d`d`ZEGdadbdbZFGdcddddZGGdedfdfZHGdgdhdhZIdS)iN)deepcopy)assert_allclose assert_equalsuppress_warnings) NumpyVersion)TransformedDensityRejectionDiscreteAliasUrnDiscreteGuideTableNumericalInversePolynomialNumericalInverseHermiteSimpleRatioUniforms UNURANError)stats)special) chisquarecramervonmises) distdiscretedistcont)check_random_statec@$eZdZddZddZddZdS)StandardNormalcCs&dtdtjtd||SN?@࿩npsqrtpiexpselfxr#]/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/scipy/stats/tests/test_sampling.pypdf s&zStandardNormal.pdfcCs,dtdtj| td||Srrr r#r#r$dpdf$s,zStandardNormal.dpdfcCs t|SN)rndtrr r#r#r$cdf' zStandardNormal.cdfN__name__ __module__ __qualname__r%r&r)r#r#r#r$rs rrdistr){Gz? ףp= ?皙?r r r r )r/modepypy) z.unsupported operand type for float\(\): 'list'zmust be real number, not listcC| Sr'r#r"r#r#r$?r:z...cCgSr'r#r9r#r#r$r:AcCtSr'foor9r#r#r$r:Cr=name 'foo' is not definedcCtjSr'rinfr9r#r#r$r:Er;cCrBr'rnanr9r#r#r$r:Gr;cCdSNrr#r#r#r#r$r:Ir=,takes 0 positional arguments but 1 was givencCrBr'rCr9r#r#r$r:Pr;cCrBr'rEr9r#r#r$r:Rr;cCr<r'r#r9r#r#r$r:Tr=cCr>r'r?r9r#r#r$r:Vr=cCrGrHr#r#r#r#r$r:Xr=cCr<r'r#r9r#r#r$r:_r=cCr>r'r?r9r#r#r$r:ar=cCrBr'rCr9r#r#r$r:cr;cCrBr'rEr9r#r#r$r:er;cCrGrHr#r#r#r#r$r:gr=z!must contain at least one elementrz0wrong number of dimensions \(expected 1, got 2\)皙?皙?r2)must contain only finite / non-nan values(must contain at least one non-zero value)r6zmust be a length 2 tupler#)rPrOz left >= right)rOrO)r7r7zonly non-nan valuesdomain, err, msgzmethod, kwargscCsVttj|}tj||d|di|d|iWddS1s$wYdS)Nmatchdomainr#)getattrrsamplingpytestraises)rTerrmsgmethodkwargsMethodr#r#r$test_bad_domains "r^c Csttj|}d}|di|d|i}|di|d|i}t|d|dtjd|di|}|d}tjd|di|ddi}|d}t||tjtj d}tj tj d} |di|d|i}|di|d| i}t|d|ddS)N{ random_statedr#) rUrrVrrvsrrandomseed RandomStateMT19937 Generator) r[r\r]rdrng1rng2rvs1rvs2seed1seed2r#r#r$test_random_states"      rncCsrttdd}tt}|dt|d|dttdd}|d}|d|d}t||dS)Nr_r`ra)rrset_random_staterrb)rhrirngrjrkr#r#r$test_set_random_states     rrcsdddGdddfdd}fdd}tj|d}tj|d}||||d d ks>Jd d ksFJdS) N)err1err2c@r)z.test_threading_behaviour..DistributioncS ||_dSr')pdf_msg)r!rvr#r#r$__init__r*z7test_threading_behaviour..Distribution.__init__cSs(d|kr dkrn|St|j|S)Ng33333H@gI@) ValueErrorrvr r#r#r$r%s z2test_threading_behaviour..Distribution.pdfcSrGNrOr#r r#r#r$r&z3test_threading_behaviour..Distribution.dpdfN)r,r-r.rwr%r&r#r#r#r$ Distribution r{c \d}t|ddd}z|dWdSty-}z|jdd<WYd}~dSd}~ww)Nr@r7ra rTr`順rrsrrbrxargsr/rqer{errorsr#r$func1z'test_threading_behaviour..func1c r})Nbarr~rPrrrrtrrrr#r$func2rz'test_threading_behaviour..func2)targetrsr@rtr) threadingThreadstartjoin)rrt1t2r#rr$test_threading_behaviours   rcCsPttj|}|di|ddi}t|}t|}t|d|ddS)Nr`r_rar#)rUrrVpickledumpsloadsrrb)r[r\r]rhobjrir#r#r$ test_pickles   rsize)Nr)rrO)r7r6)rPr6)rr)rrOcCsRtt}|durt||sJdSt|r|f}||j|ks'JdSr')rrrisscalarrbshape)rrqr#r#r$ test_rvs_sizes  rcCst}tjd}t||d}tjdddd}t|||t| || |tjddd}t||d}t|||t| || |t d d }t ||d}| }| t|d|dd}t|||dS) NrrorOranum$@@)locscaler7rK)rnormrrc default_rngr linspacecheck_cont_samplesrppfbinomrsupportpmfarangecheck_discr_samples)r/urngrqurTpvr#r#r$test_with_scipy_distributions     rcCsn|d}||f}t|drt||ddd|d}t|j|_t||jj }|dks5JdS)NrrOgHz>皙?rtolatoli) rbmeanvarrisfiniter vectorizer)rpvalue)rqr/mv_exrbmvpvalr#r#r$r(s  rc Cs|d}||f}t||ddd||}t|}tj|dd\}}||}||d|j<t ||j }|dksBJdS)NrMbP?rrT) return_counts) rbrrrsumr zeros_likeuniquerrr) rqrrrbr obs_freqs_freqsrr#r#r$r5s     rcCsd}tjt|dttdddWdn1swYtjt|dttddWddS1s;wYdS)Nz.102 : center moved into domain of distributionrRrr6r)centerrTrT)rWwarnsRuntimeWarningr r)rZr#r#r$!test_warning_center_not_in_domainEs"rr[)r r rcCsTttj|}d}tjt|d|tdddWddS1s#wYdS)Nz17 : mode not in domainrRrr)r3rT)rUrrVrWrXr r)r[r]rZr#r#r$test_error_mode_not_in_domainOs "rc@seZdZddZdejjdddejjdddgZde fdd d d gZ de fde fd gZ e j d ee j de e j de ddZddZdS)TestQRVScCsd}tjt|dttj|}|t}|jddWdn1s%wYttj|}|t}d}tjt|d|jdtj ddWddS1sUwYdS) Nz&`qmc_engine` must be an instance of...rRr) qmc_engine6`d` must be consistent with dimension of `qmc_engine`.r6rPdr) rWrXrxrUrrVrqrvsqmcHalton)r!r[rSr]genr#r#r$test_input_validation`s    "zTestQRVS.test_input_validationNrOrrdr6)rO)rO)rr)rr)rPrr)r6)r6qrngsize_in, size_outz d_in, d_outcCsHtjdko tddk}|r|dkrtdt}ttj|} | |} |durV|durV|j |krVd} tj t | d| j |||dWddS1sOwYdS|durg|durg|j d krg|j f}||} t |} | j |||d }|dur|j| ksJ| dur| t|pd }tj|| }t||d d dSdS) Nwin32r32bitr z4NumericalInversePolynomial.qrvs fails for Win 32-bitrrRrrOrrr-q=r)sysplatform architecturerWxfailrrUrrVrrXrxrrrrcrprodrrreshaper)r!rsize_insize_outd_ind_outr[w32r/r]rrSshape_expectedqrng2runiformqrvs2r#r#r$test_QRVS_shape_consistencyvs4    z$TestQRVS.test_QRVS_shape_consistencycCst}ttj|}||}d}d}tjj|dd}tjj|dd}|t|} |j |||d} tj | } t |D]} | d| f} | dd| f |}t| |ddq;dS) N)r6rrrrr.rr)rrUrrVrrrcrrrrrrangerr)r!r[r/r]rrrrrrrrisamplesample2r#r#r$test_QRVS_size_tuples    zTestQRVS.test_QRVS_size_tuple)r,r-r.rrrSobolrqrngstuplesizesdsrWmark parametrizerrr#r#r#r$r]s"     "rc @seZdZGdddZGdddZGdddZeeededgZdd gZdd gZde j gZ de j gZ eee e gZ ejd eee d d ZddedfgZee7ZejdeddZejdeddZejdeddZejdgdddZddZejd d!e je j d"d#gd$d%Ze jd&d'd(d)ggge jge j e je j gd&e jd*d"gd+d,e j ggd-ggZ ejd.e d/d0Z!d1d2Z"d3S)4TestTransformedDensityRejectionc@,eZdZddZddZddZddZd S) z%TestTransformedDensityRejection.dist0cCdd||SN?rOr#r r#r#r$r%z)TestTransformedDensityRejection.dist0.pdfcCs dd|S)Nrr#r r#r#r$r& z*TestTransformedDensityRejection.dist0.dpdfcCd||dddSNrr6gUUUUUU?r#r r#r#r$r)z)TestTransformedDensityRejection.dist0.cdfcCrGN)rOr#r!r#r#r$rrzz-TestTransformedDensityRejection.dist0.supportNr,r-r.r%r&r)rr#r#r#r$dist0  r c@r)z%TestTransformedDensityRejection.dist1cCtj|dSNrrr_pdfr r#r#r$r%rz)TestTransformedDensityRejection.dist1.pdfcCs| dtj|dS)N{Gz?rrr r#r#r$r&z*TestTransformedDensityRejection.dist1.dpdfcCrrrr_cdfr r#r#r$r)rz)TestTransformedDensityRejection.dist1.cdfNr+r#r#r#r$dist1 rc@r) z%TestTransformedDensityRejection.dist2cCrur'shiftr!rr#r#r$rwr*z.TestTransformedDensityRejection.dist2.__init__cC&||j8}dt|d}d||SNr?rabsr!r"yr#r#r$r%  z)TestTransformedDensityRejection.dist2.pdfcCs8||j8}dt|d}|||}|dkr|S| S)NrrJrr!r#r#r$r&s  z*TestTransformedDensityRejection.dist2.dpdfcC.||j8}|dkrdd|Sddd|SNrJrrrr r#r#r$r)  z)TestTransformedDensityRejection.dist2.cdfN)r,r-r.rwr%r&r)r#r#r#r$dist2s  r'rJ@?r dist, mv_excCsLt}|tt|dd}Wdn1swYt|||dSN*ro)rfilterrrr)r!r/rsuprqr#r#r$ test_basics  z*TestTransformedDensityRejection.test_basiccCrG)Nrr#r9r#r#r$r:r=z(TestTransformedDensityRejection.50 : bad construction points. pdf, err, msgcCsZGddd}||_dd|_tj||d t|WddS1s&wYdS)Nc@ eZdZdS)z:TestTransformedDensityRejection.test_bad_pdf..distNr,r-r.r#r#r#r$r/r/cSrGryr#r9r#r#r$r:r=z>TestTransformedDensityRejection.test_bad_pdf..rRr%r&rWrXrr!r%rYrZr/r#r#r$ test_bad_pdfs   "z,TestTransformedDensityRejection.test_bad_pdfzdpdf, err, msgcCs^Gddd}dd|_||_tj||dt|ddWddS1s(wYdS)Nc@r2)z;TestTransformedDensityRejection.test_bad_dpdf..distNr3r#r#r#r$r/r4r/cSs|Sr'r#r9r#r#r$r:r=z?TestTransformedDensityRejection.test_bad_dpdf..rRrOr7rr5)r!r&rYrZr/r#r#r$ test_bad_dpdf s  "z-TestTransformedDensityRejection.test_bad_dpdfrQcCBtj||dtt|dWddS1swYdSNrRr)rWrXrrr!rTrYrZr#r#r$test_inf_nan_domains"z4TestTransformedDensityRejection.test_inf_nan_domainsconstruction_points)r rrcCsBtjtddtt|dWddS1swYdS)Nz1`construction_points` must be a positive integer.rRr?rWrXrxrrr!r?r#r#r$#test_bad_construction_points_scalars "zCTestTransformedDensityRejection.test_bad_construction_points_scalarcCsg}tjtddtt|dWdn1swYgd}tjtddtt|dWdn1s=wYtjtjtjg}tjt ddtt|dWdn1scwYddg}tjtd dttd |d WddS1swYdS) NzC`construction_points` must either be a scalar or a non-empty array.rRr@)rOrOrOrOrOrOz:33 : starting points not strictly monotonically increasingr0r7z!50 : starting point out of domain)r6)rTr?) rWrXrxrrrrrrFr rBr#r#r$"test_bad_construction_points_array$s2"zBTestTransformedDensityRejection.test_bad_construction_points_arraycrrcCsFd}tjt|dttddWddS1swYdS)Nz`c` must either be -0.5 or 0.rRrH)rGrA)r!rGrZr#r#r$ test_bad_cHs"z*TestTransformedDensityRejection.test_bad_crrOirrrKrLrr6rrcCsttdd}t+}|td|td|td|td||}tj|}Wdn1s8wYt ||ddd |j |j ksMJdS) NgH.?)max_squeeze_hat_ratio$invalid value encountered in greater*invalid value encountered in greater_equal!invalid value encountered in less'invalid value encountered in less_equalrh㈵>r) rrrr-rppf_hatrrrrr)r!rrqr.resexpectedr#r#r$ test_ppf_hatRs      z,TestTransformedDensityRejection.test_ppf_hatcCsGddd}d}tjt|d t|Wdn1swYGddd}d}tjt|d t|WddS1sDwYdS)Nc@r2);TestTransformedDensityRejection.test_bad_dist..distNr3r#r#r#r$r/hr4r/z`pdf` required but not found.rRc@eZdZddZdS)rUcSs d||Sryr#r9r#r#r$r:qs zDTestTransformedDensityRejection.test_bad_dist..dist.Nr,r-r.r%r#r#r#r$r/p z`dpdf` required but not found.)rWrXrxr)r!r/rZr#r#r$ test_bad_distfs  "z-TestTransformedDensityRejection.test_bad_distN)#r,r-r.r rr'distsmv0mv1rrDmv2mv3mvsrWrrzipr/r bad_pdfsbad_pdfs_commonr7bad_dpdf_commonr9inf_nan_domainsr=rCrFrFrIrrrTrYr#r#r#r$rsD           $    rc @s0eZdZhdZejdeddZdde dfdde dfd de d fd de d fd de dfdde dfdde dfgZ ejde ddZejdgdgdgddZejdeddZej ejfejejfej ej fdejfej dfgZejded d!Zd"d#Zd$d%Zd&S)'TestDiscreteAliasUrn>randintnchypergeom_fishernchypergeom_walleniusdistname, paramsc C||jvr d}t|t|ts|}ntt|}||}|}t |d|ds1tdt |d|dd}| |}|d}t |dd} t | ||dS)NzYDAU fails on these probably because of large domains and small computation errors in PMF.rOrz$DAU only works with a finite domain.rr,ro)basic_fail_distsrWskip isinstancestrrUrrrrrrrr r!distnameparamsrZr/rTkrrrqr#r#r$r/s        zTestDiscreteAliasUrn.test_basiccCrBr'rCr9r#r#r$r:r;zTestDiscreteAliasUrn.rMcCrBr'rEr9r#r#r$r:r;cCrG)NrJr#r9r#r#r$r:r=rNcCr>r'r?r9r#r#r$r:r=rAcCr<r'r#r9r#r#r$r:r=z)setting an array element with a sequence.cCr8r'r#r9r#r#r$r:r;z50 : probability < 0cCrGrHr#r#r#r#r$r:r=rIz pmf, err, msgcCsTGddd}||_tj||dt|ddWddS1s#wYdS)Nc@r2)z/TestDiscreteAliasUrn.test_bad_pmf..distNr3r#r#r#r$r/r4r/rRr8r)rrWrXr)r!rrYrZr/r#r#r$ test_bad_pmfs "z!TestDiscreteAliasUrn.test_bad_pmfr)r1r0r2)rrg@@rg@cCs~tj|tjd}t|dd}|d}||}tdt|}tj||d}tj||d|d}||f}t |||dS)N)dtyper_rorr)weightsrP) rasarrayfloat64rrbrrlenaverager)r!rrqrbvariates m_expected v_expected mv_expectedr#r#r$test_sampling_with_pvs   z*TestDiscreteAliasUrn.test_sampling_with_pvpv, msgcC<tjt|d t|WddS1swYdSNrRrWrXrxrr!rrZr#r#r$ test_bad_pv "z TestDiscreteAliasUrn.test_bad_pvrrTcCHtjtddttdd|dWddS1swYdSNzmust be finiterRr7rKr)rWrXrxrrrr!rTr#r#r$test_inf_domain"z$TestDiscreteAliasUrn.test_inf_domaincCsDtjtddtddgddWddS1swYdS)Nzrelative urn size < 1.rRrr ) urn_factor)rWrrrr r#r#r$test_bad_urn_factors"z(TestDiscreteAliasUrn.test_bad_urn_factorcCsNd}Gddd}tjt|d t|WddS1s wYdS)NzG`domain` must be provided when the probability vector is not available.c@rV)z0TestDiscreteAliasUrn.test_bad_args..distcSs|Sr'r#r r#r#r$rrzz4TestDiscreteAliasUrn.test_bad_args..dist.pmfN)r,r-r.rr#r#r#r$r/rXr/rRr)r!rZr/r#r#r$ test_bad_argss  "z"TestDiscreteAliasUrn.test_bad_argsN)r,r-r.rkrWrrrr/rx NameErrorr TypeErrorbad_pmfrsr bad_pv_commonrrrD inf_domainrrrr#r#r#r$rexsR      $   rec @sheZdZGdddZGdddZGdddZGdddZeeeegZd d gZd d gZ d e j d de j dgZ d e j dde j dgZ ee e e gZejdeeeddZgdZgdZejjejdeddZejdeddZejdeddZejded d!Ze jd"d#d$d%ggge j ge j! e j e j!gd"e j d&d'gd(d)e j!ggd*ggZ"ejd+e"d,d-Z#e jd.d/d$d%ggge j ge j! e j e j!gd"e j d&d'gd(d)e j!ge j! d0d1gggZ$ejd2e$d3d4Z%d5d6Z&d#d7d8e j!e j gZ'd9d'e j!e j gZ(ejd:e'd;d<Z)ejd=e(d>d?Z*d@dAZ+dBdCZ,dDS)ETestNumericalInversePolynomialc@r)z$TestNumericalInversePolynomial.dist0cCrrr#r r#r#r$r%rz(TestNumericalInversePolynomial.dist0.pdfcCrrr#r r#r#r$r)rz(TestNumericalInversePolynomial.dist0.cdfcCrGr r#r r#r#r$rrzz,TestNumericalInversePolynomial.dist0.supportNr,r-r.r%r)rr#r#r#r$r rr c@seZdZddZddZdS)z$TestNumericalInversePolynomial.dist1cCrrrr r#r#r$r%rz(TestNumericalInversePolynomial.dist1.pdfcCrrrr r#r#r$r)rz(TestNumericalInversePolynomial.dist1.cdfN)r,r-r.r%r)r#r#r#r$rs rc@r)z$TestNumericalInversePolynomial.dist2cCs dddtdtj|S)N皙??rOrPrsinrr r#r#r$r%s z(TestNumericalInversePolynomial.dist2.pdfcCsDd|ddddtjd|tdtj|dtjS)NrrOg?rrrtrrcosr r#r#r$r)s ,z(TestNumericalInversePolynomial.dist2.cdfcCrGr r#r r#r#r$rrzz,TestNumericalInversePolynomial.dist2.supportNrr#r#r#r$r'r|r'c@r)z$TestNumericalInversePolynomial.dist3cCs$ddddtdtj|S)NrKrrrOrPrr r#r#r$r%s$z(TestNumericalInversePolynomial.dist3.pdfcCs>|ddddtjtdtjtdtj|S)Nrrg ףp= ?rPr7rr r#r#r$r)s$z(TestNumericalInversePolynomial.dist3.cdfcCrG)N)rr#r r#r#r$rrzz,TestNumericalInversePolynomial.dist3.supportNrr#r#r#r$dist3s rrJr)rgܿUUUUUU?gQ?rPg @r*cCst|dd}t|||dSr+)r rr!r/rrqr#r#r$r/$s z)TestNumericalInversePolynomial.test_basic) studentized_range trapezoidtriangvonmises levy_stablekappa4ksonekstwolevy_l gausshyperanglit) ncfparetochi2 fatiguelife halfgennormgibratlognormncx2tricCs||jvr td|||jvrtd|t|tr$tt|n|}||}t}| t t |dd}Wdn1sCwYt ||| |gdS)NzPINV too slow for zPINV fails for r,ro)very_slow_distsrWrl fail_distsrmrnrUrrr-rr rrr)r!rprqr/r.rqr#r#r$test_basic_all_scipy_dists3s    z9TestNumericalInversePolynomial.test_basic_all_scipy_distsr1cCXGddd}||_tj||dt|ddgdWddS1s%wYdS)Nc@r2)z9TestNumericalInversePolynomial.test_bad_pdf..distNr3r#r#r#r$r/Er4r/rRrrr)r%rWrXr r6r#r#r$r7C "z+TestNumericalInversePolynomial.test_bad_pdfzlogpdf, err, msgcCr)Nc@r2)z.distNr3r#r#r#r$r/Mr4r/rRrrr)logpdfrWrXr )r!rrYrZr/r#r#r$test_bad_logpdfKrz.TestNumericalInversePolynomial.test_bad_logpdfrQcCr:r;)rWrXr rr<r#r#r$r=Ur>z3TestNumericalInversePolynomial.test_inf_nan_domainsrrO'rrrrKrLrJrcCt}t|dd}t+}|td|td|td|td||}tj|}Wdn1s:wYt||ddd|j |j ksOJdS N+= u_resolutionrLrMrNrOgdy=r) rr rr-rrrrrrr!rr/rqr.rRrSr#r#r$test_ppfi       z'TestNumericalInversePolynomial.test_ppfrDr7r6rr"cCrr) rr rr-rr)rrrr)r!r"r/rqr.rRrSr#r#r$test_cdfrz'TestNumericalInversePolynomial.test_cdfcCsjt}t|dd}|\}}|dksJ||ksJt|dd}|\}}|dks-J||ks3JdSNg|=rr)rr u_error)r!r/rq max_errormaer#r#r$ test_u_errors       z+TestNumericalInversePolynomial.test_u_errorg@g#B ;ordercCsJt}d}tjt|dt||dWddS1swYdS)Nz2`order` must be an integer in the range \[3, 17\].rRr)rrWrXrxr )r!rr/rZr#r#r$test_bad_orderss "z.TestNumericalInversePolynomial.test_bad_ordersrcCsFd}tjt|dtt|dWddS1swYdS)Nz.`u_resolution` must be between 1e-15 and 1e-5.rRr)rWrXrxr r)r!rrZr#r#r$test_bad_u_resolutions "z4TestNumericalInversePolynomial.test_bad_u_resolutioncCsGddd}|}d}tjt|d t|}Wdn1s"wYt}t|}d}tjt|d |dWdn1sGwYGddd}|}t|}d }tjt|d |WddS1sswYdS) Nc@rV)z=TestNumericalInversePolynomial.test_bad_args..BadDistcSs tj|Sr'rr r#r#r$r)rzATestNumericalInversePolynomial.test_bad_args..BadDist.cdfN)r,r-r.r)r#r#r#r$BadDistrXrz9Either of the methods `pdf` or `logpdf` must be specifiedrRz4`sample_size` must be greater than or equal to 1000.r7c@rV)zBTestNumericalInversePolynomial.test_bad_args..DistributioncSstd||S)Nr)rrr r#r#r$r%szFTestNumericalInversePolynomial.test_bad_args..Distribution.pdfNrWr#r#r#r$r{rXr{z!Exact CDF required but not found.)rWrXrxr rr)r!rr/rZrqr{r#r#r$rs&   "z,TestNumericalInversePolynomial.test_bad_argscCshGddd}|}dd|_t|}|}dd|_t|}tjdddd }t||||dS) Nc@r2)zJTestNumericalInversePolynomial.test_logpdf_pdf_consistency..MyDistNr3r#r#r#r$MyDistr4rcSst| |dSNrP)mathrr9r#r#r$r:szLTestNumericalInversePolynomial.test_logpdf_pdf_consistency..cSs| |dSrr#r9r#r#r$r:srPgwJ?rar)r%r rrrrr)r!rdist_pdfrh dist_logpdfriqr#r#r$test_logpdf_pdf_consistencys  z:TestNumericalInversePolynomial.test_logpdf_pdf_consistencyN)-r,r-r.r rr'rrZr[r\rrr]r^r_rWrrr`r/rrxslowrrrbr7bad_logpdfs_commonrrdr=rrFrDrrr"rr bad_ordersbad_u_resolutionrrrrr#r#r#r$rsb              "        rc @seZdZGdddZGdddZeegZddejddd ejejgZd d ddejejd gZ ee gZ e j d eee e j dddgddZe j deddZddZe j de j je j deddZe j dddZddZd d!ejd!gZeejd!d efd"d#gZe j d$ee j d%ed&d'Z d(d)Z!d*d+Z"ej#d,d-d.d/gggej$gej% ej$ej%gd!ej$d0d1gd2d3ej%ggd4ggZ&e j d5e&d6d7Z'd8d9Z(d S):TestNumericalInverseHermitec@r) z!TestNumericalInverseHermite.dist0cCsddtdtj|S)Nrrrrr r#r#r$r%sz%TestNumericalInverseHermite.dist0.pdfcCstjtdtj|S)Nrrr r#r#r$r&rz&TestNumericalInverseHermite.dist0.dpdfcCs4ddtjd|tdtj|dtjS)NrrrOrtrr r#r#r$r)s4z%TestNumericalInverseHermite.dist0.cdfcCrGr r#r r#r#r$rrzz)TestNumericalInverseHermite.dist0.supportNr r#r#r#r$r rr c@r) z!TestNumericalInverseHermite.dist1cCs\|dkrtdtj|dtjS|dkrdS|dkr,tdtj|dtjSdS)NrrrrJrr r#r#r$r%sz%TestNumericalInverseHermite.dist1.pdfcCs`|dkrtdtj|tjtjS|dkrdS|dkr.tdtj|tjtjSdS)NrrrJrrrrr r#r#r$r&s  z&TestNumericalInverseHermite.dist1.dpdfcCsX|dkrddtdtj|S|dkrdS|dkr*ddtdtj|SdS)Nrg?rOrrJrrrr r#r#r$r)sz%TestNumericalInverseHermite.dist1.cdfcCrG)N)r rr#r r#r#r$r rzz)TestNumericalInverseHermite.dist1.supportNr r#r#r#r$rs  rr rPrrOrgпg?g?r*rr6rcCst||dd}t|||dS)Nr,)rr`)r r)r!r/rrrqr#r#r$r/sz&TestNumericalInverseHermite.test_basicrQcCr:r;)rWrXr rr<r#r#r$r= r>z0TestNumericalInverseHermite.test_inf_nan_domainsc Cshd}hd}||vrtd||vrtdtjdtt||}t|}tj d}t t | || |t | |}t t | | ||} |dks_J| dkseJdS) N>rrskewnormr> rnctbetawaldrinvgaussrr geninvgauss norminvgauss genhyperbolicrzDistribution is too slowz)Fails - usually due to inaccurate CDF/PDFrr7:0yE>r)rWrlrrrcrdrUrr randmaxr rr)) r!rpshapes slow_distsrr/fnir"p_tolu_tolr#r#r$basic_test_all_scipy_dists%s    0  z6TestNumericalInverseHermite.basic_test_all_scipy_distszignore::RuntimeWarning)rprcCs|||dSr'r)r!rprr#r#r$r>sz6TestNumericalInverseHermite.test_basic_all_scipy_distscCs|dddS)N truncnorm)rrPrr r#r#r$test_basic_truncnorm_gh17155Fsz8TestNumericalInverseHermite.test_basic_truncnorm_gh17155cCsd}tjt|dttddWdn1swYd}tjt|d tdWdn1s8wYd}tjt|dttdd WddS1sYwYdS) Nz"`order` must be either 1, 3, or 5.rRrPrz`cdf` required but not foundrz!could not convert string to floatekkir)rWrXrxr r)r!rSr#r#r$rJs "z1TestNumericalInverseHermite.test_input_validationNr))r)rrrrqrc Cstt}t|}t|}|j||d}|dur|j|ksJ|dur8t|}|j|d}tj |} t || dSdS)N)rr`)r) rr rrbrrrrrrr) r!rqrrr/rrirbrrkr#r#r$test_RVS\s  z$TestNumericalInverseHermite.test_RVScCsZd}d}tjt|dttj|Wdn1swYttj|dddS)N)gdsz@g@e?zg98 : one or more intervals very short; possibly due to numerical problems with a pole or very flat tailrRrr)rWrrr rr)r!rrSr#r#r$test_inaccurate_CDFms z/TestNumericalInverseHermite.test_inaccurate_CDFcCs<t}t|}t}t|}t|jdd|jdddS)Nrro)rr rrrrb)r!rfni1r'fni2r#r#r$test_custom_distribution{s z4TestNumericalInverseHermite.test_custom_distributionrJrrrrrrKrLrJrcCst}t|dd}t+}|td|td|td|td||}tj|}Wdn1s:wYt||ddd |j |j ksOJdS) NrrrLrMrNrOg& .>gLa㧝=r) rr rr-rrrrrrrr#r#r$rrz$TestNumericalInverseHermite.test_ppfcCst}t|dd}|\}}|dksJ||ksJt}|tt|dd}Wdn1s4wY|\}}|dksEJ||ksKJdSr)rr rrr-r)r!r/rqrrr.r#r#r$rs       z(TestNumericalInverseHermite.test_u_error))r,r-r.r rrZrrr[r\r_rWrrr`r/rdr=rfilterwarningsrrrrrrcrerngsappendrrrrrrrrFrDrrrr#r#r#r$rsN &               rc @s eZdZhdZddZddZddZej de d d Z e j d d d dggge jge j e je jgd e jddgdde jggdggZej deddZej deddZe j e jfe je jfe j e j fd e jfe j d fgZej deddZdS)TestDiscreteGuideTable>rfrgrhcCPgd}tj}ttt||ddWddS1s!wYdS)Nrg333333?g333333?r5r` guide_factorrrcrrWrrr r!rrr#r#r$$test_guide_factor_gt3_raises_warning   "z;TestDiscreteGuideTable.test_guide_factor_gt3_raises_warningcCr)Nrrrrrr#r#r$%test_guide_factor_zero_raises_warningrzz,TestSimpleRatioUniforms.test_inf_nan_domainscCsDtjtddttdddWddS1swYdS)Nz`pdf_area` must be > 0rRrr )r3pdf_area)rWrXrxr rr r#r#r$rJs"z%TestSimpleRatioUniforms.test_bad_argsN)r,r-r.r/rZrrDr\r]r_rWrrr`r/rdr=rr#r#r#r$r s     r )JrrrWcopyrrrrnumpyr numpy.testingrrr numpy.librscipy.stats.samplingrrr r r r r scipyrr scipy.statsrrscipy.stats._distr_paramsrrscipy._lib._utilrr all_methodsimplementationnameversionfloaterrrrrbrcrrFrDrrxbad_sized_domains bad_domainsrd nan_domainsrrr^rnrrrrrrrrrrrrrerrrr r#r#r#r$s  $                               1      [?nyWi