fume-manage-python.git

U  
®ý°dûã@sRddlZddlmZddlmZddlmZddlmZm    Z    m
Z
GdddZdS)éN)Ú_period_code_map)ÚINVALID_FREQ_ERR_MSG)ÚOutOfBoundsDatetime)ÚPeriodÚ    TimestampÚoffsetsc    @sºeZdZdZej dddddddg¡d    d
ZddZd dZ    ddZ
ddZddZddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-S).ÚTestFreqConversionz)Test frequency conversion of date objectsÚfreqÚAÚQÚMÚWÚBÚDcCsXtd|d}|j|j|jf}|d}|j|jdks8t|j|j|jf}||ksTtdS)Nú
0001-01-01©r    é)rÚyearÚhourÚdayÚordinalÚAssertionErrorÚmonth)Úselfr    ÚperZtup1ÚprevZtup2©rú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/scalar/period/test_asfreq.pyÚtest_asfreq_near_zerosz(TestFreqConversion.test_asfreq_near_zerocCshtddd}tddd}| d¡}| d¡}||ks<t| dd¡|ksPt| dd¡|ksdtdS)Nrrér ÚEÚS©rÚasfreqr)rZper1Úper2Zweek1Zweek2rrrÚtest_asfreq_near_zero_weeklys
 
z/TestFreqConversion.test_asfreq_near_zero_weeklyc    Cs6tddd}d}tjt|d| ¡W5QRXdS)Nrrrz"Out of bounds nanosecond timestamp©Úmatch)rÚpytestÚraisesrZto_timestamp)rrÚmsgrrrÚtest_to_timestamp_out_of_bounds&sz2TestFreqConversion.test_to_timestamp_out_of_boundscCsltddd}| d¡}| d¡}tddd}|j|jks<t|jdksJt|j|jksZt|jd    kshtdS)
Nr
é×©r    rZ5tÚtz2007-12-31 23:59rZ5TÚT)rr#rrZfreqstr)rÚvalÚresult1Úresult2ÚexpectedrrrÚtest_asfreq_corner-s
 
z%TestFreqConversion.test_asfreq_cornerc    CsÈtddd}tddd}tddd}tddd}tdddd    }tddd
d    }tdddd}tddd d}tddddd}    tddd dd}
tddddd}tddd dd}tddddd} tddd dd}tdddddd}tddd ddd}tddddddd}tddd dddd}tdddddddd}tddd ddddd}tddddd}tddddd}tddddd}tddd dd}tddd!dd}tddd dd}| dd¡|ks¶t| dd"¡|ksÌt| dd#¡|ksât| dd$¡|ksøt| dd¡|    kst| dd$¡|
ks$t| dd¡|ks:t| dd$¡|ksPt| dd¡| ksft| dd$¡|ks|t| dd¡|kst| dd$¡|ks¨t| d%d¡|ks¾t| d%d$¡|ksÔt| d&d¡|ksêt| d&d$¡|kst| dd¡|kst| dd$¡|ks,t| dd¡|ksBt| dd$¡|ksXt| dd¡|ksnt| dd$¡|kst| dd¡|kst| dd$¡|ks°t| d¡|ksÄtdS)'Nr
r,r-úA-JANúA-JUNzA-NOVrr©r    rZquarterér©r    rrér ©r    rrrérrÚHr©r    rrrréÚMin©r    rrrrÚminuteé;r!©r    rrrrrBÚsecondéÖéréééÚeÚsr Úminr/r")rZival_AZ    ival_AJANZ    ival_AJUNZ    ival_ANOVZival_A_to_Q_startZival_A_to_Q_endZival_A_to_M_startZival_A_to_M_endZival_A_to_W_startZival_A_to_W_endZival_A_to_B_startZival_A_to_B_endZival_A_to_D_startZival_A_to_D_endZival_A_to_H_startZival_A_to_H_endZival_A_to_T_startZival_A_to_T_endZival_A_to_S_startZival_A_to_S_endZival_AJAN_to_D_endZival_AJAN_to_D_startZival_AJUN_to_D_endZival_AJUN_to_D_startZival_ANOV_to_D_endZival_ANOV_to_D_startrrrÚtest_conv_annual7s¢ÿÿÿÿz#TestFreqConversion.test_conv_annualc    CsDtdddd}tdddd}tdddd}tdddd}tddd    }td
ddd}td
ddd}td dddd}td dddd}    tddddd}
tddddd}tddddd}tddddd} tdddddd}tdddddd}tddddddd}tddddddd}tdddddddd}tdddddddd}tddddd}tddddd}tddddd}tddddd}| d¡|kst| d¡|ks t| d
d¡|ks¶t| d
d ¡|ksÌt| d d¡|ksât| d d ¡|    ksøt| dd¡|
kst| dd ¡|ks$t| dd¡|ks:t| dd ¡| ksPt| dd¡|ksft| dd ¡|ks|t| dd¡|kst| dd ¡|ks¨t| dd¡|ks¾t| dd ¡|ksÔt| dd¡|ksêt| dd ¡|kst| dd¡|kst| dd ¡|ks,t| d¡|ks@tdS)!Nrr,rr7r8úQ-JANúQ-JUNr
r-rr9ér r;r<rrHrr=rr>r?r@rArCr!rDrFrGrIé    r r")rZival_QZival_Q_end_of_yearZ
ival_QEJANZ
ival_QEJUNZival_Q_to_AZival_Q_to_M_startZival_Q_to_M_endZival_Q_to_W_startZival_Q_to_W_endZival_Q_to_B_startZival_Q_to_B_endZival_Q_to_D_startZival_Q_to_D_endZival_Q_to_H_startZival_Q_to_H_endZival_Q_to_T_startZival_Q_to_T_endZival_Q_to_S_startZival_Q_to_S_endZival_QEJAN_to_D_startZival_QEJAN_to_D_endZival_QEJUN_to_D_startZival_QEJUN_to_D_endrrrÚtest_conv_quarterly~sÿÿÿÿz&TestFreqConversion.test_conv_quarterlyc    Cstdddd}tdddd}tdddd}tddd}td    ddd
}tddddd}tdddd d}tddddd}tdddd d}    tddddd}
tdddd d}tdddddd}tdddd dd} tddddddd}tdddd ddd}tdddddddd}tdddd dddd}| d¡|ks0t| d¡|ksDt| d    ¡|ksXt| d    ¡|kslt| dd¡|kst| dd¡|kst| dd¡|ks®t| dd¡|    ksÄt| dd¡|
ksÚt| dd¡|ksðt| dd¡|kst| dd¡| kst| dd¡|ks2t| dd¡|ksHt| dd¡|ks^t| dd¡|kstt| d¡|kstdS)Nrr,rr9r:rQr
r-rr7r r;r<rrr=rr>r?r@rArCr!rDr r")rZival_MZival_M_end_of_yearZival_M_end_of_quarterZival_M_to_AZival_M_to_QZival_M_to_W_startZival_M_to_W_endZival_M_to_B_startZival_M_to_B_endZival_M_to_D_startZival_M_to_D_endZival_M_to_H_startZival_M_to_H_endZival_M_to_T_startZival_M_to_T_endZival_M_to_S_startZival_M_to_S_endrrrÚtest_conv_monthly¾sÿÿÿÿz$TestFreqConversion.test_conv_monthlyc+    CsÖtddddd}tddddd}tddddd}tdddd    d}td
dddd}tdddd d}tddddd}tddddd}tddddd}    tddddd}
tddddd}tddddd}tddddd} tdddd    d}tddddd}tddddd}tddddd}tdddd d}tddddd}tddddd}tddddd}tddddd}tddddd}tddd dd}tddddd}tddd}tdddd}tdddd}tdddddjdkrÞtddd}ntdd d}tddd ddjdkrtdddd}ntdddd}tdddddjdkrHtdddd}ntdddd}td!dddd} td!ddd    d}!tddddd}"tddddd}#td"dddd#d$}$td"dddd%d$}%td&dddd#d#d'}&td&dddd%d(d'}'td)dddd#d#d#d*}(td)dddd%d(d(d*})| d¡|ks"t| d¡|ks6t| d¡|ksJt| d¡|ks^t| d¡|ksrt| d¡|kst| d!d)¡| kst| d!d+¡|!ks²t| dd)¡|"ksÈt| dd+¡|#ksÞt| dd)¡|    ksôt| dd+¡|
ks
t| dd)¡|ks t| dd+¡|ks6t| dd)¡| ksLt| dd+¡|ksbt| dd)¡|ksxt| dd+¡|kst| dd)¡|ks¤t| dd+¡|ksºt| dd)¡|ksÐt| dd+¡|ksæt| dd)¡|ksüt| dd+¡|kst| d"d)¡|$ks(t| d"d+¡|%ks>t| d&d)¡|&ksTt| d&d+¡|'ksjt| d)d)¡|(kst| d)d+¡|)kst| d¡|ksªtt}*tjt|*d,| d-¡W5QRXdS).Nr r,rr;rIzW-SATrzW-FRIézW-THUr8zW-WEDrQzW-TUErGzW-MONrrFr:r<rHéééér
r-rr7rr9éØrr=rr>r?r@rArCr!rDr r&ÚWK)rÚweekdayr#rrr(r)Ú
ValueError)+rZival_WZ    ival_WSUNZ    ival_WSATZ    ival_WFRIZ    ival_WTHUZ    ival_WWEDZ    ival_WTUEZ    ival_WMONZival_WSUN_to_D_startZival_WSUN_to_D_endZival_WSAT_to_D_startZival_WSAT_to_D_endZival_WFRI_to_D_startZival_WFRI_to_D_endZival_WTHU_to_D_startZival_WTHU_to_D_endZival_WWED_to_D_startZival_WWED_to_D_endZival_WTUE_to_D_startZival_WTUE_to_D_endZival_WMON_to_D_startZival_WMON_to_D_endZival_W_end_of_yearZival_W_end_of_quarterZival_W_end_of_monthZival_W_to_AZival_W_to_QZival_W_to_MZival_W_to_A_end_of_yearZival_W_to_Q_end_of_quarterZival_W_to_M_end_of_monthZival_W_to_B_startZival_W_to_B_endZival_W_to_D_startZival_W_to_D_endZival_W_to_H_startZival_W_to_H_endZival_W_to_T_startZival_W_to_T_endZival_W_to_S_startZival_W_to_S_endr*rrrÚtest_conv_weeklyïsÞÿÿÿÿz#TestFreqConversion.test_conv_weeklyc    Cs.t}tjt|dtdddddW5QRXtjt|dtdddddW5QRXtjt|dtdddd    dW5QRXtjt|dtd
ddddW5QRXtjt|dtdddd dW5QRXtjt|dtdddddW5QRXtjt|dtdddddW5QRXdS)Nr&r[r,rr;zWK-SATrzWK-FRIrUzWK-THUr8zWK-WEDrQzWK-TUErGzWK-MON)rr(r)r]r)rr*rrrÚtest_conv_weekly_legacy^sz*TestFreqConversion.test_conv_weekly_legacyc    Cs`tddddd}tddddd}tddddd}tddddd}tdddd    d}td
dd}tdddd }tdddd}tddddd}    tddddd}
tdddddd}tdddddd}tddddddd} tddddddd}tdddddddd}tdddddddd}| d
¡|ks$t| d
¡|ks8t| d¡|ksLt| d¡|ks`t| d¡|kstt| d¡|kst| d¡|    kst| d¡|    ks°t| d¡|
ksÄt| dd¡|ksÚt| dd¡|ksðt| dd¡| kst| dd¡|kst| dd¡|ks2t| dd¡|ksHt| d¡|ks\tdS)Nrr,rr;r:r<rQrHrUr
r-rr7rr9r rIrr=rr>r?r@rArCr!rDr r")rZival_BZival_B_end_of_yearZival_B_end_of_quarterZival_B_end_of_monthZival_B_end_of_weekZival_B_to_AZival_B_to_QZival_B_to_MZival_B_to_WZival_B_to_DZival_B_to_H_startZival_B_to_H_endZival_B_to_T_startZival_B_to_T_endZival_B_to_S_startZival_B_to_S_endrrrÚtest_conv_businessqs|ÿÿÿÿz%TestFreqConversion.test_conv_businessc    Cstddddd}tddddd}tddddd}tddddd}tddddd}tdddd    d}tdddd
d}tddddd}tdddd    d}    tddddd}
td dd}tddd}tddd} tddd}tdddd}tdddd}tdddd}tdddd}tddddd}tdddddd}tdddddd}tddddddd }tdddddd!d }td"ddddddd#}td"ddddd!d!d#}| d ¡|ks¤t| d¡|ks¸t| d¡| ksÌt| d¡|ksàt| d ¡|ksôt| d$¡|kst| d¡|kst| d¡|ks0t| d¡|ksDt| d¡|ksXt| d¡|kslt| d¡|kst| d¡|kst| d¡|    ks¨t| dd"¡|    ks¾t| dd%¡|
ksÔt| dd"¡|    ksêt| dd%¡|
kst| dd"¡|kst| dd%¡|ks,t| dd"¡|ksBt| dd%¡|ksXt| d"d"¡|ksnt| d"d%¡|kst| d¡|kstdS)&Nrr,rr;r:r<rQrIrUrrér
r-r5rZr6zA-DECrOr8r7rPzQ-DECrr9r r=rr>r?r@rArCr!rDrr r")rZival_DZival_D_end_of_yearZival_D_end_of_quarterZival_D_end_of_monthZival_D_end_of_weekZ ival_D_fridayZival_D_saturdayZ ival_D_sundayZ ival_B_fridayZ ival_B_mondayZival_D_to_AZival_Deoq_to_AJANZival_Deoq_to_AJUNZival_Deoq_to_ADECZival_D_to_QEJANZival_D_to_QEJUNZival_D_to_QEDECZival_D_to_MZival_D_to_WZival_D_to_H_startZival_D_to_H_endZival_D_to_T_startZival_D_to_T_endZival_D_to_S_startZival_D_to_S_endrrrÚtest_conv_daily¢s ÿÿÿÿz"TestFreqConversion.test_conv_dailyc    Cstdddddd}tdddddd}tddd    ddd}tdddddd}tdddd
dd}tdddddd}tdddddd}tddd}td ddd}    tdddd}
tdddd
d}tddddd}tddddd} tddddddd}tddddddd}tdddddddd}tdddddddd}| d¡|ks>t| d¡|ksRt| d ¡|    ksft| d ¡|    kszt| d¡|
kst| d¡|
ks¢t| d¡|ks¶t| d¡|ksÊt| d¡|ksÞt| d¡|ksòt| d¡| kst| d¡| kst| dd¡|ks0t| dd¡|ksFt| dd¡|ks\t| dd¡|ksrt| d¡|kstdS)Nr=r,rrr>r:r<r?rQrIr
r-rr7rr9r r;rrr@rArCr!rDr r")rZival_HZival_H_end_of_yearZival_H_end_of_quarterZival_H_end_of_monthZival_H_end_of_weekZival_H_end_of_dayZival_H_end_of_busZival_H_to_AZival_H_to_QZival_H_to_MZival_H_to_WZival_H_to_DZival_H_to_BZival_H_to_T_startZival_H_to_T_endZival_H_to_S_startZival_H_to_S_endrrrÚtest_conv_hourlyísÿÿÿÿz#TestFreqConversion.test_conv_hourlyc    Cstddddddd}tdddddd    d}tddd
ddd    d}tdddddd    d}tdddddd    d}tdddddd    d}tdddddd    d}tdddddd    d}tddd }    tdddd}
tdddd}tddddd}tddddd} tddddd}tdddddd}tdddddddd}tddddddd    d}| d¡|    ksJt| d¡|    ks^t| d¡|
ksrt| d¡|
kst| d¡|kst| d¡|ks®t| d¡|ksÂt| d¡|ksÖt| d¡| ksêt| d¡| ksþt| d¡|kst| d¡|ks&t| d¡|ks:t| d¡|ksNt| dd¡|ksdt| dd¡|kszt| d¡|kstdS)Nr@r,rrrAr:r<r?rCrQrIr
r-rr7rr9r r;rrr=r>r!rDr r")rZival_TZival_T_end_of_yearZival_T_end_of_quarterZival_T_end_of_monthZival_T_end_of_weekZival_T_end_of_dayZival_T_end_of_busZival_T_end_of_hourZival_T_to_AZival_T_to_QZival_T_to_MZival_T_to_WZival_T_to_DZival_T_to_BZival_T_to_HZival_T_to_S_startZival_T_to_S_endrrrÚtest_conv_minutely sÆÿÿÿÿÿÿÿÿÿz%TestFreqConversion.test_conv_minutelyc    Cstdddddddd}tdddddd    d    d}tddd
ddd    d    d}tdddddd    d    d}tdddddd    d    d}tdddddd    d    d}tdddddd    d    d}tdddddd    d    d}tddddddd    d}    tddd }
tdddd}tdddd}tddddd} tddddd}tddddd}tdddddd}tddddddd}| d¡|
ksXt| d¡|
kslt| d¡|kst| d¡|kst| d¡|ks¨t| d¡|ks¼t| d¡| ksÐt| d¡| ksät| d¡|ksøt| d¡|kst| d¡|ks t| d¡|ks4t| d¡|ksHt| d¡|ks\t| d¡|kspt|     d¡|kst| d¡|kstdS)Nr!r,rrrDr:r<r?rCrQrIr
r-rr7rr9r r;rrr=r>r@rAr")rZival_SZival_S_end_of_yearZival_S_end_of_quarterZival_S_end_of_monthZival_S_end_of_weekZival_S_end_of_dayZival_S_end_of_busZival_S_end_of_hourZival_S_end_of_minuteZival_S_to_AZival_S_to_QZival_S_to_MZival_S_to_WZival_S_to_DZival_S_to_BZival_S_to_HZival_S_to_TrrrÚtest_conv_secondly]sÄÿÿÿÿÿÿÿÿz%TestFreqConversion.test_conv_secondlyc    Cstddd}|jdkst|j}td}||ks4t|j|jdksHttdd}d}tjt|d|jW5QRXtjt|d|j    W5QRXdS)    Nz2020-01-30 15:57:27.576166ÚUrlf    vu,ièz
2300-01-01Úusr&)
rrrÚ
start_timerÚ_valuer(r)rZend_time)rrÚstartr3r$r*rrrÚtest_conv_microseconds
z(TestFreqConversion.test_conv_microsecondcCstddd}dt d¡fD]F}| |¡}tddd}||ks@t|j|jksPt|j|jkstqdt d¡fD]J}|j|dd    }tddd}||kst|j|jksªt|j|jksptqptddd}dt ¡fD]H}| |¡}td
dd}||ksút|j|jkst|j|jksÔtqÔdt ¡fD]R}|j|dd    }tddd}||ksVt|j|jksht|j|jks*tq*tddd}dt d¡fD]N}| |¡}td dd}||ksÀt|j|jksÒt|j|jkstqdt d¡fD]R}|j|dd    }tddd}||ks"t|j|jks4t|j|jksötqötddd}dt d¡fD]N}| |¡}tddd}||kst|j|jkst|j|jksdtqddt d¡fD]R}|j|dd    }tddd}||ksît|j|jkst|j|jksÂtqÂdS)Nr
r,r-Z3ArQÚ2007rr!©ÚhowZ2009Z2MrGz2007-12z2007-01z2009-12)rrZYearEndr#rrr    ZMonthEnd)rÚpr    Úresultr3rrrÚtest_asfreq_mult¨sh
 
 
 
z#TestFreqConversion.test_asfreq_multc
Cstddd}tddd}tddgddgD]B\}}|j||d    }||ksLt|j|jks\t|j|jks*tq*tdd
d}tdd
d}| d¡}| d¡}    tddd}||ks²t|j|jksÂt|j|jksÒt|    |ksÞt|    j|jksît|    j|jksþt|jddd    }|jddd    }    td dd}||ks4t|j|jksFt|j|jksXt|    |ksft|    j|jksxt|    j|jkstdS)Nrlr=rZ25HZ1D1HZ1H1Dr r!rmr,r-z
2007-01-02z
2007-01-01)rÚzipr#rrr    )
rror3r    rnrpÚp1Úp2r1r2rrrÚtest_asfreq_combinedïs6
 
z'TestFreqConversion.test_asfreq_combinedc    Cstd}|jdddtddks$tt}tjt|d|jdddW5QRXtjt|dtddW5QRXt d¡dkstdS)NZ2013rr!)r    rnz2013-01r&ZMS)    rr#rrr(r)r]rÚget)rÚinitialr*rrrÚtest_asfreq_MSsz!TestFreqConversion.test_asfreq_MSN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r(ÚmarkZparametrizerr%r+r4rNrSrTr^r_r`rbrcrdrerkrqrurxrrrrrs(
 
 
 
G@1o1K3=9G&r)r(Zpandas._libs.tslibs.dtypesrZpandas._libs.tslibs.periodrZ pandas.errorsrZpandasrrrrrrrrÚ<module>s