U
    Kvf                     @   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rd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   H/tmp/pip-unpacked-wheel-2v6byqio/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 )Nd      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__Z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   

