&]\c@`sdZddlmZmZmZddlZddlZddlm Z ddl Z ddl j Z ddlmZddlmZmZmZddlmZmZmZmZmZyddlZWnek redZnXd efd YZd efd YZd efdYZdZ dZ!dZ"ddZ$dZ%ddZ&dZ'e j(j)eeddefdYZ*dZ+dS(s Test cdflib functions versus mpmath, if available. The following functions still need tests: - ncfdtr - ncfdtri - ncfdtridfn - ncfdtridfd - ncfdtrinc - nbdtrik - nbdtrin - nrdtrimn - nrdtrisd - pdtrik - nctdtr - nctdtrit - nctdtridf - nctdtrinc i(tdivisiontprint_functiontabsolute_importN(t assert_equal(twith_metaclass(t MissingModulet check_versiontFuncData(tArgtIntArgtget_argst mpf2floattassert_mpmath_equaltmpmathtProbArgcB`s eZdZdZdZRS(s*Generate a set of probabilities on [0, 1].cC`sd|_d|_dS(Nii(tatb(tself((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt__init__-s cC`std|d}tjdtjd|}tjdd|ddtd}dtjtjdd|}tj|||f}tj|S(s5Return an array containing approximatively n numbers.iiig333333?gffffff?tendpointi(tmaxtnptlogspacetlog10tlinspacetFalsetr_tunique(Rtntmtv1tv2tv3tv((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pytvalues2s #"(t__name__t __module__t__doc__RR"(((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyR+s tEndpointFiltercB`seZdZdZRS(cC`s(||_||_||_||_dS(N(RRtrtoltatol(RRRR'R(((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyR=s   cC`stj||j|jtj|j|jk}tj||j|jtj|j|jk}tj||BttS(N( RtabsRR'R(RtwhereRtTrue(Rtxtmask1tmask2((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt__call__Cs66(R#R$RR/(((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyR&<s t_CDFDatacB`sAeZedddddddZdZdZdZRS(iic C`s||_||_||_||_||_||_||_||_| |_t |t sud|_ d|_ n| dk s| dk rt | t r| |_ n| gt|j|_ t | t r| |_ q | gt|j|_ nd|_ d|_ dS(N(tspfunctmpfunctindextargspect spfunc_firsttdpsRR'R(t isinstancetlisttNonet endpt_rtolt endpt_atoltlen( RR1R2R3R4R5R6RR'R(R:R;((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRJs*              cG`s|jr|j|}tj|r.tjSt|}|||jlib/python2.7/site-packages/scipy/special/tests/test_cdflib.pytidmapgs      cC`s|jdkr"|jdkr"dSg}xt|j|j|jD]\}}}|dkr~|dkr~|jdqDn*|dkrd}n|dkrd}n|jt|j|j||qDW|S(Ng( R:R9R;tzipR4tappendR&RR(RtfiltersR'R(tspec((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pytget_param_filter{s+     &cC`st|j|j}|j}tt|jd}|jd}tj||dd|j fj |jddf}t |j |d|d|d|j d|jdtd|jdS( Niit param_columnstresult_columnsR'R(t vectorizedt param_filter(R R4RRIR@trangetshapeRthstackR3treshapeRRDR'R(Rtcheck(RRBRMRJRK((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRRs  ; N(R#R$R+R9RRDRIRR(((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyR0Is    cO`st||}|jdS(N(R0RR(Rtkwtd((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt_assert_invertsscC`stj|tj|tj|}}}|dkrHtjdS||kratjdStjd|dt}tj|||dd|dtS(Niitexacttx2t regularized(R tmpftfsubR+tbetainc(tkRtptonemp((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt _binomial_cdfs/    cC`s|dkrtjdStj|tj|tj|}}}|||||}tj|d|dd|dt}|S(NiiRWRX(R RYR[R+(tdfntdfdR,tubRC((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt_f_cdfs   /&c C`s|dkrtjj}ntj|tj|tj|}}tjdd|dd|d |}||tjd|d9}|tjtj |tjd|}d|SWdQXdS(Ng?ig?i( R9R tmpR6R?RYthyp2f1tgammatsqrttpi(tdfttR6tfac((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt_student_t_cdfs )+cC`s_tj|ddtj||}|tj|| d|||ddd9}|S(Niiig?(R tbesseliRgtexp(RjRitncRC((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt_noncentral_chi_pdfs'4c`s|dkrtjj}ntj|tjtj}tj|-tjfdd|g}|SWdQXdS(Nc`st|S(N(Rp(Rj(RiRo(s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pytsi(R9R RdR6RYR?tquad(R,RiRoR6RC((RiRos>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt_noncentral_chi_cdfs  /$cC`s||d|||S(Ni((R]tlmbda((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt_tukey_lmbda_quantiless0.19t TestCDFlibcB`s(eZejjdedZdZdZdZ ejjdedZ dZ dZ dZ d Zejjded Zd Zd Zejjded ZejjdedZdZdZdZejjdedZRS(truncC`s8ttjtdttddtgdddS(NiiiR'g-C6?(RUtsptbdtrikR_RR (R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_bdtriks !c C`sGttjtdtddttgddddddgdS(NiiR'g-C6?R;gư>(RURxtbdtrinR_R RR9(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_bdtrins !c C`sSttjddttdddttdddtdtgdddS( NcS`stj||d|dtS(NRWRX(R R[R+(RRR,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRqsigY@t inclusive_ait inclusive_bR'gư>(RURxtbtdtriaRRR(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_btdtrias c C`sbttjddtdddtttdddtdtgddd dd d gdS( NcS`stj||d|dtS(NRWRX(R R[R+(RRR,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRqsiigY@R}R~R'gHz>R;gC]r2lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_btdtribs c C`sDttjtdtddttdddtgdddS(NiidiR}R'gHz>(RURxtfdtridfdRcR RRR(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_fdtridfds -c C`s\ttjddttdddttdddtgdddddd gdS( NcS`stj|d||dtS(NRRX(R tgammaincR+(RRR,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRqsig@@R}g@R'gHz>R;g|=(RURxtgdtriaRRRR9(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_gdtrias c C`sMttjddtdddtttdddtgdddS( NcS`stj|d||dtS(NRRX(R RR+(RRR,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRq siigY@R}g@@R'gh㈵>(RURxtgdtribRRR(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_gdtribs c C`s\ttjddtdddttdddttgdddddd gdS( NcS`stj|d||dtS(NRRX(R RR+(RRR,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRqsiig@@R}R'gHz>R;g|=(RURxtgdtrixRRRR9(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_gdtrix s 'cC`s8ttjttddtdtjgdddS(Niidg|=R'gHz>(R RxtstdtrRlR RRtinf(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_stdtrscC`s,ttjtdttgdddS(NiR'gHz>(RURxtstdtridfRlRR(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_stdtridfsc C`s>ttjtdtddtgdddddgdS(NiidR'gHz>R;g|=(RURxtstdtritRlR RR9(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_stdtrit#s !cC`s5ttjddttddgdddS(NcS`s!tj|dd|ddtS(NiRRX(R RR+(R!R,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRq-siiidR'g-C6?(RURxtchdtrivRR (R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_chdtriv*sc C`sVttjtdtdddtttdddtgddddd d dS( NiiidR}RiR'g-C6?R(gV瞯<(RURxt chndtridfRsRRR(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyttest_chndtridf0s c C`sPttjtdtdddttddtgdddd d d dS( NiiidR}iRiR'g-C6?R(gV瞯<(RURxt chndtrincRsRRR R(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyttest_chndtrinc:s -cC`s_ttjtdttddtdddtgddddd d d d ddgdS( NiiidR}RiR'g-C6?R(gV瞯( RURxtchndtrixRsRR RRR9(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyt test_chndtrixCs -c`s;tjdtdfdtgdddS(NicS`stj|dS(Ni(Rxttklmbda(R,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRqPsc`stj| S(N(R Rn(R,(tone(s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRqQsR'gHz>(R RYR R(R((Rs>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyttest_tklmbda_zero_shapeLs  c C`sJttjtdttdddtgdtdddddgdS( NiiR~R5R'gh㈵>R;g& .>(RURxRRuRRR(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyttest_tklmbda_neg_shapeTs ! c C`s>ttjtdttdddtgdtdddS(NiidR}R5R'gh㈵>(RURxRRuRRR(R((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyttest_tklmbda_pos_shape\s !(R#R$tpytesttmarktxfailRRzR|RRRRRRRRRRRRRRRR(((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyRvs$            c C`s-d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCg}tjjd#x|D]\}}tt|}gtjj|D](}t|tjtjtj f^q}xLt j |D];}||}t d$|Drt |tjqqWqzWdS(DNRiRRyR{RitchndtrRRRRtncfdtritncfdtrit ncfdtridfnt ncfdtridfdt ncfdtrincRRRtnbdtriktnbdtrintnrdtrimntnrdtrisdtpdtrikRRRtnctdtrtnctdtritt nctdtridft nctdtrincRics`s|]}tj|VqdS(N(RR=(t.0R,((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pys s(Ri(Ri(Ryi(R{i(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri(Ri( RtrandomtseedtgetattrRxtrandtfloatR>Rt itertoolstproducttanyR(tfuncstfunctnumargsR,t args_choicesRBRC((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyttest_nonfiniteesR > (,R%t __future__RRRRtnumpyRt numpy.testingRRt scipy.specialtspecialRxtscipy._lib.sixRtscipy.special._testutilsRRRtscipy.special._mptestutilsRR R R R R t ImportErrortobjectRR&R0RUR_RcR9RlRpRsRuRtslowRvR(((s>lib/python2.7/site-packages/scipy/special/tests/test_cdflib.pyts6   (  P