p7]c@sdZddlZddlmZmZmZddlmZddl m Z ddl m Z de fdYZd efd YZd efd YZed krddlZejedddgndS(sYTesting GLSAR against STATA Created on Wed May 30 09:25:24 2012 Author: Josef Perktold iN(tassert_almost_equaltassert_allcloset assert_equal(tGLSAR(t add_constant(t macrodatatCheckStataResultsMixincBseZdZRS(cCs|j|j}}t|j|jdt|j|jdt|j|jddddt|j|jdddddS(NitatolitrtolgMbp?gHz>(trestresultsRtparamstbseRttvaluestpvalues(tselfR R ((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyttest_params_tables (t__name__t __module__R(((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyRstCheckStataResultsPMixincBseZdZRS(cCs[|j|j}}t|j|jdd|j|jj}t||jdddS(NRgMb`?g-C6Z?(R R Rt fittedvaluestpredicttmodeltexog(RR R t predicted((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyttest_predicteds(RRR(((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyRst TestGLSARCorccBs2eZedZdZdZdZRS(cCstjj}dtjtj|dj}dtjtj|dj}ttj||dd jfdt }t ||d}|j d|_ dd l m}||_dS( Nitrealgdptrealinvtrealintitprependii(R (Rt load_pandastdatatnptdifftlogtvaluesRtc_tFalseRt iterative_fitR tresults.macro_gr_corc_stataR (tclstd2tg_gdptg_invtexoggtmod1R ((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyt setup_class's##*cCs?t|jjj|jjdt|jj|jjddS(Nii(RR RtrhoR tllftll(R((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyttest_rho4sc Csddlm}|jjj}|jjj}t||d}|jd}||ddd|ddddf}|jdd d dd d}t |j |j d d dgddddt |jj |j dddddt |j |j d d dgddddt t|jddt |jdd|j |jddd}t t|jddt t|jdddS(Ni(tARMAii torderiRtmethodtcsstiprinttdispiiRg{Gz?RgMbP?g?gQ?R iiR0(ii(tstatsmodels.tsa.arima_modelR4R RtendogRRR'tfitRR R0R Rtlenthistory( RR4R;RR.R tmod_armatres_armatres2((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyttest_glsar_arima:s.,&,cCs|jjj}|jjj}tjdddg}t|||}|j}|jd}|jd}t |j |j ddt |j |j ddt |jj |ddt |jj |dddS(Ng"~?g?gI +?iiRgdy=( R RR;RR!tarrayRR<R'RR R0(RR;RR0R.tres1tres0tres0b((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyttest_glsar_iter0Ps (RRt classmethodR/R3RBRG(((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyR%s  t__main__s-vvss-xs--pdb(t__doc__tnumpyR!t numpy.testingRRRt#statsmodels.regression.linear_modelRtstatsmodels.tools.toolsRtstatsmodels.datasetsRtobjectRRRRtpytesttmaint__file__(((sLlib/python2.7/site-packages/statsmodels/regression/tests/test_glsar_stata.pyts   ;