U
    Kvf                     @   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exogZfitres1r   res2)clsdata r   I/tmp/pip-unpacked-wheel-2v6byqio/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_Qk?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   ciZll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   )Znumpyr    Znumpy.testingr   Zstatsmodels.datasetsr   Z!statsmodels.emplike.originregressr   Zresults.el_resultsr   r   r   r   r   r   r   <module>   s   