U
    Ãmœd¶  ã                   @   s„  d Z ddl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 dd„ Zd	d
„ ZG dd„ dƒZedk�r€ddlmZ eƒ Ze ¡  ejZee ¡ ƒZeeƒ eZedƒ ee ¡ ƒZeeƒ edƒ ddlmZ eƒ Ze ¡  ejZee ¡ ƒZeeƒ edƒ eZddlm Z  e ƒ ZejZee ¡ ƒZeeƒ eZ!edƒ ddl"m#Z# e#ƒ ZejZee ¡ ƒZeeƒ e ¡ Z$ee$ %¡ ƒ dS )z8examples to check summary, not converted to tests yet


é    N)Úassert_equal)Ú	macrodata)Úadd_constant)ÚOLSc                  C   s^   t  ¡ j} | jddidd� tjd| d�}| ¡ }d| ¡  ¡ ksFt	‚d| ¡  
¡ ksZt	‚d S )NZcpiZCPI_T)ÚcolumnsZinplacezCPI_ ~ 1 + np.log(realgdp))ÚdatazCPI\_)r   Úloadr   Úrenamer   Zfrom_formulaÚfitÚsummaryZas_latexÚAssertionErrorZas_text)r   ÚmodÚres© r   ú]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/iolib/tests/test_summary.pyÚtest_escaped_variable_name   s    
r   c              	   C   sx   t j d¡}t j dd¡}t||ƒ ¡ }t t¡� |jdgd� W 5 Q R X t t¡� |jdddgd� W 5 Q R X d S )Néd   é   Úx1)ZxnameZx2Zx3)	ÚnpÚrandomZrandnr   r
   ÚpytestZraisesÚ
ValueErrorr   )Zreset_randomstateÚyÚxr   r   r   r   Útest_wrong_len_xname   s    r   c                   @   s   e Zd Zdd„ ZdS )ÚTestSummaryLatexc                 C   sš   d}ddddddddddddd	ddd	dd
d	dg}t |ƒ}ddd	ddd	dd
d	dddddddddddg}t||ƒ ¡ }| ¡ jd  ¡ }d| }t||ƒ d S )NaÛ  
\begin{center}
\begin{tabular}{lcccccc}
\toprule
               & \textbf{coef} & \textbf{std err} & \textbf{t} & \textbf{P$> |$t$|$} & \textbf{[0.025} & \textbf{0.975]}  \\
\midrule
\textbf{const} &       7.2248  &        0.866     &     8.346  &         0.000        &        5.406    &        9.044     \\
\textbf{x1}    &      -0.6609  &        0.177     &    -3.736  &         0.002        &       -1.033    &       -0.289     \\
\bottomrule
\end{tabular}
\end{center}
é   é   é   é   é   é   é   r   é	   z
%s
)r   r   r
   r   ZtablesZ_repr_latex_r   )ÚselfZdesiredr   r   ÚregÚactualr   r   r   Útest__repr_latex_%   s    ,,z"TestSummaryLatex.test__repr_latex_N)Ú__name__Ú
__module__Ú__qualname__r(   r   r   r   r   r   $   s   r   Ú__main__)ÚTestOLSz

)ÚTestProbitNewton)Ú
TestHampel)ÚTestGlmBinomial)&Ú__doc__Únumpyr   r   Znumpy.testingr   Zstatsmodels.datasetsr   Zstatsmodels.tools.toolsr   Z#statsmodels.regression.linear_modelr   r   r   r   r)   Z,statsmodels.regression.tests.test_regressionr-   ZaregressionZsetup_classZres1ÚresultsÚstrZsummary_oldZ	r_summaryÚprintZolsresr   Z(statsmodels.discrete.tests.test_discreter.   ZprobresZ!statsmodels.robust.tests.test_rlmr/   ZrlmresZ!statsmodels.genmod.tests.test_glmr0   ZsmryZas_csvr   r   r   r   Ú<module>   sV   

