o i 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.[/home/kim/smarthome/.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_simpler5rZ 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_nanZallow_infinity min_value max_valuer4g@@)elementsr2rr9r r,r!z<=slowrrMmutually_broadcastable_shapesrZbooleansrDrTrXZone_ofZdecimals charactersZ sampled_fromlistrr_rarcr.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)NrrE)rHr?)ri proportionsZi_parameterizationri)r@Ztypical)parameter_valuesrqriregionrbTTprJ)rqriignoredivideinvalid)Z_num_parameterizationsrLrintegers_num_parametersrMrn_drawtupleZbroadcastable_shapes _variable _parametersr;Zbroadcast_shapesrrerrstatelog)familyrDrirqr@Zn_parameterizationsiZ n_parametersrArPdistx_shaper1x_result_shapeZy_shapeyxy_result_shapeZp_domainZp_varrvlogpr.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<)Z max_examplesrrrGrDseedcCstj|}d}t||||}|\}}} } } } } }|tjdddd}tjddd)t|t || t |d|| |t j d|d }t |d|| |WddS1sWwYdS) Nffffff?皙?rrrr)Zmin_dimsr@Zmax_siderwrzrysamplerErQr) r;rgrhrrLrMZ array_shapesr check_supportcheck_moment_funcscheck_sample_shape_NaNsr Halton)r+rrDrrirqtmprr1rrvrrPrr sample_shapeZqrngr.r.r/test_support_moments_samples   "z-TestDistributions.test_support_moments_sampler2zfunc, methods, argentropylog/exp quadratureN logentropymedianicdfmode optimizationmeancachevarianceskewnesskurtosispdfr1logpdflogcdf>r complementrcdflogccdfccdfilogccdfr inversionriccdfrvcCs|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)r1rvrrw)rzryZoverrr4rrFTrZaddition)rr&skipr;rgrhrrcheck_dist_funcrstandard_deviationr isinstancer check_cdf2 check_ccdf2)r+rrDrfuncmethodsargrirqrrr1rrvrrPrrargsr.r.r/ test_funcssB    "zTestDistributions.test_funcscCsNzddlm}Wn tyYdSwtddd}|}||ks%JdS)Nr?r)Zmatplotlib.pyplotZpyplot ImportErrorrZplotZgca)r+ZpltXaxr.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_saferrErrtolr rtolr;asarraygetattrrr) r+rrrvrifuncr1p1p2r.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) NrrrrrrrrErrr)r+rrrvrrx1Zx2r.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 )rrrr2rrZsubtraction_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)rdrerfr __classcell__r.r.rr/_Normal7rrEr4r5sigmarwry logexp_saferlogexprrrrE)r rr;rrisinfallr)r+rrres1res2rSi_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)rE,)ir4-rwrrrrrrrrE)r rr;rrrrr)r+rr1rrrrSrr.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)Nr4rrrrErrwrrrrrr)r rrr;r)r+rrr1rrrrSr.r.r/test_logexp_safeZs    z"TestDistributions.test_logexp_safe)rdrerfr&rj fail_slowr rkfamiliesrrrDr{rrrrrrrrrr.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) NrZinverse_transform__formular)rrir.*)r _overridesrr Z QMCEngineaddr; broadcast_toget_valid_parametersshapertestingdtype_dtypeisscalarrisfinitenananyequal) rfnamerrPriZ full_shape sample_methodrrrRvalid_parameterssample1sample2r.r.r/ris,    rcCsl|\}}t|dd|t|dd||j|jksJ|j|jks$J|j|jks,J|j|jks4JdS)Nsupport)rcheck_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׊>rrrr)r~copyr&r'NotImplementedErrorrrrrr;r rr rrexpr ) rrrrPrrrSZ tol_overriderrRr.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)NZ_logcdf2_formularZ_logcdf_formulaZ_logccdf_formularZ _cdf_formulaZ _ccdf_formularZ _cdf2_formular+=r)r rrrr;r rr r~rr"rrr  rrr1rrPrrSrrRr.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) NrrZ ccdf2_formularrr#r$)r rrrrr;r rr r~rr"rrr 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>rrrbrur9>ilogcdfrrrrE>r>r>r>r>r>r>r(r>rr>rrrrr)r rrrr, classify_argr;broadcast_arraysrrrrr rr)rrrrRr arg_domainZclassified_argsZ valid_argZ endpoint_argZ outside_argZnan_arg all_validr r!Zoutside_arg_minusZoutside_arg_plusZendpoint_arg_minusZendpoint_arg_plusr.r.r/r s              rc 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'r!)orderkindrrSsuccessrRrrr.r/check`s "z!check_moment_funcs..checkcsDd|d}|}|sdSt|}t|dttd}||vS)NZ_moment_rForders)rrsetrange)r.r/Z formula_nameZ overridesrr3)rr.r/ has_formulais   z'check_moment_funcs..has_formular4rawrF)r0rrr-rZgeneralrZquadrature_icdfcentralrEr4 transform standardizedrVr5 normalize)NNT) reset_cacher6r-rr rr rrrr)rrPr2r7rrSvarr.r1r/rXs                    rrr)r4r5 dist_shape)r6rErrrng_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.irrErrr)ZaxisF)rg{Gz?)r;rgrhr|r&rr}r Generatorr r r r8sum)rr?rrr@riZnum_parametersrnZ sample_sizeZsample_array_shaperr1Z statisticpvalueZ p_thresholdZ num_pvaluesZnum_small_pvaluesr.r.r/test_sample_against_cdfs(        rEc 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 rs)r_parameterizationsupdate parametersr;onesrboolitemsr,r)r:rr=_invalid) rrsrJZparameterizationr,namevalue 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)NrFrGrrE) r;rKrrLzerosr)rr*r:r intisnan)rrr+Z valid_argsZ endpoint_argsZ outside_argsZnan_argsr r!rRrSZinsideonZoutsiderr.r.r/r) s*       r)cCsGdddt}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..Testr1rJN)rdrerfrrrr.r.r.r/Test=srYz`The `Test` distribution family does not accept parameters, but parameters `{'a'}` were provided.r$rEr 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...r4coconutr/c@sBeZdZededZededZeeegZededZ dS)z$test_input_validation..Test2crJrQr1N) rdrerfrrZ_p1Z_p2rrHrr.r.r.r/Test2Zs  r_zkThe 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.)r^rQzmThe 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+rrgr.r.r/rzATestAttributes.test_iv_policy..MyUniform._entropy_formulac[rtrur.)r+r.rr.r.r/_moment_raw_formularxzDTestAttributes.test_iv_policy..MyUniform._moment_raw_formulaN)rdrerfrryr.r.r.r/ MyUniforms rzrvr8)r/r.z/Attribute `validation_policy` of `MyUniform`...r$rz) rrrsr;rrr-r&r'ra)r+rr r!rzr-r.r.r/test_iv_policys "zTestAttributes.test_iv_policycCstjddd}tjdgdd}|jdksJ|jdksJ|jddks&J|jddks/Jtt d|_Wdn1sBwYd|jd<|jddksUJdS)NrEr4murr5rr2)r rr}rr&r'AttributeError)r+rYr.r.r/ test_shapess  zTestAttributes.test_shapesN)rdrerfrlrqr{rr.r.r.r/ris ( ric@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>ZargusexponpowZ powernorm kstwobign halfgennormZ exponweibZgenexponkstwoZ powerlognorm johnsonsbZ recipinvgaussZgompertzZkappa4Zstudentized_rangeksoneZ vonmises_lineZ SCIPY_XSLOW0zSkipping as XSLOW>vonmises levy_stablegh㈵>)rrrZpearson3rr>rchirrinvgammar>Z skewcauchyZarcsineZ trapezoidZtriang>alphaZ halfcauchyZ foldcauchyZlevy_lZlevyZparetorr3r^rEls4XFr2r rigHz>rrwrxZmvskr)rrexcessr`rVr8r]r5r6r;l1XS+-)r8 random_stater.))rVosenvironr`r&rrr rAr*zipsplitr;rgrhmake_distributionrrrrrrrrrrrrrrrlogsfrsfrppfrisfr6r-rvs)r+rZdistdatadistnamermZcustom_tolerancesZ skip_entropyZ skip_skewnessZ skip_kurtosisZ skip_logccdfZskip_rawZskip_standardizedrrriZCustomDistributionrrr1rvrrmvskr.rr.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'r!r rrrrarbr+r-r.r.r/rcTs"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,'r5rE)lbubr5r2rr%?rrr5r2)'r;rgrhminimummaximumr truncater truncnormrrrrrrrrrrr>rsqrtrrrrrrrrrrrrrr(rrr) r+rirrrY0rrvrr.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__rrwrr.r/rs zATestTransforms.test_loc_scale..TransformedNormal.__init__)rdrerfrrr.r.rr/TransformedNormalrr)rErrrrE)rqr@)locscalerwrr4rrVr8r9r;)'r;rgrhrrrrrr normrrrr"rrrrrrrrsignrrrrrrrrr(rrrr6r-rUr)r+rDrrirrrr1rrvrrPrrrrZdist0Zdist_refZx0Zy0r r!Za0Zb0rzr.r.r/test_loc_scales         !     $zTestTransforms.test_loc_scalerVexp_powr"powc 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) Nrrr")rrrrrrr5M[rr)%r;rgrhrr r"eZlognormrrrrrrrrrrr>rrrrrrrrrrrrrrrr(rr) r+rrir}rrrrrrvrr.r.r/ test_exp_pows>      $zTestTransforms.test_exp_powr)rEr4rz/`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) NrrrZrrrrwrxrrr)r;rgrhrrnr rrrrrrrrrrrrrrrrrrrr(r) r+rrir rZInvGammarrrrrvrrr.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)NrrrZrrrrrr5rw)rzrrr)$r;rgrhrr Zloggammarrrrrrrrrrr>rrrrrrrrrrrrrrrrr(rr) r+rir rrrrrvrr.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)NrEr4rzUniform(a=1.0, b=2.0)zlog(r]z1/(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...r?z=Raising a random variable to the power of an argument is onlyr)rrWr rr"r&r'r!)r+rZX_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;rgrhrlrrrr) r+rir r!rrr1rrr.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) Nrrrrr%rrrr5r2r)%r;rgrhr rnrZfoldnormrrrrrrrrrrr>rrrrrrrrrrrrrrrr(rrr)r+rirrrrrvrr.r.r/test_abss6     zTestTransforms.test_abscCsttj}|dd}t| }t|d|dt|d|dt|d|dt|d|dt| d| dt| d| ddS)Nr4)r^rE) r rZ weibull_minrnrrrrrrr)r+ZWeibullrrr.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)Nrr4rE)Zdfr2rr%rrrrr5r)$r;rgrhrr Zchi2rrrrrrrrrr>rrrrrrrrrrrrrrrrr(rrr)r+rirrrrvrr.r.r/test_pows4      zTestTransforms.test_powN)rdrerfrr&rjrrrrDr{rrkrZxfail_on_32bitrrrrrrrr.r.r.r/rss& "  d  &  ( !" rc@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=rrErrVr5)rCrrrrrr$rrrrrrrwrz0`r` and `n` must contain only positive integers.r$rr)&r;rgrhrrr order_statisticrrrrr"rrrrr>rrrrrrrrrrrrrrr(rrrr&r'ra) r+rirrCrrrrrvr-r.r.r/test_order_statisticsR   $z'TestOrderStatistic.test_order_statisticcCsDttj}|}|dd}tj|ddd}t||dS)NrVr4r5rrC)r rrlrrr)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) NrrrEr5r6rVg333333ӿrrr) r rrrrrr;r rr) r+rZTruncatedNormalr r!rrCr1roZY1ZZ1rpZZ2r.r.r/test_composition_gh22037*s   z+TestOrderStatistic.test_composition_gh22037N) rdrerfr&rjrrrrr.r.r.r/rs   * rc@sfeZdZddZddZejdej ej fej ej fgddZ dd Z d d Zd d ZddZdS)TestFullCoveragecCstttddWdn1swYtttddWdn1s1wYtttdWddS1sLwYdSNr)r&r'r!rr=r)__str__r+r.r.r/ test_Domain?s    "zTestFullCoverage.test_DomaincCs<tttddWddS1swYdSr)r&r'r!rvalidaterr.r.r/test_ParameterGs "zTestFullCoverage.test_Parameter)dtype_in dtype_outc Csbtd}td|d}tjddg|d}|j|id\}}}t||||ks(Jt|dd gdS) N)rrEr1rJrg@rFrGTF)rrr;rrr) r+rrr,rQr1Zarrr rTr.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"rJc_rt)Nrr.r+r1rrgr.r.r/_logpdf_formula_rxz^TestFullCoverage.test_ContinuousDistribution_set_invalid_nan..TestDist._logpdf_formulaN)rdrerfrrrrr.r.r.r/TestDist]s rrrF)rr;float32r rrr )r+rrr r1r.r.r/+test_ContinuousDistribution_set_invalid_nanXs z.assert_allclose)r8r9r;rVr]rr$)r2rrVrrg?)!r;rgrhrrrrrrrrrrrrrrrrrrr(rrrr6r-rr r Zks_1sampZravelrD) r+r rirrr1rr/r.rr.r r/ test_basicsJ     "zTestMixture.test_basiccCsZd}ttj}||d}t|| f}tddd}t||tj|d|dS)NrrZr6r) r rrrr;rNrrZdgamma)r+r rrrr1r.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) Nr rr|rrr rr2rEr)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;rgrhr rZcauchyrr rrrr(r)r+riZCauchyZX0rrvr.r.r/ test_inverseKs   zTestMixture.test_inverseN) rdrerfrcr&rjrkrrrrr.r.r.r/r s! + r )r)Mrrdr rrr;rr&Z numpy.testingrrZ hypothesisrrrr Zhypothesis.extra.numpyextrarMZscipyr Zscipy.stats._fitr Zscipy.stats._ksstatsr rr Zscipy.stats._distr_paramsrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrjrkr~rgrArZSobolrEr r)rcrhrirrrrrrr r.r.r.r/sl       0 g&= 6,K~'CzjHRN