U
|
®ý°d#@ ã @ s| d dl m Z mZ d dlZd dlZd dlZd dlmZmZm Z m
|
Z
|
m Z m Z m Z mZ d dlmZ G dd dZG dd dZdS )é ©ÚdatetimeÚ timedeltaN)Ú CategoricalÚ DataFrameÚIndexÚSeriesÚ
|
date_rangeÚoption_contextÚ period_rangeÚtimedelta_rangec @ sN e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
dd Z dd Z dd Z dd Zej ddddddd d d d d d!g ¡d"d# Zd$d% Zej d&d'd(g¡d)d* Zd+d, Zej d-dd.gd/d0gg¡d1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Zd?d@ ZdAdB ZdCdD Z dEdF Z dGdH Z dIdJ Z dKdL Z dMdN Z!d/S )OÚTestSeriesReprc C sT |}t tt||dd}ddddddd d
|
d d d dg }d |¡}t||ksPtd S )NZsth©ÚindexÚnamez first secondzfoo one 0z two 1z three 2zbar one 3z two 4zbaz two 5z three 6zqux one 7z two 8z three 9zName: sth, dtype: int64Ú
|
)r ÚrangeÚlenÚjoinÚreprÚAssertionError)ÚselfZ%lexsorted_two_level_string_multiindexr ÚserÚexpected© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/test_repr.pyÚtest_multilevel_name_print s" ô
|
z)TestSeriesRepr.test_multilevel_name_printc C s> t dddg}d|_dt|ks$td |_dt|ks:td S )Nr é é Útestú
|
Name: testúName:)r r r r ©r Úsr r r Útest_small_name_printing, s
|
z'TestSeriesRepr.test_small_name_printingc C s< t td}d|_dt|ks"td |_dt|ks8td S )Néè r r r! )r r r r r r" r r r Útest_big_name_printing6 s
|
z%TestSeriesRepr.test_big_name_printingc C s( t tdddtd}dt|ks$td S )NZ20010101Z20020101r )r r Údtyper )r r Úobjectr r r" r r r Útest_empty_name_printing@ s z'TestSeriesRepr.test_empty_name_printingÚargsr )r éÿÿÿÿc C s& t ttj d¡tjd| d d S )Nr% ©r )r% )Ústrr ÚnpÚrandomÚrandnÚarange)r r* r r r Útest_float_rangeD s z TestSeriesRepr.test_float_rangec C s t ttd d S )N©r' )r- r r( ©r r r r Útest_empty_objectH s z TestSeriesRepr.test_empty_objectc C s0 t | t | t¡ tj|dd
< t | d S )Né é )r- ÚastypeÚintr. ÚNaN)r Ú string_seriesr r r Ú test_stringL s zTestSeriesRepr.test_stringc C s t | d S ©N)r- )r Z object_seriesr r r Ú test_objectT s zTestSeriesRepr.test_objectc C s, t | | d¡}d |d d d
< t| d S )NÚOr )r- r8 r )r Zdatetime_seriesZotsr r r Ú test_datetimeW s
|
z TestSeriesRepr.test_datetimer Ú r g333333ó?Úfoou αβγZ7loooooooooooooooooooooooooooooooooooooooooooooooooooong©rB ÚbarZbaz)r r )rB r gffffff@)õ αu βu γ)rE rD c C s ||_ t| d S r= )r r )r r r; r r r Útest_various_names^ s z!TestSeriesRepr.test_various_namesc C s( t tj d¡t d¡dd}t| d S )Nr% rC r )r r. r/ r0 r1 r )r Zbiggier r r Útest_tuple_names s
|
ÿz TestSeriesRepr.test_tuple_nameÚargéd éé c C s, t tj |¡dd}t|}d|ks(td S )Nr ©r zName: 0)r r. r/ r0 r r )r rH r Zrep_strr r r Útest_tidy_repr_name_0y s z$TestSeriesRepr.test_tidy_repr_name_0c C sF t dgddgd}dt|ks"tdt|ks2tdt|ksBtd S )Nza
|
bza
|
dza
|
f)r r ú ú za
|
)r r r )r r r r r Ú test_newline s zTestSeriesRepr.test_newlinezname, expectedz#Series([], Name: foo, dtype: int64)NzSeries([], dtype: int64)c C s$ t g tj|d}t||ks td S )N)r' r )r r. Úint64r r )r r r r# r r r Útest_empty_int64 s z TestSeriesRepr.test_empty_int64c C s t dgd }d|_t| d S )Nõ ×r% Ztitle1)r r r )r Úar r r Útest_tidy_repr s z TestSeriesRepr.test_tidy_reprc C s8 t dd tdD }t| | ¡ }|jdks4td S )Nc S s g | ]}t tj d d ¡qS )r )r r. r/ r0 ©Ú.0Úir r r Ú
|
<listcomp> s z7TestSeriesRepr.test_repr_bool_fails.<locals>.<listcomp>r6 rA )r r r Z
|
readouterrÚerrr )r Zcapsysr# Zcapturedr r r Útest_repr_bool_fails s z#TestSeriesRepr.test_repr_bool_failsc C s2 t dddgt d¡d}t| d|_t| d S )Nr r é rK )rR rR )r r. rP r r r" r r r Ú!test_repr_name_iterable_indexable s z0TestSeriesRepr.test_repr_name_iterable_indexablec C s< ddddg}ddddg}t ||d}t| ¡ tks8td S ) Né r6 r[ u Ïu Ïu Ï
u Ïr, )r ÚtypeÚ__repr__r- r )r ÚdataZindex1Údfr r r Útest_repr_should_return_str© s z*TestSeriesRepr.test_repr_should_return_strc C s* t dd tttd W 5 Q R X d S )Núdisplay.max_rowsrJ )r
|
r- r r r4 r r r Útest_repr_max_rows´ s z!TestSeriesRepr.test_repr_max_rowsc C s t dgdd}t| d S )NrR u ×rK )r r- )r ra r r r Ú test_unicode_string_with_unicode¹ s z/TestSeriesRepr.test_unicode_string_with_unicodec C s8 t dgdd}d}tjt|d t| W 5 Q R X d S )NÚabcrK z2^'str' object cannot be interpreted as an integer$)Úmatch)r ÚpytestZraisesÚ TypeErrorÚbytes)r ra Úmsgr r r Útest_str_to_bytes_raises½ s z'TestSeriesRepr.test_str_to_bytes_raisesc C s t dd tdD td}ttj t|¡|}t| t
|
d¡}t| ¡ d d¡s\t |jtj dt|d d ¡ }t| ¡ d d S )
|
Nc S s g | ]}t d ddt| qS )éÐ r r rU r r r rX Æ s zDTestSeriesRepr.test_timeseries_repr_object_dtype.<locals>.<listcomp>r% r3 r+ zFreq:r r i )r r r( r r. r/ r0 r r ÚtmZmakeTimeSeriesÚ
|
splitlinesÚ
|
startswithr ZilocÚrandint)r r ÚtsZts2r r r Ú!test_timeseries_repr_object_dtypeÄ s ÿ
|
z0TestSeriesRepr.test_timeseries_repr_object_dtypec C sZ t d¡ d}tdd dd$ tdddg}|| ¡ ks<tW 5 Q R X | ¡ d ksVtd S ) NZjinja2zj\begin{tabular}{ll}
|
\toprule
|
& 0 \\
|
\midrule
|
0 & $\alpha$ \\
|
1 & b \\
|
2 & c \\
|
\bottomrule
|
\end{tabular}
|
zstyler.format.escapezstyler.render.reprZlatexz$\alpha$ÚbÚc)rh Z importorskipr
|
r Z _repr_latex_r )r Úresultr# r r r Útest_latex_reprÑ s
|
|
ÿz TestSeriesRepr.test_latex_reprc C s6 t dtjg}tddg|d}d}t||ks2td S )Nr r r, z 1.0 1
|
NaN 2
|
dtype: int64)r r. Únanr r r )r rW r# Úexpr r r Ú!test_index_repr_in_frame_with_nanå s z0TestSeriesRepr.test_index_repr_in_frame_with_nanc C s. t dddd}| ¡ td|d}t| d S )Nz1/1/1850z1/1/1950zA-DEC)Úfreqr r, )r Úformatr r )r Úrngrr r r r Útest_format_pre_1900_datesí s z)TestSeriesRepr.test_format_pre_1900_datesc C s4 t dddtjjgdd}t|}d}||ks0td S )Nr r% rm zM8[ns]r3 z0 1970-01-01 00:00:00.000000
|
1 1970-01-01 00:00:00.000001
|
2 1970-01-01 00:00:00.000002
|
3 NaT
|
dtype: datetime64[ns])r ÚpdÚNaTÚ_valuer r )r Zseriesrv r r r r Útest_series_repr_nató s
|
ÿz#TestSeriesRepr.test_series_repr_natc C s( t dg t¡}d}t||ks$td S )Ng ð?z0 1.0
|
dtype: object)r r8 r( r r )r r r r r r Útest_float_repr s z TestSeriesRepr.test_float_reprc C s: t ddddgdd tjtjg}t|}d}||ks6td S )Nr r r[ é Tz4True 1
|
None 2
|
NaN 3
|
NaT 4
|
dtype: int64)r r. rx r r r r )r r rv r r r r Útest_different_null_objects s z*TestSeriesRepr.test_different_null_objects)"Ú__name__Ú
|
__module__Ú __qualname__r r$ r& r) rh ÚmarkZ parametrizer2 r5 r< r> r@ rF rG rL rO rQ rT rZ r\ rb rd re rl rs rw rz r~ r r r
r r r r r sf
|
|
|
õþ
|
|
þþ
|
r c @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dS )ÚTestCategoricalReprc sL G dd d t fddtdD }t|}| ¡ }t| t| d S )Nc @ s" e Zd ZdZdZedddZdS )zATestCategoricalRepr.test_categorical_repr_unicode.<locals>.Countyu San SebastiánZPR)Úreturnc S s | j d | j S )Nz, )r Ústater4 r r r r_ s zJTestCategoricalRepr.test_categorical_repr_unicode.<locals>.County.__repr__N)r r r r r r- r_ r r r r ÚCounty s r c s g | ]
|
} qS r r )rV Ú_©r r r rX s zETestCategoricalRepr.test_categorical_repr_unicode.<locals>.<listcomp>é= )r r r Z to_seriesr r- )r ÚcatÚidxr r r r Ú test_categorical_repr_unicode s z1TestCategoricalRepr.test_categorical_repr_unicodec C s t tddddg}d}|| ¡ ks(tt tddgd }d }td
|
d |t|ks\tW 5 Q R X td }t tddg|d d}d}|| ¡ kstd S )Nr r r[ r zO0 1
|
1 2
|
2 3
|
3 4
|
dtype: category
|
Categories (4, int64): [1, 2, 3, 4]rS rt é zf0 a
|
1 b
|
..
|
48 a
|
49 b
|
Length: 50, dtype: category
|
Categories (2, object): ['a', 'b']rc r6 ÚabcdefghijklmnopqrstuvwxyzT)Ú
|
categoriesÚorderedzh0 a
|
1 b
|
dtype: category
|
Categories (26, object): ['a' < 'b' < 'c' < 'd' ... 'w' < 'x' < 'y' < 'z'])r r Ú__str__r r
|
r Úlist)r rS ry Zlevsr r r Útest_categorical_repr! s ÿÿ ÿz)TestCategoricalRepr.test_categorical_reprc C s^ t tdddg}d}t||ks&tt tt d¡}dt ¡ j d}t||ksZtd S )Nr r r[ zE0 1
|
1 2
|
2 3
|
dtype: category
|
Categories (3, int64): [1, 2, 3]é
|
úf0 0
|
1 1
|
2 2
|
3 3
|
4 4
|
5 5
|
6 6
|
7 7
|
8 8
|
9 9
|
dtype: category
|
Categories (10, z ): [0, 1, 2, 3, ..., 6, 7, 8, 9]©r r r r r. r1 Úint_r' ©r r# ry r r r Ú test_categorical_series_repr< s õ z0TestCategoricalRepr.test_categorical_series_reprc C sf t tdddgdd}d}t||ks*tt tt d¡dd}dt ¡ j d }t||ksbtd S )
|
Nr r r[ T©r zG0 1
|
1 2
|
2 3
|
dtype: category
|
Categories (3, int64): [1 < 2 < 3]r r z$): [0 < 1 < 2 < 3 ... 6 < 7 < 8 < 9]r r r r r Ú$test_categorical_series_repr_orderedV s õ z8TestCategoricalRepr.test_categorical_series_repr_orderedc C sb t dddd}tt|}d}t||ks.tt ddddd}tt|}d}t||ks^td S ) Nú2011-01-01 09:00ÚHr6 ©r{ Úperiodsa2 0 2011-01-01 09:00:00
|
1 2011-01-01 10:00:00
|
2 2011-01-01 11:00:00
|
3 2011-01-01 12:00:00
|
4 2011-01-01 13:00:00
|
dtype: category
|
Categories (5, datetime64[ns]): [2011-01-01 09:00:00, 2011-01-01 10:00:00, 2011-01-01 11:00:00,
|
2011-01-01 12:00:00, 2011-01-01 13:00:00]ú
|
US/Eastern©r{ r¦ Útza³ 0 2011-01-01 09:00:00-05:00
|
1 2011-01-01 10:00:00-05:00
|
2 2011-01-01 11:00:00-05:00
|
3 2011-01-01 12:00:00-05:00
|
4 2011-01-01 13:00:00-05:00
|
dtype: category
|
Categories (5, datetime64[ns, US/Eastern]): [2011-01-01 09:00:00-05:00, 2011-01-01 10:00:00-05:00,
|
2011-01-01 11:00:00-05:00, 2011-01-01 12:00:00-05:00,
|
2011-01-01 13:00:00-05:00]©r r r r r ©r r r# ry r r r Ú%test_categorical_series_repr_datetimep s
|
z9TestCategoricalRepr.test_categorical_series_repr_datetimec C sj t dddd}tt|dd}d}t||ks2tt ddddd }tt|dd}d
|
}t||ksftd S ) Nr£ r¤ r6 r¥ Tr¡ a6 0 2011-01-01 09:00:00
|
1 2011-01-01 10:00:00
|
2 2011-01-01 11:00:00
|
3 2011-01-01 12:00:00
|
4 2011-01-01 13:00:00
|
dtype: category
|
Categories (5, datetime64[ns]): [2011-01-01 09:00:00 < 2011-01-01 10:00:00 < 2011-01-01 11:00:00 <
|
2011-01-01 12:00:00 < 2011-01-01 13:00:00]r§ r¨ a· 0 2011-01-01 09:00:00-05:00
|
1 2011-01-01 10:00:00-05:00
|
2 2011-01-01 11:00:00-05:00
|
3 2011-01-01 12:00:00-05:00
|
4 2011-01-01 13:00:00-05:00
|
dtype: category
|
Categories (5, datetime64[ns, US/Eastern]): [2011-01-01 09:00:00-05:00 < 2011-01-01 10:00:00-05:00 <
|
2011-01-01 11:00:00-05:00 < 2011-01-01 12:00:00-05:00 <
|
2011-01-01 13:00:00-05:00]rª r« r r r Ú-test_categorical_series_repr_datetime_ordered s
|
zATestCategoricalRepr.test_categorical_series_repr_datetime_orderedc C s` t dddd}tt|}d}t||ks.tt dddd}tt|}d}t||ks\td S ) Nr£ r¤ r6 r¥ a 0 2011-01-01 09:00
|
1 2011-01-01 10:00
|
2 2011-01-01 11:00
|
3 2011-01-01 12:00
|
4 2011-01-01 13:00
|
dtype: category
|
Categories (5, period[H]): [2011-01-01 09:00, 2011-01-01 10:00, 2011-01-01 11:00, 2011-01-01 12:00,
|
2011-01-01 13:00]ú2011-01ÚMz0 2011-01
|
1 2011-02
|
2 2011-03
|
3 2011-04
|
4 2011-05
|
dtype: category
|
Categories (5, period[M]): [2011-01, 2011-02, 2011-03, 2011-04, 2011-05]©r r r r r r« r r r Ú#test_categorical_series_repr_period¨ s z7TestCategoricalRepr.test_categorical_series_repr_periodc C sh t dddd}tt|dd}d}t||ks2tt dd dd}tt|dd}d
|
}t||ksdtd S ) Nr£ r¤ r6 r¥ Tr¡ a 0 2011-01-01 09:00
|
1 2011-01-01 10:00
|
2 2011-01-01 11:00
|
3 2011-01-01 12:00
|
4 2011-01-01 13:00
|
dtype: category
|
Categories (5, period[H]): [2011-01-01 09:00 < 2011-01-01 10:00 < 2011-01-01 11:00 < 2011-01-01 12:00 <
|
2011-01-01 13:00]r® r¯ z0 2011-01
|
1 2011-02
|
2 2011-03
|
3 2011-04
|
4 2011-05
|
dtype: category
|
Categories (5, period[M]): [2011-01 < 2011-02 < 2011-03 < 2011-04 < 2011-05]r° r« r r r Ú+test_categorical_series_repr_period_ordered s z?TestCategoricalRepr.test_categorical_series_repr_period_orderedc C s\ t ddd}tt|}d}t||ks,tt ddd}tt|}d}t||ksXtd S )Nú1 daysr6 ©r¦ z0 1 days
|
1 2 days
|
2 3 days
|
3 4 days
|
4 5 days
|
dtype: category
|
Categories (5, timedelta64[ns]): [1 days, 2 days, 3 days, 4 days, 5 days]ú1 hoursr aÍ 0 0 days 01:00:00
|
1 1 days 01:00:00
|
2 2 days 01:00:00
|
3 3 days 01:00:00
|
4 4 days 01:00:00
|
5 5 days 01:00:00
|
6 6 days 01:00:00
|
7 7 days 01:00:00
|
8 8 days 01:00:00
|
9 9 days 01:00:00
|
dtype: category
|
Categories (10, timedelta64[ns]): [0 days 01:00:00, 1 days 01:00:00, 2 days 01:00:00,
|
3 days 01:00:00, ..., 6 days 01:00:00, 7 days 01:00:00,
|
8 days 01:00:00, 9 days 01:00:00]©r r r r r r« r r r Ú&test_categorical_series_repr_timedeltaÜ s z:TestCategoricalRepr.test_categorical_series_repr_timedeltac C sd t ddd}tt|dd}d}t||ks0tt ddd}tt|dd}d }t||ks`td S )
|
Nr³ r6 r´ Tr¡ z0 1 days
|
1 2 days
|
2 3 days
|
3 4 days
|
4 5 days
|
dtype: category
|
Categories (5, timedelta64[ns]): [1 days < 2 days < 3 days < 4 days < 5 days]rµ r aÑ 0 0 days 01:00:00
|
1 1 days 01:00:00
|
2 2 days 01:00:00
|
3 3 days 01:00:00
|
4 4 days 01:00:00
|
5 5 days 01:00:00
|
6 6 days 01:00:00
|
7 7 days 01:00:00
|
8 8 days 01:00:00
|
9 9 days 01:00:00
|
dtype: category
|
Categories (10, timedelta64[ns]): [0 days 01:00:00 < 1 days 01:00:00 < 2 days 01:00:00 <
|
3 days 01:00:00 ... 6 days 01:00:00 < 7 days 01:00:00 <
|
8 days 01:00:00 < 9 days 01:00:00]r¶ r« r r r Ú.test_categorical_series_repr_timedelta_orderedü s zBTestCategoricalRepr.test_categorical_series_repr_timedelta_orderedN) r r r r r r r¢ r¬ r r± r² r· r¸ r r r r r s r )r r Únumpyr. rh Zpandasr r r r r r r
|
r r Zpandas._testingZ_testingrn r r r r r r Ú<module> s (
|
y
|
