p7]c@sdZddlmZddlmZmZddlZddlZddlZ ddl m Z m Z ddl ZddlmZddlmZddlmZddljjjZd Ze\ZZZe jed eZee jeeg\Z Z!e j"e j#kst$e!j"e j#ks5t$e j%d d d ddddddddddg Z&e j%ddddddddd d!d"d#d$g Z'e j%d%d&d'd(d)d*d+d,d-d.d/d0d1g Z(e j%d2d3d4d5d6d7d8d9d:d;d<d=d>g Z)e j%d?d@dAdBdCdDdEdFdGdHdIdJdKg Z*dLZ+dMZ,dNZ-ee.e'Z/e/dOe/dO Z/e dde/fZ0dPe1fdQYZ2dRe1fdSYZ3dTe3fdUYZ4dVe3fdWYZ5dXe3fdYYZ6dZe3fd[YZ7d\e3fd]YZ8d^e3fd_YZ9d`e3fdaYZ:dbe3fdcYZ;dde3fdeYZ<dfe3fdgYZ=dhe1fdiYZ>dje1fdkYZ?dle?fdmYZ@dnZAdoZBdS(ps> Created on Fri Oct 04 13:19:01 2013 Author: Josef Perktold i(tprint_function(tlrangetlmapN(tassert_allcloset assert_equal(tiolib(t add_constant(tOLSc Csddl}|jjtd}|jj|d}tjt#tj |dt j dt }WdQX|dj }|jd}x}|D]u}t j||d|g:5>g mj?gĊÛ?g j6E?gW?g ]k?gg p?g ?g{=?cCstjttt}|j}tj\}}t|jtddddt|j t ddddt |d s~t dS(NtrtolgHz>tatolg& .>igv !>t_results( tgmmtIV2SLStendogtexogt instrumenttfitR+RtparamstbsethasattrtAssertionError(tmodtrestntk((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_iv2sls_rYs  c Cstj\}}tj\}}tjtjt|}tjj|}tj t tt}|j tj tjdt ddd|dddidd 6dd 6}t|jtd d d d t|jtd dd d|jj|j|}t|tj|jd dd ddS(Nitmaxiterit inv_weightst optim_methodtbfgst optim_argsg:0yE>tgtoltdispR;g-C6?R<g ףp= ?gh㈵>(RAR+RBR'tdottTtlinalgtinvR>tIVGMMR@RCtonestfloatRRDREtmodeltscoreR,( RJRKtnobstk_instrtw0invtw0RHRIR\((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_ivgmm0_rms+c Cstjddddddddd d d d d g }tjdddddddddddddg }tj\}}tj\}}tjtjt|}tjj|}t t tj j }t jt tt} | j |ddd|ddd id!d"6d#d$6} dS(%Ngr^RkP"@gU?gAAJhg9L,xRRiRS(R'tarrayRAR+RBRTRURVRWRR@RCRDR>RX( t params_statat bse_stataRJRKR]R^R_R`tstartRHRI((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyttest_ivgmm1_statasit TestGMMOLScBs)eZedZdZdZRS(c Cst}tt|j}|j\}}tj|j||}tj t||}|jtj |jdt ddd|dddidd 6dd 6}||_ ||_ dS( NiRMiRNRORPRQgư>RRRS(texog_stRR@RCR+R'RTRUR>RXRYRZtres1tres2(tclsRAtres_olsR]R^R_RHRI((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt setup_classs+ cCs|j|j}}t|j|jddddt|j|jdddd|jjjd}d}t|j||jddddt|j||jdddddS( NR;gMb@?R<igh㈵>igh㈵>gHz>( RiRjRRDR[RAR+REtHC0_se(tselfRiRjRJtdffac((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_basics#cCs|j|j}}dS(N(RiRj(RoRiRj((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_others(t__name__t __module__t classmethodRmRqRr(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRgs tCheckGMMcBsYeZddgZddgZdZdZdZejj dZ dZ RS(gh㈵>gƠ>cCs|j|j}}|j\}}t|j|jd|ddt|j|jddd||jjjd}d}|j\}}t|j ||j d|ddt|j ||j ddd|dS(NR;R<ii( RiRjt params_tolRRDR[RAR+tbse_tolRE(RoRiRjR;R<RJRp((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRqs#cCsU|j|j}}t|j|jddddt|j|jdddddS(NR;gh㈵>R<ig-C6 ?(RiRjRtqtQtjvaltJ(RoRiRj((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt _est_othersc Cs|j|j}}tjt|j}|j|}t|j|j ddddt|j |j dddd|j \}}t|j|j d|dd|t|j |j d|dd||j |d }|j|d }dS(NR;g-q=R<ii i(RiRjR'teyetlenRDtt_testRttvaluettvaluestpvaluetpvaluesRxtf_testt wald_test( RoRiRjt restrictiontres_tR;R<tres_ftres_wald((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyttest_hypothesiss##cCs@|j}|j}tt|jdt|jddS(Ni(RitsummaryRRttablesRD(RoRitsumm((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_summarys  cCs[tj|j}t|_|j}dt|ks?tdt|ksWtdS(NsP>|t|sP>|z|(tcopytdeepcopyRiR)tuse_tRtstrRG(RoRiR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_use_ts   ( RsRtRwRxRqR}RR#tmarktsmokeRR(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRvs     t TestGMMSt1cBseZedZRS(c Cst}tt|jj}tj\}}tjtj t|}t j t|t}|j|ddd|dddidd6d d 6d it d 6}||_ d dlm}||_dS(NRMi RNRORPRQgư>RRiRStwargstcenteredi(tresults(RhRR@RCRDRBR+R'RTRUR>RXtFalseRitresults_gmm_griliches_iterRRj( RkRAReR]R^R_RHtres10R((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms (RsRtRuRm(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRstTestGMMStTwostepcBseZedZRS(c Csddg|_ddg|_t}tt|jj}tj\}}t j tj t|}t j t|t}|j|ddd|ddd id d 6d d 6ditd6}||_ddlm}||_dS(Ng-C6 ?gh㈵>gƠ>RMiRNRORPRQgư>RRiRSRRi(tresults_twostep(RwRxRhRR@RCRDRBR+R'RTRUR>RXRRitresults_gmm_grilichesRRj( RkRAReR]R^R_RHRR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms (RsRtRuRm(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRstTestGMMStTwostepNOcBseZedZRS(c Csddg|_ddg|_t}tt|jj}tj\}}t j tj t|}t j t|t}|j|ddd|ddd idd 6d d 6d itd6dt}||_ddlm}||_dS(Ng-C6 ?gh㈵>gư>RMiRNRORPRQRRiRSRRthas_optimal_weightsi(R(RwRxRhRR@RCRDRBR+R'RTRUR>RXRRiRRRj( RkRAReR]R^R_RHRR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRm5s (RsRtRuRm(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyR2stTestGMMStOnestepcBs eZedZdZRS(c Csddg|_ddg|_t}tt|jj}tj\}}t j tj t|}t j t|t}|j|ddd|ddd id d 6dd 6}||_d dlm}||_dS(NgMb@?g-C6 ?gy&1|?RMiRNRORPRQgư>RRRSi(tresults_onestep(RwRxRhRR@RCRDRBR+R'RTRUR>RXRiRRRj( RkRAReR]R^R_RHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRmLs cCsk|j|j}}tjtj|jdt}|jjj|jj tj j |jj }dS(NR( RiRjR'tsqrttdiagt _cov_paramsRR[t gmmobjectiveRDRVRWtweights(RoRiRjRERy((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyttest_bse_other`s$(RsRtRuRmR(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRJstTestGMMStOnestepNOcBseZedZRS(c Csddg|_ddg|_t}tt|jj}tj\}}t j tj t|}t j t|t}|j|ddd|dd d idd 6dd 6d itd6dt}||_ddlm}||_dS(Ngh㈵>gư>gh㈵>gƠ>RMiRNRORPRQRRRSRRRi(R(RwRxRhRR@RCRDRBR+R'RTRUR>RXRRiRRRj( RkRAReR]R^R_RHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRmps (RsRtRuRm(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRmstTestGMMStOneitercBs eZedZdZRS(c Csddg|_ddg|_t}tt|jj}tj\}}t j tj t|}t j t|t}|j|ddd|ddd id d 6d d 6}||_ddlm}||_dS(NgMb@?g-C6 ?gy&1|?RMiRNRORPRQgư>RRiRS(R(RwRxRhRR@RCRDRBR+R'RTRUR>RXRiRRRj( RkRAReR]R^R_RHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms c Cs|j|j}}|jj|j}|jj|}tjtj|j dt d|j }tjtj|j dt }t j d}tjt jt |}|jjj|jj|}dS(NRRi(RiRjR[tmomcondRDtcalc_weightmatrixR'RRRRRRBR+RTRUR( RoRiRjtmomstwRER]R_Ry((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRs$ (RsRtRuRmR(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRstTestGMMStOneiterNOcBseZedZRS(c Csddg|_ddg|_t}tt|jj}tj\}}t j tj t|}t j t|t}|j|ddd|dd d idd 6d d 6ditd6dt}||_ddlm}||_dS(Ngh㈵>gư>gh㈵>gƠ>RMiRNRORPRQRRiRSRRR(R(RwRxRhRR@RCRDRBR+R'RTRUR>RXRRiRRRj( RkRAReR]R^R_RHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms (RsRtRuRm(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRstTestGMMStOneiterNO_LinearcBseZedZRS(c CsMddg|_ddg|_t}tt|jj}tj\}}t j tj t|}t j t|t}|j|ddd|dd d id d 6d d6ditd6dt}||_t jt|t}|j|ddd|dd d idd 6d d6ditd6dt}||_ddlm}||_dS(Ng:0y5>g& .>g& .>g|=RMiRNRORPRQg:0yE>RRiRSRRRgư>(R(RwRxRhRR@RCRDRBR+R'RTRUR>t LinearIVGMMRRiRXtres3RRRj( RkRAReR]R^R_RHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms$  (RsRtRuRm(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRstTestGMMStOneiterNO_NonlinearcBs eZedZdZRS(c CsYddg|_ddg|_t}tt|jj}tj\}}t j tj t|}d}t j t|t|}|j|ddd|dd d id d 6d d6ditd6dt}||_t jt|t}|j|ddd|dd d idd 6d d6ditd6dt}||_ddlm} | |_dS(Ng-C6 ?gh㈵>g?cSstj||S(N(R'RT(RDRA((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pytfuncsRMiRNRORPRQg:0yE>RRiRSRRRgư>(R(RwRxRhRR@RCRDRBR+R'RTRUR>tNonlinearIVGMMRRiRXRRRRj( RkRAReR]R^R_RRHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms&   cCs|jjd}|jj}|jjj||}t|jjj|jjj||}t||ddddt||dddd|jjj|jj|}t|tj t |dddddS(Ng?R;gư>R<igHz>g:0yE>( RiRDRR[R\tsupert __class__RR'R,R(RoRDRtsc1tsc2((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_score s ! (RsRtRuRmR(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRstTestGMMStOneiterOLS_LinearcBseZedZRS(c Csddg|_ddg|_t}tt|j}tjt|j }t j \}}tj |j ||}tjt||}|j|ddd|didd6dd d itd 6d d 6dt}t|_|jt|j |_||_||_dS(Ngdy=g-q=RMiRNRQRStweights_methodtiidRRtk_paramstddofR(RwRxRhRR@RCR'RYRRDRBR+RTRUR>RRR)RR]tdf_residRiRj( RkRARlReR]R^R_RHRI((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms"    (RsRtRuRm(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRst TestGMMSt2cBs eZedZdZRS(c Cs1t}tt|jj}tj\}}tjtj t|}t j t|t}|j|ddd|didd6t d6dd d id d 6dd 6}||_ ddlm}||_t j t|t}|j|ddd|didd6t d6dd d id d 6dd 6}||_dS(NRMiRNRiRRRORPRQgư>RRRSi(tresults_gmm2s_robust(RhRR@RCRDRBR+R'RTRUR>RXRRitresults_ivreg2_grilichesRRjR( RkRAReR]R^R_RHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRm@s   c Cs|j|j}}t|j|jddddt|j|jdddd|jjjd}tjdd}t|j ||j ddddt|j ||j ddddtjtj |j d t d |j }t|j |j dd ddtjtj |j d t d |j d t }t|j |j dd ddt|jj |j ddddt|jj |j dddddS(NR;g-C6 ?R<igh㈵>gH@ig{Gzt?RRg?t use_weightsg?(RiRjRRDR[RAR+R'RRERRR)RR(RoRiRjRJtdffactRE((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRq[s ## "(RsRtRuRmRq(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyR<st CheckIV2SLScBsAeZdZdZdZdZejjdZ RS(cCs|j|j}}t|j|jddddt|j|jdddd|jjjd}t|j|jddddt|j|jddddt|j|jdddddS(NR;g& .>R<ig|=gdy=g& .>( RiRjRRDR[RAR+RER(RoRiRjRJ((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRqscCs]|j|j}}t|j|jddddt|j|jddddt|j|jddddt|j |j ddddtt j |j |jddddt|j|jddddt|j|jddddt|j|jddddt|j|jddddt|j|jt|j|jdS(NR;gHz>R<ig|=g:0yE>g& .>(RiRjRtrsquaredtr2t rsquared_adjtr2_atfvaluetFtf_pvaluetFpR'Rt mse_residtrmsetssrtrsstuncentered_tsstyyt centered_tsstyyctesstmssRtdf_modeltdf_mRtdf_r(RoRiRj((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRrs(cCs |j|j}}tjt|j}|j|}t|j|j ddddt|j |j dddd|j |d }t|j |j ddddt|j |jddddt|j |jddddt|j |jdddddS(NR;g-q=R<iig|=g:0yE>(RiRjR'R~RRDRRRRRRRRRRR(RoRiRjRRR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRscCsk|j|j}}|j}t|d|jdddddt|d|jddddd dS( NitDWHR;gdy=R<itDWHpg|=g}:(RiRjt spec_hausmanRthausman(RoRiRjthausm((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyt test_hausmans $cCs@|j}|j}tt|jdt|jddS(Ni(RiRRRRRD(RoRiR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRs  ( RsRtRqRrRRR#RRR(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRs   t TestIV2SLSSt1cBs eZedZdZRS(cCstt}tt|jj}tj\}}tjt|t}|j}||_ ddl m }||_ dS(Ni(t results_small( RhRR@RCRDRBR+R>R?RiRRRj(RkRAReR]R^RHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRms  cCshtjjd}|jdd}|jd}tjd|dddftjd||dddfR?RCtsizeR+RRD(Rotrstxtztety_1dtyty_2dty_seriesty_dftx_1dtx_2dtx_dft x_df_singletx_seriestz_2dtz_seriestz_1dtz_dftystxstzsRjRit res1_2sintrt_yt_xt_zRI((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyttest_input_dimensionssLD8=2      (RsRtRuRmR (((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRscCstddddf}tjt|t}|j}t|jtj |j }tt |j dt |j ddS(Nii(RhR>R?R@RBRCRRR'tnanRRRRD(RARHRIR((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyttest_noconstants   c Cstjddddddddddddddddd dd dg}tjd d d ddd dddd dddddddddd g}tjtjt||f}dtjfdY}tjddddg}|d|d|d|dd d!d j|d"id d#6}|j}t t|j dt|j dgt t|j D]}d$|^qt}t |j j||d|d|d|dd d!d } d%d&d'd(g}| j|| j|d"id d#6} t | j j|dS()Ng?g333333?g@g@g?gffffff?g?g?gffffff?gQ?ig?g?gɿg?g333333ӿggٿgg333333?gQѿgQ?tGMMbasecBseZdZRS(c Ss|\}}}}|jdddf}|j}|j}|||||} ||||d||d|d|} tj| | f} | S(Nii (R@RRARBR'R( RoRDtp0tp1tp2tp3R@RAtinsttmom0tmom1tg((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyRs  (RsRtR(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyR sg?g{Gz?iR@RARBtk_momsiRRQRSsp%2dtbetatgammatpsitphi(R'RbRRYRR>tGMMRCRRRRDR-R[t exog_namestset_param_names( tcdtdcdRR tbeta0RIRR5tpnamesRHRi((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pyttest_gmm_basic s&*!'$$  ',  (Ct__doc__t __future__Rtstatsmodels.compat.pythonRRRR#tnumpyR't numpy.testingRRR Rt statsmodelsRtstatsmodels.tools.toolsRt#statsmodels.regression.linear_modelRt"statsmodels.sandbox.regression.gmmtsandboxt regressionR>R9tyg_dftxg_dftzg_dftasarrayRZR@RARBR:tfloat64RGRbtvarnamesRDRERRRLRaRfRtidxRhtobjectRgRvRRRRRRRRRRRRRR R"(((sLlib/python2.7/site-packages/statsmodels/sandbox/regression/tests/test_gmm.pytsp     $'       &=#-0#E?>