
    'YHh=                     z    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y)    N)assert_almost_equal)heart)
emplikeAFT)add_constant   )AFTResc                       e Zd Zed        Zy)GenResc                    t        j                         }t        j                  |j                        |_        t        j                  |j
                        |_        t        j                  |j                        }t        |j
                        }t        |||j                        | _
        | j                  j                         | _        t               | _        y )N)r   loadnpasarrayendogexoglog10r   r   censorsmod1fitres1r   res2)clsdatar   r   s       [/var/www/html/planif/env/lib/python3.12/site-packages/statsmodels/emplike/tests/test_aft.pysetup_classzGenRes.setup_class   s    zz|ZZ

+
JJtyy)	$DII&eT4<<888<<>8    N)__name__
__module____qualname__classmethodr    r   r   r
   r
      s     r   r
   c                   \    e Zd Zd Zd Zd Zd Zej                  j                  d        Z
y)Test_AFTModelc                 x    t        | j                  j                         | j                  j                  d       y )N   decimal)r   r   paramsr   test_paramsselfs    r   r(   zTest_AFTModel.test_params   s(    DII,,.		0E0E$%	'r   c                     t        | j                  j                  dgdg      | j                  j                  d       y )Nr$   r   r%   )r   r   	test_betar   
test_beta0r)   s    r   r-   zTest_AFTModel.test_beta0   s0    DII//aS9 II00!	=r   c                     t        | j                  j                  dgdg      | j                  j                  d       y )Ng{Gzr   r$   r%   )r   r   r,   r   
test_beta1r)   s    r   r/   zTest_AFTModel.test_beta1#   s0    DII//< II00!	=r   c                     t        | j                  j                  ddgddg      | j                  j                  d       y )Ng      @gQ롿r   r   r$   r%   )r   r   r,   r   
test_jointr)   s    r   test_beta_vectzTest_AFTModel.test_beta_vect'   s4    DII//eq!fE II00!	=r   c                    | j                   j                  ddd      }|d   }|d   }| j                   j                  |gdg      d   }| j                   j                  |gdg      d   }t        |dd       t        |dd       y )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    YYq$*UU))%%rdQC03))%%rdQC03GWa8GVQ7r   N)r   r   r   r(   r-   r/   r2   pytestmarkslowr:   r    r   r   r"   r"      s4    '=== [[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      s0     -  & 1 * &
 
8F 8r   