p7]c @sddlZddlZddlmZmZmZmZddlm Z ddl Z ddl Z e ZyddljZWnek rnXdZejd5Zejd6Zejd7Zejd8ZejdddejfZejd9Zejd:Zejd;Zejd<Zejd=Zejd>Z ejdddejfZ!ejd?Z"ejd@Z#ej$j%ej$j&e'Z(ej$j)e(ddZ*e j+e*Z,dZ-d Z.d!Z/d"Z0d#Z1d$Z2e j3j4d%Z5d&Z6d'Z7d(Z8d)Z9d*Z:d+Z;d,Z<d-Z=d.Z>d/Z?d0Z@d1ZAe j3jBd2ZCd3ZDe j3jBd4ZEdS(AiN(t SurvfuncRighttsurvdifft plot_survfunctCumIncidenceRight(tassert_allclosecCstr|j|ndS(N(t pdf_outputtsavefig(tpdftfig((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyt close_or_savesiiiiig?g:gUU?g?gg.?gC&ϫ?g?iiig?ii gh"q?gc@?gc@?gG TѺ?gb?gE ;I?tresultssbmt.csvcCs{ttt}t|jtddddt|jtddddt|jt t|j t t|j t dS(Ntatolgh㈵>trtol(Rtti1tst1Rt surv_probt surv_prob1t surv_prob_set surv_prob_se1t surv_timesttimes1tn_risktn_risk1tn_eventst n_events1(tsr((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survfunc1;s cCs{ttt}t|jtddddt|jtddddt|jt t|j t t|j t dS(NR gh㈵>R (Rtti2tst2RRt surv_prob2Rt surv_prob_se2Rttimes2Rtn_risk2Rt n_events2(R((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survfunc2Fs cCstjttf}tjttf}tjt|}d|dtt+t|||\}}t |dddddt |ddddddS(Nig:ǀ,@R g-C6?R g-lI?( tnpt concatenateR RRRtonestlenRR(ttitsttgroupstztp((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survdiff_basicQsc Cstjtddkddf}t|d|d}|jdd\}}|jdd\}}|jj}gdD]}|j|^q}t||tj dt||tj dt||tj dt||tj d dS(!NtGrouptALLtTtStatust transformtlogtarcsininiziigNB?g7?g)(i?gm?gi@ֶ?g?g*^"?g׾s|?g0՜Q?gNFS?g2]_V.?g)1?gFw?gBh ?g-U?g9j?(inizii(gNB?g7?g)(i?gm?(gi@ֶ?g?g*^"?g׾s|?(g0՜Q?gNFS?g2]_V.?g)1?(gFw?gBh ?g-U?g9j?( tbmttlocRtsimultaneous_cbRttolisttindexRR#tr_( tdftsftlcb1tucb1tlcb2tucb2R'txtix((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_simultaneous_cb]s#"c CsZiddgd6ddgd6ddgd6d dgd 6d dgd 6}ttjd k}tjjtjjt}tjj|d d}tj |}t |dj |j j }t |j|jt |j|jddddt |j|jddddxGdD]?}|jdd|\}}t ||tj||fqWdS(NikitlineariVitcloglogiLR2ihtasinsqrttlogitR.R sbmt_results.csvR/R g-C6?R g?tmethod(RCRDR2RFRE(R4R-tostpathtdirnametabspatht__file__tjointpdtread_csvRtvaluesR0RRttRtsRtset quantile_ciR#R9( tcbtdfatcur_dirtfptrsltR;RGtlcbtucb((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_bmtts     c Csrttjdkj}t|d|j|j\}}t|dddddt|d|j|jdd\}}t|d ddddt|d|j|jdd \}}t|d ddddt|d|j|jdd d d\}}t|dddddt|d|j|jdd d d\}}t|dddddtj|jdd}||dyX5)@g-*@( R4R-tcopyRR0RR#tarangetshapeRb(R:tstatR+Rb((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyt test_survdiffs@"(( +""+""cCs tr%ddlm}|d}nd}ttt}ttt}t |}t ||t |}t ||t ||g}t ||t j d}g}xD|D]<}t|dd|ddd|d } |j | qWt |}|jd } | jd d d d g| j\} } |jgdD]} | | ^qJgdD]} | | ^qddt ||t jdk}tt j|dft j|df}|j}|jd } | jd d d d g| j\} } |jdd\}}tj|j||dd|jdd\}}tj|j|ddtj|j|ddtj|j|jd |jddtj|j|jd |jddtjddt ||tr|jndS(Ni(tPdfPagesstest_survfunc.pdfR-iR/R0ttitleig?g{Gz?g?iis center rightR.R1R2tcolort lightgreyR3tdarkgreytredidiX(iii(iii(Rtmatplotlib.backends.backend_pdfRhtNoneRR RRRRR R4tgroupbytappendtget_axest set_positiontget_legend_handles_labelstlegendR-R5tplotR6tpltt fill_betweenRRRtxlimtclose(t close_figuresRhRtsr1tsr2RR^tsvtgts0taxthatlbtktiiR;RZR[((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyt test_plot_kmsT      )   ) '' cCstjd}tjd}tjd}t||d |}t|jtjdt|jtjdt|jtjddS(Niiiiiii iiit freq_weightsg?g?gvWUU?g&>?gguh5y?gXa?g Q[?gi< ?g?( iiiiiiii iiiii( iiiiiiiiiiiii( iiiiiiiiiiiii(iiiii (g?g?gvWUU?g&>?g(guh5y?gXa?g Q[?gi< ?g?(R#R9RRRRR(ttmR(twtR;((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyt test_weights1s     cCstjd}tjd}tjd}tjd}tjd}t||d |}t||}t|j|jt|j|jt|jtjddS(Niiiiiiiii iRgb?gL?g3?g0^?g!?gyx?gm+?goG!?( iiiiiiiiii ( iiiiiiiiii( iiiiiiiiii(iiiiiiiiii iiiii (iiiiiiiiiiiiiii(gb?gL?g3?g0^?g!?gyx?gm+?goG!?(R#R9RRRRR(RR(Rttm0tst0tsf0tsf1((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyt test_weights2 s      cCs tjd}tjd}t||}tjd d d d d d d gtjd d d d d d d gtjd d d ddddgg}t|d |jd t|d|jdt|d|jdtjdddddddgtjd d dddddgtjd d d ddddgg}t|d |jd t|d|jdt|d|jdtjt|}t||d|}t|jd |jd t|jd|jdt|jd|jddS(Niiiiiii iig#q?g쿭??ggruS?gl?gS?g!?g\?gm7)?gā?gL^?R(iiiiiiiiiii i i iiiii(iiiiiiiiiiiiiiiiii( R#R9RtarrayRtcinctcinc_seR%R&(tftimetfstattciRRtweightstciw((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_incidence%s4   c Cstjd}tjd}tjd}t||d |}t|jtjd t|jtjd!t|jtjd"ddt|jtjd#dddS($Niiiiii i iiitentryig?gz6?g( ?g y?g镲 q?gD?g37?R g-C6?gea?g( y?gH.?g9#J?gAǘ?g/$?g=yX?( iiiiiiiiii i i ( iiiiiiiiiiii( iiiiiiiiiiii(iii iiii(iiiiii i (g?gz6?g( ?g y?g镲 q?gD?g37?(gea?g( y?gH.?g9#J?gAǘ?g/$?g=yX?(R#R9RRRRRR(ttimeststatusRR;((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survfunc_entry_1Ss   c Cstjd }tjd }tjd }t||d |}t||}t|j|jt|j|jt|j|jt|j|jdS( Niiiiii i iR( iiiiiiiiii i i ( iiiiiiiiiiii( iiiiiiiiiiii(R#R9RRRRRR(RRRR;R((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survfunc_entry_2js   c Cstjd}tjd}tjd}t||d |}t|jtjdt|jtjdt|jtjdd dt|jtjdd ddS(Niiiii iiiRig(m?g_#I?R gh㈵>gHK?g1q ?( iiiiiiiii ( iiiiiiiii( iiiiiiiii(ii(ii(g(m?g_#I?(gHK?g1q ?(R#R9RRRRRR(RRRR;((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survfunc_entry_3zs   c Cstjd }tjd }tjd }tjd}t|||d |\}}t|||\}}t||t||dS(Niiiiiiii iR( iiiiiiiiii ( iiiiiiiiii( iiiiiiiiii( iiiiiiiiii(R#R9RR(R'R(tgrRtz1tp1tz2tp2((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survdiff_entry_1s     c Cstjd}tjd}tjd}tjd}t|||\}}t|d t|d t|||d |\}}t|d t|ddS(Niiiiiii iig:@gZj΃?Rg@g $#P?( iiiiiiiiii ( iiiiiiiiii( iiiiiiiiii( iiiiiiiiii(R#R9RR(R'R(RRR*R+((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survdiff_entry_2s       c Cstjd}tjd}tjd}tjd}t|||\}}t|d t|d t|||d |\}}t|d t|ddS(Niiiiii iiig-4@g[TR?Rg&x@gb:$0?( iiiiiiii ii ( iiiiiiiiii( iiiiiiiiii( iiiiiiiiii(R#R9RR(R'R(RRR*R+((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_survdiff_entry_3s       c Cstjjdd}tjtjjd| }tjjddd|}tj|}||}||}t||}d|dk}t||}d|j }|j d|j dtj |}t ||dS(Niw itsizeiii( R#trandomtseedR2tuniformtrandinttargsortRRRRt flatnonzeroR( tnttimeRRRtstatusaR;R@ty((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_incidence2s   %cCsd}tjjdtjjd|df}tjjd|}tjjddd|}t||d|}tjd}tjd}t|j dd!|t|j dd!|dS(Nidij RiiitexoggsX?g؝Ic?g>t(?g C ?gN`ז ?g,`#ש?gNti?g|#?g\?gV[Ch?i(gsX?g؝Ic?g>t(?g C ?gN`ז ?(g,`#ש?gNti?g|#?g\?gV[Ch?( R#RRtnormalRRRR9RRR(RR@RRtresultttimextsprob((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_kernel_survfunc1s  cCsd}tjjdtjjd|df}tjjddd|}tjjddd|}t||}t||d|d d }t|j|jt|j |j d d d d dS(Nidij Riii iRt bw_factori'R gư>R ( R#RRRRRRRRR(RR@RRtresultkmR((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_kernel_survfunc2sc Csd}tjjdtjjd|df}tjjddd|}tjjddd|}t||d|d d t||d|d tjd dS( Nidij Riii iRRi'(i'i'(R#RRRRRR9(RR@RR((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_kernel_survfunc3sc Csd}tjjdtjjd|df}tjjddd|}tjjddd|}t||}xzttfD]l}t||d|dd d |}t |j |j x/dD]'}t |j ||j |d d qWqWdS(Nidij Riii RRi'tdimrediR gh㈵>(ii( R#RRRRRRtFalsetTrueRRR(RR@RRtresult1Rtresult2R((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_kernel_cumincidence1s  cCsd}tjjdtjjd|df}tjjddd|}tjjddd|}t||d|dd dS( Nidij Riii RRi'(R#RRRRR(RR@RR((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pyttest_kernel_cumincidence2s (iiiiiiii(iiiiiiii(iiii(g?g:gUU?g?g(iiii(g?g?g?g?( iiiiiiiii ( iiiiiiiii(iiii (gh"q?gc@?gc@?g(i iii(g?g?g?g?(FRHtnumpyR#tstatsmodels.duration.survfuncRRRRt numpy.testingRtpandasRNtpytestRRtmatplotlib.pyplottpyplotRwt ImportErrorR R9R RRRtnanRRRRRRRRR R!RIRJRKRLRWRMRXROR4RR"R,RBR\Rgtmarkt matplotlibRRRRRRRRRRRRRtsmokeRRR(((sGlib/python2.7/site-packages/statsmodels/duration/tests/test_survfunc.pytsd  "                   -9   .