U Kv f@#h@sdZddlZddlmZddlmZeddddgd d d d gd dddgddddgddddgddddgdddd gd!d"d#d$gd%d&d'd(gd)d*d+d,gd-d.d/d0gd1d2d3d4gd5d6d7d8gd9d:d;dd?d@gdAdBdCdDgdEdFdGdHgdIdJdKdLgdMddNdOgdPdQdRdSgdTdUdVdWgdXdYdZd[gd\d]d^d_gd`daddagdbdcdddegdfdgdhdigdjdkdldmgdndodpdqgdrdsdtdugdvdwdxdygdzd{d|d}gd~dddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddddgddd+dgddddgddddgddddgdd d d gd d ddgddddgddddgddddgddddgdd d!d"gd#d$d%d&gd'd(d)d*gd+d,d-d.gdd/d0d1gd2d3d4d5gd6d7d8d9gd:d;d<d=gd>d?d@dgdAdBdCdDgdEdFdGdHgdIddJdKgdLdMddNgdedOdPdQgdRdSdTdqgdUdVdWdXgdYdZd[d\gd]d^dDd_gd`dadbdcgdddedfdggdhdidjdkgdldmdndogdpdqdrdsgdtd6dudvgdwdxdydzgd{d|d}d~gddddggdZedZGdddZdddZ dddZ ddZ e dkrddl mZddlZejjZedddZedZdZedZde_ejegddd\e_e_e_ejegddd\e_e_e_ejeddedZde_ej egddd\e_!e_e_ej egddd\e_"e_e_ejeddedZ#de#_ej$eddeddgddd\e#_%e#_e#_ej$eddeddgddd\e#_&e#_e#_e#jeddedZ'de'_e(ee)ddZ*ee*j+,e'_-e(ee)d2dZ*ee*j+,e'_.e'jedddS(zlgenerate py modules with test cases and results from mlabwrap currently matlab: princomp, garchar, garchma N)assert_array_almost_equal)arrayi]i%iiii8iKiiiiiii iiaipiiiii1i{iiiiiii iNiKi>ii=iii5i&iiiii`ii=ii i=iipiViHi-iii4ii4`6iiii}iiiiRiiiiViiii ii{ii>i5iSi6isiiiiii iiiiuiYiDiiiixiiiiii iiii0iiKiiii+i1Diiieiiiiiiiiiai(i=iGii+idi+iIi-i iys zHoldIt.save..z,import numpy as np from numpy import array z.from statsmodels.tools.testing import Holder z %s = Holder()z%s.z%scomment = '%s'z %s%s = %sza+zutf-8)encoding ) __dict__appendr+reprgetattrextendopenwritejoin) r,whatfilenameheaderZ useinstantcommenttxtprefixxfdr-r-r.savevs(   z HoldIt.save)NNTTN)__name__ __module__ __qualname__r/rGr-r-r-r.r*ssr* testsave.pycCsddlm}tjdddtd}||_|jddd td }|jtd d \|_ |_ |_ |j|d ddtd}|jtddfd d \|_ |_ |_ tjddd|j|d ddtd}|jtddftddf dd d \|_ |_ |_ tjddd|j|d dddS)Nrmlabd)Z precisionZ linewidthdatarKzgenerated data, divide by 1000)r@rBZ princomp1ZnoutFzmlab.princomp(x, nout=3)r@rArBZ princomp2r#zmlab.princomp(x[:20,], nout=3)Z princomp3z.mlab.princomp(x[:20,]-x[:20,].mean(0), nout=3)) mlabwraprMnpZset_printoptionsr*xorGZprincomprEZcoefZfactorsvaluesmean)rVfilenrMrPZ res_princompr-r-r.generate_princomps2  *rZcCsddlm}td}tdddg|_tdddg|_||jd d |jd dd |_| |jd d |jd dd |_ |j |d d d dS)NrrLZarmarepg?gg?g333333g{Gz?r#Fzw''mlab.garchma(-res_armarep.ar[1:], res_armarep.ma[1:], 20) mlab.garchar(-res_armarep.ar[1:], res_armarep.ma[1:], 20)''rS) rTrMr*rUrarmaZgarchmamarepZgarchararreprG)rYrM res_armarepr-r-r.generate_armareps &&racCsnddlm}|j|j|jdddd}|j|j|jdddd}t|j|dt|j |ddS)Nr)tsa)Znobsr[rN) Zstatsmodels.sandboxrbZarma_impulse_responser]r\rr^ravelr_)r`rbr_r^r-r-r. exampletests  re__main__rLiztestsavetls.pyZmlpacfzmlab.parcorr(x, [], 2, nout=3)rQrRT)r@rAZmlacfzmlab.autocorr(x, [], 2, nout=3)FZmlccfz,mlab.crosscorr(x[4:], x[:-4], [], 2, nout=3)Zmlywarz-mlab.ar(x100-x100.mean(), 10, 'yw').a.ravel() Zyw)rK)rK)/__doc__ZnumpyrUZ numpy.testingrrrVrEr*rZrarerHrTrMZsavedrvsZrvsdataZxar2Zx100Zx1000rYZres_pacfrBZparcorrZpacf100Zlags100Z bounds100Zpacf1000Zlags1000Z bounds1000rGZres_acfZautocorrZacf100Zacf1000Zres_ccfZ crosscorrZccf100Zccf1000Zres_ywarr\rXZmbarywardZ arcoef100Z arcoef1000r-r-r-r.s                                                                    e      ,,