U
|
ý°d5 ã @ sH d dl mZm Z d dlZd dlZd dlZd dlZd dlm m Z
|
d dl m Z m Z mZmZmZmZmZ d dlmZ d dlmZmZ d dlmZmZm Z m Z zd dl m Z W n e k
|
r¾ Y nX e !d¡ e !d¡Z"d d
|
Z#d d Z$G d d dZ%G dd dZ&G dd dZ'G dd dZ(ej) *dddddddd d d g ¡d d Z+dS )!é )ÚdateÚdatetimeN)ÚIndexÚPeriodÚ PeriodIndexÚSeriesÚ TimestampÚarraysÚ
|
date_range)Ú deregister_matplotlib_convertersÚ register_matplotlib_converters)ÚDayÚMicroÚMilliÚSecond)Ú converterúmatplotlib.pyplotúmatplotlib.datesc C s d} t jd| g}t |¡ d S )Nzúimport matplotlib.units as units; import matplotlib.dates as mdates; n_conv = len(units.registry); import pandas as pd; pd.plotting.register_matplotlib_converters(); pd.plotting.deregister_matplotlib_converters(); assert len(units.registry) == n_convú-c)ÚsysÚ
|
executableÚ
|
subprocessÚ check_output)ÚcodeÚcall© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/test_converter.pyÚtest_registry_mpl_resets. s ÿ r c C s t d¡t d¡kstd S )Nz00:01)r Ztime2numÚAssertionErrorr r r r Ú test_timtetonum_accepts_unicode= s r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestRegistrationc C s& d}t jd|g}t |¡dks"td S )Nzwimport matplotlib.units; import pandas as pd; units = dict(matplotlib.units.registry); assert pd.Timestamp not in unitsr r )r r r Ú
|
check_callr )Úselfr r r r r Ú test_dont_register_by_defaultB s ÿ z.TestRegistration.test_dont_register_by_defaultc C sP t d¡}ttdtdddd}| ¡ \}}t | |j|j ¡ |
|
¡ d S )Nr é Ú2017©Zperiods©Úindex) ÚpytestÚ importorskipr Úranger
|
Úsubplotsr Úplotr( ÚvaluesÚclose)r" ÚpltÚsÚ_Úaxr r r Útest_registering_no_warningM s
|
z,TestRegistration.test_registering_no_warningc C sd t d¡}ttdtdddd}t d ¡}| ¡ W 5 Q R X zt |dksRt
|
W 5 | ¡ X d S )Nr r$ r% r& r' r ) r) r* r r+ r
|
ÚtmZassert_produces_warningr- r/ Úlenr )r" r0 r1 Úwr r r Útest_pandas_plots_registerW s
|
z+TestRegistration.test_pandas_plots_registerc
|
C sZ t d¡}t dd¡: t dd¡ t|jks4tW 5 Q R X t|jksLtW 5 Q R X d S )Númatplotlib.unitsú'plotting.matplotlib.register_convertersTF)r) r* ÚcfÚoption_contextr Úregistryr )r" Úunitsr r r Útest_matplotlib_formattersc s
|
|
z+TestRegistration.test_matplotlib_formattersc C s t d¡ t dd¡}t d¡}ttdtdddd}| ¡ \}}| | |j |j
|
¡ W 5 Q R X t | | |j |j
|
¡ W 5 Q R X | ¡ d S )Nr r: Fr$ r% r& r' ) r) r* r; r< r r+ r
|
r, r- r( r. r r/ )r" Úctxr0 r1 r2 r3 r r r Útest_option_no_warningo s
|
|
z'TestRegistration.test_option_no_warningc
|
C s¦ t d¡}t d¡}t|j}zZ|j ¡ | ¡ }||jt< ||jt< t |jt |k s^t
|
t |jt |ksvt
|
W 5 |j ¡ | ¡ D ]\}}||j|< qX d S )Nr9 r ) r) r* Údictr= ÚclearÚitemsZ DateConverterr r r r r )r" r> ÚdatesÚoriginalÚkÚvZdate_converterr r r Útest_registry_resets s
|
|
|
|
|
|
|
z%TestRegistration.test_registry_resetsN) Ú__name__Ú
|
__module__Ú __qualname__r# r4 r8 r? rA rI r r r r r A s
|
r c @ s e Zd Zejdd Zdd Zdd Zdd Zd d
|
Z ej
|
d d d dddg¡dd Z ej
|
dd¡dd Z ej
|
de e edg¡dd Zdd Zd S ) ÚTestDateTimeConverterc C s t ¡ S ©N)r ZDatetimeConverter©r" r r r Údtc s zTestDateTimeConverter.dtcc C s0 | dd d ¡}| dd d ¡}||ks,tdd S )Nz2000-01-01 12:22z/DatetimeConverter.convert should accept unicode©Úconvertr )r" rP Úr1Úr2r r r Ú test_convert_accepts_unicode s z2TestDateTimeConverter.test_convert_accepts_unicodec C s¼ | dgd d ¡d }t tddd¡}||ks2t| dd d ¡}||ksLt| tdddd d ¡}||ksnt| dd d ¡}||kst| tdd d ¡}||ks¦t| dd d ¡}||ksÀt| dd d ¡}||ksÚt| t ddg¡d d ¡}|d |ksttd d¡
|
d ¡}| |d d ¡}||ks4t| | ¡ d d ¡}||ksTt| t |t d |gd d ¡}|d |kst| t |t d |g ¡ d d ¡}|d |ks¸td S )
|
Nú2012-1-1r éÜ é ú
|
2012-01-01ú2012-01-01 00:00:00+0000z2012-01-02 00:00:00+0000ÚUTCz
|
US/Eastern)rR rE Údate2numr r r r ÚnpÚarrayZ tz_localizeZ
|
tz_convertZ to_pydatetimer r )r" rP ÚrsÚxpÚtsr r r Útest_conversion¤ s< ý "z%TestDateTimeConverter.test_conversionc C s d}| tdddd d ¡}tj tddd¡}tj|||d | tdddd d ¡}tj|||d | tdd d d d
|
d d d ¡}tj|||d d S ) NgÖ&è .>z2012-1-1 01:02:03r[ )Útz©Úrtolz2012-1-1 09:02:03zAsia/Hong_KongrW rX é é )rR r r Úmdatesr\ r5 Úassert_almost_equalr )r" rP re r_ r` r r r Útest_conversion_floatÒ s
|
ÿ z+TestDateTimeConverter.test_conversion_floatc C sà t dddt dddg}| |d d ¡}tj |¡}t ||¡ | |d d d ¡}tj |d ¡}||kslttddddtddddg}| |d d ¡}tj |¡}t ||¡ | |d d d ¡}tj |d ¡}||ksÜtd S )Ni rX rf r r$ ) r rR r rh r\ r5 Zassert_numpy_array_equalr r )r" rP r. r_ r` r r r Ú$test_conversion_outofbounds_datetimeá s z:TestDateTimeConverter.test_conversion_outofbounds_datetimeztime,format_expected)r z00:00)góþÿÿõ@z23:59:59.999999)i_ z01:00)i z01:02:03)gfffffeã@z 11:02:03.200c C s t d ¡|}||kstd S rN )r Z TimeFormatterr )r" ÚtimeÚformat_expectedÚresultr r r Útest_time_formatteró s z)TestDateTimeConverter.test_time_formatterÚfreq)ÚBÚLÚSc C sD d}t jd|d}| |d d ¡}tj | ¡ ¡}t j|||d d S )NgÖ&è .>é
|
)rG rp rd )r5 Z makeDateIndexrR r rh r\ Z _mpl_reprri )r" rp rP re Z dateindexr_ r` r r r Útest_dateindex_conversion s
|
z/TestDateTimeConverter.test_dateindex_conversionÚoffseté2 c C sL t d}|| }| |d d ¡}| |d d ¡}||k sHt| d| dd S )NrV z is not less than Ú.)r rR r )r" rv rP Zts1Zts2Zval1Zval2r r r Útest_resolution
|
s z%TestDateTimeConverter.test_resolutionc sR t dt dg}||g} |d d ¡} fdd|D }t |¡|k ¡ sNtd S )Nz
|
2017-01-01z
|
2017-01-02c s g | ]} |d d ¡qS rN ©rR ©Ú.0Úx©rP r r Ú
|
<listcomp> s z=TestDateTimeConverter.test_convert_nested.<locals>.<listcomp>)r rR r] r^ Úallr )r" rP ÚinnerÚdatarn Úexpectedr r~ r Útest_convert_nested s
|
z)TestDateTimeConverter.test_convert_nestedN)rJ rK rL r) ÚfixturerP rU rb rj rk ÚmarkÚ parametrizero ru r r r ry r r r r r rM s*
|
.ûþ
|
|
|
|
|
rM c @ sH e Zd Zejdd Zejdd Zdd Zdd Zd d
|
Z d d Z
|
d S )ÚTestPeriodConverterc C s t ¡ S rN )r ZPeriodConverterrO r r r Úpc s zTestPeriodConverter.pcc C s G dd d}| }d|_ |S )Nc @ s e Zd ZdS )z&TestPeriodConverter.axis.<locals>.AxisN)rJ rK rL r r r r ÚAxis$ s r ÚD©rp )r" r Úaxisr r r r " s zTestPeriodConverter.axisc C s, | dd |¡}| dd |¡}||ks(td S )NrV rQ ©r" r r rS rT r r r rU + s z0TestPeriodConverter.test_convert_accepts_unicodec C s8 | dgd |¡d }tdj}||ks*t| dd |¡}||ksDt| tdddgd |¡d }||kslt| tdddd |¡}||kst| tdgd |¡d }||ks²t| tdd |¡}||ksÐt| dd |¡}||ksêt| dd |¡}||kst| tjddgd d
|
d |¡}|d |ks4td S ) NrV r rW rX rY rZ z2012-01-01 00:00:00z2012-01-02 00:00:00zdatetime64[ns])Zdtype)rR r Zordinalr r r r] r^ ©r" r r r_ r` r r r rb 0 s4
|
þúz#TestPeriodConverter.test_conversionc C s* | ddgd |¡}ddg}||ks&td S )Nr rX rQ r r r r Útest_integer_passthroughT s z,TestPeriodConverter.test_integer_passthroughc sD ddg gd ¡} fddtdD }||ks@td S )NrV z2012-1-2c s g | ]} d ¡qS rN rz )r| r2 ©r r r r r r ] s z;TestPeriodConverter.test_convert_nested.<locals>.<listcomp>rf )rR r+ r r r r r r Z s z'TestPeriodConverter.test_convert_nestedN) rJ rK rL r) r
r r rU rb r r r r r r r s
|
|
$r c @ sH e Zd ZdZej dddddg¡dd Zej d d
|
d g¡d d ZdS )ÚTestTimeDeltaConverterzTest timedelta converterzx, decimal, format_expected)ç r z00:00:00)l xqAsrX z
|
01:06:12.3)l 6¶2º" rf z8 days 06:07:13.43)l KVôué z 09:00:23.4320c C s& t j}|j|d |d}||ks"td S )N)ÚposZ
|
n_decimals)r Ú TimeSeries_TimedeltaFormatterZformat_timedelta_ticksr )r" r} Údecimalrm Útdcrn r r r Útest_format_timedelta_ticksd s
|
z2TestTimeDeltaConverter.test_format_timedelta_ticksÚ view_interval)rX rf )rf rX c s8 G fddd}t ¡ }| |d| ¡ |dd d S )Nc s e Zd Z fddZdS )zNTestTimeDeltaConverter.test_call_w_different_view_intervals.<locals>.mock_axisc s S rN r rO ©r r r Úget_view_intervalv s z`TestTimeDeltaConverter.test_call_w_different_view_intervals.<locals>.mock_axis.get_view_intervalN)rJ rK rL r r r r r Ú mock_axisu s r r r r )r r Úsetattr)r" r Z monkeypatchr r r r r Ú$test_call_w_different_view_intervalsr s z;TestTimeDeltaConverter.test_call_w_different_view_intervalsN) rJ rK rL Ú__doc__r) r r r r r r r r r a s üþ
|
r Ú year_spang &@é éP é i i iÜ iÄ i¬ c C s d}|| d }|| d }|dk r(d S |d }t |¡\}}t ||d¡}ttjt dd |D ¡dd}t d d |D ¡} t d
|
d |D ¡}
|
|| } ||
|
} | j| d k} | j| d k}| j dk}| j dk}t
|
| ¡sät t
|
|¡sòt t
|
|¡st t
|
|¡st d S ) Niüÿÿr rX é- ÚQc S s g | ] }|d qS )r r r{ r r r r s z)test_quarterly_finder.<locals>.<listcomp>r c S s g | ] }|d qS )rX r r{ r r r r s c S s g | ] }|d qS )rf r r{ r r r r s r ) r Z_get_default_annual_spacingZ_quarterly_finderr r Z PeriodArrayr] r^ ÚyearZquarterr r )r¡ ZvminZvmaxÚspanZnyearsZ
|
min_anndefZ
|
maj_anndefrn ZquartersZmajorsZminorsZmajor_quartersZminor_quartersZcheck_major_yearsZcheck_minor_yearsZcheck_major_quartersZcheck_minor_quartersr r r Útest_quarterly_finder~ s. ÿ
|
|
r© ),r r r r Únumpyr] r) Zpandas._config.configÚ_configÚconfigr; Zpandasr r r r r r r
|
Zpandas._testingZ_testingr5 Zpandas.plottingr r Zpandas.tseries.offsetsr r r r Zpandas.plotting._matplotlibr Ú ImportErrorr* rE r r r rM r r r r r© r r r r Ú<module> s0 $
|
|
Y D
|
