ó
¦–Õ\c           @   s³   d  d l  Z d  d l m Z d  d l j Z d  d l j Z	 d  d l
 Z d Z d „  Z d „  Z d „  Z d „  Z d „  Z d	 „  Z d
 „  Z d „  Z d „  Z d „  Z d „  Z d S(   iÿÿÿÿN(   t   dedentså   <style scoped>
    .dataframe tbody tr th:only-of-type {
        vertical-align: middle;
    }

    .dataframe tbody tr th {
        vertical-align: top;
    }

    .dataframe thead th {
        text-align: right;
    }
</style>
c          C   sš   t  j i t t d ƒ ƒ d 6ƒ }  t j |  d ƒ } x_ | | j | j g D]H } t | ƒ j	 t
 | ƒ k sq t ‚ t | j ƒ t
 | ƒ k sJ t ‚ qJ Wd  S(   Nid   t   xi   (   t   pdt	   DataFramet   listt   ranget   ddt   from_pandast   indexR   t   typet   __name__t   reprt   AssertionErrort   strt   npartitions(   t   dft   ddfR   (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt	   test_repr   s
    "!c          C   sœ   t  j i t d ƒ d 6d d d d d g d 6ƒ }  t j |  d d ƒ} t | ƒ } t | ƒ | k sk t ‚ | j j d	 ƒ | _ t | ƒ | k s˜ t ‚ d  S(
   Ni   t   at   bt   ct   dt   eR   i   t   category(	   R   R   R   R   R   R   R   R   t   astype(   R   R   t   s1(    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_repr_meta_mutation#   s    c       
   C   s  t  j i d d d d d d d d g d	 6t d
 ƒ d 6t  j t d ƒ ƒ d 6ƒ }  t j |  d ƒ } d } t | ƒ | k s t ‚ t | ƒ | k s™ t ‚ d } | j	 ƒ  | k s· t ‚ d } d j
 d | ƒ } | j ƒ  | k sç t ‚ d j
 d t d | ƒ } | j ƒ  | k st ‚ d  S(   Ni   i   i   i   i   i   i   i   t   At   ABCDEFGHt   Bt   AAABBBCCt   CsM  Dask DataFrame Structure:
                   A       B                C
npartitions=3                                
0              int64  object  category[known]
3                ...     ...              ...
6                ...     ...              ...
7                ...     ...              ...
Dask Name: from_pandas, 3 taskss                     A       B                C
npartitions=3                                
0              int64  object  category[known]
3                ...     ...              ...
6                ...     ...              ...
7                ...     ...              ...s­  <table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th></th>
      <th>A</th>
      <th>B</th>
      <th>C</th>
    </tr>
    <tr>
      <th>npartitions=3</th>
      <th></th>
      <th></th>
      <th></th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <th>0</th>
      <td>int64</td>
      <td>object</td>
      <td>category[known]</td>
    </tr>
    <tr>
      <th>3</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th>6</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th>7</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
  </tbody>
</table>sl   <div><strong>Dask DataFrame Structure:</strong></div>
{exp_table}
<div>Dask Name: from_pandas, 3 tasks</div>t	   exp_tables€   <div><strong>Dask DataFrame Structure:</strong></div>
<div>
{style}{exp_table}
</div>
<div>Dask Name: from_pandas, 3 tasks</div>t   style(   R   R   R   t   CategoricalR   R   R   R   R   t	   to_stringt   formatt   to_htmlR!   t   _repr_html_(   R   R   t   expR    (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_dataframe_format.   s    (+c       
   C   s	  t  j i d d d d d d d d g d	 6t d
 ƒ d 6t  j t d ƒ ƒ d 6d t d
 ƒ ƒ}  t j |  d ƒ } d } t | ƒ | k s t ‚ t | ƒ | k s¥ t ‚ d } d j	 d | ƒ } | j
 ƒ  | k sÕ t ‚ d j	 d t d | ƒ } | j ƒ  | k st ‚ d  S(   Ni   i   i   i   i   i   i   i   R   R   R   R   R   R   sM  Dask DataFrame Structure:
                   A       B                C
npartitions=3                                
A              int64  object  category[known]
D                ...     ...              ...
G                ...     ...              ...
H                ...     ...              ...
Dask Name: from_pandas, 3 taskss­  <table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th></th>
      <th>A</th>
      <th>B</th>
      <th>C</th>
    </tr>
    <tr>
      <th>npartitions=3</th>
      <th></th>
      <th></th>
      <th></th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <th>A</th>
      <td>int64</td>
      <td>object</td>
      <td>category[known]</td>
    </tr>
    <tr>
      <th>D</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th>G</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th>H</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
  </tbody>
</table>sl   <div><strong>Dask DataFrame Structure:</strong></div>
{exp_table}
<div>Dask Name: from_pandas, 3 tasks</div>R    s€   <div><strong>Dask DataFrame Structure:</strong></div>
<div>
{style}{exp_table}
</div>
<div>Dask Name: from_pandas, 3 tasks</div>R!   (   R   R   R   R"   R   R   R   R   R   R$   R%   R!   R&   (   R   R   R'   R    (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt    test_dataframe_format_with_index   s    (+c       
   C   s7  t  j i d d d d d d d d g d	 6t d
 ƒ d 6t  j t d ƒ ƒ d 6ƒ }  t j |  d ƒ } | j ƒ  } | j s t ‚ d } t	 | ƒ | k s t ‚ t
 | ƒ | k sµ t ‚ d } | j ƒ  | k sÓ t ‚ d } d j d | ƒ } | j ƒ  | k st ‚ d j d t d | ƒ } | j ƒ  | k s3t ‚ d  S(   Ni   i   i   i   i   i   i   i   R   R   R   R   R   sM  Dask DataFrame Structure:
                   A       B                C
npartitions=3                                
               int64  object  category[known]
                 ...     ...              ...
                 ...     ...              ...
                 ...     ...              ...
Dask Name: from_pandas, 3 taskss                     A       B                C
npartitions=3                                
               int64  object  category[known]
                 ...     ...              ...
                 ...     ...              ...
                 ...     ...              ...s©  <table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th></th>
      <th>A</th>
      <th>B</th>
      <th>C</th>
    </tr>
    <tr>
      <th>npartitions=3</th>
      <th></th>
      <th></th>
      <th></th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <th></th>
      <td>int64</td>
      <td>object</td>
      <td>category[known]</td>
    </tr>
    <tr>
      <th></th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th></th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th></th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
  </tbody>
</table>sl   <div><strong>Dask DataFrame Structure:</strong></div>
{exp_table}
<div>Dask Name: from_pandas, 3 tasks</div>R    s€   <div><strong>Dask DataFrame Structure:</strong></div>
<div>
{style}{exp_table}
</div>
<div>Dask Name: from_pandas, 3 tasks</div>R!   (   R   R   R   R"   R   R   t   clear_divisionst   known_divisionsR   R   R   R#   R$   R%   R!   R&   (   R   R   R'   R    (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt'   test_dataframe_format_unknown_divisionsÉ   s     (+c       
   C   s'  t  j i d d d d d d d d g d	 d
 6t d ƒ d	 d 6t  j t d ƒ d	 ƒ d 6ƒ }  t j |  d	 ƒ } d } t | ƒ | k s t ‚ t | ƒ | k s¥ t ‚ d } | j	 ƒ  | k sÃ t ‚ d } d j
 d | ƒ } | j ƒ  | k só t ‚ d j
 d t d | ƒ } | j ƒ  | k s#t ‚ d  S(   Ni   i   i   i   i   i   i   i   i
   R   R   R   R   R   sƒ  Dask DataFrame Structure:
                    A       B                C
npartitions=10                                
0               int64  object  category[known]
8                 ...     ...              ...
...               ...     ...              ...
72                ...     ...              ...
79                ...     ...              ...
Dask Name: from_pandas, 10 taskssH                      A       B                C
npartitions=10                                
0               int64  object  category[known]
8                 ...     ...              ...
...               ...     ...              ...
72                ...     ...              ...
79                ...     ...              ...s  <table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th></th>
      <th>A</th>
      <th>B</th>
      <th>C</th>
    </tr>
    <tr>
      <th>npartitions=10</th>
      <th></th>
      <th></th>
      <th></th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <th>0</th>
      <td>int64</td>
      <td>object</td>
      <td>category[known]</td>
    </tr>
    <tr>
      <th>8</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th>...</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th>72</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
    <tr>
      <th>79</th>
      <td>...</td>
      <td>...</td>
      <td>...</td>
    </tr>
  </tbody>
</table>sm   <div><strong>Dask DataFrame Structure:</strong></div>
{exp_table}
<div>Dask Name: from_pandas, 10 tasks</div>R    u   <div><strong>Dask DataFrame Structure:</strong></div>
<div>
{style}{exp_table}
</div>
<div>Dask Name: from_pandas, 10 tasks</div>R!   (   R   R   R   R"   R   R   R   R   R   R#   R$   R%   R!   R&   (   R   R   R'   R    (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_dataframe_format_long  s    , 	1c       	   C   s  t  j d d d d d d d d g d	 t d
 ƒ ƒ}  t j |  d ƒ } d } t | ƒ | k sc t ‚ t | ƒ | k s{ t ‚ d } | j ƒ  | k s™ t ‚ t  j d d d d d d d d g d	 t d
 ƒ d d ƒ}  t j |  d ƒ } d } t | ƒ | k st ‚ t | ƒ | k st ‚ d  S(   Ni   i   i   i   i   i   i   i   R   R   s}   Dask Series Structure:
npartitions=3
A    int64
D      ...
G      ...
H      ...
dtype: int64
Dask Name: from_pandas, 3 taskss9   npartitions=3
A    int64
D      ...
G      ...
H      ...t   namet   XXXsˆ   Dask Series Structure:
npartitions=3
A    int64
D      ...
G      ...
H      ...
Name: XXX, dtype: int64
Dask Name: from_pandas, 3 tasks(	   R   t   SeriesR   R   R   R   R   R   R#   (   t   st   dsR'   (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_series_formatv  s    $$c          C   s«   t  j d d d d d d d d d	 d
 g
 d
 d t d ƒ d
 ƒ}  t j |  d
 ƒ } d } t | ƒ | k sq t ‚ t | ƒ | k s‰ t ‚ d } | j ƒ  | k s§ t ‚ d  S(   Ni   i   i   i   i   i   i   i   i	   i
   R   t
   ABCDEFGHIJsŠ   Dask Series Structure:
npartitions=10
A    int64
B      ...
     ...  
J      ...
J      ...
dtype: int64
Dask Name: from_pandas, 10 taskssE   npartitions=10
A    int64
B      ...
     ...  
J      ...
J      ...(	   R   R0   R   R   R   R   R   R   R#   (   R1   R2   R'   (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_series_format_long›  s    .c          C   s-  t  j d d d d d d d d g d	 t d
 ƒ ƒ}  t j |  d ƒ } d } t | j ƒ | k sf t ‚ t | j ƒ | k s t ‚ t  j d d d d d d d d g d	 t  j	 d d d d d d d d g d d ƒƒ}  t j |  d ƒ } t
 d ƒ } t | j ƒ | k st ‚ t | j ƒ | k s)t ‚ d  S(   Ni   i   i   i   i   i   i   i   R   R   s   Dask Index Structure:
npartitions=3
A    object
D       ...
G       ...
H       ...
dtype: object
Dask Name: from_pandas, 6 tasksR.   t   YYYsÒ       Dask Index Structure:
    npartitions=3
    1    category[known]
    4                ...
    7                ...
    8                ...
    Name: YYY, dtype: category
    Dask Name: from_pandas, 6 tasks(   R   R0   R   R   R   R   R   R   R   t   CategoricalIndexR    (   R1   R2   R'   (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_index_format©  s    $$0	c          C   s…   t  j d d d g ƒ j d ƒ }  t j |  d d ƒ} | j j ƒ  } d } t | ƒ | k sc t ‚ d } t | ƒ | k s t ‚ d  S(	   NR   R   R   R   R   i   s~   Dask Series Structure:
npartitions=1
0    category[known]
2                ...
dtype: category
Dask Name: from_pandas, 1 taskss‚   Dask Series Structure:
npartitions=1
0    category[unknown]
2                  ...
dtype: category
Dask Name: from_pandas, 1 tasks(	   R   R0   R   R   R   t   catt
   as_unknownR   R   (   R1   t   knownt   unknownR'   (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_categorical_formatÈ  s    !c          C   sS   t  j t j d ƒ j d d ƒ d d ƒ}  t j |  d d d g ƒ} t | ƒ d  S(   Ni
   i   i   t   chunkst   columnsR   (   i   i   (   t   dat
   from_arrayt   npt   aranget   reshapeR   t   from_dask_arrayR   (   t   arrt   frame(    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   test_duplicate_columns_reprÜ  s    *(   t   pandasR   t   textwrapR    t   dask.dataframet	   dataframeR   t
   dask.arrayt   arrayR@   t   numpyRB   R!   R   R   R(   R)   R,   R-   R3   R5   R8   R=   RH   (    (    (    s?   lib/python2.7/site-packages/dask/dataframe/tests/test_format.pyt   <module>   s    				Q	J	T	Y	%			