U
|
ý°dt ã @ sP d dl mZ d dlmZ d dlZd dlZd dlZd dl m
|
Z
|
d dl m Z m Z mZmZ dd Zdd Zej d
|
d d g¡d d Zdd Zdd Zdd Zejje
|
dddd Zejje
|
dddd Zej d ed d d d gd d fed ed!d d fed e e d e d g¡d d fed e e d d"d#gg¡d d fg¡d$d% Z d&d' Z dS )(é )ÚStringIO)Úascii_uppercaseN)ÚPYPY)ÚCategoricalIndexÚ
|
MultiIndexÚSeriesÚ
|
date_rangec C sp d} t td¡ t jjdd| d¡}t| d¡}| ¡ t }|j
|
|d ||dk }t }|j
|
|d d S ) NiÄ Z
|
abcdefghijr é
|
©ÚsizeÚcategory©ÚbufÚd) ÚnpÚarrayÚlistZtakeÚrandomÚrandintr ÚastypeZisnar Úinfo)ÚnÚdataÚsr Ús2© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/formats/test_series_info.pyÚ'test_info_categorical_column_just_works s " r c C s4 t ddg} tt d¡| d}t }|j|d d S )NÚaÚbé ©Úindexr )r r r Úzerosr r )Úidxr r r r r Útest_info_categorical s r% ÚverboseTFc C s| | }t tt||dd}t }|j||d | ¡ }t d¡}|rR|t d¡7 }|t d| ¡ d¡7 }||ksxt d S )NZsth)r" Úname)r& r zw <class 'pandas.core.series.Series'>
|
MultiIndex: 10 entries, ('foo', 'one') to ('qux', 'three')
|
z Series name: sth
|
Non-Null Count Dtype
|
-------------- -----
|
10 non-null int64
|
z/ dtypes: int64(1)
|
memory usage: z.0+ bytes
|
)
|
r ÚrangeÚlenr r ÚgetvalueÚtextwrapÚdedentÚ memory_usageÚAssertionError)Z%lexsorted_two_level_string_multiindexr& r" Zserr ÚresultÚexpectedr r r Útest_info_series' s$ ÿÿþÿr1 c C sX t ddgdd} t }| j|d | ¡ }t| ¡ }t d| d¡}||ksTtd S )Né r Úi8)Údtyper zÕ <class 'pandas.core.series.Series'>
|
RangeIndex: 2 entries, 0 to 1
|
Series name: None
|
Non-Null Count Dtype
|
-------------- -----
|
2 non-null int64
|
dtypes: int64(1)
|
memory usage: z bytes
|
) r r r r* Úfloatr- r+ r, r. )r r r/ Z memory_bytesr0 r r r Útest_info_memoryG s øÿ r6 c C s> t tj d¡} d}tjt|d | jdd W 5 Q R X d S )Née zIArgument `max_cols` can only be passed in DataFrame.info, not Series.info)Úmatchr2 )Zmax_cols)r r r ÚrandnÚpytestZraisesÚ
|
ValueErrorr )r Úmsgr r r Útest_info_wide\ s r= c C st dddddddg} d}| D ]T}t tjjd |d
|
|¡}t }|j|d | ¡ }|d d | }||kstqd S )NÚint64Úfloat64zdatetime64[ns]ztimedelta64[ns]Z
|
complex128ÚobjectÚboolr r r
|
r r z non-null ) r r r r r r r r* r. )Zdtypesr r4 r r Úresr' r r r Útest_info_shows_dtypesc s ù rC z'on PyPy deep=True doesn't change result)Úreasonc C s\ t ddgidgd} | jddd| jddks2tt ddgi}|jdd| ¡ ksXtd S ©Nr r2 Úfoor! T)r" Údeep©rG ©r r- r. ©Zs_with_object_indexZs_objectr r r Ú$test_info_memory_usage_deep_not_pypyw s ÿ
|
þrK z(on PyPy deep=True does not change resultc C s\ t ddgidgd} | jddd| jddks2tt ddgi}|jdd| ¡ ksXtd S rE rI rJ r r r Ú test_info_memory_usage_deep_pypy s ÿ
|
þrL z series, plusr2 r é r! ÚABCrF Úbarc C s< t }| j|d |r(d| ¡ ks8tnd| ¡ ks8td S )Nr ú+)r r r* r. )ZseriesÚplusr r r r Ú test_info_memory_usage_qualified s
|
rR c C s° d} t t}tjtttd| dgddgd}ttj | | ¡|d}|
|
d¡}|j j |j j ksdt |jdd |jdd ¡ kst |jdd ¡ |jdd }|d
|
k s¬t d S ) Néd Z20160101)ZperiodsÚidÚdate)Únamesr! TrH iÐ )r) Ú uppercaser Ú from_productr r r r r r9 ZunstackÚvaluesÚnbytesr. r- Úsum)ÚNÚMr" r Z unstackedZdiffr r r Ú(test_info_memory_usage_bug_on_multiindex¢ s þ
|
r^ ) Úior Ústringr rW r+ Únumpyr r: Z pandas.compatr Zpandasr r r r r r% ÚmarkZ parametrizer1 r6 r= rC ZxfailrK rL r rX r( rR r^ r r r r Ú<module> s:
|
|
|
|
|
þüþ
|
|
