o ?Hh 6 @snddlZddlZddlmZddlZddlmZddlZddlm Z m Z ddl m Z m Z mZmZddlmmZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)m*Z*m+Z+dd lm,Z,m-Z-GdddZ.dDddZ/e)e,e-e*gZ0GdddZ1ddZ2ddZ3ddZ4ddZ5ddZ6ddZ7d d!Z8ej9:d"e,fej9:d#e;d$gej9:d%e;d&gej9:d'd(gej9:d)ejej?gd*d+Z@d,d-ZAd.d/ZBd0d1ZCd2d3ZDGd4d5d5ZEGd6d7d7ZFGd8d9d9ZGGd:d;d;ZHGdd?d?ZJGd@dAdAe#ZKGdBdCdCZLdS)EN)deepcopy)inf)assert_allclose assert_equal) strategiesgivenreproduce_failuresettings)stats)_kolmogorov_smirnov)kolmogn)qmc)distcont) _Domain _RealDomain _Parameter_Parameterization_RealParameterContinuousDistributionShiftedScaledDistribution_fiinfo_generate_domain_supportMixture)StandardNormal _LogUniform_Gamma)NormalUniformc @s*eZdZejdZddZej dej ddddej ej gdd Zej jeejd d d eeed ddZej dej ej dddfdgddZej jeeejddejddeeejeejddejddeeejeedddZddZ dS) Test_RealDomainl@x cCsLtdd}d}tjt|d|tWddS1swYdS)Nab endpointsz0The endpoints of the distribution are defined...match)rpytestraises TypeErrorget_numerical_endpointsdict)selfdomainmessager.a/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/stats/tests/test_continuous.pytest_ivs  "zTest_RealDomain.test_ivx sizecCs@tj tj}}t||fdd}t||||k||k@dS)N)FTr# inclusive)nprpirrcontains)r+r1r r!r,r.r.r/test_contains_simple#s z$Test_RealDomain.test_contains_simpler5r num_shapesmin_side)shapes inclusive_a inclusive_bdatac Cs$|\}}|\}}} tddddd} tddddd} |tjt|| d} |tjt|| d} | | }tt| || dt| | dt| | |dg}td ||fd }| t d td t d td | |t| | d }|rxdnd}|r~dnd}t d|d|d}t ||dS)NFg@) allow_nanallow_infinity min_value max_valuer4g@@)elementsr2rr9r r,r!z<=slowrrOmutually_broadcastable_shapesrbooleansrErbrfone_ofdecimals characters sampled_fromlistrrnrrrvr.r.r.r/rsF    (      rcCsR|}|dkr/|tjd|dd}||}|tj||d\}} |j||||d} n |j|d} t} |tj | |d} | j j| | j ||dd } t | | } |tj | |d}| j j|| j ||dd }t || }td d }td |d }|j| ||d}t jddd t |}Wdn1swY| | |||| | |fS)NrrF)rJr?)r| proportionsi_parameterizationr|)rAtypical)parameter_valuesrr|regionrsTTprL)rr|ignoredivideinvalid)_num_parameterizationsrNrintegers_num_parametersrOr_drawtuplebroadcastable_shapes _variable _parametersr;broadcast_shapesrrerrstatelog)familyrEr|rrAn_parameterizationsi n_parametersrBrWdistx_shaper1x_result_shapey_shapeyxy_result_shapep_domainp_varrlogpr.r.r/draw_distribution_from_familysF            rc@seZdZejdeddejdee e e j ddddd Z ejd ejdeejd d d dhdfdd dhdfddhdfddhdfddhdfddhdfddhdfddhdfdd hdfdd hdfdhddfdhddfd hddfd!hddfd"d#d$hd%fd&d#d$hd'fgedde e e j dddd(d)Zd*d+Zejd,dd!gd-d.Zejd,dd!gd/d0Zd1d2Zd3d4Zd5d6Zejd,dd gd7d8ZdS)9TestDistributions<) max_examplesrrrIrEseedcCstj|}d}t||||}|\}}} } } } } }|tjdddd}tjddd)t|t || t |d|| |t j d|d }t |d|| |WddS1sWwYdS) Nffffff?皙?rrrr)min_dimsrAmax_siderrrsamplerFr]r) r;rzr{rrNrO array_shapesr check_supportcheck_moment_funcscheck_sample_shape_NaNsr Halton)r+rrErr|rtmprr1rrrrWrr sample_shapeqrngr.r.r/test_support_moments_samples   "z-TestDistributions.test_support_moments_sampler2zfunc, methods, argentropylog/exp quadratureN logentropymedianicdfmode optimizationmeancachevarianceskewnesskurtosispdfr1logpdflogcdf> complementrrcdflogccdfccdfilogccdfr inversionriccdfrcCs|tkr |dkr tdtj|}d}t||||} | \} } } } }}}}| | | d}tjdddd#|durBt| |d||n||vrPt| |||||Wdn1sZwY|dkrnt | d| tjddddYt | t s|d krd h}t| d | | ||t| d | | ||n1|d krdh}t| d | | ||t| d | | ||WddSWddSWddSWddS1swYdS)Nrz'Mode is not unique; `method`s disagree.r)r1rrr)rroverrr4rrFTraddition)rr&skipr;rzr{rrcheck_dist_funcrstandard_deviationr isinstancer check_cdf2 check_ccdf2)r+rrErfuncmethodsargr|rrrr1rrrrWrrargsr.r.r/ test_funcssB    "zTestDistributions.test_funcscCsNzddlm}Wn tyYdSwtddd}|}||ks%JdS)Nr?r)matplotlib.pyplotpyplot ImportErrorrplotgca)r+pltXaxr.r.r/ test_plots  zTestDistributions.test_plot method_namec Cst}d|_tddg}t||}t|d|}||dd}||dd}||dd}t|d |d |d |d ks?Jt|d |d |jd dS) N-q=-C6?MbP?rformulamethodcomplement_saferrFrrtolr rtolr;asarraygetattrrr) r+rrrrifuncr1p1p2r.r.r/test_complement_safes    z&TestDistributions.test_complement_safecCst}d|_tddg}t||}t|d|}||dd}||dd}t|d|d|d |d ks9Jt||d |d |jd dS) NrrrrrrrrFrrr)r+rrrrrx1x2r.r.r/test_icomplement_safes    z'TestDistributions.test_icomplement_safecCst}d|_gd}gd}|j||dd}|j||dd}|j||dd}t||t|||jdtgd }tgd }|d ||}|j||dd}|j||dd}t|d t|||jddS) Nr)r2 )r r r r2rrsubtraction_safe subtractionr)#B ǻO 㒻#B ;O ;r)rrrrrr) r rrrrrr;rr)r+rr1rp0rrr.r.r/test_subtraction_safes   z'TestDistributions.test_subtraction_safecCsGdddtj}|gdd}tjdd|jdd}|jd d}Wdn1s-wY|jd d}d d g}t||sGJ|d |d ksQJt||||dS)NceZdZfddZZS)z7TestDistributions.test_logentropy_safe.._Normalcs.ttjdi|}d|d<tj|d<|S)Nrr.)r;rsuper_entropy_formular)r+paramsout __class__r.r/r8s zHTestDistributions.test_logentropy_safe.._Normal._entropy_formula)rwrxryr __classcell__r.r.rr/_Normal7rrFr4r5sigmarr logexp_saferlogexprrrrF)r rr;rrisinfallr)r+rrres1res2rai_flr.r.r/test_logentropy_safe5s  z&TestDistributions.test_logentropy_safecCstjgdd}gd}gd}tjdd|j||dd}|j||d d}Wdn1s1wY|j||d d}d d g}t||sMJ|d |d ksWJt||||dS)Nrr )rF,)ir4-rr"r#rr$rrrrF)r rr;rrr%r&r)r+rr1rr'r(rar)r.r.r/test_logcdf2_safeJsz#TestDistributions.test_logcdf2_safecCstjdd}|dkrddgnddg}t||}tjdd||d d }||d d }Wdn1s4wY||d d }|d |d ksIJ|d |d ksSJ|d|dks]J|d|dksgJdS)Nr4r rr+rFr-rr"r#rr$rr)r rrr;r)r+rrr1rr'r(rar.r.r/test_logexp_safeZs    z"TestDistributions.test_logexp_safe)rwrxryr&r} fail_slowr r~familiesrrrErrrrrrrr*r.r/r.r.r.r/rsP                    !   rc Cs ||}|dkr |j}dh}|d|dr"t|tjs"|d|D]_}||||d} tt|| j } t | j |tj | j |j |dkrPt| sPJtt| | s\Jt | | tj|||dd} |||dd} tt| | r~Jt | | q$dS) Nrinverse_transform__formular)rr|r.*)r _overridesrr QMCEngineaddr; broadcast_toget_valid_parametersshapertestingdtype_dtypeisscalarr&isfinitenananyequal) rfnamerrWr| full_shape sample_methodrrr^valid_parameterssample1sample2r.r.r/ris,    rcCsl|\}}t|dd|t|dd||j|jksJ|j|jks$J|j|jks,J|j|jks4JdS)Nsupport)rJcheck_nans_and_edgesr;_shaper=r>)rr r!r.r.r/rs rc Cs|durtn|f}|}d|vr1ttt|||ddiWdn1s,wYt|||}t||||ddi}|dvrLddi}n|dvrUddi}n|dvr]d d i}|d |d rk|d t j |j ||tkrt |sJ|D]J}t|||d|i} d|vrt j jt | t |fi|n t j j| |fi|t j | j|jt j | j ||tkrt | sJqdS)NrratolV瞯<>rlogmeanr logskewness>rư>>rrgH׊>r3r4rr)rcopyr&r'NotImplementedErrorrrKr6r8r;r<rr;r?rexpr=) rrDrrWrrra tol_overriderr^r.r.r/rsB      rc Cs|}|r/|dr|d|ds|dr|d|ds)|dr.|dn(|d r9|d|dsC|drH|d|dsR|drW|d||||}tj|j||tkrut |suJ|D]K}|rt |j |||d n|j|||d }tjj ||d d |rtj|j |d j n tj|j |j tj|j||tkrt |sJqwdS)N_logcdf2_formular_logcdf_formula_logccdf_formular  _cdf_formula _ccdf_formular _cdf2_formular+=rM)rRr6r8rr;r<rr;rr?rTrrr= rrr1rrWrrarr^r.r.r/rsH            rc Cs|}|d|r dnddr|d||||}tj|j||t kr4t |s4J|D]=}|rEt |j |||dn|j|||d}tjj ||ddtj|j|jtj|j||t krst |ssJq6dS) Nr3r ccdf2_formularrr\r])rRr6r8rrr;r<rr;rr?rTrrr=r_r.r.r/rs$   rcCst|}|dvrtddd}n|dvrtt dfdd}n|jj}t|||}tj|g|R^}}|\}}} } ||@} t||tj t|| tj | \} } t | |j } t | |j } | dk|@}| dk|@}|dk|@}|dk|@}|d vrt||tj t||tj t|||@tj t|||@tj n|d vrt||dt||dt|||@dt|||@dn|d vrt||t t||dt||t t||dn|d vrt||dt||dt||dt||dn|d vr9t||dt||t t||dt||t n|dvr[t||dt||dt||dt||dng|dvrt|| dktj t|| dktj t||dk| |dkt||dk| |dkn3|dvrt|| dktj t|| dktj t||dk| |dkt||dk| |dk|dvrt || |dk@sJdSdS)N>rrrsrr9>ilogcdfrrrrF>r>r>r>r>r>r>rrb>rr>rrOrJrrP)r:rrrr, classify_argr;broadcast_arraysrrArJr9r;r@r&)rrDrr^rG arg_domainclassified_args valid_arg endpoint_arg outside_argnan_arg all_validr r!outside_arg_minusoutside_arg_plusendpoint_arg_minusendpoint_arg_plusr.r.r/rK s              rKc sddfdd }fdd}tdD]O}||dd d d j|dd d }tdd||j|ks8J||dd |dd ||dd|||dd ||dd||dkd jtjkrg||dd|dd qddddtddD](}j|dd d }||dd|||dp||dd |d||dd|q}tdD]x}||dd d d j|dd d }|j|ksJ||dd |dd ||dd|||dd ||dd||dkd jtjkr||dd|dd jtjkr|dks||dd|||dp|dkd ||ds&|d||dd|qddddddtddD]$}j|dd d }||dd|||dd |d||dd|qBjddd d }tdD]?}||dd d d j|dd d ||d}|j|ksJ||dd|||dd ||dd||dkd ||dd|qwt t rdSdS)Ng& .>Tcs~|rj|||d}t||d|d|j|jksJdSttj|||dWddS1s8wYdS)Nrr2r])momentrr;r&r'rS)orderkindrrasuccessr^rMrr.r/check`s "z!check_moment_funcs..checkcsDd|d}|}|sdSt|}t|dttd}||vS)N_moment_r4Forders)r6rsetrange)rqrr formula_name overridesrrw)rr.r/ has_formulais   z'check_moment_funcs..has_formularxrawrF)rsrrrprgeneralrquadrature_icdfcentralrFr4 transform standardizedrdr5 normalize)NNT) reset_cacherzrprKr;rr rrrr)rrWrur}rravarr.rtr/rXs                    rrr)r4r5 dist_shape)r6rFrDrrng_typecCstjd}|}|r|dkrtd|||}d}|f|} | |} |dkr-|j} |tjjkr9|d|d}| | |d} | j | ksFJt || dd } t | j d| d d }d }|j }t ||k}|||kskJdS) NlMLyrz3Distribution can't have a shape without parameters.irrFrr)axisF)rg{Gz?)r;rzr{rr&rrr Generatorr;r r r8sum)rrrrDrr|num_parametersrn sample_sizesample_array_shaperFr1 statisticpvalue p_threshold num_pvaluesnum_small_pvaluesr.r.r/test_sample_against_cdfs(        rc Cs|j}i}|jD]}||jqtj|jtd}|D]A\}}||vr&q||}|j j |d\}} |j j \} } | r?||kn||k} | | rJ|| kn|| kM} t | |j j ||d|| M}qt ||j|S)Nr=r)r_parameterizationsupdate parametersr;onesrLboolitemsr,r)r:rr=_invalid) rrrparameterizationrknamevalue parameterr r! a_included b_includedvalidr.r.r/r:s*    r:cCs.|dur*tj|jtd}tj|jtd}tj|jtd}tj|jtd}||||fS|j|jd\}}t|||\}}}|j\} } | rH||kn||k} | | rS||kn||kM} tj|j t d} d| ||k<d| ||k<tj|j t d} d| | r|||kn||k<d| | r||kn||k<t |}| | | |fS)NrrrrF) r;rrLrzerosr)rrdr:r;intisnan)rrre valid_args endpoint_args outside_argsnan_argsr r!rrinsideonoutsiderAr.r.r/rc s*       rccCsGdddt}d}tjt|d |ddWdn1s!wYd}tjt|d|tgdWdn1sBwYtjt|d|gd dWdn1s`wYtjt|d|tjdWdn1s}wYtjt|d |d dWdn1swYd }tjt|d|d Wdn1swYtjt|d|tjWdn1swYd }tjt|d|jd ddWdn1swYGdddt}d}tjt|d |ddWdn 1swYd}tjt|d |Wdn 1s;wYd}tjt|d|dd ggd dWdn 1s_wYd}tjt|d|gd gd d}| dd gWdn 1swYd}tjt |d|dt gdd gdWdn 1swYd}tjt|d|gd gd d}|j ddWddS1swYdS)Nc@seZdZededZdS)z#test_input_validation..Testr1rLN)rwrxryrrrr.r.r.r/Test=srz`The `Test` distribution family does not accept parameters, but parameters `{'a'}` were provided.r$rFr zAAttribute `tol` of `Test` must be a positive float, if specified.rrrzEArgument `order` of `Test.moment` must be a finite, positive integer.z2Argument `kind` of `Test.moment` must be one of...r4coconutrrc@sBeZdZededZededZeeegZededZ dS)z$test_input_validation..Test2crLr]r1N) rwrxryrr_p1_p2rrrr.r.r.r/Test2Zs  rzkThe provided parameters `{a}` do not match a supported parameterization of the `Test2` distribution family.zLThe `Test2` distribution family requires parameters, but none were provided.zjThe parameters `{c, d}` provided to the `Test2` distribution family cannot be broadcast to the same shape.)rr]zmThe argument provided to `Test2.pdf` cannot be be broadcast to the same shape as the distribution parameters.z$Parameter `c` must be of real dtype.z.MyUniformc_dSN incorrectr.r+rrr.r.r/rzATestAttributes.test_iv_policy..MyUniform._entropy_formulac[rrr.)r+rqrr.r.r/_moment_raw_formularzDTestAttributes.test_iv_policy..MyUniform._moment_raw_formulaN)rwrxryrrr.r.r.r/ MyUniforms rrr~)rrrqz/Attribute `validation_policy` of `MyUniform`...r$r) rrrr;rrrpr&r'r)r+rr r!rr-r.r.r/test_iv_policys "zTestAttributes.test_iv_policycCstjddd}tjdgdd}|jdksJ|jdksJ|jddks&J|jddks/Jtt d|_Wdn1sBwYd|jd<|jddksUJdS)NrFr4mur!r5rr2)r rrr!r&r'AttributeError)r+rYr.r.r/ test_shapess  zTestAttributes.test_shapesN)rwrxryrrrrr.r.r.r/rs ( rc@s6eZdZejdeeddZddZ ddZ dS) TestMakeDistributionz i, distdatac Cs|d}hd}ttjdds||vrtd|dvr dSddd}d d h}d d h}hd }hd} hddhdhd} d d h} tt|} | jrXt t | j d|dni} t j d}t| }|d)i| }| d)i| }|jd|d}||}||d}d}t jddd.|d\}}}}t||||vrt|||dt|||dt||||dt||||d||vrt||||d||vrt|jdd |||dt|||||dt|||||dt|||||dt|||||d|| vr4t|||||dt| ||!||dt|"||#||dt|$||%||dt&d!D]}|| |ivrzt|j'|d"d#|'|||dq_t&d$d%D]}|| vrt|j'|d&d#|d|d||dqd'}t|jd|d|j(dt j |d(|dWddS1swYdS)*Nr>argusksonekstwokappa4exponpowgenexpongompertz exponweib johnsonsb kstwobign powernorm halfgennorm powerlognorm recipinvgauss vonmises_linestudentized_range SCIPY_XSLOW0zSkipping as XSLOW>vonmises levy_stablegh㈵>)rrrpearson3rr>chirrrinvgammar>triangarcsine trapezoid skewcauchy>levyalphalevy_l foldcauchy halfcauchyparetorr3rmrFls4XFr2r;r|gHz>rrrmvskr)rrMexcessrrdr~rr5r6rl1XS+-)r8 random_stater.))rosenvironror&rrr rBr*zipsplitr;rzr{make_distributionrrrrrJrrrrrrrrrrlogsfrsfrppfrisfrzrprvs)r+rdistdatadistnamercustom_tolerances skip_entropy skip_skewness skip_kurtosis skip_logccdfskip_rawskip_standardizedrrr|CustomDistributionrrr1rrrMmvskrqrr.r.r/test_make_distribution s   $        $z+TestMakeDistribution.test_make_distributioncCsd}tjt|dttjWdn1swYd}tjt|dttjWdn1s9wYd}tjt|dttWddS1sYwYdS)Nz`levy_stable` is not supported.r$z`vonmises` is not supported.z4The argument must be an instance of `rv_continuous`.) r&r'rSr r rrrrr+r-r.r.r/rTs"z*TestMakeDistribution.test_input_validationcCsddlm}|D] }tt|sJq ttj}t|dddks%Jtj dkr5t |dddks5Jd|j vstjd}|d}|d}t||t||}}tjt||d}t||}|j d|d}| |}t | t |dt ||t ||dt ||t ||t |t|t ||dt ||d d t ||t ||||t | || |t ||||t ||||t ||||t ||||t | || |t |!||"|t |#t |||t |$t ||||%d }t&||k||k@sJdS) Nb2,'r5rF)lbubr5r2rr^?rrr5r2)'r;rzr{minimummaximumr truncater truncnormrrrrrrrrrrrrsqrtrrrJrrrrrrrrrr rbrrr&) r+r|r3r4rY0rrrr.r.r/ test_truncatevs:       zTestTransforms.test_truncater2rrrcCsRtj|}Gdddt}t|||ddd}|\}}}} } } } } |j}|j}t}tj ||d}|||}|||}| \}}| \}}t |||t |||tj ddd`t t ||t ||t |||t |||t |||t |||d t ||d t ||t|t ||t ||||t|t |||||t ||||t ||||t ||||t ||||t ||||||t ||||||t ||||||t ||||||t || || ||t | | | | ||t |!| |!| ||t |"| |"| ||t#dd D]4}t |$|d |$|t |$|d |$|d ||t |$|d|$|dt||qWdn 1swY|d ||}||d |}t%|j&}| \}}| \}}t |||t |||tj dddt |||t |||t |||t |||t |||t |||t |||t |||t |||t |||||t |||||t |||||t |||||t |||||t |||||t || || |t | | | | |t |!| |!| |t |"| |"| |t#dd D]*}t |$|d |$|d t |$|d |$|d t |$|d|$|dqWddS1s"wYdS)Ncr)z8TestTransforms.test_loc_scale..TransformedNormalcs tjtg|Ri|dS)N)r__init__rrrr.r/r?s zATestTransforms.test_loc_scale..TransformedNormal.__init__)rwrxryr?rr.r.rr/TransformedNormalrr@)rFrrrrF)rrA)locscalerrr4r7rdr~rr)'r;rzr{rrrArBrr normrJrrrTrrrrrrrrsignrrrrrrrrrbrrrrzrprrL)r+rErr|r@rrr1rrrrWrrrArBdist0dist_refx0y0r r!a0b0rzr.r.r/test_loc_scales         !     $zTestTransforms.test_loc_scalerdexp_powrTpowc Cs@tjd}|d}|d}t||}|dkr!t|}ntj|}tj|t|d}|jd|d}| |} t | t | t | | t ||dt ||t ||t |t|t ||dt ||d d t ||t ||||t | || |t ||||t || || t || || t ||||t ||||t | ||!|t |"t | || t |#t | || d } t |j$| d t|j$| d dS) Nr1r2rT)rBr5r6r7rrr5M[r)%r;rzr{rr rTelognormrrrrrrrrrrrrr<rrrJrrrrrrrrrr rbrr) r+rMr|rr!rrr=rrrr.r.r/ test_exp_pows>      $zTestTransforms.test_exp_powrB)rFr4rz/`scale=-1` fails on 32-bit; needs investigationc Cstjd}|d}t|t|}}ttj}||||d}|dkr-t|dnt|d }||}|j d|d} | | } t | } t | t |t ||t ||t ||t || || t | | | | t || || t || || t || || t || || t || || t || || tjdddt || || t || || Wdn1swYd } t |j | d ||j | d dS) Nr1r2rrr5rrrrOr)r;rzr{rDrr r rrrrrrrrrrJrrrrrrrrrbr) r+rBr|r rDInvGammar=rrrrrrr.r.r/test_reciprocal&s8     "zTestTransforms.test_reciprocalc CsBtjd}|d}t|d}t|}t|}|jd|d}||}t | t| t | | t | | dt ||t ||t |t|t ||dt ||dd t ||t ||||t ||||t ||||t ||| |t ||||t ||||t ||||t ||||tj d d $t |!t|| |t |"t|||Wdn 1swYd }t |j#|d t|j#|d dS)Nr1r2rr5r6r7rrr5r)rrOr)$r;rzr{rr loggammarrrrrrrrrrrrr<rrrJrrrrrrrrrr rrbrr) r+r|r rr=rrrrr.r.r/test_logQs<      $zTestTransforms.test_logcCsFtddd}d}tt|d|dksJtd|d|dks%Jtt|d|dks4Jtd dd}d }tjt|d  d|Wdn1sRwYd }tjt|d  t|Wdn1spwYd }tjt|d  d|Wdn1swYtjt|d  d|Wdn1swYtjt|d ddg|Wdn1swYd}tjt|d  |dWdn1swYtjt|d  |dWdn1swYtjt|d |ddgWddS1swYdS)NrFr4rzUniform(a=1.0, b=2.0)zlog(rlz1/(zexp(rz=Division by a random variable is only implemented when the...r$zBThe logarithm of a random variable is only implemented when the...z@Raising an argument to the power of a random variable is only...r7?z=Raising a random variable to the power of an argument is onlyr)rrer rrTr&r'rS)r+rX_strr-r.r.r/test_monotonic_transformsssD        $z(TestTransforms.test_monotonic_transformsc Cstjd}d\}}}}|ddd}t||d}|||}t|||||||||}t|||||||||}t|||||||t||d|}t|||| ||dS)NllO[5)g7A`?gq= ףp?g= ףp=?g(\?r5dr)r;rzr{rrrrr) r+r|r r!rArBr1rrr.r.r/test_arithmetic_operatorss      $z(TestTransforms.test_arithmetic_operatorscCstjd}|d}tt|}t|}|jd|d}||}t | t | dt | | t | |dt ||t ||t |t|t ||dt ||dd t ||t ||||t ||||t ||||t ||||t ||||t ||||t ||||t ||| |t |!t |||t |"t ||||#d }t$|d ksJdS) Nr1r2r5r6r^r7rrr5r2r)%r;rzr{r rrfoldnormrrrrrrrrrrrrr<rrrJrrrrrrrrrr rbrrr&)r+r|rArr=rrrr.r.r/test_abss6     zTestTransforms.test_abscCsttj}|dd}t| }t|d|dt|d|dt|d|dt|d|dt| d| dt| d| ddS)Nr4)rrF) r r  weibull_minrrrrrrrr)r+Weibullrrr.r.r/test_abs_finite_supports   z&TestTransforms.test_abs_finite_supportcCstjd}td}tjdd}|jd|d}||}t| t | ddd t| | dd t| | t| | t||t|t|t||d t||d d t||t||||t||||t||||t||||t||||t||||t||||t||||t| t |||t|!t ||||"d}t#|d ksJdS)Nr1r4rF)dfr2r6r^rQrrrr5r)$r;rzr{rr chi2rrrrrrrrrrrr<rrrJrrrrrrrrrrr rbrrr&)r+r|rr=rrrr.r.r/test_pows4      zTestTransforms.test_powN)rwrxryr>r&r}r0rrrErrLr~rRxfail_on_32bitrTrVrZr]r_rbrer.r.r.r/r0ss& "  d  &  ( !" r0c@s0eZdZejdddZddZddZdS) TestOrderStatisticrc CsXtjd}tddd}d}tdgdgdgg}tj|||d}t||d|}|jd|d }| |}t t | | t | | t ||t ||t ||d d d t ||d dd d t ||dt ||t ||||t |j |dd|j |ddt |j|dd|j|ddt |j|dd|j|ddt |j|dd|j|ddt ||||t ||||t ||||tjddd$t |t |||t |!t ||"|Wdn 1s"wYd} t#j$t%| dtj||ddWdn 1sDwYt#j$t%| dtj|d|dWdn 1sdwYt#j$t%| dtj||ddWdn 1swYt#j$t%| dtj|d|dWddS1swYdS)NlCvu=rrFrrdr5)rrr5r6rrNr]rr7rrrrrrz0`r` and `n` must contain only positive integers.r$rrX)&r;rzr{rrr order_statisticr%rrrrTrrrrrrrrrrJrrrrrrrr rrbrrrr&r'r) r+r|rrrhrr=rrr-r.r.r/test_order_statisticsR   $z'TestOrderStatistic.test_order_statisticcCsDttj}|}|dd}tj|ddd}t||dS)Nrdr4r5rhr)r r rrirrJ)r+rrrZr.r.r/test_support_gh22037 s  z'TestOrderStatistic.test_support_gh22037c Csttj}ttj}ddgd}}ddgdgg}}dggdggg}|}t|||} tj| ||d } |||d } tj| ||d } tj| || |dS) NrWrrFr5r6rdg333333ӿrrkr) r r rCr;r:rir;r<rr) r+rTruncatedNormalr r!rhrr1rY1Z1rZ2r.r.r/test_composition_gh22037*s   z+TestOrderStatistic.test_composition_gh22037N) rwrxryr&r}r0rjrmrrr.r.r.r/rgs   * rgc@sfeZdZddZddZejdej ej fej ej fgddZ dd Z d d Zd d ZddZdS)TestFullCoveragecCstttddWdn1swYtttddWdn1s1wYtttdWddS1sLwYdSNr)r&r'rSrr=r)__str__r+r.r.r/ test_Domain?s    "zTestFullCoverage.test_DomaincCs<tttddWddS1swYdSrt)r&r'rSrvalidatervr.r.r/test_ParameterGs "zTestFullCoverage.test_Parameter)dtype_in dtype_outc Csbtd}td|d}tjddg|d}|j|id\}}}t||||ks(Jt|dd gdS) N)rrFr1rLr7g@rrTF)rrr;rrxr) r+rzr{r,rr1arrr=rr.r.r/"test_RealParameter_uncommon_dtypesKs   z3TestFullCoverage.test_RealParameter_uncommon_dtypescCsJGdddt}|}tj}||_tjdg|d}||j|ks#JdS)Nc@s&eZdZededddZddZdS)zNTestFullCoverage.test_ContinuousDistribution_set_invalid_nan..TestDistr1)rrr"rLc_r)Nrr.r+r1rrr.r.r/_logpdf_formula_rz^TestFullCoverage.test_ContinuousDistribution_set_invalid_nan..TestDist._logpdf_formulaN)rwrxryrrrrr.r.r.r/TestDist]s rr7r)rr;float32r>rrr=)r+rrr=r1r.r.r/+test_ContinuousDistribution_set_invalid_nanXs z.assert_allclose)r~rrrdrrNr])r2rrdrg?)!r;rzr{rrrrrrrrrrrrrrrrrrrbrrrrzrprr;r ks_1sampravelr) r+r;r|rrr1rrrrqrr.rr/ test_basicsJ     "zTestMixture.test_basiccCsZd}ttj}||d}t|| f}tddd}t||tj|d|dS)Nrrr6r,) r r r*rr;rSrrdgamma)r+r r$rrr1r.r.r/test_default_weights1s   "z TestMixture.test_default_weightscCstdddtdddg}d}t||d}tt d|_Wdn1s'wYtt d|_Wdn1s?wY|d |jd <|jd |d ksVJ|d |jd <|jd |d kshJdS) Nrrrr7rrrr2rFr)rrr&r'r componentsr)r+rrrr.r.r/test_properties9s   zTestMixture.test_propertiescCstjd}ttj}|}t||g}|jdd}tj| || |tj| || |tj| || |tj| || |dS)Nl nppRr2r7) r;rzr{r r cauchyrr<rrrrbr)r+r|CauchyX0rrr.r.r/ test_inverseKs   zTestMixture.test_inverseN) rwrxryrr&r}r~rrrrr.r.r.r/rs! + r)r)MrrrRrrr;rr& numpy.testingrr hypothesisrrrr hypothesis.extra.numpyextrarOscipyr scipy.stats._fitr scipy.stats._ksstatsr rr scipy.stats._distr_paramsrr'rrrrrrrrrrrrrrrrrrr1rrrrrrrKrr}r~rrzrrSobolrr:rcrrrrr0rgrsrrrr.r.r.r/sl       0 g&= 6,K~'CzjHRN