U
    Kvf=  ã                   @   sn   d dl Z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G d
d„ deƒZdS )é    N)Úassert_almost_equal)Úheart)Ú
emplikeAFT)Úadd_constanté   )ÚAFTResc                   @   s   e Zd Zedd„ ƒZdS )ÚGenResc                 C   sb   t  ¡ }t |j¡|_t |j¡|_t |j¡}t|jƒ}t|||j	ƒ| _
| j
 ¡ | _tƒ | _d S )N)r   ÚloadÚnpZasarrayÚendogÚexogÚlog10r   r   ZcensorsZmod1ZfitÚres1r   Úres2)ÚclsÚdatar   r   © r   úF/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/emplike/tests/test_aft.pyÚsetup_class   s    
zGenRes.setup_classN)Ú__name__Ú
__module__Ú__qualname__Úclassmethodr   r   r   r   r   r      s   r   c                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zejj	d	d
„ ƒZ
dS )ÚTest_AFTModelc                 C   s   t | j ¡ | jjdd d S )Né   ©Údecimal)r   r   Úparamsr   Útest_params©Úselfr   r   r   r      s    ÿzTest_AFTModel.test_paramsc                 C   s$   t | j dgdg¡| jjdd d S )Nr   r   r   )r   r   Ú	test_betar   Ú
test_beta0r   r   r   r   r"      s     ÿzTest_AFTModel.test_beta0c                 C   s$   t | j dgdg¡| jjdd d S )Ng{®Gáz¤¿r   r   r   )r   r   r!   r   Ú
test_beta1r   r   r   r   r#   #   s     ÿzTest_AFTModel.test_beta1c                 C   s(   t | j ddgddg¡| jjdd d S )Ng      @gìQ¸…ë¡¿r   r   r   r   )r   r   r!   r   Z
test_jointr   r   r   r   Útest_beta_vect'   s     ÿzTest_AFTModel.test_beta_vectc                 C   sl   | j  ddd¡}|d }|d }| j  |gdg¡d }| j  |gdg¡d }t|ddd t|ddd d S )Nr   g¸…ëQ¸®¿r   gš™™™™™©?r   r   )r   Zci_betar!   r   )r    ÚciZllÚulZll_pvalZul_pvalr   r   r   Útest_betaci+   s    zTest_AFTModel.test_betaciN)r   r   r   r   r"   r#   r$   ÚpytestÚmarkZslowr'   r   r   r   r   r      s   r   )Znumpyr
   Znumpy.testingr   r(   Zstatsmodels.datasetsr   Zstatsmodels.emplike.aft_elr   Zstatsmodels.toolsr   Zresults.el_resultsr   r   r   r   r   r   r   Ú<module>   s   