U
|
ý°dq< ã @ s¸ d dl mZ d dlZd dlmZ d dlZd dlZd dlZ d dl
|
Z
|
d dl m Z m Z d dlmZmZmZmZmZmZ d dlmZ e
|
jdd Zdd Zd
|
d Ze
|
j d d dddg¡dd Z e
|
j ddddddg¡dd Z dd Z e
|
j d d d d d d!g¡d"d# Z!d$d% Z"d&d' Z#d(d) Z$d*d+ Z%d,d- Z&d.d/ Z'e
|
j j(e d0d1d2d3 Z)e
|
j j*e d4d1d5d6 Z+e
|
j j(e d7d1d8d9 Z,d:d; Z-d<d= Z.d>d? Z/e
|
j j*e d@d1dAdB Z0dCdD Z1dS )Eé )ÚStringION)Úascii_uppercase)ÚIS64ÚPYPY)ÚCategoricalIndexÚ DataFrameÚ
|
MultiIndexÚSeriesÚ
|
date_rangeÚoption_contextc C s t tj dd¡ddddgdS )z&Dataframe with duplicate column names.iÜ é ÚaÚb©Úcolumns)r ÚnpÚrandomÚrandn© r r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/formats/test_info.pyÚduplicate_columns_frame s r c C s: t } t }| j|d | ¡ }t d¡}||ks6td S )N©Úbufzd <class 'pandas.core.frame.DataFrame'>
|
RangeIndex: 0 entries
|
Empty DataFrame
|
©r r ÚinfoÚgetvalueÚtextwrapÚdedentÚAssertionError©Údfr ÚresultÚexpectedr r r Útest_info_empty s ÿr# c C s d} t dtjjd| di}tt td¡ tjjdd| d¡ d¡|d< | ¡ t
|
}|j |d ||d d
|
k }t
|
}|j |d d S ) NiÄ Úint64éd ©ÚsizeZ
|
abcdefghijr é
|
Úcategoryr Úd) r r r Úrandintr ÚarrayÚlistZtakeÚastypeZisnar r )Únr r Zdf2r r r Ú'test_info_categorical_column_smoke_test/ s ÿþ r0 Úfixture_func_nameZ int_frameZ float_frameZdatetime_framec C s< | | ¡}t }|j|d | ¡ ¡ }t|dks8td S )Nr r( )Zgetfixturevaluer r r Ú
|
splitlinesÚlenr )r1 ÚrequestÚframer r! r r r Útest_info_smoke_test> s
|
|
|
r6 z&num_columns, max_info_columns, verbose)r( r% T)r( é T)r( r( T)r( é F)r( é Fc C sn t tj d| ¡}td|H t }|j|d | ¡ }t }|j||d | ¡ }||ks`tW 5 Q R X d S )Né údisplay.max_info_columnsr ©r Úverbose) r r r r r r r r r )Z num_columnsÚmax_info_columnsr= r5 Z
|
io_defaultr! Z io_explicitr" r r r Ú#test_info_default_verbose_selectionO s r? c
|
C sÆ t } d}d}ttj d|¡}|jd| d | ¡ }d}||ksFt|jd| d | d¡ | ¡ }t
|
|dksvtt |D ]B\}}|| kr|| k r~n q~d|| d} | | ¡s~tq~d S ) Néé r: é T)r= r z- # Column Dtype
|
--- ------ ----- r ú ) r r r r r r r r ÚseekÚ readlinesr3 Ú enumerateÚ
|
startswith)
|
r r' Ústartr5 ÚresÚheaderÚlinesÚiÚlineZline_nrr r r Ú-test_info_verbose_check_header_separator_bodyg s
|
rM z>size, header_exp, separator_exp, first_line_exp, last_line_exp)r ú$ # Column Non-Null Count Dtype ú$--- ------ -------------- ----- ú$ 0 0 3 non-null float64z$ 3 3 3 non-null float64)r7 rN rO rP z$ 10 10 3 non-null float64)ée z% # Column Non-Null Count Dtype z%--- ------ -------------- ----- z% 0 0 3 non-null float64z% 100 100 3 non-null float64)r@ z& # Column Non-Null Count Dtype z&--- ------ -------------- ----- z& 0 0 3 non-null float64z& 1000 1000 3 non-null float64)i' z' # Column Non-Null Count Dtype z'--- ------ -------------- ----- z' 0 0 3 non-null float64z' 10000 10000 3 non-null float64c C s t tj d| ¡}t "}|jdd|d | ¡ ¡ }W 5 Q R X |dd
}|^} }
|
} } } | |ksht|
|
|kstt| |kst| |kstdS )zETest header column, spacer, first line and last line in verbose mode.rA T)r= Ú show_countsr éþÿÿÿN) r r r r r r r r2 r )r' Z
|
header_expZ separator_expZfirst_line_expZ last_line_expr5 r Z all_linesÚtablerI Ú separatorÚ
|
first_lineÚrestÚ last_liner r r Ú%test_info_verbose_with_counts_spacing} s , rY c C sd t dtddgddi} t }| j|d | ¡ }t| ¡ ¡ }t d| d¡}||ks`t
|
d S ) Nr r9 é Úi8)Údtyper a <class 'pandas.core.frame.DataFrame'>
|
RangeIndex: 2 entries, 0 to 1
|
Data columns (total 1 columns):
|
# Column Non-Null Count Dtype
|
--- ------ -------------- -----
|
0 a 2 non-null int64
|
dtypes: int64(1)
|
memory usage: z bytes
|
) r r r r r ÚfloatÚ memory_usageÚsumr r r )r r r! Úbytesr" r r r Útest_info_memory· s øÿ ra c C s t } ttj dd¡}|j| d t } |j| dd | ¡ }t| ¡ dksTt |}t
|
dd, t } |j| d | ¡ }||kst W 5 Q R X d S )Nr: rQ r ©r Úmax_colsr% r; ) r r r r r r r r3 r2 r r )Úior r! r" r r r Útest_info_wideÍ s re c C sZ t } tddggddgd}|j| d | ¡ d¡}d|d ksFtd |d
|
ksVtd S ) Nr9 g @r r r Tz% 0 a 1 non-null int64
|
r: z% 1 a 1 non-null float64
|
é )r r r r r2 r )rd r5 rJ r r r Ú0test_info_duplicate_columns_shows_correct_dtypesß s rg c
|
C s¼ dddddddg} i }d}t | D ]"\}}tjjd |d
|
|¡||< q"t|}t }|j|d | ¡ }d }||ksxt t | D ]6\}}d |dd|dd|dd| } | |kst qd S )Nr$ Úfloat64údatetime64[ns]útimedelta64[ns]Ú
|
complex128ÚobjectÚboolr( rZ r& r zY # Column Non-Null Count Dtype
|
--- ------ -------------- ----- rB r* z z z non-null )
|
rE r r r+ r. r r r r r )
|
ÚdtypesÚdatar/ rK r\ r r rH rI Únamer r r Ú test_info_shows_column_dtypesé s, ù ÿ "rq c
|
C st t tj dd¡} dD ]T\}}tdd< t }| j||d | ¡ }t| ¡
|
d¡|ks`t W 5 Q R X qdD ]T\}}tdd< t }| j||d | ¡ }t| ¡
|
d¡|ksºt W 5 Q R X qpd D ]¤\}}tdd> t }| j||d
|
| ¡ }t| ¡
|
d¡|kst W 5 Q R X tdd> t }| j||d
|
| ¡ }t| ¡
|
d¡|ksdt W 5 Q R X qÊd S ) Nr( r: ))r: N©r: F©é Tr> r r< Ú
|
))rt Nrr rs ))rt r: )r: r rb ) r r r r r r r r r3 ÚstripÚsplitr )r Zlen_r= r rH rc r r r Útest_info_max_cols s0 & & & rx c C sê dddddddg} i }d}t | D ]"\}}tjjd |d
|
|¡||< q"t|}t }|j|d d | ¡ ¡ }d |d ks~t
|
|j|dd | ¡ ¡ }d |d ks¨t
|
|j|d d | ¡ ¡ }t d|d ¡sÖt
|
|j d d
d d
f j|d d | ¡ ¡ }t d|d ¡rt
|
ddddg} i }d}t | D ]$\}}tjjd |d
|
|¡||< q4t|}| |_tddgidgd}|j|d d | ¡ ¡ }t d|d ¡s¬t
|
|j|dd | ¡ ¡ }t d|d ¡sÜt
|
| ¡ ¡ } t| | d |jj }
|
| |
|
kst
|
t |jj¡d } | t | ¡ ¡ks8t
|
| ¡ ¡ |jd d ¡ ksZt
|
tddgdgd jd d tddgdgd jj tdt dgtd g¡dgd }|jj |jd d |jjj |jd d ¡ } | d ksæt
|
d S ) Nr$ rh ri rj rk rl rm r( rZ r& T)r r^ zmemory usage: éÿÿÿÿFzmemory usage: [^+]+\+r: r% r r9 Úfoo©ÚindexÚdeepzmemory usage: [^+]+$é ©r} ÚA©r| r éè ©ro r| r r )rE r r r+ r. r r r r r2 r ÚreÚmatchZilocr r^ r_ r3 r| Únbytesr' Úvaluesr Ú from_productÚrange) rn ro r/ rK r\ r r rH Údf_with_object_indexZdf_sizeZexp_sizeZsize_dfÚmemr r r Útest_info_memory_usage' st ù " ÿ
|
r z'on PyPy deep=True doesn't change result)Úreasonc C sl t ddgidgd} | jddd ¡ | jdd ¡ ks:tt ddgi}|jdd ¡ | ¡ ¡ kshtd S ©Nr r9 rz r{ T)r| r} r ©r r^ r_ r ©r Z df_objectr r r Ú$test_info_memory_usage_deep_not_pypy| s ÿÿr z(on PyPy deep=True does not change resultc C sl t ddgidgd} | jddd ¡ | jdd ¡ ks:tt ddgi}|jdd ¡ | ¡ ¡ kshtd S r r r r r r Ú test_info_memory_usage_deep_pypy s ÿÿr z PyPy getsizeof() fails by designc C sR t dt dgtdg¡dgd} | jdd ¡ }|t | ¡ }t|dk sNt d S ) Nr9 r r r r Tr r% )
|
r r r r r^ r_ ÚsysÚ getsizeofÚabsr )r r Zdiffr r r Útest_usage_via_getsizeof s ÿr c C s t } tdtddddgd}|j| d d| ¡ ks:tt } tdtdtdd}|j| d d| ¡ ksrtt } tdtdt tdtdg¡d}|j| d d| ¡ ks¸tt } tdtdt tdd d
|
gg¡d}|j| d d| ¡ ksþtd S ) Nr9 ÚabrZ rA )r r| r ú+ÚABCrz Úbar) r r r- r r r r r r )r r r r r Ú test_info_memory_usage_qualified s0 ÿ ÿ r c C s dd } d}t t}tjtttd|dgddgd}td tj || ¡i|d
|
}|
|
d¡}|j j |j j kspt | || |kst | || | d k st d S ) Nc S s | j dd ¡ S )NTr )r^ r_ )Úfr r r r^ ¾ s z>test_info_memory_usage_bug_on_multiindex.<locals>.memory_usager% Z20160101)ZperiodsÚidÚdate)ÚnamesÚvaluer{ iÐ )r3 Ú uppercaser r r- r
|
r r r r Zunstackr r r )r^ ÚNÚMr| r Z unstackedr r r Ú(test_info_memory_usage_bug_on_multiindexº s þ
|
r¤ c C s6 t ddg} tt d¡| | d}t }|j|d d S )Nr r )rZ rZ r r )r r r Úzerosr r )Úidxr r r r r Útest_info_categoricalÑ s r§ z GH 36579: fail on 32-bit systemc C sT t ddgddgdddgd} t }| jd|d | ¡ }t d
|
¡}||ksPtd S ) Nr9 rZ rA )r9 rZ r ÚBr{ T)rR r af <class 'pandas.core.frame.DataFrame'>
|
Index: 2 entries, A to B
|
Data columns (total 2 columns):
|
# Column Non-Null Count Dtype
|
--- ------ -------------- -----
|
0 1 2 non-null int64
|
1 2 2 non-null int64
|
dtypes: int64(2)
|
memory usage: 48.0+ bytes
|
r r r r r Útest_info_int_columnsÚ s ÿ r© c C sR t ddgd} t d ¡ | ¡ }W 5 Q R X ttr6dnddgd}t ||¡ d S )Nr r r{ é r~ ZIndex)r ÚtmZassert_produces_warningr^ r r Zassert_series_equal)r r! r" r r r Ú"test_memory_usage_empty_no_warningñ s
|
r¬ )2rd r r Ústringr r¡ r r Únumpyr ZpytestZ pandas.compatr r Zpandasr r r r r
|
r Zpandas._testingZ_testingr« Zfixturer r# r0 ÚmarkZ parametrizer6 r? rM rY ra re rg rq rx r Zskipifr Zxfailr r r r¤ r§ r© r¬ r r r r Ú<module> sz
|
üþ
|
ûþ
|
|
ãþ(
|
|
"U
|
|
|
|
|
