p7]c@sdZddlmZmZddlZddlZddlmZddl m Z m Z m Z m Z ddlZddlmZmZddlZddlmZddlmZdd lmZejd d Zd efd YZdZdefdYZdefdYZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'defdYZ(defdYZ)dZ*dZ+d Z,d!Z-dS("s! Test functions for models.tools i(tlrangetrangeN(tstandard_normal(t assert_equaltassert_array_equaltassert_almost_equaltassert_string_equal(tassert_frame_equaltassert_series_equal(tlongley(ttools(t pinv_extendedtmodulecCstjdd!tjdd!tjdd!tjdd!tjdd!g}|d9}tjt|}tj|dd}|S( Niii iiitnamet string_var(tstringtascii_lowercasetnptasarraytsortedtpdtSeries(Rtseries((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs     t TestToolscBs}eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d ZRS( cCs_tdd}tj|}tjddddgddddggj}t||dS(Niiiig@(RR t add_constantRRtTR(tselftxty((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_add_constant_list$s0cCsbtjdd}tj|}tjddddgddddggj}t||dS(Niiiig@(RtarangeR RRRR(RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_add_constant_1d*s0cCstjd}tj|dd}t|tjdtjttj|ddWdQXttj|ddtjd dS( Nit has_constanttskipitraisetaddi(ii(ii(RtonesR RRtpytesttraisest ValueError(RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_add_constant_has_constant1d0scCstjddddgddddggj}tj|dd}t||tjttj|ddWdQXttj|ddtj tj d |fdS( Niiig@R R!R"R#i( RRRR RRR%R&R't column_stackR$(RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_add_constant_has_constant2d;s0 cCstjdtfd dtjfdtjfg}tjd ddg|}|jtj}tj |}t |d tjdddgx4|jj D]&}||j||jkst qWdS(NtsiR?i R@ii( RtzerosR RRR$RRRRARBRCR(RR>R;R:R<RDRE((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_add_constant_zeros`s"&3  3 cCs\tjddgddgg}tj|}t|tjddgddggdS(Niiiig?(RR2R treciprR(RtXtY((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_reciprvs!cCs\tjddgddgg}tj|}t|tjddgddggdS(Niiiig?gп(RR2R trecipr0R(RRJRK((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_recipr0{s!cCsftd}tjj|}tjj|dd}t|\}}t||t||dS(Ni(i i(i(i (RRtlinalgtpinvtsvdR R(RRJtnp_invt np_sing_valstsm_invt sing_vals((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_extendedpinvs   cCstd}|dddf|dddf|dddf(R%R&t TypeErrorR RR~(R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_array1d_col_errorscCsztj|j|j|jf}tj|dddt}t|ddddf|jt |j dddS(NRitdropiii ( RR)R|R~R RtTrueRRRR](RR|((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_array2d_drops!&cCs@tj|jdt}t||jt|jdddS(NRii(R RR~RRRRR](RR|((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_array1d_dropscCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRRxii ( R RRRR)R.tnamesRRRtlen(RR|t_ttest_des((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_recarray2ds0cCsLtj|j|jf}tjtddtj|ddWdQXdS(NRscol is None and the inputR( Rtc_RR%R&t IndexErrorR RR(Rtarr((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_recarray2d_errorscCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRiii ( R RRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_recarray2dints0cCs|jdjtj}tj|}tjg|jjdD]}||^q?}t ||j t t |jjddS(NRxii( RR3RR4R RR)R.RRRRR(RR~tdumRttest_dum((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_recarray1d s 0cCs|jdjtj}tj|dt}tjg|jj D]}||^qA}t ||j t t |jj ddS(NRxRi(RR3RR4R RRR)R.RRRRR(RR~RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_recarray1d_drops ,cCs{tj|jdddt}tjg|jjdD]}||^q5}t||j t t |jjddS(NRRxRii( R RRRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_recarray2d_drops0cCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRRxii ( R RRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_structarray2ds0cCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRiii ( R RRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_structarray2dint$s0cCs|jdjddg}tj|}tjg|jjdD]}||^qB}t||j t t |jjddS(NRxR.RuRvii(RuRv( RR3R RRR)R.RRRRR(RR~RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_structarray1d*s 0cCs{tj|jdddt}tjg|jjdD]}||^q5}t||j t t |jjddS(NRRxRii( R RRRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_structarray2d_drop1s0cCs|jdjddg}tj|dt}tjg|jjD]}||^qD}t ||j t t |jjddS(NRxR.RuRvRi(RuRv( RR3R RRRR)R.RRRRR(RR~RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_structarray1d_drop7s ,treasonsCall to tools.categorical raises AttributeError. Previously this was commented-out with the comment 'comment out until we have type coercion'.tstrictR&cCsbtj|jjdd}|ddddf}t||jt|jdddS(NRiiii (R RRRhRRRR](RR|R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_arraylike2d>scCsf|jdj}tj|}|ddddf}t||jt|jdddS(NRxiii(RRhR RRRRR](RR~RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_arraylike1dIs cCshtj|jjdddt}|ddddf}t||jt|jdddS(NRiRiii( R RRRhRRRRR](RR|R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_arraylike2d_dropUs$cCsP|jdj}tj|dt}t||jt|jdddS(NRxRii( RRhR RRRRRR](RR~R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_arraylike1d_drop`s(R_R`t classmethodRRRRRRRRRRRRRRRRRR%tmarktxfailRtAttributeErrorRRRR(((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRrs2                tTestCategoricalStringcBsmeZejjdddZejjdddZejjdddZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZejjdddededZejjdddededZejjdddededZejjdddededZRS(RswNo idea! But xfailing instead of leaving this commented out with the comment 'comment out until we have type coercion'cCsttj|j|j|jf}tj|dd}t|ddddf|jt|j dddS(NRiiii ( RR)R|R~R RRRRR](RR|((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRns!&cCsPtj|j}t|ddddf|jt|jdddS(Niii(R RR~RRRR](RR|((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRws&cCsztj|j|j|jf}tj|dddt}t|ddddf|jt |j dddS(NRiRiii ( RR)R|R~R RRRRRR](RR|((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs!&cCs@tj|jdt}t||jt|jdddS(NRii(R RRRRRRR](RR|((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRscCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRRyii ( R RRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs0cCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRiii ( R RRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs0cCs|jdjtj}tj|}tjg|jjdD]}||^q?}t ||j t t |jjddS(NRyii( RR3RR4R RR)R.RRRRR(RR~RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs 0cCs|jdjtj}tj|dt}tjg|jj D]}||^qA}t ||j t t |jj ddS(NRyRi(RR3RR4R RRR)R.RRRRR(RR~RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs ,cCs{tj|jdddt}tjg|jjdD]}||^q5}t||j t t |jjddS(NRRyRii( R RRRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs0cCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRRyii ( R RRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs0cCsutj|jdd}tjg|jjdD]}||^q/}t||jt t |jjddS(NRiii ( R RRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs0cCs|jdjddg}tj|}tjg|jjdD]}||^qB}t||j t t |jjddS(NRyR.RuRzii(RuRz( RR3R RRR)R.RRRRR(RR~RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs 0cCs{tj|jdddt}tjg|jjdD]}||^q5}t||j t t |jjddS(NRRyRii( R RRRRR)R.RRRRR(RR|RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs0cCs|jdjddg}tj|dt}tjg|jjD]}||^qD}t ||j t t |jjddS(NRyR.RuRzRi(RuRz( RR3R RRRR)R.RRRRR(RR~RRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs ,s-Test has not been implemented for this class.RR&cCs tdS(N(tNotImplementedError(R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRscCs tdS(N(R(R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRscCs tdS(N(R(R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRscCs tdS(N(R(R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs(R_R`R%RRRRRRRRRRRRRRRRRRRRRR(((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRls,            cCs}tjjdgdggdd}tj|}tjjd d gddtfdtfd tfg}t||dS( Ni i Rtgroupg?gR.tgroup_10tgroup_11(i g?g(i gg?( Rtrect fromrecordsR RR2R/tfloatR(RtactualR<((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_rec_issue302s $$cCstjjddgddggdddg}tj|ddg}tjjddgd dtfdtfd tfd tfg}t||dS(Ni i i i RRtwhateverRg?gR.RR(i i g?g(i i gg?( RRRR RR2R/RR(RRR<((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_issue302s 0cCs`tj}|jd}tj|dt}td|jdt|j ddddS(NtGNPtprependR-ii( R t load_pandastexogR RtFalseRtcolumnsRtvar(tdtaR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_pandas_const_seriess   cCs`tj}|jd}tj|dt}td|jdt|j ddddS(NRRR-i( R RRR RRRRRR(RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_pandas_const_series_prepends   cCsVtjj}tj|dt}td|jdt|j ddddS(NRR-ii( R RRR RRRRRR(R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_pandas_const_dfscCsptjj}|dc|dj:s !cCsStj}tj|j|j}tjddgddgg}t||dS(Ng(RRR RRRR2R(RRRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_32Ds !cCsStj}tj|j|j}tj|dgddgg}t||dS(Ngg@(RRR RRRR2R(RRRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_24Js !cCsStj}tj|j|j}tjd|gddgg}t||dS(Ngg@(RRR RRRR2R(RRRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_25Ps !cCsStj}tj|j|j}tjddgddgg}t||dS(Ng@g$@g.@g6@(RRR RRR2R(RRRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_66Vs !(R_R`RRRRRRRRRRRR(((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyRs         t TestEnsure2dcBs2eZedZdZdZdZRS(cCshtjdjd }tj|dddddg|_|jjddd f|_||_dS( Ngy@idiRR>tbtctdi(idi( RRRRRARDtilocRtndarray(RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyR]s$cCstj|jt}t|d|jt|d|jjtj|jt}t|d|jdddgft|d|jjddS(Nii(R t _ensure_2dRDRRRRR(Rtresults((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_enfore_numpyds 'cCstj|jt}t|d|jt|d|jjtj|jt}t|d|jjdddgft |d|jjddS(Nii( R RRDRRRRRRR(RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_pandasls *cCstj|j}t|d|jt|ddtj|jdddf}t|d|jdddgft|dddS(Nii(R RRRRR(RR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyt test_numpyus "'(R_R`RRRRR(((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyR\s  c Cstjtddtj|dWdQXtj|}tjtddtj|dWdQXtjtddtj|dWdQXdS(NRsdata.name does not match coltunknownscol must be a str or ints"Column 'unknown' not found in data( R%R&R'R RRRARR(RRD((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_categorical_pandas_errorsscCstj|dt}tjtj|}t||tj|dt}t|j |_ t|j ddddf|t |j dddf|tj|dtdt\}}xNt tj|j D]4\}}||kst|||kstqWdS(NRit dictnames(R RRRt get_dummiest CategoricalRRtlistRRRt enumeratet categoriesR6(RtdesigntdummiesRRtiR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_categorical_seriess & !%cCstj|}tj|ddt}tjtj|}t||tji|d6|d6}tj|ddt}t||dS(NRRtappletban(RRAR RRRRR(RRDRR((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_categorical_dataframes c Csctjtddtj|dWdQXtjtddtj|idd6WdQXdS(NRsCan only convert one columniisdata.name does not match colR>(ii(R%R&R'R R(R((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyttest_categorical_errorss(.t__doc__tstatsmodels.compat.pythonRRRtnumpyRt numpy.randomRt numpy.testingRRRRtpandasRtpandas.util.testingRRR%tstatsmodels.datasetsR tstatsmodels.toolsR tstatsmodels.tools.toolsR tfixtureRtobjectRRqRrRRRRRRRRRRRRRR(((sAlib/python2.7/site-packages/statsmodels/tools/tests/test_tools.pyts:  "    w      F#