U
/e7 @ s d dl mZ d dlZd dlZd dlZd dlmZ d dl
m Z dZ
dd Zdd Zdd Zd
d Zd d
Zdd Zdd Zdd Zdd Zdd Zdd Zdd ZdS ) )dedentNz
c C sf t dttdi} t| d}||j|jfD ]0}t|j t
|ksJt t |j
t
|ks0t q0d S )Nxd )pd DataFramelistrangedd from_pandasindexr type__name__reprAssertionErrorstr npartitions)dfddfr r D/tmp/pip-unpacked-wheel-dbjnr7gq/dask/dataframe/tests/test_format.py test_repr s
r c C sf t tddddddgd} tj| dd }t|}t||ksDt|jd
|_t||ksbtd S ) N abcde)r r r category) r r r r
r r r r astype)r r s1r r r test_repr_meta_mutation# s r# c
C s t d tdddddddd gtd
ttd d } t| d}d
}t||ksZt t
||ksjt d}| |ks~t d}dj |d}|
|kst dj t|d}| |kst d S )Njinja2 r r r ABCDEFGHAAABBBCCABCaS Dask DataFrame Structure:
A B C
npartitions=3
0 int64 object category[known]
3 ... ... ...
6 ... ... ...
7 ... ... ...
Dask Name: from_pandas, 1 graph layera A B C
npartitions=3
0 int64 object category[known]
3 ... ... ...
6 ... ... ...
7 ... ... ...a
|
A |
B |
C |
| npartitions=3 |
|
|
|
| 0 |
int64 |
object |
category[known] |
| 3 |
... |
... |
... |
| 6 |
... |
... |
... |
| 7 |
... |
... |
... |
rDask DataFrame Structure:
{exp_table}
Dask Name: from_pandas, 1 graph layer
exp_tableDask DataFrame Structure:
{style}{exp_table}
Dask Name: from_pandas, 1 graph layer
styler2 pytest importorskipr r r Categoricalr
r r r r to_stringformatto_htmlr5 _repr_html_r r expr2 r r r test_dataframe_format- s2
+ r@ c C s t d tjdddddddd gtd
ttd d td
d
} t| d}d}t||ksbt t
||ksrt d}dj |d}| |kst dj t
|d}| |kst d S )Nr$ r% r r r& r r' r( r) r* r+ r, r aS Dask DataFrame Structure:
A B C
npartitions=3
A int64 object category[known]
D ... ... ...
G ... ... ...
H ... ... ...
Dask Name: from_pandas, 1 graph layera
|
A |
B |
C |
| npartitions=3 |
|
|
|
| A |
int64 |
object |
category[known] |
| D |
... |
... |
... |
| G |
... |
... |
... |
| H |
... |
... |
... |
r0 r1 r3 r4 )r7 r8 r r r r9 r
r r r r r; r< r5 r= r> r r r test_dataframe_format_with_index s.
+ rB c
C s t d tdddddddd gtd
ttd d } t| d}| }|j rXt
d
}t ||kslt
t ||ks|t
d}|
|kst
d}dj|d}| |kst
djt|d}| |kst
d S )Nr$ r% r r r& r r' r( r) r* r+ r, aS Dask DataFrame Structure:
A B C
npartitions=3
int64 object category[known]
... ... ...
... ... ...
... ... ...
Dask Name: from_pandas, 1 graph layera A B C
npartitions=3
int64 object category[known]
... ... ...
... ... ...
... ... ...a
|
A |
B |
C |
| npartitions=3 |
|
|
|
|
int64 |
object |
category[known] |
|
... |
... |
... |
|
... |
... |
... |
|
... |
... |
... |
r0 r1 r3 r4 )r7 r8 r r r r9 r
r Zclear_divisionsZknown_divisionsr r r r: r; r< r5 r= r> r r r 'test_dataframe_format_unknown_divisions s6
+ rC c
C s t d tdddddddd gd
td d
ttd d
d
} t| d
}d}t||ksft t
||ksvt d}| |kst d}dj |d}|
|kst dj t|d}| |kst d S )Nr$ r% r r r& r r' r( r)
r* r+ r, a Dask DataFrame Structure:
A B C
npartitions=10
0 int64 object category[known]
8 ... ... ...
... ... ... ...
72 ... ... ...
79 ... ... ...
Dask Name: from_pandas, 1 graph layeraH A B C
npartitions=10
0 int64 object category[known]
8 ... ... ...
... ... ... ...
72 ... ... ...
79 ... ... ...a
|
A |
B |
C |
| npartitions=10 |
|
|
|
| 0 |
int64 |
object |
category[known] |
| 8 |
... |
... |
... |
| ... |
... |
... |
... |
| 72 |
... |
... |
... |
| 79 |
... |
... |
... |
r0 r1 r3 r4 r6 r> r r r test_dataframe_format_longA s2
1 rE c C s t jddddddddgtd d
} t| d}d }t||ksBtt||ksRtd }| |ksftt jddddddddgtd d
d} t| d}d}t||kstt||kstd S )Nr% r r r& r r' r( r) r* rA zDask Series Structure:
npartitions=3
A int64
D ...
G ...
H ...
dtype: int64
Dask Name: from_pandas, 1 graph layerz9npartitions=3
A int64
D ...
G ...
H ...ZXXX)r namezDask Series Structure:
npartitions=3
A int64
D ...
G ...
H ...
Name: XXX, dtype: int64
Dask Name: from_pandas, 1 graph layer r Seriesr r
r r r r r: sZdsr? r r r test_series_format s " $ rK c C sv t jddddddddd d
g
d
td d
d } t| d
}d
}t||ksNtt||ks^td}| |ksrtd S )Nr% r r r& r r' r( r) rD Z
ABCDEFGHIJrA zDask Series Structure:
npartitions=10
A int64
B ...
...
J ...
J ...
dtype: int64
Dask Name: from_pandas, 1 graph layerzEnpartitions=10
A int64
B ...
...
J ...
J ...rG rI r r r test_series_format_long s . rM c C s t jddddddddgtd d
} t| d}d }t|j|ksDtt|j|ksVtt jddddddddgt j ddddddddgd d
d
} t| d}t
d}t|j|kstt|j|kstd S )Nr% r r r& r r' r( r) r* rA zDask Index Structure:
npartitions=3
A object
D ...
G ...
H ...
dtype: object
Dask Name: from_pandas-index, 2 graph layersZYYY)rF z Dask Index Structure:
npartitions=3
1 category[known]
4 ...
7 ...
8 ...
Name: YYY, dtype: category
Dask Name: from_pandas-index, 2 graph layers) r rH r r
r r r r r ZCategoricalIndexr rI r r r test_index_format s " rN c C sZ t dddgd} tj| dd}|j }d}t||ksBtd}t||ksVtd S ) Nr r r r r% r zDask Series Structure:
npartitions=1
0 category[known]
2 ...
dtype: category
Dask Name: from_pandas, 1 graph layerzDask Series Structure:
npartitions=1
0 category[unknown]
2 ...
dtype: category
Dask Name: from_pandas, 1 graph layer) r rH r! r
r catZ
as_unknownr r )rJ Zknownunknownr? r r r test_categorical_format s
rQ c C s: t jtddddd} tj| ddgd}t| d S )NrD r r )r r )chunksr )columns)daZ
from_arraynpZarangeZreshaper
Zfrom_dask_arrayr )Zarrframer r r test_duplicate_columns_repr s rW c C sZ t d t } tj| dd}d}t||ks4td}djt |d}|
|ksVtd S )Nr$ r% r zaEmpty Dask DataFrame Structure:
Columns: []
Divisions: [, ]
Dask Name: from_pandas, 1 graph layera r3 r4 ) r7 r8 r r r
r r r r; r5 r= r> r r r test_empty_repr s
rX )textwrapr ZnumpyrU Zpandasr r7 Z
dask.arrayarrayrT Zdask.dataframeZ dataframer
r5 r r# r@ rB rC rE rK rM rN rQ rW rX r r r r s$
^Uaf#"