ó áp7]c@sÜdZddlZddlmZddlmZddlmZm Z m Z m Z defd„ƒYZ de fd „ƒYZd efd „ƒYZd efd „ƒYZdefd„ƒYZde fd„ƒYZdS(s> Created on Fri Jun 28 14:19:26 2013 Author: Josef Perktold iÿÿÿÿN(tstats(tGenericLikelihoodModel(tassert_array_lesstassert_almost_equaltassert_allclosetassert_tMyParetocBs2eZdZd„Zd„Zd„Zd„ZRS(smMaximum Likelihood Estimation pareto distribution first version: iid case, with constant parameters cCs?tt|ƒjƒtjd|jjƒddgƒ|_dS(Ngø?gð?(tsuperRt initializetnptarraytendogtmint start_params(tself((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRscCs||| dS(Ni((Rtxtb((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pytpdfscCs|j|ƒjdƒ S(Ni(t nloglikeobstsum(Rtparams((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pytloglike!scCs¢|jdk r!|j|ƒ}n|d}|d}|d}|j}|||}tj|ƒ|dtj|ƒ}|tj|ƒ8}d||dk<| S(Niiigð?iðØÿÿ(t fixed_paramstNonet expandparamsR R tlog(RRRtloctscaleR Rtlogpdf((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyR*s    $(t__name__t __module__t__doc__RRRR(((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRs    tCheckGenericMixincBs,eZd„Zd„Zd„Zd„ZRS(cCsC|jjƒ}|jjr!dnd}|t|ƒks?t‚dS(NsP>|t|sP>|z|(tres1tsummarytuse_ttstrtAssertionError(Rtsummt check_str((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyt test_summaryAscCsO|jj}t|j_|jjƒ}dt|ƒks?t‚||j_dS(NsP>|t|(R!R#tTrueR"R$R%(Rtorig_valR&((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyttest_use_t_summaryFs   cCs)|jjtjt|jjƒƒƒdS(N(R!tt_testR teyetlenR(R((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyt test_ttestMscCs|jj}tjdddgƒ}|jjjdk rO||jjj}nt||ddƒt|tjt |ƒƒddƒt|jj tjt |ƒƒddƒ|j st|jj |jj dddd ƒt|jj |jj ddddƒndS( Niitatolgø?igà?trtolgš™™™™™©?g333333Ã?(R!RR R tmodeltfixed_paramsmaskRRtzerosR.tbset skip_bsejactbsejactbsejhj(RRt params_true((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyt test_paramsPs "( (RRR(R+R/R:(((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyR >s   t TestMyPareto1cBs eZed„ƒZd„ZRS(cCsÄdddg}d}tjjdƒtjj|td|ƒŽ}t|ƒ}d|_ d|_ d|_ |j j d|j |_ddd g|j_||_|jd dƒ|_t|_dS( NiiidiÒtsizeitshapeRRtdisp(R trandomtseedRtparetotrvstdictRRRR3tdf_modelR R=tdf_residtdatatxnamestmodtfitR!R)R6(tclsRtnobsRBtmod_par((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyt setup_classjs     cCsT|jj}|jjjƒ}|d|d}t||ƒt||ddƒdS(Niitdecimal(R!RR R RR(RRtx_mintp_min((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyttest_minsupports   (RRt classmethodRMRQ(((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyR;hstTestMyParetoRestrictioncBseZed„ƒZRS(cCsýdddg}d}tjjdƒtjj|td|ƒŽ}t|ƒ}tjtj dƒ}d|d<||_ tj |ƒ|_ |j |j |_ d|_|jjd|j|_d d g|j_||_|jd dƒ|_t|_dS( Niii2iÒR<igš™™™™™¹¿iR=RR>(R R?R@RRARBRCRtnantonesRtisnanR3R RDR R=RERFRGRHRIRR!tFalseR6(RJRRKRBRLtfixdf((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRMŠs      (RRRRRM(((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRS‡stTwoPeakLLHNoExogcBskeZdZddgZdddgZddgZdgZd d d d „Zd „Z d „Z d „Z RS(s*Fit height of signal peak over background.i ièR tsignalt backgroundtn_signalt n_backgroundtalphacOs;||_||_tt|ƒjd|d|||ŽdS(NR texog(RZR[RRYt__init__(RR R_RZR[targstkwargs((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyR`©s  cCs|j|ƒ S(N(tnloglike(RR((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyR²scCs|j}|j||ƒS(N(R tnlnlike(RRR ((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRcµs c Csš|d}|d}|dks,|dkr3tjS||}|}|jj|ƒ}|jj|ƒ}tjtj||||ƒƒ} | |8} | S(Nii(R tinfRZRR[RR( RRR tn_sigtn_bkgtn_totR^tsigtbkgtsumlogl((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRd¹s   $ N( RRRR t cloneattrt exog_namest endog_namesRR`RRcRd(((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRY¢s     tTestTwoPeakLLHNoExogcBs eZed„ƒZd„ZRS(cCsÀtjjdƒtjddddƒ}tjddddƒ}d}d}||g}tj|jd |ƒ|jd |ƒgƒdd…tjf}||_ ||_ ||_ ||_ dS( Ni*RiRiidi2iÈR<( R R?R@Rt halfcauchytuniformt concatenateRBtnewaxistXRtpdf_atpdf_b(RJRuRvtn_atn_bRRt((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRMÉs     cCs¯tjjdƒt|jd|jd|jƒ}|jƒ}t|j |j ddƒt tj |j ƒƒ|j ddƒ}t|djd ƒ|j ddƒ|jƒdS( Ni*RZR[R1gš™™™™™¹?tnrepi2ii(R R?R@RYRtRuRvRIRRRRVREt bootstraptmeanR"(Rt llh_noexogtrestres_bs((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyttest_fitÛs    #(RRRRRMR(((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyRoÇs(RtnumpyR tscipyRtstatsmodels.base.modelRt numpy.testingRRRRRtobjectR R;RSRYRo(((sLlib/python2.7/site-packages/statsmodels/miscmodels/tests/test_generic_mle.pyts "-*%