p7]c@ sddlmZddlmZddljZddljjZ ddl Z ddlm Z ddl Z ddl Z ddlmZddlmZdZdZd Zd Zd efd YZd efdYZdefdYZdZdefdYZdZdZdZdZdefdYZ de fdYZ!de fdYZ"de fdYZ#d e fd!YZ$d"e fd#YZ%d$e fd%YZ&d&e fd'YZ'd(e!fd)YZ(d*e$fd+YZ)dS(,i(tdivision(tdedentN(t assert_equal(tros(tStringIOcC s_td}tj|}|d|jdddf<|ddk|jdddf<|S(Nsres,qual 2.00,= 4.20,= 4.62,= 5.00,ND 5.00,ND 5.50,ND 5.57,= 5.66,= 5.75,ND 5.86,= 6.65,= 6.78,= 6.79,= 7.50,= 7.50,= 7.50,= 8.63,= 8.71,= 8.99,= 9.50,ND 9.50,ND 9.85,= 10.82,= 11.00,ND 11.25,= 11.25,= 12.20,= 14.92,= 16.77,= 17.81,= 19.16,= 19.19,= 19.64,= 20.18,= 22.97,= trestconctqualtNDtcensored(Rtpandastread_csvtloc(traw_csvtdf((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pytload_basic_datas  #c&C sPtjitd6dd6dd6dd6itd6dd6dd6dd6itd6dd6dd6dd6itd6d d6d d6dd6itd6d d6d d6dd6itd6d d6d d6dd6itd6d d6dd6dd6itd6dd6dd6dd6itd6dd6dd6dd6itd6dd6dd6d d6itd6dd6dd6dd6itd6dd6dd6dd6itd6dd6d d6dd6itd6dd6d d6dd6itd6dd6d d6d d6itd6dd6d d6d d6itd6dd6d d6dd6itd6dd6d d6dd6itd6dd6d d6dd6itd6dd6d d6dd6itd6dd6d d6dd6itd6d d6d d6d!d6itd6d"d6d d6dd6itd6d#d6d d6dd6itd6d$d6dd6dd6itd6d$d6dd6dd6itd6d%d6dd6d d6itd6d&d6dd6d d6itd6d'd6dd6dd6itd6d(d6dd6dd6itd6d)d6dd6dd6itd6d*d6dd6dd6itd6d+d6dd6dd6itd6d,d6dd6d!d6itd6d-d6dd6d.d6g#}|S(/NR g@Ritdet_limit_indextrankig@g@ig#@ig&@ig@ig@g{Gz@gHzG@gp= ף@gq= ףp@g@gQ@g)\(@g@iig(\B!@igQk!@i g{G!@i g33333#@gp= ף%@g&@gffffff(@gףp= -@gQ0@g(\1@g)\(3@gq= ף03@gp= ף3@gGz.4@gQ6@i (R t DataFrametTruetFalse(R((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pytload_intermediate_data sJ(c&C s:tjidd6td6dd6dd6dd 6dd 6id d6td6dd6dd6d d 6d d 6idd6td6dd6d d6dd 6dd 6idd6td6dd6dd6dd 6dd 6idd6td6dd6dd6dd 6dd 6idd6td6dd6dd6dd 6d d 6idd6td6dd6dd6dd 6dd 6idd6td6d d6d!d6d"d 6dd 6id d6td6d#d6d!d6d d 6d d 6id$d6td6d%d6d!d6d&d 6dd 6id'd6td6d(d6d d6d)d 6dd 6id*d6td6d+d6d d6d,d 6d d 6id-d6td6d.d6dd6d/d 6dd 6id0d6td6d1d6dd6d2d 6d d 6id3d6td6d4d6dd6d5d 6dd 6id6d6td6d7d6dd6d8d 6dd 6id9d6td6d:d6dd6d;d 6dd 6id<d6td6d:d6dd6d=d 6d>d 6id?d6td6d:d6dd6d@d 6dAd 6idBd6td6dCd6dd6dDd 6dEd 6idFd6td6dGd6dd6dHd 6dId 6idJd6td6dKd6dd6dLd 6dMd 6idNd6td6dOd6dd6dPd 6dd 6idQd6td6dRd6dd6dSd 6d d 6idTd6td6dUd6dd6dVd 6dd 6idWd6td6dUd6dd6dXd 6d d 6idYd6td6dZd6dd6d[d 6dd 6id\d6td6d]d6dd6d^d 6dd 6id_d6td6d`d6dd6dad 6dd 6idbd6td6dcd6dd6ddd 6d>d 6ided6td6dfd6dd6dgd 6dAd 6idhd6td6did6dd6djd 6dEd 6idkd6td6dld6dd6dmd 6dId 6idnd6td6dod6dd6dpd 6dMd 6idqd6td6drd6dd6dsd 6dtd 6g#}|S(uNgCB!tZprelimR g@RiRg$F?tplot_posRgXHg6iqx?igJ@g@g$F?gI@g@ig$F?gXg#@igs?gW'>ٿg__?gz0zʿg&@igs?g{g@ig6iqx?g@g?g{Gz@g)YUZ?gL{bgHzG@g<8 E?gMQngp= ף@g!Rk f?g#ܿgq= ףp@gGGt)?g!6|Rf׿g@gj` ?g@>(xҿgQ@gL_?g? $R˿g)\(@gr8-?g@Dg@g+$?g?V$kmg{?ig.+NƷ?gϋ~K3?igȎE?g(\B!@g+(?ig[ T?gQk!@gj?i go4%?g{G!@g?i g?g33333#@gU ?g?糼k?gp= ף%@gyڽN?g?g&@g~?g&.Ef?gyy?gXu ?gffffff(@gPuPu?gU?gףp= -@gKK?gK4?gQ0@g""""""?g?g(\1@g?gM?g)\(3@g<s   c C stjidgdd6tdtdd6ttdd6}tj|dd}tjdddddd dddd dd g d d }tj|j t |j t dS( Nii tdl_idxtAABCCCDEtDCBAtparamstvaluesiiitnameR( R RtlisttrangeRt_ros_group_ranktSeriesR+tassert_series_equaltastypetint(RR-texpected((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyttest__ros_group_ranks 9tTest__ros_plot_poscB s5eZdZdZdZdZdZRS(cC st|_dS(N(R#R"(R(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyR)scC sDitd6dd6dd6}tj|d|j}t|ddS(NR iRiRg<8 E?(RRt _ros_plot_posR"R(R(trowR-((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyttest_uncensored_1scC sDitd6dd6dd6}tj|d|j}t|ddS(NR iRi Rg}?(RRRYR"R(R(RZR-((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyttest_uncensored_2scC sDitd6dd6dd6}tj|d|j}t|ddS(NR iRiRg__?(RRRYR"R(R(RZR-((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyttest_censored_1!scC sDitd6dd6dd6}tj|d|j}t|ddS(NR iRiRgs?(RRRYR"R(R(RZR-((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyttest_censored_2&s(R3R4R)R[R\R]R^(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyRXs     cC sJtjddddg}tjddddg}tj||dS( NiiiigT]?g`>?gO:`?gyj?(Rt_norm_plot_posRtarraytnpttassert_array_almost_equal(R-RV((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyttest__norm_plot_pos,sc$C st}t}tj|d|}tjdddddddddd d d d d ddddddddddddddddddd d!d"g#}tj||dS(#NR g@,U?gUqx?g^c@?g?g_?g?gaˀqx?gB UZ?geL$E?gKn f?g*?gh ?gޮ_?gqo-?g>?g{X={?gK3?g m(?g(?g1^?g?g}N?g"}?gCy?g&(Nu?gC" K?g`C#""?gp?g!Rd?d?d@gdA6ZRS(Bis res cen 0.090 True 0.090 True 0.090 True 0.101 False 0.136 False 0.340 False 0.457 False 0.514 False 0.629 False 0.638 False 0.774 False 0.788 False 0.900 True 0.900 True 0.900 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 False 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.000 True 1.100 False 2.000 False 2.000 False 2.404 False 2.860 False 3.000 False 3.000 False 3.705 False 4.000 False 5.000 False 5.960 False 6.000 False 7.214 False 16.000 False 17.716 False 25.000 False 51.000 FalseRR}tseps\s+g6 j?g}%?gM "?gB`"۹?g rh?g(\?gsh|??g r?gT㥛 ?g"~j?gS㥛?gK7?gJ c?gπz3*?gN?g&V?g7F?gV?gq C/7?gɹEGϹ?g?gzc?gfD̙+?gc T?g-?g",(?gG4L?gL6S?g?ΰ(W~?gR*?gZKfd?gJE?g?g@gZd;@gzG@g@gp= ף @g@g@gףp= @g@gB`"@g0@gK1@g9@gI@g"@gg2@Rg.@g@@Rg1@RgzG?g ףp= ?iR(R3R4RoRRt datastringRqRrR R RRR`RtRR!Rx(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyRs,3  tTest_NoOp_ZeroNDcB s eZdZejjddZejjdeZe geZ dZ dZ e jiee 6e e 6Zejdddd d d d d ddddddddddddgZe jiejgd6ejgd6ejgd6ejgd6ZRS(iiitsizeR|R}gRQ?gQ?gQ?gQ?g?gGz?gffffff?gQ?gffffff?gףp= ?g)\(?g(\?gQ@gQ@gHzG@gQ@gףp= @g\(\@gzG@g"@RRRR(R3R4RoRtrandomtseedtNt lognormalRRR}RqRrR RRR`RtRx(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyRs"  tTest_ROS_OneNDcB sveZdZejdddddddddddddddd d d gZejeeeeeeeeeeeeeeeeeegZd Z d Z e j iee 6ee 6Z ejdddddddddddddddd d d gZe j iejdejgd6ejdejgd6ejdejgd6ejddgd6ZRS(ig?g$@g@g@g"@g(@g.@g4@g;@g@@gI@RR}gQ?g1@RRRg~!<8?gR(R3R4RoRR`RRRR}RqRrR RRRtR!Rx(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyR2s&$$ tTest_HalfDLs_80pctNDscB seZdZejdddddddddddddddd d d gZejeeeeeeeeeeeeeeeeeegZd Z d Z e j iee 6ee 6Z ejdddddddddddddddd d d gZe j iejddddddddejg d6ejddddddddejg d6ejddddddddejg d6ejdgddgd6ZRS(ig?g$@g@g@g"@g(@g.@g4@g;@g@@gI@tvalueRg?g@g?g @g@g@g@gRg @g*@g,@RRgKqU?iR(R3R4RoRR`RRRR}RqRrR RRRtR!Rx(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyRMs&$$ ...tTest_HaflDLs_OneUncensoredcB seZdZejddddgZejeeeegZdZ dZ e j iee 6ee 6Z ejddddgZe j iejddejgd 6ejd d ejgd 6ejd dejgd 6ejdddgd6ZRS(ig?g(@g.@RRg?g@gRg@g@RRg?R(R3R4RoRR`RRRR}RqRrR RRRtR!Rx(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyRhs tTest_ROS_MaxCen_GT_MaxUncencB seZejddddddddddddddddd d d d gZejeeeeeeeeeeeeeeeeeeeegZRS( g?g$@g@g@g"@g(@g.@g4@g;@g@@gI@i<iF(R3R4RR`RRRR}(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyRystTest_ROS_OnlyDL_GT_MaxUncencB s}eZejjddZdddddddd d d d d ddddddddddgZegeeegZ RS(iigRQ?gQ?gQ?gQ?g?gGz?gffffff?gQ?gffffff?gףp= ?g)\(?g(\?gQ@gQ@gHzG@gQ@gףp= @g\(\@gzG@g"@g$@( R3R4RRRRRRRR}(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyRs (*t __future__RttextwrapRt numpy.testingttestingRatpandas.util.testingtutilR+RRR REtstatsmodels.imputationRtstatsmodels.compat.pythonRRRRR#tobjectR$R5R>RWRXRcRfRkRmRnR{RRRRRRRR(((sDlib/python2.7/site-packages/statsmodels/imputation/tests/test_ros.pyts@     * M '#      'N