U
|
ý°dük ã @ sÌ d dl mZm Z d dlZd dlZd dlZd dlmZ d dlm Z d dl
|
Z d dl
|
m Z m Z mZmZmZ d dlmZ dd Zdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Z ej de j dddd e j d d dd fed ddd d!ed d dd"d!fe j d#ddd e j d$d dd fg¡d%d& Z!d'd( Z"d)d* Z#d+d, Z$ej d-d.d.gd.d/gd.gd/gd.d/ggd0d0gd0d1gd0gd1gd2d3d.gd/fd.d/gfd0d4d5d0gd1fd2d/gd.d6gd4d1gd7d8d9g¡d:d; Z%ej d<d.d/dgd0d1gfg d0d1gfd.d/dgg fg¡d=d> Z&d?d@ Z'dAdB Z(dCdD Z)dEdF Z*dGdH Z+dIdJ Z,dKdL Z-dMdN Z.dOdP Z/ej dQg g fdRdSdTgg fg d0d1dUgfg¡dVdW Z0ej dXe1e2dY¡dZd[ Z3ej d\d.d.gd.d/gd.gd/gd0d0gd0d1gd0gd1gg¡d]d^ Z4d_d` Z5dadb Z6ej dcdddeg¡ej dfdgdh didh djdh g¡dkdl Z7dmdn Z8dodp Z9ej dqed.d/dgdRdred0d1gdSdrdRdSgfed.d/dgdRdrd0d1gdRdgfd.d/dgd0d1gdfg¡dsdt Z:dudv Z;dwdx Z<dydz Z=d{d| Z>ej j?ed}d~dd Z@ej ded.d/ddYgd.d/ddYgd.d/gddYgddgge d.d/ddYge Ad.d/gddYgddgg¡dg¡d
d ZBdd ZCej ddddgfddgddgfg¡dd ZDej ddd0d1dUgdfg¡dd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdS )¡é )ÚdateÚdatetimeN)Úpa_version_under7p0)Ú'construct_1d_object_array_from_listlike)ÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚ
|
date_rangec C sl t ddddggddddggd gd
|
} t| t s2ttddddgd d }t | jd |¡ | jd gkshtd S ) NÚfooÚbarÚbazÚquxr é é é Úfirst©ÚlevelsÚcodesÚnames©Úname)r Ú
|
isinstanceÚAssertionErrorr ÚtmÚassert_index_equalr r ©ÚresultÚexpected© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_constructors.pyÚ test_constructor_single_level s ÿr" c C sz d} t jt| d tg g d W 5 Q R X d} t jt| d tg d W 5 Q R X t jt| d tg d W 5 Q R X d S )Nz non-zero number of levels/codes©Úmatch©r r z Must pass both levels and codes©r )r )ÚpytestÚraisesÚ
|
ValueErrorr Ú TypeError©Úmsgr r r! Útest_constructor_no_levels# s r- c C sê ddgddgg} ddddgddddgg}dgdgf}d}t jt|d t| ||d
|
W 5 Q R X tddgddggddddgddddggd d
|
}d gd gg}t jt|d |j|d W 5 Q R X t jt|d |j|d W 5 Q R X d S )Nr r ÚoneÚtwor r r z(MultiIndex\.name must be a hashable typer# r )r r ZfoorZbarr©r )r' r( r* r ÚrenameZ set_names)r r r r, ÚmiZrenamedr r r! Ú"test_constructor_nonhashable_names/ s ý r3 c C s¦ t dg¡t dg¡t dg¡g}dg}d}tjt|d t||d W 5 Q R X d}d }d
|
}tjt|d. tdgd ggd dddgd dd dggd W 5 Q R X tjt|d* tdgd ggd d d d gd d ggd W 5 Q R X tjt|d | ¡ dgd gg¡ W 5 Q R X tjt|d$ | ¡ d d d d gd d gg¡ W 5 Q R X | ¡ jd d d d gd d ggdd tjt|d& tdgd ggd dgd d ggd W 5 Q R X d S )Nr r r Úaz+Length of levels and codes must be the samer# r% zbOn level 0, code max \(3\) >= length of level \(1\)\. NOTE: this index is in an inconsistent statez Unequal code lengths: \[4, 2\]z"On level 0, code value \(-2\) < -1Úbr é F)r Zverify_integrityéþÿÿÿ) ÚnpÚarrayr' r( r) r ÚcopyÚ
|
set_levelsÚ set_codes)Úidxr r r, Z length_errorZ label_errorZcode_value_errorr r r! Ú(test_constructor_mismatched_codes_levelsG s( "ÿ2. ("r> c C sV t tjd tjddggddddddggd} t tjd tjddggddddddggd}t | |¡ t tjd tjdd ggddddddggd} t tjd tjdd ggddddddggd}t | |¡ t ddddd
|
ggddddddggd tjd tjdd gg¡} t | |¡ t tjd tjdd ggddddddggd ddddddgg¡} t | |¡ d S ) Né r r éÿÿÿÿr r r6 r% Úsé ) r r8 ÚnanÚpdÚNaTr r r; r< r r r r! Útest_na_levelsm s@ ÿ ÿ ÿ ÿ ÿþ ÿþrF c C s¤ t dddg¡} t dddddddg¡}|d }t| | g||gdd}|jd d |ks\td |d< |jd d |kszt| d }d
|
| d< |jd d |ks td S ) Nr4 r5 Úcr r r T)r r r: é ZPANDA)r8 r9 r r r r )r r Úvalr2 r r r! Útest_copy_in_constructor s rJ c C s dd t | j| jD }tj|| jd}t || ¡ t tj t
|
dgddgg¡}|jd t t
|
dg¡snt |jd t ddg¡st d S ) Nc S s g | ]\}}t |¡ |¡qS r ©r8 ÚasarrayÚtake©Ú.0ÚlevZ level_codesr r r! Ú
|
<listcomp> s ÿz$test_from_arrays.<locals>.<listcomp>r0 Z20130101r4 r5 r r )Úzipr r r Ú from_arraysr r r rD rE r Úequalsr r ©r= Úarraysr r r r! Útest_from_arrays s þ rW c C sd dd t | j| jD }tjt|| jd}t || ¡ d}t j
|
t |d t d¡ W 5 Q R X d S )Nc S s g | ]\}}t |¡ |¡qS r rK rN r r r! rQ ¯ s ÿz-test_from_arrays_iterator.<locals>.<listcomp>r0 z/Input must be a list / sequence of array-likes.r# r ) rR r r r rS Úiterr r r r' r( r* )r= rV r r, r r r! Útest_from_arrays_iterator s þ rY c C s< t dd t| j| jD }tj|| jd}t || ¡ d S )Nc s s& | ] \}}t t |¡ |¡V qd S ©N)Útupler8 rL rM rN r r r! Ú <genexpr>¿ s ÿz*test_from_arrays_tuples.<locals>.<genexpr>r0 ) r[ rR r r r rS r r r rU r r r! Útest_from_arrays_tuples¾ s
|
þr] )Úidx1Úidx2ú
|
2011-01-01ÚDr ©ÚfreqÚperiodsz
|
2015-01-01ÚHú2015-01-01 10:00ú
|
US/Eastern©rc rd Útzú
|
Asia/Tokyoú1 daysz2 hoursc C s| t | |g¡}t | d¡| ¡ t | d¡|¡ t t| t|g¡}t | d¡| ¡ t | d¡|¡ t ||¡ d S )Nr r )r rS r r Úget_level_valuesr )r^ r_ r Úresult2r r r! Ú=test_from_arrays_index_series_period_datetimetz_and_timedeltaÉ s rn c C s t ddddd} t dddd}tjdddd}tjd ddd}t | |||g¡}t | d
|
¡| ¡ t | d ¡|¡ t | d ¡|¡ t | d¡|¡ t t | t |t |t |g¡}t | d
|
¡| ¡ t | d ¡|¡ t | d ¡|¡ t | d¡|¡ t ||¡ d S ) Nrf ra r rg rh re rb rk r` r r r )
|
r
|
rD Útimedelta_rangeÚ period_ranger rS r r rl r )r^ r_ Zidx3Zidx4r rm r r r! Ú)test_from_arrays_index_datetimelike_mixedæ s" ÿrq c C sÖ t jtdtddd} t jtdtddd}t | |g¡}t | d¡| ¡ t | d¡|¡ t t| t|g¡}t | d¡| ¡ t | d¡|¡ t | j |j g¡}t | d¡| ¡ t | d¡|¡ d S )NÚabcaabÚbacF©Ú
|
categoriesÚorderedTr r )
|
rD ÚCategoricalIndexÚlistr rS r r rl r Úvalues)r^ r_ r rm Zresult3r r r! Ú)test_from_arrays_index_series_categoricalý s rz c C sÔ d} t jt| d tjg d W 5 Q R X tjg gdgd}t|tsJttg dd}t |j
|
d |¡ |j dgksxtdD ]R}g g| }t d d |
}tj||d}tg g| g g| |d
|
}t ||¡ q|d S ) Nú)Must pass non-zero number of levels/codesr# ©rV ÚA©rV r r r )r r ÚABCr ) r' r( r) r rS r r r r r r r rx )r, r r ÚNrV r r r r! Útest_from_arrays_empty s
|
r Úinvalid_sequence_of_arraysr r r4 r5 ©r ©r r ©r4 ©r4 r5 ©r )r r )r r )r
r5 c C s. d}t jt|d tj| d W 5 Q R X d S )Nz.Input must be a list / sequence of array-likesr# r| )r' r( r* r rS )r r, r r r! Ú test_from_arrays_invalid_input% s r z
|
idx1, idx2c C s0 d}t jt|d t | |g¡ W 5 Q R X d S )Nz ^all arrays must be same length$r# )r' r( r) r rS )r^ r_ r, r r r! Ú"test_from_arrays_different_lengthsG s r c C st t dddgdd} t dddgd d}tj| |gd d
|
}tdddgdddggd ddgd ddggd d }t ||¡ d S ) Nr r r r r r4 r5 rG r r0 r r )r r rS r r ©r4 r5 r r r r r! Ú$test_from_arrays_respects_none_namesQ s ÿr c C st d} t jt| d t g ¡ W 5 Q R X tddgddggddgddggdd gd
|
}tjd dd gd }t ||¡ d S ) Nz-Cannot infer number of levels from empty listr# r r r r6 r r4 r5 r )r )r r6 r0 )r' r( r* r Ú from_tuplesr r )r, r r r r r! Útest_from_tuplesa s ÿr c C s t ddgddggddgddggddgd} t jtddgddgddgd }t || ¡ d
|
}tjt|d t d¡ W 5 Q R X d S ) Nr r r r6 r r4 r5 r r0 z/Input must be a list / sequence of tuple-likes.r# )r r rR r r r' r( r* )r r r, r r r! Útest_from_tuples_iteratoro s ÿ r c C s8 t jg ddgd} t jg g gddgd}t | |¡ d S )Nr4 r5 r0 r~ )r r rS r r r r r r! Útest_from_tuples_empty s r c C s" t | ¡}|j| jk ¡ s td S rZ )r r ry Úallr )r= r r r r! Ú test_from_tuples_index_values s
|
r c C s^ dddg} d}t jt|d t| dd W 5 Q R X t jt|d t| dd W 5 Q R X d S ) N)r r r )r r r )r r r ú*Names should be list-like for a MultiIndexr# Úabcr r4 )r' r( r) r )Úlir, r r r! Ú test_tuples_with_name_string s
|
r c C sf t jdddgdddggdddgd ddg¡} tjd
|
d gd d }t jddgdg|d}t | |¡ d S )Nr r r6 r r r4 r5 rG ©Úcolumns)r r )r6 r r r0 )r Úindex)rD Ú DataFrameZ set_indexr r r Zassert_frame_equal)r r= r r r r! Ú!test_from_tuples_with_tuple_label s ÿþr c C s, d} t jt| d t g ¡ W 5 Q R X d S )Nr{ r# )r' r( r) r Ú from_productr+ r r r! Ú#test_from_product_empty_zero_levels£ s r c C sD t jg gdgd} tg dd}t | jd |¡ | jdgks@td S )Nr} r0 r r )r r r r r r r r r r r r! Ú!test_from_product_empty_one_levelª s r z first, secondr r r rG c C s@ ddg}t j| |g|d}t | |gg g g|d}t ||¡ d S )Nr} ÚBr0 r ©r r r r )r Úsecondr r r r r r! Ú"test_from_product_empty_two_levels± s r¡ r r6 c C sT dddg}t t| }tjg |g g|d}tg |g gg g g g|d}t ||¡ d S )Nr} r ÚCr0 r )rx Úranger r r r )r r Zlvl2r r r r r! Ú$test_from_product_empty_three_levels» s
|
|
r¤ Ú invalid_inputc C s. d}t jt|d tj| d W 5 Q R X d S )NzDInput must be a list / sequence of iterables|Input must be list-liker# )Ú iterables)r' r( r* r r )r¥ r, r r r! Ú test_from_product_invalid_inputÅ s r§ c C s` t ddd} t ddg| g¡}tdtdfdtdfdtdfdtdfg}t |j|¡ d S )Nz
|
2000-01-01r )rd r z
|
2000-01-02)r
|
r r r r r Zassert_numpy_array_equalry )Zdt_indexr2 Zetalonr r r! Ú test_from_product_datetimeindexÎ s
|
|
|
|
üÿr¨ c C s< t td} ddg}t | |g¡}tj|jd | dd d S )NrB r4 r5 r T)Úexact)r r£ r r r r Z_levels)ÚrngÚotherr2 r r r! Ú test_from_product_rangeindexÜ s r¬ rv FTÚfc C s | S rZ r ©Úxr r r! Ú<lambda>å ó r° c C s t | S rZ )r r® r r r! r° å r± c C s | j S rZ )ry r® r r r! r° å r± c C sh ddg}t jtdtd| d}t jtdtd td| d}t |||g¡}t | d¡|¡ d S )Nr r rr rs rt r )rD rw rx r r r r rl )rv r r r= r r r r r! Ú*test_from_product_index_series_categoricalä s ÿr² c C sb dddg} dddg}ddg}t j| |g|d }d
|
d d d dddddg }t j||d }t ||¡ d S )Nr r Úbuzr4 r5 rG r r r0 ©r r4 ©r r5 ©r rG ©r r4 ©r r5 ©r rG ©r³ r4 ©r³ r5 ©r³ rG )r r r r r )r r r r Útuplesr r r r! Útest_from_productó s
|
|
÷ r¾ c C s dddg} dddg}ddg}d d
|
d d d ddddg }t j||d}t jt| |g|d}t ||¡ d}tjt|d t d¡ W 5 Q R X d S )Nr r r³ r4 r5 rG r r r´ rµ r¶ r· r¸ r¹ rº r» r¼ r0 z-Input must be a list / sequence of iterables.r# r ) r r r rX r r r' r( r* )r r r r½ r r r, r r r! Útest_from_product_iterator s&
|
|
÷ r¿ za, b, expected_namesr c C sV t | |g¡}t dddgddggddddddgddddddgg|d}t ||¡ d S )Nr r r r4 r5 r r r )r4 r5 Zexpected_namesr r r r r! Ú test_from_product_infer_names% s ýrÀ c C s| t dddgdd} t ddgdd}tj| |gd d }tdddgddggd
|
d
|
ddddgd
|
dd
|
dd
|
dggd d }t ||¡ d S ) Nr r r r r r4 r5 r r0 r r )r r r r r r r r r! Ú%test_from_product_respects_none_names< s ýrÁ c C sN t td¡} ddg}t | |g¡}| jdd t | |g¡}t ||¡ d S )Nr r4 r5 F)Úwrite)r8 r9 r£ r r Zsetflagsr r )r4 r5 r r r r r! Útest_from_product_readonlyJ s rà c C sz | }ddg|_ t|}ttddddddgd d
|
}t ||¡ t|d d }ttddddddgd d
|
d d }t ||¡ d S ) Nr r )r r. )r r/ )r r. )r r/ )r r. )r r/ Úobject©Údtyper} r )r r r r )r= r r r r r r! Ú test_create_index_existing_nameU s>
|
ú÷ÿ ú÷ ôrÇ c C s^ t jddgddgddgddggddgd} tjdddd gddgd
|
}t | ¡}t ||¡ d S ) Nr4 r5 ÚL1ÚL2r )r4 r4 r )r5 r4 )r5 r5 r0 )rD r r r Ú
|
from_framer r )Údfr r r r r! Útest_from_frame s ÿ
|
ÿ
|
rÌ z minimum pyarrow not installed)Úreasonc
|
C s dd l } t tddd gddt ¡ | dtjd g¡¡d¡}t
|
|¡}t j tddd g d¡t ¡ | dtjd g¡¡gdd gd
|
}t ||¡ d S ) Nr r r ZInt64rÅ gÉ?r r4 r5 r0 )ZpyarrowrD r r Z Float64DtypeZ__from_arrow__r9 r8 rC r rÊ rS Zastyper r )ÚparË Z multi_indexedr r r r! Ú)test_from_frame_missing_values_multiIndex s þÿ
|
þûrÏ Ú non_framerB é é c C s( t jtdd t | ¡ W 5 Q R X d S )NzInput must be a DataFramer# )r' r( r* r rÊ )rÐ r r r! Útest_from_frame_error¡ s rÓ c sê t tddddddddddgt jddddd d gd
|
d d d d dd d gd¡} | j ¡ }tjtddddddddddgt jddddd d gd
|
d d d d dd d ggdddd gd}t | ¡ fddt j
|
D }t | ¡ ||ksæt d S )NZ19910905rÑ rg )rd ri r r r4 r5 rG T)rv r¯ ÚyÚz)Údatesr4 r5 rG rÖ r0 c s i | ]\}}| j | jqS r )r rÆ )rO Úir ©r2 r r! Ú
|
<dictcomp>È s z2test_from_frame_dtype_fidelity.<locals>.<dictcomp>)rD r r
|
Z CategoricalÚdtypesÚto_dictr rS rÊ Ú enumerater r r r )rË Zoriginal_dtypesZ expected_miZ mi_dtypesr rØ r! Ú test_from_frame_dtype_fidelity² s( üÿ
|
ü
|
ù
|
rÝ znames_in,names_out©rÈ r¯ ©rÉ rÔ r¯ rÔ c C sP t jddgddgddgddggt ddg¡d}tj|| d}|j|ksLtd S )Nr4 r5 rÞ rß r r0 )rD r r r rÊ r r )Znames_inZ names_outrË r2 r r r! Útest_from_frame_valid_namesÎ s þrà znames,expected_error_msg)Z bad_inputr z9Length of names must match number of levels in MultiIndexc C s\ t jddgddgddgddggt ddg¡d}tjt|d tj|| d W 5 Q R X d S )Nr4 r5 rÞ rß r r# r0 )rD r r r r' r( r) rÊ )r Zexpected_error_msgrË r r r! Ú test_from_frame_invalid_namesÛ s þrá c C s@ t g g gg g gddgd} t jg g gddgd}t | |¡ d S )Nr4 r5 r r~ )r rS r r r r r r! Ú test_index_equal_empty_iterableì s râ c
|
C sÈ ddgdddgg} t | ddddddgddddddggdd tjtdd0 t | ddddddgddddddggdd W 5 Q R X tjtdd0 t | ddddddgddddddggdd W 5 Q R X d S )Nr r r )r r Z sortorderz%.* sortorder 2 with lexsort_depth 1.*r# z%.* sortorder 1 with lexsort_depth 0.*)r r' r( r) r& r r r! Ú test_raise_invalid_sortorderó s$ ÿ ÿ ÿrã c C s6 t jdddgd dd} tddd d
|
d }t | |g¡}t jdddgdd}t |jd |¡ t |jd |¡ t d¡}t
|
¡ }t ¡ }t |||g|||g¡D ]F\}}t |g|gg¡} t| jd t jsÐtt| jd t js tq t ¡ }
|
t |
|
g|gg¡} t| jd t jrtt| jd t js2td S )Nz2013-04-01 9:00z2013-04-02 9:00z2013-04-03 9:00r rj )ri z
|
2010/01/01rÑ ÚMrg )rd rc ri r r Útoday)rD Ú DatetimeIndexr
|
r rS r r r r8 Z
|
datetime64r rå r Ú itertoolsÚproductr r r r ) r^ r_ r= Z expected1Zdate1Zdate2Zdate3Úd1Zd2r Zdate4r r r! Útest_datetimeindex s. ÿ ÿ
|
rê c C s¼ t jddgddd} t jddgdd d}t | |g¡}|jddgksHtt |jd
|
| ¡ t |jd |¡ t t | t |g¡}|jddgkstt |jd
|
| ¡ t |jd |¡ d S ) Nz2013/01/01 09:00z2013/01/02 09:00Zdt1z
|
US/Pacific)r ri z2014/01/01 09:00z2014/01/02 09:00Zdt2rj r r )
|
rD ræ r rS r r r r r r )r r r r r r! Útest_constructor_with_tz) s$ ÿ ÿrë c C s t ¡ } | | g}t|}|jtks&tt |g¡}|jd }|jtksJtt |g¡}|jd }|jtksntt
|
dd |D ¡}|jd }|jtkstd S )Nr c S s g | ]
|
}|fqS r r )rO r¯ r r r! rQ P s z9test_multiindex_inference_consistency.<locals>.<listcomp>) r rå r rÆ rÄ r r rS r r r )ÚvZarrr= r2 rP r r r! Ú%test_multiindex_inference_consistency> s
|
|
|
rí c C sJ t jdgdggddgd} | j}tddgt ddg¡d}t ||¡ d S ) Nr r4 r )rG Údr0 Úint64rÄ )r )r rS rÚ r r r Zassert_series_equal)Zpmidxr r r r r! Útest_dtype_representationU s ÿrð )Lr r rç Únumpyr8 r' Z pandas.compatr Zpandas.core.dtypes.castr ZpandasrD r r r r r
|
Zpandas._testingZ_testingr r" r- r3 r> rF rJ rW rY r] ÚmarkZ parametrizerp ro rn rq rz r r r r r r r r r r r r r¡ rx r£ r¤ r§ r¨ r¬ r² r¾ r¿ rÀ rÁ rà rÇ rÌ ZskipifrÏ r9 rÓ rÝ rà rá râ rã rê rë rí rð r r r r! Ú<module> s
|
& þþþ÷þ
|
éþ
|
(ÿ
|
ÿ
|
|
*ÿ
|
ý ùþ
|
*
|
|
úþ
|
ÿ
|
|
þþ
|
|
|
