U md<@sddlZddlZddlZddlmZddlmZmZddl m Z ddl m mZddlmZddlmZd1ddZd d Zd d Zd dZddZddZddZddZddZddZddZddZ dd Z!d!d"Z"d#d$Z#d%d&Z$d'd(Z%d)d*Z&d+d,Z'd-d.Z(d/d0Z)dS)2N)assert_allclose assert_equal)raises)getfullargspec_no_self)statsFcCsHt|D]:\}}|r,t||t||qt||t||qdSN) enumeratema_nptrgetattrnpt)res attributesmaiattrrW/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scipy/stats/tests/common_tests.pycheck_named_resultssrc Cs|jd |}t|d|dkr,d\}}nd\}}|jdd|d}tj|d|||d d |j|\}}|j|f|} t| ddS) Nr?rv_histogram_instance)h㈵>r)Hz>rcSsdS)Nrxrrr z%check_normalization..)argsT)atolrtolerr_msgverbose)r)momentr rexpectsupportcdf) distfnrdistnameZ norm_momentrrZnormalization_expectZ_a_bZnormalization_cdfrrrcheck_normalizations   r)cCs|jd |}|jd |}t|s>tj||d|ddntt||dt|t|stj||||d|ddntt||dt|dS) Nr z - 1st momentdecimalr z - 1st moment -infinite, m1=%sz - 2ndt momentz - 2nd moment -infinite, m2=%s)r)r*)r"npisinfr assert_almost_equalassert_str)r&argmvmsgm1m2rrr check_moment)s        r9cCs4t|r0|dd|}tj||d|dddS)NcSs|Srrrrrrr=rz#check_mean_expect..z - 1st moment (expect)r,)r.isfiniter#r r0)r&r3r4r6r7rrrcheck_mean_expect;s   r<cCsJ|dkrddini}t|rF|dd|}tj||||f|dS)Nrrgh㈵>cSs||SrrrrrrrErz"check_var_expect..)r.r;r#r r)r&r3r4r5r6kwargsr8rrrcheck_var_expectBs r>csVt|rB|fdd|}tj||t|dd|ddntt|dS)Ncst|dS)Nr.powerrr4rrrKrz#check_skew_expect..g?r:z - skewr,)r.r;r#r r0rAr1isnan)r&r3r4r5sr6Zm3errBrcheck_skew_expectIs rEcsft|rH|fdd|}tj||dt|ddd|ddnt|sbtt|dS)Ncst|dS)Nr@rrBrrrTrz#check_kurt_expect..g@r*rz - kurtosis)rrr ) r.r;r#r rrAZisposinfr1rC)r&r3r4r5kr6Zm4errBrcheck_kurt_expectRs  rHcCs&|j|}tt| |ddS)Nztest Entropy is nan)entropyr r1r.rC)r&r3r6entrrr check_entropy[s rKcCs"t|j||j|f|dSr)r rZ_entropy)r&r superclassrrrcheck_private_entropy`s rMcstddgddgg}jd|i}fdd|D}t||j}t||dd ddd g}jd|i}fd d|D}t||j}t||dd dS) Nrr*r?rFscalecsg|]}jd|iqSrNrI.0rDr3r&rr jsz,check_entropy_vect_scale..g+=)rcsg|]}jd|iqSrOrPrQrSrrrTqs)r.asarrayrIZravelZreshapeshaper)r&r3scZv_entZs_entrrSrcheck_entropy_vect_scalefs rYcCs:|j|}t|tjr*|dd|df}t|j|f|ddgt|j|f|ddg|jdkrt|j |f|t j dgt|j |f|dt j gt|j ddgf||t|jddgf||dddtt |jddgf|tt |j ddgf|dS)Nrrgr)ZskellamZdlaplacer*)r$ isinstancerZ rv_discreter rr%sfnamelogcdfr.inflogsfppfisfr1rCall)r&rrrrrcheck_edge_supportvs     &$rdc st|j}t|jdkt|jdkt|j tt|jt|k|j dt | }|j r~|j dd }nd}tt ||jktt |t |ktfdd|D}ttt||ddddi} | rl| ifdd|D} t|| dkrt|jd |jd qd d itt|jfdS) N, csg|]}|fqSrrrRmeth) shape_argsrrrrTsz$check_named_args..csg|]}|fqSrrrh)arGrrrrTsnrZkaboom*)r)r)_getfullargspec _parse_argsr r1varargsvarkw kwonlyargslistdefaultsrlenshapesreplacesplitZnumargsr.rcr;updatepopZassert_array_equalkeysrr" assert_raises TypeErrorr%) r&rrjrtmeths signatureZshape_argnamesZshapes_valsnamesr5r)rkrGrjrrcheck_named_argss2     rc Cs|j}tjdd|_|j|ddi}d|_|j|ddi}t||tjd|_|j|ddi}t||ttjdrtj d}|j|d|dd|_|j }|j|dtjdd}t||t|j |||_dS)Nsize default_rngr)r random_stater*) rr.randomseedrvsr rZ RandomStatehasattrrZ get_state) r&rrndmr0r1r2rngZ orig_stateZr3rrrcheck_random_state_propertys&       rcsdddg}|j|f|fddtjtjtjtjfD}|D]P}|j|||j|k||jk@}|D]$}||f|}t |j tj kqjq@dS)N???csg|]}|qSrZastyperRtpZx0rrrTsz$check_meth_dtype..) rar.int_float16float32float64 _argcheckrkbr r1dtypefloat_)r&r3r~q0x_castrrivalrrrcheck_meth_dtypes   rcsntdddgfddtjtjtjfD}|D]6}|j|jfD]$}||f|}t|j tj kqBq2dS)Nrrrcsg|]}|qSrrrrrrrTsz#check_ppf_dtype..) r.rVrrrrarbr r1rr)r&r3Zq_castqrirrrrcheck_ppf_dtypes rcs\dd}|jdddgf|fddtjtjtjtjfD}|D]}|j|||j|k||jk@}|j |f||j |f||j |f|}}}t ||j |f||dd t ||j |f|||dd t ||j |f|| dd t ||j|f|| |dd t ||j|f|||j |f||j |f|dd qDdS) NcWs*t|}d}|||df||jS)Ng|=y?)r.rVimag)frr3hrrrderivs z check_cmplx_deriv..derivrgRQ?rcsg|]}|qSrrrrrrrTsz%check_cmplx_deriv..rr)rar.rrrrrrkrpdfr%r\rr^r`Zlogpdf)r&r3rrrrr%r\rrrcheck_cmplx_derivs"   4 " rc CsJ|j}d|_|j|ddit|}|j|ddi}t|}|j|ddi}t|||jd ||jd |g}t|d|dt|j|df||j|df|||}t|} t| }|jdd}|jdd}t||t |dr@|j } t| } t| } | j | j kr:dks@nt ||_dS) Nrrrrrr)rfit)r)r) rrpickledumpsloadsr rrar%rr__name__AssertionError) r&rrrDrZ unpickledrZmediansZ frozen_distZpklZ fit_functionZpickled_fit_functionZunpickled_fit_functionrrrcheck_picklings2           "rcCsVt|tjrddd}nddi}|||}|j||jks>t|j||jksRtdS)Nrr*)locrNr)r[rZ rv_continuousrkrr)r&rZlocscalervrrrcheck_freezing;s    rc shtjdj|}t|j|d||sdtjfdd|d}tjd||}t||dddS)N{z%s: rvs failed to broadcastcs j|Sr)r)allargsdistfuncrrrMrz%check_rvs_broadcast..)ZotypesgvIh%<=r)r.rrrrrWZ vectorizer) rr'rrWZ shape_onlyotypesamplerexpectedrrrcheck_rvs_broadcastHs   rc Cs|jdkrgn |jd}tt||p&g}|dd}d}d}d}d|kr |jfd|i|} |jdd|i|} t| | tt t |d|jfd|i|W5QRXtt t |d|jd |W5QRX|jd |} t| | tt t |d|jf|W5QRXn|jf|} tt t |d|jd ddi|} W5QRXt j t$|jfddi|} t| | W5QRX|jd |} t| | tt t |d|jf|W5QRXdS) N, rlz/moment() missing 1 required positional argumentz9_parse_args() missing 1 required positional argument: 'n'z/moment() got multiple values for first argumentrmatch)r)r)r)r)r)rvrxdictziprzZmeanr"rr|r}reescaper.testing assert_warnsDeprecationWarning) r&r3r'rv kwd_shapesrlmessage1message2message3rr rrr'check_deprecation_warning_gh5982_momentSs8      rc sjdkrgn jd}tt||p&g}|dd}fdd}d}d}d} d|kr*|fd|i|} jd d|i|} t| | ttt |d jfd|i|W5QRXttt |d jd |W5QRXjd |} t| | ttt |d jf|W5QRXn|f|} ttt | d jddd i|} W5QRXt j t$jfddi|} t| | W5QRXjd|} t| | ttt |d jf|W5QRXdS)Nralphacs jd||jd||fS)Nrr)r)r)ra)rkwdsr&rr my_intervalsz>check_deprecation_warning_gh5982_interval..my_intervalz1interval() missing 1 required positional argumentz=_parse_args() missing 1 required positional argument: 'alpha'z1interval() got multiple values for first argumentrrr)r)r)r)r)r)rvrxrrrzintervalrr|r}rrr.rrr) r&r3r'rvrrrrrrrr rrr)check_deprecation_warning_gh5982_intervals:       r)F)*rrnumpyr.Z numpy.testingrr rrZpytestrr|Znumpy.ma.testutilsrZ testutilsr Zscipy._lib._utilrrnZscipyrrr)r9r<r>rErHrKrMrYrdrrrrrrrrrrrrrrs:       ') +  7