Ë
|
oñúh> ã ó¬ d dl Z d dlZd dlZd dlmZ d dlmZmZm Z m
|
Z
|
m Z m Z d dl mZ d dlmZ ej$ d « Z G d d« Z G d d « Zy)
|
é N)ÚSettingWithCopyError)Ú DataFrameÚIndexÚ
|
IndexSliceÚ
|
MultiIndexÚSeriesÚconcat)ÚBDayc ó¼ t j g d¢g d¢g d¢g« } t ddgddgg d¢g d ¢gg d
|
¢g d ¢g d ¢g d ¢gg d¢¬« }t | |t d« ¬« S )N©gEØðôJYà¿grùé·¯Àgî|?5^ºÝ¿gÌÑã÷6ý©?g²ï§ÆK×?©g à- Ü?gÉTÁ¨¤ö?gúíëÀ9#Ò?g9]ð?g46<½RÆ?)gá ©Qå¿g^)ËÇà¿g¶óýÔxéÖ¿gI&Âì?g=U«@ÚaÚxÚbÚq)çı.n£$@ç 4@g >@)é é é )r r é )r r r )r r é )r r r )ÚoneÚtwoÚthreeÚfour©ÚlevelsÚcodesÚnamesÚABCDE©ÚindexÚcolumns)ÚnpÚarrayr r Úlist)Úarrr# s úVH:\Change_password\venv_build\Lib\site-packages\pandas/tests/frame/indexing/test_xs.pyÚfour_level_index_dataframer* sc ä
|
((â 7Ú 5Ú 7ð
|
ó Cô Øc
|
S #JÒ(=ºyÐIÚ)¢Y² Ð:Ú-ô Eô
|
S ¬t°G«}Ô =Ð=ó c ó* e Zd Zd Zd Zd Zd Zd Zy)ÚTestXSc ó |j « }|j d }|j |« }|j « D ]E \ }} t j
|
| « r t j
|
|| | « r7J | || | k( rEJ dddddddd }
|
t |
|
« } | j d« }|j t j k( sJ |d
|
dk( sJ |d dk( sJ t j t t j d « ¬ « 5 |j |j d t « z
|
« d d d « |j d
|
d¬« } |d
|
} t j | | « |j d
|
d¬« } t j" |« 5 d| d d d d d « |r2t j |d
|
|d
|
« | dk( j% « rJ y | dk( j% « sJ y # 1 sw Y ·xY w# 1 sw Y _xY w)Nr r r )Ú1Ú2r/ r0 Ú3)r/ r0 r1 )ÚAÚBr2 r3 z Timestamp('1999-12-31 00:00:00')©Úmatchr )Úaxis)Úcopyr# ÚxsÚitemsr% Úisnanr ÚdtypeÚobject_ÚpytestÚraisesÚKeyErrorÚreÚescaper
|
ÚtmÚassert_series_equalÚassert_cow_warningÚall)ÚselfÚ float_frameÚdatetime_frameÚusing_copy_on_writeÚwarn_copy_on_writeÚfloat_frame_origÚidxr8 ÚitemÚvalueÚ test_dataÚframeÚseriesÚexpecteds r) Útest_xszTestXS.test_xs' s ð '×+Ñ+Ó-ÐØ× Ñ Ñ"Ø ^^CÓ Ø 88:ò 7KD%ÜxxÜxx ¨DÑ 1°#Ñ 6Õ7Ð7Ð7à ¨DÑ 1°#Ñ 6Ó6Ð6Ð6ð 7ð !"¨Ñ*°sÀÈ3Ñ1OÑP Ü)Ó$Ø XXc]Øxx2::Ò%Ð%Ð%Ø#w!|Ð|Ø#w#~Ð ~ä ]]Ü BIIÐ&HÓ Iô
|
ñ @ð × Ñ n× 2Ñ 2°1Ñ 5¼»Ñ >Ô ?÷ @ð ¨! Ó,Ø sÑ#Ü
|
× Ñ v xÔ0ð ¨! Ó,Ü × "Ñ "Ð#5Ó 6ñ ØF1I÷ á ä × "Ñ " ;¨sÑ#3Ð5EÀcÑ5JÔ KØ A ×*Ñ*Ô,Ð ,Ð,Ð,à M×&Ñ&Ô(Ð (Ñ(÷' @ð @ú÷ ð ús Ä*G2ÆG>Ç2G;Ç>Hc ó t dgt g d¬« ¬« }d|d< d|d< d |d
|
< d |d < d |d< |j d« }t g d¢t d« d¬« }t j ||« t g d¢¬« }|j d« }t g dt j ¬« }t j ||« y )Nr Ústr)r; r" ç ð?r2 Úfoor3 ç @ÚCÚbarÚDç @ÚE)rV rW rX rZ r\ r! )r# Úname©r r Úc©r# r )r^ r; ) r r r8 r r' rB rC r% Úfloat64)rF Údfr8 ÚexpÚresultrR s r) Útest_xs_cornerzTestXS.test_xs_cornerP s¯ ä aS¬%°¸%Ô*@Ô AØ3Ø3Ø3Ø3Ø3à UU1XÜÒ2¼$¸w»-ÈaÔPÜ
|
× Ñ r 3Ô'ô _Ô -ØsÜ" 3¬b¯j©jÔ9Ü
|
× Ñ v xÕ0r+ c óæ t t j j d« j d« g d¢¬« }|j d« }|j d }t j ||« y )Nr )r r )r r r` r r ra r` ) r r% ÚrandomÚ default_rngÚstandard_normalr8 ÚilocrB rC )rF rc Úcrossrd s r) Útest_xs_duplicateszTestXS.test_xs_duplicatesc sX Ü Ü II× !Ñ ! !Ó $× 4Ñ 4°VÓ <Ú+ô
|
ð
|
c
|
ØggajÜ
|
× Ñ u cÕ*r+ c ó t ddddddddddddd« j g d ¢« }|j dd
|
d ¬ « }|d d }t j ||« |j ddd
|
gd ¬ « }t j ||« y )NÚsatÚsun)r r Ú
|
strawberryé
|
é éØ )ÚdayÚflavourÚsalesÚyear)rx rv ru ru F)ÚlevelÚ
|
drop_levelr )rt ro rx )r Ú set_indexr8 rB Úassert_frame_equal©rF rc re rR s r) Útest_xs_keep_levelzTestXS.test_xs_keep_levelm s Ü à UÑ+Ø +° Ñ=Ø B Ø TÑ*ñ ó
|
÷ )Ò.Ó
|
/ð ð u E°eÓ<Øbq6Ü
|
× Ñ f hÔ/à}¨V°U¨OÈÓNÜ
|
× Ñ f hÕ/r+ c óò t t j d« j dd« t d« t d« ¬« }|j « }|rHt j « 5 d|j d« d d d d d « t j ||« y |r_d}t j t |¬« 5 d|j d« d d d d d « |j d« dk( j « rJ y t j « 5 d|j d« d d d d d « |j d« dk( j « sJ y # 1 sw Y ×xY w# 1 sw Y xY w# 1 sw Y FxY w) Nr r r r" é r zC\nA value is trying to be set on a copy of a slice from a DataFramer4 )r r% ÚarangeÚreshapeÚranger7 rB Ú raises_chained_assignment_errorr8 r| r= r> r ÚanyrE )rF Úusing_array_managerrI rJ ÚdmÚdf_origÚmsgs r) Ú test_xs_viewzTestXS.test_xs_view} s1 ô ryy ×.Ñ.¨q°!Ó4¼EÀ!»HÌeÐTUËhÔ WØ'')á Ü×3Ñ3Ó5ñ !Ø a ÷ !ä × !Ñ ! " gÕ .Ù ð YCÜ Ô 3¸3Ô?ñ !Ø a ÷ !àa B×+Ñ+Ô-Ð -Ð-Ð-ä×3Ñ3Ó5ñ !Ø a ÷ !àEE!H N×'Ñ'Ô)Ð )Ñ)÷ !ð !ú÷ !ð !ú÷ !ð !ús$ Á*EÂ=E!ÄE-ÅE Å!E*Å-E6N)Ú__name__Ú
|
__module__Ú __qualname__rS rf rm r~ r © r+ r) r- r- & s ò')òR1ò&+ò0ó *r+ r- c óÐ e Zd Zd Zd Zd Zd Zd Zd Ze j j ddd gd
|
gfg« d « Z d Z e j j d ddd eddgd¬« fddd edgd¬« fg« d « Ze j j dd d g« d « Ze j j dd d g« d « Zd Zd Zd Zd! Zd" Zd# Zd$ Zd% Zy&)'ÚTestXSWithMultiIndexc ó\ g d¢g d¢g}t t | « }t j |ddg¬« }t t
|
j j d« j d« g d¢|¬ « }|j d
|
d d ¬ « }|j d d
dgf }t j ||« y )N)rZ rZ Úbazr rW rW Úquxr )r r r r r r r r ÚfirstÚsecond©r r )r é )r2 r3 rY r" )r rZ )r r r ©ry r6 r ) r' Úzipr Ú from_tuplesr r% rh ri rj r8 rk rB r| )rF ÚarraysÚtuplesr# rc re rR s r) Útest_xs_doc_examplez(TestXSWithMultiIndex.test_xs_doc_example s ò EÚ Dð
|
ô c6lÓ#ä×&Ñ& v°g¸xÐ5HÔIÜ Ü II× !Ñ ! !Ó $× 4Ñ 4°VÓ <Ú!Øô
|
ð ~Ð-@ÀqÓIà771qc6?Ü
|
× Ñ f hÕ/r+ c ól t dd« }t d« }t j ||gddg¬« }t t
|
j j d« j t |« df« |g d ¢« }|j dd¬
|
« }|j dd d
f }t j ||« y ) Niaß2ieß2ÚabcdeÚdateÚsecidr r r )ÚXÚYÚZ©ry )r r' r Ú from_productr r% rh ri rj Úlenr8 ÚlocrB r| )rF ÚdatesÚidsr# rc re rR s r) Útest_xs_integer_keyz(TestXSWithMultiIndex.test_xs_integer_key s äh Ó)Ü7mÜ×'Ñ'¨°¨ ¸VÀWÐ<MÔNÜ Ü II× !Ñ ! !Ó $× 4Ñ 4´c¸%³jÀ!°_Ó EØ Ú ó
|
ð x vÓ.Ø66(¢A+Ñ&Ü
|
× Ñ f hÕ/r+ c óÄ |}|j dd¬« }||j j d« dk( }t g d¢d¬« |_ t j
|
||« y )Nr r r¥ r )rW rZ r r r ©r^ )r8 r# Úget_level_valuesr rB r| ©rF Ú multiindex_dataframe_random_datarc re rR s r) Ú test_xs_levelz"TestXSWithMultiIndex.test_xs_level¼ sV Ø -Øu HÓ-Øbhh×/Ñ/°Ó2°eÑ;Ñ<Ü Ò ;À'ÔJÜ
|
× Ñ f hÕ/r+ c ó2 t j j d« j d« }t g d¢g d¢g d¢gg d¢g d¢g d¢g¬« }t ||¬« }t |d d d
|
gd gg¬« }|j d d¬ « }t j ||« y )Nr )r r )r Úpr )r r Úy)r` ÚrÚz)r r r ©r r ra r r r r` r¥ © r% rh ri rj r r r8 rB r| )rF r( r# rc rR re s r) Útest_xs_level_eq_2z'TestXSWithMultiIndex.test_xs_level_eq_2à s Üii×#Ñ# AÓ&×6Ñ6°vÓ>ÜÚ#¢_²oÐFÚ iªÐ3ô
|
ô s %Ô (Ü S 1X¨s¨e°c°U¨^Ô<Øs !Ó$Ü
|
× Ñ f hÕ/r+ c óü |}|j « }|j dd¬« }|s|rd|d d n+d}t j t |¬« 5 d|d d d d d « t j ||« y # 1 sw Y xY w)Nr r r¥ rr úAA value is trying to be set on a copy of a slice from a DataFramer4 ©r7 r8 r= r> r rB r| )rF r° rI rJ rc r re r s r) Ú test_xs_setting_with_copy_errorz4TestXSWithMultiIndex.test_xs_setting_with_copy_errorÎ sz ð .Ø'')Øu HÓ-á Ñ"4ØF1Ið VCÜ Ô 3¸3Ô?ñ Ø q ÷ ä
|
× Ñ b 'Õ*÷ ð ús Á A2Á2A;c ó |}|j « }|j dddg¬« }|s|rd|d d n+d}t j t |¬« 5 d|d d d d d « t j ||« y # 1 sw Y xY w)N©r r r r r¥ rr r» r4 r¼ )rF r* rI rJ rc r re r s r) Ú(test_xs_setting_with_copy_error_multiplez=TestXSWithMultiIndex.test_xs_setting_with_copy_error_multipleã s~ ð (Ø'')Øx¨¨v Ó7á Ñ"4ØF1Ið VCÜ Ô 3¸3Ô?ñ Ø q ÷ ä
|
× Ñ b 'Õ*÷ ð ús ÁA4Á4A=z
|
key, level)r r r r c óF |}t |gdz « }|j j du sJ t |j dd¬« gdz « }t |t
|
« r |j t |« |¬« }n|j ||¬« }t j ||« y )Nr Fr r r¥ ) r r# Ú is_uniquer8 Ú
|
isinstancer' ÚtuplerB r| )rF Úkeyry r° rP rc rR re s r) Útest_xs_with_duplicatesz,TestXSWithMultiIndex.test_xs_with_duplicatesõ s ð 1Ü UGaKÓ Øxx×!Ñ! UÑ*Ð*Ð*Ü588 E°8Ó:Ð;¸aÑ?Ó@ä c4Ô ØUU5 :¨UUÓ3FàUU3 eUÓ,FÜ
|
× Ñ f hÕ/r+ c óú g d¢}t |g d¢¬« j ddg« }t dg d¢it dt j d d
|
gd¬ « ¬ « }|j d d¬« }t j ||« y )N))r r r )r Úbbcder )r´ Úyzcdeé )r¶ Úxbcdeé )r¶ Né )r¶ ÚzbcderÊ )r¶ ÚybcderÍ )Úa1Úa2Úcnt©r$ rÐ rÑ rÒ )rÌ rÍ rÊ rÍ rË rÎ rÏ r ra r¶ r¥ )r r{ r r% Únanr8 rB r| )rF Úaccrc rR re s r) Ú test_xs_missing_values_in_indexz4TestXSWithMultiIndex.test_xs_missing_values_in_index su ò
|
ô sÒ$7Ô 8× BÑ BÀDÈ$À<Ó PÜ Ø Ò$Ð %ܤ"§&¡&¨'°7Ð;À$ÔGô
|
ð
|
s $Ó'Ü
|
× Ñ f hÕ/r+ z key, level, exp_arr, exp_indexr Úlvl0c ó | d d
dd
f S )Nr r r ©r s r) ú<lambda>z TestXSWithMultiIndex.<lambda> s A¢a¨¨1¨ f¡I r+ rZ rW Úlvl1r c ó | d d
dd
f S )Nr r r rÙ s r) rÚ z TestXSWithMultiIndex.<lambda> s aª¨1¨Q¨3¨¡i r+ c ó, t j j d« j d« }t ddgg d¢gg d¢g d¢gdd g¬
|
« }t ||¬ « }|j ||d ¬ « }t ||« |¬ « } t j || « y )Nr )r r r r )rZ rW ÚhelloÚworld)r r r r )r r r r r× rÛ r rÓ r r r¸ )
|
rF rÅ ry Úexp_arrÚ exp_indexr( r# rc re rR s
|
r) Ú test_xs_named_levels_axis_eq_1z3TestXSWithMultiIndex.test_xs_named_levels_axis_eq_1 s ô ii×#Ñ# AÓ&×6Ñ6°vÓ>ÜØ#JÒ @ÐAÚ ¢ Ð.Ø6Ð"ô
|
ô
|
s EÔ *Øs %¨aÓ0Ü W S\°9Ô=Ü
|
× Ñ f hÕ/r+ Úindexerc ó, | j dddg¬« S )Nr¿ r r r¥ ©r8 ©rc s r) rÚ z TestXSWithMultiIndex.<lambda>/ s ruuX¨e°V¨_uÓ= r+ c óF | j d« j dd¬« S )Nr r r r¥ rå ræ s r) rÚ z TestXSWithMultiIndex.<lambda>0 s ruuSz}} Q¨f}Ó5 r+ c ó¬ |}g d¢g}t dgdggdgdggddg¬« }t ||t d« ¬ « } ||« }t j ||« y )
|
Nr r r r r r r r! r" ©r r r' rB r| ©rF rã r* rc Úexpected_valuesÚexpected_indexrR re s r) Útest_xs_level_multiplez+TestXSWithMultiIndex.test_xs_level_multiple, si ð (ÚDÐEÜ#ØED6?¨A¨3°°¨*¸UÀGÐ<Lô
|
ô Ø >¼4À»=ô
|
ñ Ü
|
× Ñ f hÕ/r+ c ó( | j dd¬« S )Nr r r¥ rå ræ s r) rÚ z TestXSWithMultiIndex.<lambda>@ s ruu S°uÓ 2 r+ c ó$ | j d« S )Nr rå ræ s r) rÚ z TestXSWithMultiIndex.<lambda>@ s ¸r¿u¹uÀS»z r+ c óÆ |}g d¢g d¢g}t ddgddgddggd d
|
gd d
|
gd
|
d ggg d ¢¬ « }t ||t d « ¬« } ||« }t j ||« y )Nr r r r r r r r r r )r r r r r! r" ré rê s r) Útest_xs_level0z#TestXSWithMultiIndex.test_xs_level0? s ð (â 7Ú 5ð
|
ô $Ø#J ¨$ °!°Q°Ð8Øq6Aq6 A q 6Ð*Ú*ô
|
ô
|
Ø >¼4À»=ô
|
ñ Ü
|
× Ñ f hÕ/r+ c ó |}|j d« j }|j d }t j ||« y )N©rZ r r )r8 ÚvaluesrB Úassert_almost_equalr¯ s r) Útest_xs_valuesz#TestXSWithMultiIndex.test_xs_valuesT s8 Ø -Ø~Ó&×-Ñ-Ø99Q<Ü
|
× Ñ v xÕ0r+ c ót |}|j d« }|j d }t j ||« y )Nró )r8 r¨ rB rC r¯ s r) Útest_xs_loc_equalityz)TestXSWithMultiIndex.test_xs_loc_equalityZ s3 Ø -Ø~Ó&Ø66.Ñ)Ü
|
× Ñ v xÕ0r+ c ó t g d¢ddggg d¢g d¢g¬« }t t j j d« j d« |¬ « }|t u r|d }|j d
|
d j d« }|j t d d d
f « }t j ||« |j t d d d
f }t j ||« y ) N))rW rZ r )rW r r ©rW r r r r )r r r r r r )r r r r r r r· r )é r ra éþÿÿÿrú )r r r% rh ri rj r rk Ú droplevelr8 r rB Ú assert_equalr¨ )rF Úframe_or_seriesr# ÚobjrR re s r) Ú+test_xs_IndexSlice_argument_not_implementedz@TestXSWithMultiIndex.test_xs_IndexSlice_argument_not_implemented` sÇ ô ÚMÐPQÐSTÈvÐVÚ%Ò'9Ð:ô
|
ô
|
×-Ñ-¨aÓ0×@Ñ@ÀÓHÐPUÔVØ fÑ $Øa&Cà88BC=×*Ñ*¨1Ó-à
|
Ð#4²aÐ#7Ñ8Ó9Ü
|
Ô)àÐ$5²qÐ$8Ñ9Ñ:Ü
|
Õ)r+ c óÆ t dg d¢i« }|t u r|d }d}t j t |¬« 5 |j dd¬« d d d « y # 1 sw Y y xY w)Nr2 ©r r r zIndex must be a MultiIndexr4 r Úasr¥ )r r r= r> Ú TypeErrorr8 )rF rÿ r r s r) Útest_xs_levels_raisesz*TestXSWithMultiIndex.test_xs_levels_raisest sZ ÜiÐ(Ó)Ø fÑ $Øc(Cà*Ü ]]9¨CÔ 0ñ "Ø FF1DFÔ !÷ "÷ "ñ "ús ºAÁA c ó t j g d¢ddg¬« }t g d¢g|¬« }|j ddd ¬
|
« }t dd ggt j d d gddg¬« ¬« }t j
|
||« y )N)©r r ©r r´ )r r Úlevel1Úlevel2r r rÓ r r F©r6 rz r r r )r r r r8 rB r| )rF Úmirc re rR s r) Ú"test_xs_multiindex_droplevel_falsez7TestXSWithMultiIndex.test_xs_multiindex_droplevel_false} s ä × #Ñ #Ú 0¸À8Ð8Lô
|
ô {¨BÔ /Øs ¨uÓ5Ü ØVHÜ ×*Ñ*ØZÐ(°¸8Ð0Dôô
|
ô × Ñ f hÕ/r+ c óª t g d¢gt g d¢« ¬« }|j ddd¬« }t ddgi« }t j ||« y )Nr r_ rÓ r r Fr )r r r8 rB r| r} s r) Útest_xs_droplevel_falsez,TestXSWithMultiIndex.test_xs_droplevel_false sI ä {¬E²/Ó,BÔ CØs ¨uÓ5Ü c A 3ZÓ(Ü
|
× Ñ f hÕ/r+ c óâ t g d¢gt g d¢« ¬« }|j ddd¬« }t j |j
|
d d
df j |j
|
d d
df j « sJ t j |« 5 d |j
|
d
|
< d d d « |rt ddgi« }nt dd gi« }t j ||« t g d ¢gt g d¢« ¬« }|j ddd¬« }d |j
|
d
|
< |rt ddgi« }n |rt dd gi« }nt ddgi« }t j ||« y # 1 sw Y ÃxY w) Nr r_ rÓ r r Fr r r )r r )r g @r )
|
r r r8 r% Ú shares_memoryrk Ú_valuesrB rD r| )rF r rI rJ rc re rR s r) Ú test_xs_droplevel_false_viewz1TestXSWithMultiIndex.test_xs_droplevel_false_view sJ ô {¬E²/Ó,BÔ CØs ¨uÓ5ä× Ñ § ¡ ªA¨q¨DÑ 1× 9Ñ 9¸2¿7¹7Â1ÀaÀ4¹=×;PÑ;PÔQÐQÐQä × "Ñ "Ð#5Ó 6ñ Ø BGGDM÷ á ä #¨ s Ó,Hô ! #¨ s Ó,HÜ
|
× Ñ f hÔ/ô ´²oÓ0FÔ GØs ¨uÓ5Ø Ù ä #¨ s Ó,HÙ ä #¨ s Ó,Hô ! #¨ s Ó,HÜ
|
× Ñ f hÕ/÷3 ð ús ÂE%Å%E.c óà t j g d¢« }t g d¢g d¢g|¬« }t j t
|
d¬« 5 |j ddd ¬
|
« d d d « y # 1 sw Y y xY w) N))r Úmr )r Únr )r´ Úor` r )r r rû rÓ r´ r4 )r r´ Fr )rz r6 )r r r r= r> r? r8 )rF r rc s r) Ú$test_xs_list_indexer_droplevel_falsez9TestXSWithMultiIndex.test_xs_list_indexer_droplevel_false· sY ä × #Ñ #Ò$WÓ XÜ ¢9Ð-°rÔ :Ü ]]8¨3Ô /ñ 8Ø EE*¨°QEÔ 7÷ 8÷ 8ñ 8ús ÁA$Á$A-N)r r r r r« r± r¹ r½ rÀ r= ÚmarkÚ parametrizerÆ rÖ r râ rí rñ rö rø r r r r r r r r+ r) r r s< ò0ò* 0ò 0ò 0ò+ò*+ð$ [[× Ñ \Ð,=ÀÀÈ(ÈÐ?TÐ+UÓVñ 0ó Wð 0ò0ð* [[× Ñ Ø(à &Ñ-©u°e¸U°^È&Ô/QÐ RØ FÑ /±¸°uÀ6Ô1JÐ Kð
|
óñ 0óð 0ð [[× Ñ Øá =Ù 5ð
|
óñ
|
0óð
|
0ð [[× Ñ ØÑ2Ñ4IÐJóñ0óð0ò$1ò 1ò *ò("ò 0ò 0ò"0óH8r+ r )r@ Únumpyr% r= Ú pandas.errorsr Úpandasr r r r r r Úpandas._testingÚ_testingrB Úpandas.tseries.offsetsr
|
Úfixturer* r- r r r+ r) ú<module>r# sT ðÛ ã Û å .÷÷ õ å 'ð ñ >ó ð >÷ n*ñ n*÷be8ò e8r+
|
