
    M/Ph=                         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	          Z G d
 de          ZdS )    N)assert_almost_equal)heart)
emplikeAFT)add_constant   )AFTResc                   $    e Zd Zed             ZdS )GenResc                    t          j                    }t          j        |j                  |_        t          j        |j                  |_        t          j        |j                  }t          |j                  }t          |||j	                  | _
        | j
                                        | _        t                      | _        d S )N)r   loadnpasarrayendogexoglog10r   r   censorsmod1fitres1r   res2)clsdatar   r   s       b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/emplike/tests/test_aft.pysetup_classzGenRes.setup_class   s    z||Z
++
Jty))	$$DI&&eT4<888<<>>88    N)__name__
__module____qualname__classmethodr    r   r   r
   r
      s-          [  r   r
   c                   P    e Zd Zd Zd Zd Zd Zej        j	        d             Z
dS )Test_AFTModelc                 l    t          | j                                        | j        j        d           d S )N   decimal)r   r   paramsr   test_paramsselfs    r   r(   zTest_AFTModel.test_params   s=    DI,,..	0E$%	' 	' 	' 	' 	' 	'r   c                 t    t          | j                            dgdg          | j        j        d           d S )Nr$   r   r%   )r   r   	test_betar   
test_beta0r)   s    r   r-   zTest_AFTModel.test_beta0   sE    DI//aS99 I0!	= 	= 	= 	= 	= 	=r   c                 t    t          | j                            dgdg          | j        j        d           d S )Ng{Gzr   r$   r%   )r   r   r,   r   
test_beta1r)   s    r   r/   zTest_AFTModel.test_beta1#   sE    DI//<< I0!	= 	= 	= 	= 	= 	=r   c                 x    t          | j                            ddgddg          | j        j        d           d S )Ng      @gQ롿r   r   r$   r%   )r   r   r,   r   
test_jointr)   s    r   test_beta_vectzTest_AFTModel.test_beta_vect'   sI    DI//eq!fEE I0!	= 	= 	= 	= 	= 	=r   c                 2   | j                             ddd          }|d         }|d         }| j                             |gdg          d         }| j                             |gdg          d         }t          |dd           t          |dd           d S )Nr   gQr   g?r$   r%   )r   ci_betar,   r   )r*   cillulll_pvalul_pvals         r   test_betacizTest_AFTModel.test_betaci+   s    Yq$**UU)%%rdQC003)%%rdQC003GWa8888GVQ777777r   N)r   r   r   r(   r-   r/   r2   pytestmarkslowr:   r    r   r   r"   r"      sq        ' ' '= = == = == = = [8 8 8 8 8r   r"   )numpyr   numpy.testingr   r;   statsmodels.datasetsr   statsmodels.emplike.aft_elr   statsmodels.toolsr   results.el_resultsr   r
   r"   r    r   r   <module>rD      s        - - - - - -  & & & & & & 1 1 1 1 1 1 * * * * * * & & & & & &
 
 
 
 
 
 
 
8 8 8 8 8F 8 8 8 8 8r   