U
    ÃmœdÏ  ã                >   @   st  d Z ddlZddlZddlmZ ddlmZ ddlm	Z	 ddl
mZ e	ƒ Zddd	d
dddddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBg<e_e dCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~g<¡e_e dd€d�d‚dƒd„d…d†d‡dˆd‰dŠd‹dŒd�dŽd�d�d‘d’d“d”d•d–d—d˜d™dšd›dœd�dždpdŸd d¡d¢d“d£d¤d¥d¦d¤d§d¨d©dªd«d¬d­d®d¯d°d±d²d³d´dµd¶d·g<¡e_e d¸d¹dºd»d¼d½d¾d¿dÀdÁdÂdÃdÄdÅdÆdÇdÈdÉdÊd”dËdÌdÍdÎdÏdÐdÑdÒdÓdÔdÕdÖd×dØdÙdÚdÛdÜdÝdÞdßdàdádâdãdädådsdædçdèdédêd‚dëdìdídîdïdðg<¡e_e dñdòdódôdõdöd÷dødùdúdûdüdýdþdÿ�d �d�d�d�d�d�d�d�d�d	�d
�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d�d �d!�d"�d#�d$�d%�d&�d'�d(�d)�d*�d+�d,g<¡e_G �d-�d.„ �d.ƒZG �d/�d0„ �d0eƒZG �d1�d2„ �d2ƒZdS (3  z>

Created on Sun Jun 30 20:25:22 2013

Author: Josef Perktold
é    N)Úassert_allclose)Úadd_constant)ÚHolder)ÚTLinearModelzApr.1982zApr.1983zApr.1984zApr.1985zApr.1986zAug.1982zAug.1983zAug.1984zAug.1985zAug.1986zDec.1982zDec.1983zDec.1984zDec.1985zDec.1986zFeb.1284zFeb.1982zFeb.1983zFeb.1985zFeb.1986zJan.1982zJan.1983zJan.1984zJan.1985zJan.1986zJul.1982ZJuly1983ZJuly1984ZJuly1985ZJuly1986ZJune1982ZJune1983ZJune1984ZJune1985ZJune1986zMar.1982zMar.1983zMar.1984zMar.1985zMar.1986ZMay1982ZMay1983ZMay1984ZMay1985ZMay1986zNov.1982zNov.1983zNov.1984zNov.1985zNov.1986zOct.1982zOct.1983zOct.1984zOct.1985zOct.1986z	Sept.1982z	Sept.1983z	Sept.1984z	Sept.1985z	Sept.1986gF¶óýÔxÁ¿grùé·¯³¿gq=
×£p­¿g	Šcîª?géH.ÿ!ý¦¿gZõ¹ÚŠýµ¿g­ú\mÅþ²¿gˆc]ÜFæ?gƒÀÊ¡E¶³¿gÃõ(\�Âµ?çú~j¼t“h?gvOjM³?g¾0™*¥¿gbX9´È¶¿g� ø1æ®Í?gO@aÃÓ»?g333333£?gØðôJY†¸?g¨5Í;NÑ‘?gû:pÎˆ²¿g333333³?g%uš®¿gX9´Èv¾¯¿g˜Ý“‡…Z£¿gX¨5Í;N‘?gjMóŽSt´¿g%ušŽ¿g¡g³êsµ…?g<½R–!Ž¥¿g’ËH¿}m¿gw-!ôl¶?g=
×£p=º?gx$(~Œ©¿g�St$—ÿÈ?g×4ï8EG²¿gˆc]ÜF¨?g€·@‚âÇÀ?g|ò°Pkš§?g	ž^)Ë ¿ga2U0*©ƒ¿g¹�ðHÀ?gÒ Þ	Š¿gL7‰A`å�?g¸…ëQ¸Ž¿g	ŠcîÂ¿g9´Èv¾Ÿz¿g+‡ÙÎ÷£?g„žÍªÏÕ–?gD‹lçû©±¿gã6À[ Á?gÓÞà“©Â?g Òo_Îy?geâX·±?g·bÙ=y˜¿g†ZÓ¼ã­¿gf÷äa¡Ö´?gRI�€&Â¶¿g0»'µ¶¿g_˜LŒº?gKY†8ÖÅ½¿g¸…ëQ¸ž¿gŸ<,Ôšæ­¿gû:pÎˆ’¿gKÈ=›UŸ?gjMóŽS¤¿gh‘í|?5ž¿gvqà- ¿g%uš¾?g¸…ëQ¸~?gÎQÚ¼?gˆ…ZÓ¼ã¤?gÛù~j¼tƒ?gû:pÎˆÒž?gþe÷äa¡–?g–²q¬‹›?gøSã¥›Ä°?gú~j¼t“h¿g¤p=
×£ ?g¥½Á&£¿gn£¼t?g/Ý$�…?gp_ÎQš¿gÎˆÒÞà“?gX9´Èv¾�¿gL7‰A`å�¿gºI+‡¦¿gUÁ¨¤N@ƒ?gÇº¸�ðf¿gŠcîZB®¿gvqà-�?gtF”ö_˜¿g¢´7øÂdº?gS–!Žuq{¿gÒo_Î‘¿gUÁ¨¤N@“?gûËîÉÃ²?gyé&1¬|¿gõÛ×�sF„¿g[B>èÙ¬ª?ggÕçj+ö‡?gˆ…ZÓ¼ã„¿g' ‰°áé¥¿gUÁ¨¤N@£?g2U0*©°?gMŒJê¤?g-Cëâ6*?g"ýöuàœ±?gÃÓ+eâ¨?g¨5Í;NÑ‘¿gZd;ßO�§?g{®Gáz„?gïÉÃB­i®¿gœÄ °rh±?g
h"lxzµ¿gà-� ¨?g_ÎQÚ‹?g?ÆÜµ„| ¿g2æ®%äƒ®¿gÃõ(\�ÂÅ¿glxz¥,Cœ?gàœ¥½Áw?g¡g³êsµ…¿g
×£p=
§?gÃÓ+eâ˜¿gu“V½?gmÅþ²{ò ¿ggÕçj+ö§?g„žÍªÏÕ–¿g”ö_˜Le¿g�St$—ÿ�?g o�Å�¡?gôlV}®¦?gòAÏfÕçÂ?g®Ø_vOÆ?gÎˆÒÞà£¿g,eâX§¿gÈ˜»–�ª?gyé&1¬l?gÁ¨¤N@Á?gHPüs§¿g¨WÊ2Ä±¾?g@¤ß¾œ³¿gµ¦yÇ)ª¿gèj+ö—Ý“¿g6<½R–±¿gàœ¥½Á·?g/n£¼²¿g^ºI+·?gàœ¥½Á—?g®Ø_vOv¿g	ž^)Ë€?g€·@‚âÇ˜?gKÈ=›U�?g-Cëâ6Z?gyé&1¬œ?g/n£¼‚?g'Â†§WÊ²?gTã¥›Ä  ?glxz¥,Cœ¿g‚âÇ˜»–�?gö(\�Âõ¸¿gÞ“‡…ZÓ¬?gßà“©‚Á?g}Ð³Yõ¹º?g©ÐDØðÄ?gÛù~j¼t“¿gðHPü³¿gÏfÕçj+¦?g¹�ðH°?g(µ¦y·?g8øÂdª`´¿g	ù g³ê³?gPüs×¢¿gæ?¤ß¾Œ¿é   é   é$   é   é)   é   é   é   é8   é3   é.   é   é   é   é%   é   é*   é    é   é   é9   é4   é/   é   é   é   é&   é   é+   é!   é   é   é:   é5   é0   é   é   é   é'   é   é,   é"   é   é	   é;   é6   é1   é   é   é   é(   é   é-   é#   é   é
   é<   é7   é2   é   c                   @   sD   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zej	j
dd„ ƒZdS )ÚCheckTLinearModelMixinc                 C   s  | j }| j}t|jd d… |jjdd� t|jd d… |jjd d …df ddd� t|jd d… |jjd d …df ddd� t|j	d d… |jjd d …d	f d
dd� t|jd |j
dd� t|jd |jddd� t |jj ¡ ¡}t|jd |dd� t|j|jdd� d S )NéþÿÿÿgiUMuÿ>©Úatolr
   r   gñhãˆµøä>©ÚrtolrF   r   r"   g;ßO�—n‚?ç-Cëâ6
?©rH   g{®GázÄ?éÿÿÿÿ)Úres1Úres2r   ÚparamsÚloc_fitZcoefficientsÚbseÚtableZtvaluesZpvaluesZdofZdofseÚnpÚsqrtZ	scale_fitÚfitted_valuesZmeanZllfZlogLik)ÚselfrL   rM   Z	scale_est© rV   úa/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/miscmodels/tests/test_tmodel.pyÚ
test_basicI   s    ***z!CheckTLinearModelMixin.test_basicc                 C   s2   | j }t|j|jddd� t|j|jddd� d S )Ng333333Ã?gü©ñÒMb`?rG   gš™™™™™¹?gü©ñÒMbp?)rL   r   ZbsejacrP   )rU   rL   rV   rV   rW   Útest_bse[   s    zCheckTLinearModelMixin.test_bsec                 C   sH   | j }| j}| ¡ }|jj| }t||jjdd� t||jjdd� d S )Ngü©ñÒMb0?rJ   g�íµ ÷ÆÀ>rE   )	rL   rM   ZpredictÚmodelÚendogr   rO   rT   Z	residuals)rU   rL   rM   ZfittedvaluesZresidrV   rV   rW   Útest_fitteda   s    z"CheckTLinearModelMixin.test_fittedc                 C   s`   | j }| j}t|j|jdd� t|j|jdd� t|jj|jjdd� t|jj|jjdd� d S )Ng-Cëâ6?rE   rI   rJ   g»½×Ùß|Û=)rL   Úresfr   rN   rP   rZ   r[   Úexog)rU   rL   r]   rV   rV   rW   Útest_formulam   s    z#CheckTLinearModelMixin.test_formulac                 C   sb   | j }t| ddƒ}|jjj\}}|j|| | ks6t‚|j|d ksHt‚t|j	ƒ|| ks^t‚d S )NÚk_extrar   r
   )
rL   ÚgetattrrZ   r^   ÚshapeZdf_residÚAssertionErrorZdf_modelÚlenrN   )rU   Úresr`   ZnobsZk_varsrV   rV   rW   Útest_dfw   s    zCheckTLinearModelMixin.test_dfc                 C   sX   | j }| j}t t|jƒ¡}| ¡  | |¡ | |¡ | ¡  | |¡ | |¡ d S ©N©	rL   r]   rR   Úeyerd   rN   ÚsummaryZt_testZf_test©rU   rL   r]   ZcontrrV   rV   rW   Ú
test_smoke   s    


z!CheckTLinearModelMixin.test_smokeN)Ú__name__Ú
__module__Ú__qualname__rX   rY   r\   r_   rf   ÚpytestÚmarkÚsmokerl   rV   rV   rV   rW   rC   G   s   
rC   c                   @   s   e Zd Zedd„ ƒZdS )Ú
TestTModelc                 C   sv   t j}tt jƒ}t||ƒ}|jddd�}tjdt jt jdœd�}|jddd�}ddlm} || _	|| _
|| _d	| _d S )
NÚbfgsF©ÚmethodZdispúprice ~ CRSP©ZpriceÚCRSP)Údatar
   )Úres_t_dfestr   )ÚmmÚ
m_mariettar   ry   r   ÚfitÚfrom_formulaZresults_tmodelr{   rM   rL   r]   r`   )Úclsr[   r^   Úmodre   Úmodfr]   rM   rV   rV   rW   Úsetup_class‘   s    

ÿzTestTModel.setup_classN)rm   rn   ro   Úclassmethodrƒ   rV   rV   rV   rW   rs   �   s   rs   c                   @   s(   e Zd Zedd„ ƒZejjdd„ ƒZdS )ÚTestTModelFixedc                 C   sj   t j}tt jƒ}t||dd�}|jddd�}tjdt jt jdœdd�}|jddd�}|| _|| _d	| _	d S )
Nr"   )Úfix_dfrt   Fru   rw   rx   )rz   r†   r
   )
r|   r}   r   ry   r   r~   r   rL   r]   r`   )r€   r[   r^   r�   re   r‚   r]   rV   rV   rW   rƒ   £   s    
þzTestTModelFixed.setup_classc                 C   sX   | j }| j}t t|jƒ¡}| ¡  | |¡ | |¡ | ¡  | |¡ | |¡ d S rg   rh   rk   rV   rV   rW   rl   ´   s    


zTestTModelFixed.test_smokeN)	rm   rn   ro   r„   rƒ   rp   rq   rr   rl   rV   rV   rV   rW   r…   ¡   s   
r…   )Ú__doc__rp   ÚnumpyrR   Znumpy.testingr   Zstatsmodels.tools.toolsr   Zstatsmodels.tools.testingr   Zstatsmodels.miscmodels.tmodelr   r|   Z
date_labelÚarrayr}   ry   Zam_canÚdaterC   rs   r…   rV   rV   rV   rW   Ú<module>   sv  
                                           ô                                                    ø
                                                    ø
                                                    ø
                                                        üH