U
    Ãmœd¢  ã                   @   sZ   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
G d	d
„ d
e
ƒZdS )é    N)Úassert_almost_equal)Úcancer)ÚELOriginRegressé   )ÚOriginResultsc                   @   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   s(   t  ¡ }t|j|jƒ ¡ | _tƒ | _d S )N)	r   Úloadr   ZendogZexogÚfitÚres1r   Úres2)ÚclsÚdata© r   ú^/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/emplike/tests/test_origin.pyÚsetup_class   s    zGenRes.setup_classN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úclassmethodr   r   r   r   r   r   
   s   r   c                   @   s0   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )Ú
TestOriginzB
    See OriginResults for details on how tests were computed
    c                 C   s   t | jj| jjdƒ d S ©Né   )r   r
   Úparamsr   Útest_params©Úselfr   r   r   r      s    zTestOrigin.test_paramsc                 C   s   t | jj| jjdƒ d S r   )r   r
   Zllf_elr   Ztest_llf_hatr   r   r   r   Útest_llf   s    zTestOrigin.test_llfc                 C   s&   t | j dgdg¡d | jjdƒ d S )Ng_ÎQÚk?r   r   r   )r   r
   Úel_testr   Ztest_llf_hypothr   r   r   r   Útest_hypothesis_beta1   s     ÿz TestOrigin.test_hypothesis_beta1c              	   C   sŒ   | j  d¡}|d }|d }t t | j j|gdgdd�d ¡¡}t t | j j|gdgdd�d ¡¡}t|| jjdƒ t|| jjdƒ d S )Nr   r   )Zreturn_weightsé   r   )	r
   Zconf_int_elÚnpÚsumÚlogr   r   r   Ztest_llf_conf)r   ÚciÚllZulZllf_lowZllf_highr   r   r   Útest_ci_beta#   s    ÿÿÿÿzTestOrigin.test_ci_betaN)r   r   r   r   r   r   r   r&   r   r   r   r   r      s
   r   )Únumpyr!   Znumpy.testingr   Zstatsmodels.datasetsr   Z!statsmodels.emplike.originregressr   Zresults.el_resultsr   r   r   r   r   r   r   Ú<module>   s   