
p7]c           @   s   d  d l  Z d  d l m Z d  d l m Z m Z d  d l m Z d  d l m Z d  d l	 Z	 d  d l
 m Z e j   Z e j   Z d e f d     YZ d S(	   iN(   t   assert_equal(   t   SimpleTablet   default_txt_fmt(   t   default_latex_fmt(   t   default_html_fmt(   t   OLSt   TestSimpleTablec           B   sG   e  Z d    Z d   Z d   Z d   Z d   Z d   Z d   Z RS(   c         C   si   d } d d g d d g g } d } d } t  | | | d
 t } d | j   } t | t |   d  S(   Ns   
=====================
      header1 header2
---------------------
stub1 1.30312 2.73999
stub2 1.95038 2.65765
---------------------
ga7l[?g@g\4?g B@t   stub1t   stub2t   header1t   header2t   txt_fmts   
%s
(   R   R   (   R	   R
   (   R   R   t   as_textR    t   str(   t   selft   desiredt	   test1datat
   test1stubst   test1headert   actual(    (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_simple_table_1   s    		c         C   s   d } d d g d d g g } d d g d d	 g g } d
 d g } d d g } d d g } d d g } t  | | | d t } t  | | | d t }	 | j |	  d | j   }
 t | t |
   d  S(   Nsu  
=============================================================
           header s1 header d1            header s2 header d2
-------------------------------------------------------------
stub R1 C1  10.30312  10.73999 stub R1 C2  50.95038  50.65765
stub R2 C1  90.30312  90.73999 stub R2 C2  40.95038  40.65765
-------------------------------------------------------------
gm2$@g"z%@gްmQfV@g~W[V@gbEyI@gq-TI@gbEyD@gq-TD@s
   stub R1 C1s
   stub R2 C1s
   stub R1 C2s
   stub R2 C2s	   header s1s	   header d1s	   header s2s	   header d2R   s   
%s
(   R   R   t   extend_rightR   R    R   (   R   R   t   data1t   data2t   stubs1t   stubs2R	   R
   t   actual1t   actual2R   (    (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_simple_table_2    s    	c         C   s   d } d d g d d g g } d d g d d	 g g } d
 d g } d d g } d d g } d d g } t  | | | d t } t  | | | d t }	 | j |	  d | j   }
 t | t |
   d  S(   Ns7  
==============================
           header s1 header d1
------------------------------
stub R1 C1  10.30312  10.73999
stub R2 C1  90.30312  90.73999
           header s2 header d2
------------------------------
stub R1 C2  50.95038  50.65765
stub R2 C2  40.95038  40.65765
------------------------------
gm2$@g"z%@gްmQfV@g~W[V@gbEyI@gq-TI@gbEyD@gq-TD@s
   stub R1 C1s
   stub R2 C1s
   stub R1 C2s
   stub R2 C2s	   header s1s	   header d1s	   header s2s	   header d2R   s   
%s
(   R   R   t   extendR   R    R   (   R   R   R   R   R   R   R	   R
   R   R   R   (    (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_simple_table_36   s    c      !      s6  t  d d d g d d d d d d	 d
 d d d d d d d d d d d d d d d d d d d d d d d  } t j   } t j   } d } d } | | g } d d g } | | g } d* }	 d+ }
 t | |
 |	 d$ | d% | d& |     f d'   }   f d(   }   f d)   } | |   | |   | |   d  S(,   Nt	   data_fmtss   %3.2fs   %dt
   empty_cellt    t	   colwidthsi   t   colseps    * t   row_pres   * t   row_posts    *t   table_dec_abovet   *t   table_dec_belowt   header_dec_belowt
   header_fmts   %st   stub_fmtt   title_alignt   rt   header_alignt   data_alignst   stubs_alignt   lt   fmtt   txtg        i   gDl
@R   R   R	   R
   R   t   ltx_fmtt   html_fmtc            s'   d } d   j    } t | |  d  S(   Ns   
*****************************
*       * header1 * header2 *
*****************************
* stub1 *    0.00 *       1 *
* stub2 *    2.00 *       3 *
*****************************
s   
%s
(   R   R    (   R   R   R   (   t   tbl(    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_txt_fmt1o   s    	c            s[   d } d   j  d t  } t | |  d | d d !} d   j    } t | |  d  S(   Ns   
\begin{tabular}{lcc}
\toprule
               & \textbf{header1} & \textbf{header2}  \\
\midrule
\textbf{stub1} &       0.0        &        1          \\
\textbf{stub2} &        2         &      3.333        \\
\bottomrule
\end{tabular}
s   
%s
t   centers    
\begin{center}
%s
\end{center}
i   i(   t   as_latex_tabulart   FalseR    (   R   R   R   t   desired_centeredt   actual_centered(   R6   (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_ltx_fmt1~   s    c            s'   d } d   j    } t | |  d  S(   Ns   
<table class="simpletable">
<tr>
    <td></td>    <th>header1</th> <th>header2</th>
</tr>
<tr>
  <th>stub1</th>   <td>0.0</td>      <td>1</td>   
</tr>
<tr>
  <th>stub2</th>    <td>2</td>     <td>3.333</td> 
</tr>
</table>
s   
%s
(   t   as_htmlR    (   R   R   R   (   R6   (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_html_fmt1   s    (   R   R   (   R	   R
   (   t   dictR   t   copyR   R   (   R   t   txt_fmt1t   ltx_fmt1t	   html_fmt1t	   cell0datat	   cell1datat   row0datat   row1datat
   table1dataR   R   R7   R=   R?   (    (   R6   sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_simple_table_4P   sB    	

c   	         sq   d } d } | | g } d d g } | | g } d } d } t  | | | d	 t     f d
   } | |   d  S(   Ni   i  i   gfffffj@s   >stub1%t   stub_2s	   #header1$s   header&|R4   c            s-   d } d   j  d t  } t | |  d  S(   Ns  
\begin{tabular}{lcc}
\toprule
                    & \textbf{\#header1\$} & \textbf{header\&$|$}  \\
\midrule
\textbf{$>$stub1\%} &          22          &         1053          \\
\textbf{stub\_2}    &          23          &        6250.4         \\
\bottomrule
\end{tabular}
s   
%s
R8   (   R9   R:   R    (   R   R   R   (   t   tbl_c(    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_ltx_special_chars   s    (   s   >stub1%RK   (   s	   #header1$s   header&|(   R   RC   (	   R   t   cell0c_datat   cell1c_datat
   row0c_datat
   row1c_datat   table1c_datat   test1c_stubst   test1c_headerRM   (    (   RL   sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_simple_table_special_chars   s    c      	   C   s  t  j d d d d g d d d } t  j d d d d d g d	 d d
 } t  j d d d d d d d d	 g d d d } t  j d | j  } | j |  } | j |  } | j |  } | j d  } | j d  } | j t j t j	 g  }	 | j t j t j	 g  }
 t
 |	 d g |
  j   } g  } d d  l } | j   U | j d  x@ t | j    j   D]& } d | k ru| j | d   ququWWd  QXd d d g } t t |  t |   d  S(   Ni   i   i   i   i
   t   namet   ii   i   t   yi   i   t   xt   indext   sumit   ignoret   _i&   s&   Dep. Variable:                  x_sum s&   y_sum          1.4595      0.209      s&   y_amax         0.2432      0.035      (   RY   R[   (   t   pandast   Seriest	   DataFrameRZ   t   joint   groupbyt   aggt   npR[   t   maxR   t   fitt   warningst   catch_warningst   simplefilterR   t   summaryt
   splitlinest   appendR    t   sorted(   R   RW   RX   RY   t   dft   endot   exot   endo_groupst
   exo_groupst   exo_dft   endo_dft   regt   interesting_linesRg   t   lineR   (    (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_regression_with_tuples   s.    %(1	c         C   si   d } d d g d d g g } d } d } t  | | | d
 t } d | j   } t | t |   d  S(   Ns   
=====================
      header1 header2
---------------------
stub1 1.30312    2.73
stub2 1.95038     2.6
---------------------
ga7l[?gףp=
@g\4?g@R   R   R	   R
   R   s   
%s
(   R   R   (   R	   R
   (   R   R   R   R    R   (   R   R   R   R   R   R   (    (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   test_default_alignment   s    	(	   t   __name__t
   __module__R   R   R   RJ   RU   Rx   Ry   (    (    (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyR      s   				\		(   t   numpyRd   t   numpy.testingR    t   statsmodels.iolib.tableR   R   R   R   R^   t#   statsmodels.regression.linear_modelR   RA   RC   RD   t   objectR   (    (    (    sA   lib/python2.7/site-packages/statsmodels/iolib/tests/test_table.pyt   <module>   s   