U
|
ý°d\ ã @ sª d dl Z d dlZd dlZd dlZd dlmZmZmZm Z d dl
|
m Z ej dd Zej dd Zej dd Zej d d
|
ZG d d d ZG d d dZG dd dZdS )é N)Ú DataFrameÚlreshapeÚmeltÚ wide_to_longc C sD t ¡ d d
} | d dk tj¡| d< | d dk tj¡| d< | S )Né
|
ÚAr Úid1ÚBÚid2)ÚtmZmakeTimeDataFrameÚastypeÚnpÚint64©Úres© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reshape/test_melt.pyÚdf s r c C sB t dddgdddgddd gg} td
|
td g| _d d g| j_| S )Nç5ì÷Ä:ñ?çS°ÆÙtÄñ¿gKÔÔ²µÊ?çU¼y$õ¿çê> ©M×?gK®âð¿çiÿ¬Õé¿ç<Ú8b->µ?ghBÄòë¿ÚABCÚabcÚCAPÚlow)r ÚlistÚcolumnsÚnamesr r r r Údf1 s ýÿ r! c C s dS )NÚvarr r r r r Úvar_name& s r# c C s dS )NÚvalr r r r r Ú
|
value_name+ s r% c @ sÞ e Zd Zdd Zdd Zdd Zdd Zej d e
|
e e j f¡d
|
d Zd d Zej ddgdgdedddddddddddddfdgdgd eddddddddddddd fg¡d d Zej d d!d"ggd!gd"gd!d"gg¡d#d$ Zd%d& Zd'd( Zd)d* Zej d+dd,g¡d-d. Zd/d0 Zej d1e ejd2d3d4d5¡ejddd6dd7gd8d9e dd dddg¡g¡d:d; Zd<d= Z d>d? Z d@dA Z dBdC Z dDdE Z dFdG Z!dHdI Z"dJdK Z#ej dLdMdNg¡dOdP Z$dQS )RÚTestMeltc C s" t |}|j ¡ ddgks td S )NÚvariableÚvalue©r r ÚtolistÚAssertionError)Úselfr Úresultr r r Útest_top_level_method1 s z TestMelt.test_top_level_methodc C s t | ¡ t|¡ t |jddgddgdt|ddgddgd¡ t |j||dt|||d¡ t |jddt|dd¡ d S ) Nr r
|
r r ©Úid_varsÚ
|
value_vars©r# r% r ©Ú col_level)r Úassert_frame_equalr )r, r r! r# r% r r r Útest_method_signatures5 s þ þz TestMelt.test_method_signaturesc C sr | ¡ }|j ¡ ddgks t|j dgd}|j ¡ dddgksDt|j ddgd}|j ¡ ddddgksntd S )Nr' r( r )r0 r
|
r) )r, r r- Zresult1Zresult2r r r Útest_default_col_namesD s z TestMelt.test_default_col_namesc C s¢ |j ddgdd}t|dks"t|j ddgddgd}t|d ¡ d |d ¡ d dgd dgd |d ¡ |d ¡ dddd d
|
gd }t ||¡ d S ) Nr r
|
r r/ r r é ©r r
|
r' r( r' r( ©r )r Úlenr+ r r* r r5 )r, r Zresult3Zresult4Z expected4r r r Útest_value_varsN s ü
|
ù zTestMelt.test_value_varsÚtype_c C s t |d ¡ d |d ¡ d dgd dgd |d ¡ |d ¡ ddddd gd
|
}|jddg|d d }t ||¡ d S ) Nr r8 r
|
r r r r9 r' r( r: ©r r r/ )r r* r r r5 )r, r= r Úexpectedr- r r r Útest_value_vars_types^ s ü
|
ù z TestMelt.test_value_vars_typesc C s\ t |d dgt| dgt| |d dddddgd }|jdgdgd
|
}t ||¡ d S ) N©r Úar Úb©r rC )rA r r r( r r r( r: r/ )r r; r r r5 )r, r! r? r- r r r Ú test_vars_work_with_multiindexm s ü
|
ù
|
z'TestMelt.test_vars_work_with_multiindexz(id_vars, value_vars, col_level, expectedr r r r r r ©r é r8 r r r )r r r( rB rC rG )rB r r( c C s |j |||d}t ||¡ d S )Nr3 )r r r5 )r, r0 r1 r4 r? r! r- r r r Ú%test_single_vars_work_with_multiindex{ s z.TestMelt.test_single_vars_work_with_multiindexzid_vars, value_varsrA rD c C s0 d}t jt|d |j||d W 5 Q R X d S )NzF(id|value)_vars must be a list of tuples when columns are a MultiIndex©Úmatchr/ )ÚpytestÚraisesÚ
|
ValueErrorr )r, r0 r1 r! Úmsgr r r Ú$test_tuple_vars_fail_with_multiindex s z-TestMelt.test_tuple_vars_fail_with_multiindexc C s, |j |d}|j ¡ ddgks"t|j dg|d}|j ¡ dddgksJt|j ddg|d}|j ¡ ddddgksvt|j ddgd|d}|j ¡ ddddgks¤t|j ddgdd g|d}td|d ¡ d
|
d|d ¡ d
|
|dgd d gd d|d ¡ |d ¡ idd|dgd }t ||¡ d S ) N)r# r" r( r )r0 r# r
|
r )r0 r1 r# r r8 r r: ©r r r* r+ r r r5 ) r, r r# Zresult5Zresult6Zresult7Zresult8Zresult9Z expected9r r r Útest_custom_var_name® s4 ÿ ü
|
ù z TestMelt.test_custom_var_namec C s, |j |d}|j ¡ ddgks"t|j dg|d}|j ¡ dddgksJt|j ddg|d}|j ¡ ddddgksvt|j ddgd|d}|j ¡ ddddgks¤t|j ddgdd g|d}td|d ¡ d
|
d|d ¡ d
|
ddgd d gd ||d ¡ |d ¡ iddd|gd }t ||¡ d S ) N)r% r' r$ r ©r0 r% r
|
r )r0 r1 r% r r8 r r: rP ) r, r r% Zresult10Zresult11Zresult12Zresult13Zresult14Z
|
expected14r r r Útest_custom_value_nameÉ s< ÿ ÿ ü
|
ù z TestMelt.test_custom_value_namec C sf |j ||d}|j ¡ ddgks$t|j dg||d}|j ¡ dddgksNt|j ddg||d}|j ¡ ddddgks|t|j ddgd||d}|j ¡ ddddgks¬t|j ddgdd g||d}td|d ¡ d
|
d|d ¡ d
|
|dgd d gd ||d ¡ |d ¡ idd||gd } t || ¡ | ¡ }
|
d |
|
j_|
|
¡ } | j ¡ d dgksbtd S )Nr2 r" r$ r ©r0 r# r% r
|
r )r0 r1 r# r% r r8 r r: Úfoor( ) r r r* r+ r r r5 ÚcopyÚname) r, r r% r# Zresult15Zresult16Zresult17Zresult18Zresult19Z
|
expected19Zdf20Zresult20r r r Ú test_custom_var_and_value_nameæ sP ÿüü ü
|
ù z'TestMelt.test_custom_var_and_value_namer4 r c C s& |j |d}|j ¡ ddgks"td S )Nr3 r r( r) )r, r4 r! r r r r Útest_col_level s zTestMelt.test_col_levelc C s$ | ¡ }|j ¡ dddgks td S )Nr r r( r) )r, r! r r r r Útest_multiindex s zTestMelt.test_multiindexÚcolZ2010é z
|
US/Pacific)ZperiodsÚtzÚcÚdÚcategory©Údtypec C s² t td|dddddg|d}tjt dddddg¡|gdd}t|ddgd d
|
d }t ttdd tj|gd ddd gd dgd |d}ddd d
|
g|_t ||¡ d S )Nr\ rG r )Úklassr[ Úattr1Úattr2T©Z ignore_indexrc r[ Ú attributer( rT r8 rd re )r rG r8 é )
|
r ÚrangeÚpdÚconcatÚSeriesr r r r r5 )r, r[ r Zexpected_valuer- r? r r r Útest_pandas_dtypes s&
|
ÿ" ÿüÿzTestMelt.test_pandas_dtypesc C s^ t ddgt ddg¡d}t|dgdg}t t ddg¡ddgddgd}t ||¡ d S ) NrG r8 ÚXÚYr> r r )r r' r( )r rj Z Categoricalr r r5 )r, Údatar- r? r r r Útest_preserve_category5 s ÿz TestMelt.test_preserve_categoryc C sT t tj dd¡tdd}d}tjt|jdddd | d
|
d gd d g¡ W 5 Q R X tjt|jdddd | dd gdd g¡ W 5 Q R X tjt|jdddd | d
|
d ddgdd g¡ W 5 Q R X |
|
¡ }tdtdg|_ tjt|jdddd | dgdg¡ W 5 Q R X tjt|jdddd |j dgdgdd W 5 Q R X d S ) Nr\ é Zabcdr: z=The following '{Var}' are not present in the DataFrame: {Col}r1 z\['C'\])ZVarZColrI rB rC ÚCr_ r0 z\['A'\]r r^ z\['not_here', 'or_there'\]Znot_hereZor_thereÚABCDz\['E'\])ÚErB rD z\['F'\]ÚFr r3 ) r r ÚrandomÚrandnr rK rL ÚKeyErrorÚformatr rV r )r, r rN Zmultir r r Ú test_melt_missing_columns_raises? s. ÿ þ" ÿz)TestMelt.test_melt_missing_columns_raisesc C sf t dgdgdgdgd}t|ddgdd gd
|
}t dgd dgd td ddgd }t ||¡ d S ) NrU ÚbarrG r8 )r rB rC r_ r rB rC r_ r/ Zbd)r rB r' r( )r r r r r5 ©r, r r- r? r r r Ú test_melt_mixed_int_str_id_varsd s ÿz(TestMelt.test_melt_mixed_int_str_id_varsc C sH t dgdgd}t|ddgd}t ddgddgd}t ||¡ d S )NrU r| )r rB r rB )r1 ©r' r( ©r r r r5 r} r r r Ú"test_melt_mixed_int_str_value_varsm s z+TestMelt.test_melt_mixed_int_str_value_varsc C sR t dgdgddgd}t|dd}t dd gddgd
|
ddgd}t ||¡ d S ) Nr rG ©rU r| Úfirst©ÚindexFrf rU r| r r r} r r r Útest_ignore_indext s ÿzTestMelt.test_ignore_indexc C s t jjddgddgd}tddgdd gd
|
|d }t|d d }t jjddgd ddgd}tdgd dgd dddd gd|d }t ||¡ d S )N)r Úsecond)r ÚthirdÚbazZfoobar)r r rG r8 rh r r Frf rU r| r )rj Z
|
MultiIndexÚ from_tuplesr r r r5 ©r, r
r r- Zexpected_indexr? r r r Útest_ignore_multiindex} s ÿ
|
ÿ þz TestMelt.test_ignore_multiindexc C s t jddgddd}tddgdd gd
|
|d }t|d d }t jddgd ddd}tddddgdddd gd|d }t ||¡ d S )NrU r| r` r )rb rW r rG r8 rh )ÚxÚyr Frf r r r )rj ÚIndexr r r r5 r r r r Ú test_ignore_index_name_and_type s þz(TestMelt.test_ignore_index_name_and_typec C s^ t dddggdddgd}|jdgdgd}t dddgdddggddd gd}t ||¡ d S )
|
NÚidr8 rh rB rC r: r/ r' r( r r} r r r Ú test_melt_with_duplicate_columns s ÿz)TestMelt.test_melt_with_duplicate_columnsrb ÚInt8ZInt64c C sj t tjddgddtjddg|dd}| ¡ }t ddd d gtjddddg|dd
|
}t ||¡ d S ) NrG r8 r ra rh rr ©rB rC rB rC r )r rj rl r r r5 )r, rb r r- r? r r r Útest_melt_ea_dtype¦ s þÿ
|
þÿzTestMelt.test_melt_ea_dtypeN)%Ú__name__Ú
|
__module__Ú __qualname__r. r6 r7 r< rK ÚmarkÚ parametrizeÚtupler r Úarrayr@ rE r rH rO rQ rS rX rY rZ rj rl Z
|
date_rangerm rq r{ r~ r r r r r r r r r r r& 0 s
|
|
|
|
|
ýÿü
|
|
|
ýÿüóþ
|
ýþ
|
*
|
ýþ
|
|
% r& c @ s e Zd Zdd ZdS )Ú TestLreshapec C sÒ dddddgdddd d
|
gd d d ddgdddddgdddddgdt jdddgdt jt jdd gd d d d d!gd"t jd#d$d%gd&t jt jd'd(gd)
|
}t|}d*d+ td,d-D d.d+ td,d-D d/}t||}ddddddddddddg dddd d
|
ddd d
|
dd d
|
g d d d ddd d ddd ddg ddddddddddddg dddddddddddd g d0d$d1d2d3d"d#d$d%d&d'd(g d4}t||jd5}t ||¡ t||d6d7}dddddddddddddddgdddd d
|
dddd d
|
dddd d
|
gd d d ddd d d ddd d d ddgdddddddddddddddgddddddt jddddt jt jdd gd0d$d1d2d3d"t jd#d$d%d&t jt jd'd(gd4}t||jd5}t ||¡ d8d+ td,d9D d:d+ td,d-D d/}d;}tj t
|
|d< t|| W 5 Q R X d S )=NZ 08jan2009Z 20dec2008Z 30dec2008Z 21dec2008Z 11jan2009iæ iå i® iC i% ée éf ég éh éi ZMaleZFemaleZ 22dec2008Z 04jan2009Z 29dec2008Z 20jan2009Z 21jan2009Z 22jan2009Z 31dec2008Z 03feb2009Z 05feb2009Z 02jan2009Z 15feb2009i i
|
i iâ iÒ g l@g @g ª@g ß±@g ê¡@g bª@g Ų@)
|
ÚbirthdtÚbirthwtr ÚsexZvisitdt1Zvisitdt2Zvisitdt3Zwt1Zwt2Zwt3c S s g | ]}d |dqS ©Úvisitdtr_ r ©Ú.0Úir r r Ú
|
<listcomp>× s z+TestLreshape.test_pairs.<locals>.<listcomp>rG rr c S s g | ]}d |dqS ©Úwtr_ r r¨ r r r r« Ø s )r§ r g |@g 4@g Ä©@g Ò°@)r£ r¤ r r¥ r§ r r: F)Zdropnac S s g | ]}d |dqS r¦ r r¨ r r r r« s rh c S s g | ]}d |dqS r¬ r r¨ r r r r« s z$All column lists must be same lengthrI ) r Únanr ri r r r r5 rK rL rM )r, rp r Úspecr- Úexp_dataÚexprN r r r Ú
|
test_pairsº s û û êþ
|
ôôôôôÆI ññññññªh þzTestLreshape.test_pairsN)r r r r² r r r r r ¹ s r c @ s¶ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zej d%d&d'g¡d(d) Zd*S )+ÚTestWideToLongc
|
C sô t j d¡ t j d¡}tddddddd dd
|
d d dd dddtttd|d}|j|d< | ¡ | ¡ dddddd gd
|
d d d ddgddddddgddddddgd}t|}|
|
ddg¡dddg }t |ddgddd }t ||¡ d S ) Né{ rh rB rC r^ rF r_ ÚeÚfç @ç333333ó?çffffffæ?ç @çÍÌÌÌÌÌô?ç¹?)ZA1970ZA1980ZB1970ZB1980rn r é² é¼ r rG r8 ©rn r r Úyearr rÀ rn r r ©rª Új©r rw Úseedrx r ÚdictÚzipri r
r* Ú set_indexr r r5 ©r, r r r° r? r- r r r Ú test_simple s*
|
|
|
|
ûÿ
|
ûzTestWideToLong.test_simplec C s\ t dddddgdddd d
|
gg}d d d ddg|_ddg}t||d dd |ddgksXtd S )Nr rG r8 rh é rr r\ é é é r Zinc1Zinc2Zedu1Zedu2ÚincÚeduÚagerÁ )r r r r+ )r, r Zstubsr r r Ú
|
test_stubs· s
|
zTestWideToLong.test_stubsc
|
C sö t j d¡ t j d¡}tddddddd dd
|
d d dd dddtttd|d}|j|d< | ¡ | ¡ dddddd gd
|
d d d ddgddddddgddddddgd}t|}|
|
ddg¡dddg }t |ddgddd d }t ||¡ d S ) Nr´ rh rB rC r^ rF r_ rµ r¶ r· r¸ r¹ rº r» r¼ )zA.1970zA.1980zB.1970zB.1980rn r r½ r¾ r rG r8 r¿ rÀ rn r r Ú.©rª r Úseprà rÈ r r r Útest_separating_characterÁ s*
|
|
|
|
ûÿ
|
ûz(TestWideToLong.test_separating_characterc
|
C sô t j d¡ t j d¡}tddddddd dd
|
d d dd dddtttd|d}|j|d< | ¡ | ¡ dddddd gd
|
d d d ddgddddddgddddddgd}t|}|
|
ddg¡dddg }t |ddgddd }t ||¡ d S ) Nr´ rh rB rC r^ rF r_ rµ r¶ r· r¸ r¹ rº r» r¼ )zA(quarterly)1970zA(quarterly)1980zB(quarterly)1970zB(quarterly)1980rn r r½ r¾ r rG r8 )rn ú A(quarterly)ú B(quarterly)rÀ r rÀ rn rÖ r× rÁ rà rÈ r r r Útest_escapable_charactersÛ s.
|
|
|
|
ûÿ
|
û ÿz(TestWideToLong.test_escapable_charactersc C s¬ t ddgddgddgddgd }|j|d
|
< ddddgddddgddtjtjgd d d d gd d ddgd}t |}| d
|
dg¡dddg }t|ddgd
|
dd}t ||¡ d S )Nç ð?ç @ç @ç @ç @ç @ÚX1ÚX2©ÚA2010ÚA2011ÚB2010rn r r rG iÚ iÛ )rn r r r rÀ rÀ rn r r rÁ )r r
r r® rÇ r r r5 ©r, r r° r? r- r r r Útest_unbalancedö s$ üÿ
|
|
|
|
|
ûz TestWideToLong.test_unbalancedc C s t dddgdddgdddgd d
|
d gd d dgdddgdddgdddgd}|j|d< t ddddddgddddddgddddddgdddd d
|
d gd d ddddgdd d dd d gddddddgd}| ddg¡d d d d d g }t|d d d gddd!}t |jd d"|jd d"¡ d S )#NÚa11Úa22Úa33Úa21Úa23Úb11Úb12Úb13Úb21Úb22Úb23rG r8 rh rr r\ rË é[ é\ é] )ÚA11ÚA12ÚB11ÚB12ÚBB11ÚBB12ÚBBBXÚBBBZr r é é )rû rü r r ÚBBr rÀ rÀ rû rü r r rÿ rÁ ©Zaxis©r r
rÇ r r r5 Ú
|
sort_index©r, r r? r- r r r Útest_character_overlap s2 øÿ
|
ùÿ z%TestWideToLong.test_character_overlapc C sÄ d}t ddgddgddgdd gd
|
}|j|d < d g g g g g g g d }t | dtji¡}| d dg¡ddddddg }|jjddgdd|_t|ddgd d|d}t |j
|
dd|j
|
dd¡ d S )Nznope!rÙ rÚ rÛ rÜ rÝ rÞ rß rà rá r Ú )rn râ rã rä r rÀ r r rÀ rn râ rã rä r r r rG ©ÚlevelrÓ r © r r
r r r rÇ Ú
|
set_levelsr r r5 r )r, rÔ r r° r? r- r r r Útest_invalid_separator+ s2 üÿ
|
ø
|
ÿz%TestWideToLong.test_invalid_separatorc C s t dddgdddgdddgd d
|
d gd d dgdddgdddgdddgd}|j|d< t ddddddgddddddgddddddgdddd d
|
d gd d ddddgdd d dd d gddddddgd}| ddg¡d d d d d g }t|d d d gddd!}t |jd d"|jd d"¡ d S )#Nrç rè ré rê rë rì rí rî rï rð rñ rG r8 rh rr r\ rË rò ró rô )rõ rö r÷ rø rù rú ÚAratingÚ Arating_oldr r rý rþ )r r r r rÿ r rÀ rÀ r r r r rÿ rÁ r r r r r r Ú test_num_string_disambiguationI s6 øÿ
|
ùÿ ÿz-TestWideToLong.test_num_string_disambiguationc C s® t ddgddgddgddgd }|j|d
|
< d g g g g g g g d }t | d tji¡}| d
|
d g¡}|jjddgdd|_t|ddgd
|
d d}t |j
|
dd|j
|
dd¡ d S )NrÙ rÚ rÛ rÜ rÝ rÞ rß rà )ÚAoneÚAtwoÚBonern r r )rn r r r r rÀ r r rÀ r rG r r r rÁ r r rå r r r Útest_invalid_suffixtypej s, üÿ
|
ø
|
z&TestWideToLong.test_invalid_suffixtypec C s6 t dddddddddg dddddddddg dddddddd d
|
g d d dd dddd dg d}t dd dd dddd ddddddd d d
|
dgddddddddddddddddddgddddddddddddddddddgddddddddddddddddddgd}| dddg¡dg }t|dddgdd}t ||¡ d S )NrG r8 rh gffffff@g333333@g@gÍÌÌÌÌÌü?gffffffþ?gffffff@çÍÌÌÌÌÌ @g333333 @gffffff@rº g333333@gffffff
|
@)ÚfamidÚbirthZht1Zht2rÚ )Úhtr r rÐ r r rÐ r rÁ ©r rÇ r r r5 r r r r Útest_multiple_id_columns sF üÿî&&&éÿz'TestWideToLong.test_multiple_id_columnsc C sd t dddddgdddddgdddddgd}d}tjt|d t|d d
|
gd d d W 5 Q R X d S )NrG r8 rh rr r\ )ZA_A1ZB_B1r z3the id variables need to uniquely identify each rowrI ZA_AZB_Br ÚcolnamerÁ ©r rK rL rM r ©r, r rN r r r Útest_non_unique_idvars° s (ÿz%TestWideToLong.test_non_unique_idvarsc
|
C s¤ t dddgdddgddd gd
|
d d gd ddgd}t ddddddgddd d
|
d d gddddddgd ddd ddgd ddg¡}t|ddgdddd}t ||¡ d S )Nz CCH Pounderz Johnny DeppzChristoph WaltzzJoel David Moorez Orlando Bloomz Rory Kinnearg @@g ã@g |Å@g @@g ³@g x@ZAvatarzPirates of the CaribbeanZSpectre)Zactor_1Zactor_2Zactor_fb_likes_1Zactor_fb_likes_2ÚtitlerG r8 )ÚactorÚactor_fb_likesÚnumr r r r r Ú_rÓ r r r r r Útest_cast_j_intº sJ ûÿ
|
úúõÿë ÿz TestWideToLong.test_cast_j_intc C sX t ddgddgddgddgd }d
|
}tjt|d t|d d gd dd W 5 Q R X d S )NrÙ rÚ rÛ rÜ rÝ rÞ rß rà )râ rã rä r z,stubname can't be identical to a column namerI r r r rÁ r r r r r Útest_identical_stubnamesá s üÿz'TestWideToLong.test_identical_stubnamesc C s t ddgddgddgddgd }t ddddgd
|
d
|
d d gddtjtjgddddgd }| d dg¡}t|ddgd dddd}t ||¡ d S )NrÙ rÚ rÛ rÜ rÝ rÞ rß rà )Ztreatment_placeboZtreatment_testZresult_placebor ZplaceboÚtest©r r r- Ú treatmentr r r- r% z[a-z]+r ©rª r ÚsuffixrÔ ©r r r® rÇ r r r5 r r r r Útest_nonnumeric_suffixî s0 üÿ
|
|
|
üÿ ÿz%TestWideToLong.test_nonnumeric_suffixc C s t ddgddgddgddgd d
|
gd }t ddddgd d d d gddddgddd d
|
gd ddg¡}t|ddgddddd}t ||¡ d S )Nrß rà r rÍ rÝ rÞ rÙ rÚ rÛ rÜ )r Úresult_1Z
|
result_fooZ treatment_1Z treatment_fooÚ1rU ç ç "@r$ r r r- r% z.+r r& r r r r r Útest_mixed_type_suffix s4 ûÿ
|
|
|
|
üÿù ÿz%TestWideToLong.test_mixed_type_suffixc C sÀ t ddgddgddgddgd d
|
gd }t d d
|
d d
|
d d
|
d d
|
gd d ddd d ddgddddtjtjtjtjgtjtjtjtjddddgd}| ddg¡}t|ddgddddd}t ||¡ d S )NrÙ rÚ rÛ rÜ rÝ rÞ r rÍ rß rà )z treatment_1.1z treatment_2.1z
|
result_1.2r* r r¸ gñ?r r, r- r$ r r r- r% z[0-9.]+r r& r( r r r r Útest_float_suffix s2 ûÿ üÿ ÿz TestWideToLong.test_float_suffixc
|
C s ddddddddd d
|
d dd d d dddddddddddddddd}t |¡}t|dgddgdd}t|dddgdd}t ||¡ d S ) Nr rG r8 rh rr )r rG r8 rh rr gé?r, g Ð?rÙ gìQ¸
ëé?g®Gáz®ç?gìQ¸
ëá?g\Âõ(\ï?g333333ã?g¤p=
|
×£è?g{®Gázä?g¤p=
|
×£à?gq=
|
×£på?gÃõ(\ÂÕ?r¹ )Únode_idr ZPA0ZPA1ZPA3ZPAr0 r Útime)Ú stubnamesrª r )r Ú from_dictr r r5 )r, Z wide_dataZwide_dfr? r- r r r Ú test_col_substring_of_stubname4 s û
|
z-TestWideToLong.test_col_substring_of_stubnamec C sL t tdtdddd}tjtt d¡d |jddd W 5 Q R X d S )
|
Nr r é r8 )r[ r( z value_name (value) cannot matchrI r( rR ) r r ri rK rL rM ÚreÚescaper )r, r r r r Ú test_raise_of_column_name_valueD s ÿz.TestWideToLong.test_raise_of_column_name_valuerb ÚOÚstringc C s¬ t ddgddgddgd}| d|i¡}t|d d
|
gdd d d }tjddddgdd}t ddddgtjgd d|d}|jjd |¡}|jj |dd|_t
|
||¡ d S )Nr+ Ú2éd éÈ i, i )r za-1za-2r rB rC r ú-)r2 rª r rÔ )r+ rG )r; rG )r+ r8 )r; r8 )r r )rW rr r r r r ) r r r rj r r r® r
Úlevelsr r r5 )r, rb r r- r
r? Z new_levelr r r Útest_missing_stubnameO s( û
|
þþz$TestWideToLong.test_missing_stubnameN)r r r rÉ rÑ rÕ rØ ræ r r
|
r r r r r! r" r) r. r/ r4 r8 rK r r r@ r r r r r³ s(
|
! )
|
' r³ )r6 Únumpyr rK Zpandasrj r r r r Zpandas._testingZ_testingr Zfixturer r! r# r% r& r r³ r r r r Ú<module> s(
|
|
|
|
e
|
