U
    Ãmœd„  ã                   @   s~   d dl m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
 G dd	„ d	ƒZejjG d
d„ deƒƒZG dd„ deƒZdS )é    )Úassert_almost_equalN)Ú	stackloss)ÚOLS)Úadd_constanté   )ÚRegressionResultsc                   @   s   e Zd ZdZedd„ ƒZdS )ÚGenResz<
    Loads data and creates class instance ot be tested
    c                 C   s4   t  ¡ }t|jƒ|_t|j|jƒ ¡ | _tƒ | _	d S )N)
r   Úloadr   Zexogr   ZendogÚfitÚres1r   Úres2)ÚclsÚdata© r   úb/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/emplike/tests/test_regression.pyÚsetup_class   s    zGenRes.setup_classN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úclassmethodr   r   r   r   r   r      s   r   c                   @   s€   e Zd ZdZejjdd„ ƒZejjdd„ ƒZdd„ Z	dd	„ Z
ejjd
d„ ƒZejjdd„ ƒZejjdd„ ƒZejjdd„ ƒZdS )ÚTestRegressionPowellú·
    All confidence intervals are tested by conducting a hypothesis
    tests at the confidence interval values.

    See Also
    --------

    test_descriptive.py, test_ci_skew
    c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Néâÿÿÿr   r   Úpowell©Zreturn_weightsÚmethodé   é   ©r   Úel_testr   r   Z
test_beta0©ÚselfZbeta0resr   r   r   Útest_hypothesis_beta0#   s
    ÿ z*TestRegressionPowell.test_hypothesis_beta0c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Nç      à?r   r   r   r   r   ©r   r    r   r   Z
test_beta1©r"   Zbeta1resr   r   r   Útest_hypothesis_beta1*   s
    ÿ z*TestRegressionPowell.test_hypothesis_beta1c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Nr   r   r   r   r   ©r   r    r   r   Z
test_beta2©r"   Zbeta2resr   r   r   Útest_hypothesis_beta21   s
    ÿ z*TestRegressionPowell.test_hypothesis_beta2c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Nr   é   r   r   r   r   r   ©r   r    r   r   Z
test_beta3©r"   Zbeta3resr   r   r   Útest_hypothesis_beta37   s
    ÿ z*TestRegressionPowell.test_hypothesis_beta3c                 C   s(   | j jddddd�}t|| jjdƒ d S )Nr   g33333sJÀgš™™™™8Àr   ©Úlower_boundÚupper_boundr   r+   ©r   Úconf_int_elr   r   Útest_ci_beta0©r"   Zbeta0cir   r   r   r4   >   s
    
 ÿz"TestRegressionPowell.test_ci_beta0c                 C   s(   | j jddddd�}t|| jjdƒ d S )Nr   g�—nƒÀÚ?gZd;ßO�ï?r   r/   r   ©r   r3   r   r   Útest_ci_beta1©r"   Zbeta1cir   r   r   r7   E   s    ÿz"TestRegressionPowell.test_ci_beta1c                 C   s(   | j jddddd�}t|| jjdƒ d S )Nr   çáz®Gáâ?çš™™™™™@r   r/   é   ©r   r3   r   r   Útest_ci_beta2©r"   Zbeta2cir   r   r   r=   K   s
    
 ÿz"TestRegressionPowell.test_ci_beta2c                 C   s(   | j jddddd�}t|| jjdƒ d S )Nr+   gö(\�ÂõØ¿g{®Gáz„?r   r/   é   ©r   r3   r   r   Útest_ci_beta3©r"   Zbeta3cir   r   r   rA   Q   s    ÿz"TestRegressionPowell.test_ci_beta3N)r   r   r   r   ÚpytestÚmarkÚslowr#   r'   r*   r.   r4   r7   r=   rA   r   r   r   r   r      s   





r   c                   @   s€   e Zd ZdZdd„ Zdd„ Zejjdd„ ƒZ	ejjdd	„ ƒZ
ejjd
d„ ƒZejjdd„ ƒZejjdd„ ƒZejjdd„ ƒZdS )ÚTestRegressionNMr   c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Nr   r   r   Únmr   r   r   r   r!   r   r   r   r#   c   s
    ÿ z&TestRegressionNM.test_hypothesis_beta0c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Nr$   r   rG   r   r   r   r%   r&   r   r   r   r'   i   s
    ÿ z&TestRegressionNM.test_hypothesis_beta1c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Nr   r   rG   r   r   r(   r)   r   r   r   r*   o   s
    ÿ z&TestRegressionNM.test_hypothesis_beta2c                 C   sT   | j jdgdgddd�}t|d d… | jjd d… dƒ t|d | jjd dƒ d S )Nr   r+   r   rG   r   r   r   r,   r-   r   r   r   r.   v   s
    ÿ z&TestRegressionNM.test_hypothesis_beta3c                 C   s$   | j jddd�}t|| jjdƒ d S )Nr   rG   ©r   r?   r2   r5   r   r   r   r4      s    zTestRegressionNM.test_ci_beta0c                 C   s$   | j jddd�}t|| jjdƒ d S )Nr   rG   rH   r?   r6   r8   r   r   r   r7   �   s    zTestRegressionNM.test_ci_beta1c                 C   s(   | j jddddd�}t|| jjdƒ d S )Nr   r9   r:   rG   r/   r?   r<   r>   r   r   r   r=   ’   s    ÿzTestRegressionNM.test_ci_beta2c                 C   s$   | j jddd�}t|| jjdƒ d S )Nr+   rG   rH   r?   r@   rB   r   r   r   rA   ˜   s    zTestRegressionNM.test_ci_beta3N)r   r   r   r   r#   r'   rC   rD   rE   r*   r.   r4   r7   r=   rA   r   r   r   r   rF   X   s   





rF   )Znumpy.testingr   rC   Zstatsmodels.datasetsr   Z#statsmodels.regression.linear_modelr   Zstatsmodels.toolsr   Zresults.el_resultsr   r   rD   rE   r   rF   r   r   r   r   Ú<module>   s   @