a < b0@s0ddlmZmZddlZddlZddlZddlZddlmm Z ddl m m ZddlmZmZmZmZmZddlmZddlmZmZddlmZmZmZmZzddl m!Z!Wne"yYn0e#de#dZ$ej%j&Z'd d Z(d d Z)Gd ddZ*GdddZ+GdddZ,GdddZ-dS))datedatetimeN)IndexPeriodSeries Timestamp date_range) deregister_matplotlib_convertersregister_matplotlib_converters)DayMicroMilliSecond) convertermatplotlib.pyplotmatplotlib.datescCsd}tjd|g}t|dS)Nzimport 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)codecallrt/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/plotting/test_converter.pytest_registry_mpl_resets1s rcCstdtdksJdS)Nz00:01)rZtime2numrrrrtest_timtetonum_accepts_unicode@src@sXeZdZddZejdddddZdd Zd d Zejdddd d Z ddZ dS)TestRegistrationcCs&d}tjd|g}t|dks"JdS)Nzwimport matplotlib.units; import pandas as pd; units = dict(matplotlib.units.registry); assert pd.Timestamp not in unitsrr)rrr check_call)selfrrrrrtest_dont_register_by_defaultEs z.TestRegistration.test_dont_register_by_defaultZ matplotlibz3.1.3) min_versioncCsPtd}ttdtdddd}|\}}t||j|j | dS)Nr 2017Zperiodsindex) pytest importorskiprrangersubplotsr plotr&valuesclose)rplts_axrrrtest_registering_no_warningPs   z,TestRegistration.test_registering_no_warningcCstd}ttdtdddd}td}|Wdn1sJ0Yzt|dksfJW| n | 0dS)Nrr"r#r$r%r) r'r(rr)rtmZassert_produces_warningr+lenr-)rr.r/wrrrtest_pandas_plots_register[s  &z+TestRegistration.test_pandas_plots_registerc Cstd}tddXtddt|jvs4JWdn1sH0Yt|jvs`JWdn1st0YdS)Nmatplotlib.units'plotting.matplotlib.register_convertersTF)r'r(cfoption_contextrregistry)runitsrrrtest_matplotlib_formattersgs  ,z+TestRegistration.test_matplotlib_formatterscCstdtdd}td}ttdtdddd}|\}}| ||j |j Wdn1sn0Yt | ||j |j Wdn1s0Y| dS)Nrr8Fr"r#r$r%) r'r(r9r:rr)rr*r+r&r,r r-)rctxr.r/r0r1rrrtest_option_no_warningss    ..z'TestRegistration.test_option_no_warningc Cstd}td}t|j}z|j|}||jt<||jt<t|jt|us^Jt |jt|usvJW|j| D]\}}||j|<qn(|j| D]\}}||j|<q0dS)Nr7r) r'r(dictr;clearZ DateConverterrrr r items)rr<datesoriginalZdate_converterkvrrrtest_registry_resetss$        z%TestRegistration.test_registry_resetsN) __name__ __module__ __qualname__r tdZ skip_if_nor2r6r=r?rGrrrrrDs     rc@sfeZdZddZddZddZddZd d Zej d gd d dZ ddZ ddZ ddZdS)TestDateTimeConvertercCst|_td|_dSN)rZDatetimeConverterdtcZ TimeFormattertc)rmethodrrr setup_methods z"TestDateTimeConverter.setup_methodcCs4|jddd}|jddd}||ks0JddS)Nz12:22z/DatetimeConverter.convert should accept unicoderNconvertrZr1Zr2rrrtest_convert_accepts_unicodesz2TestDateTimeConverter.test_convert_accepts_unicodecCs|jdgddd}ttddd}||ks4J|jddd}||ksPJ|jtddddd}||kstJ|jddd}||ksJ|jtddd}||ksJ|jddd}||ksJ|jddd}||ksJ|jtddgdd}|d|ksJtd d d }|j|dd}||ksFJ|j| dd}||kshJ|jt |t d|gdd}|d|ksJ|jt |t d|g dd}|d|ksJdS) N2012-1-1r 2012-01-012012-01-01 00:00:00+00002012-01-02 00:00:00+0000UTCz US/Eastern)rNrSrCdate2numrrrnparrayZ tz_localizeZ tz_convertZ to_pydatetimerr )rrsxptsrrrtest_conversions<         $z%TestDateTimeConverter.test_conversionc Csd}|jtddddd}tjtddd}tj|||d|jtddddd}tj|||d|jtdd d d d d dd}tj|||ddS) Ng& .>z2012-1-1 01:02:03r\)tzrtolz2012-1-1 09:02:03zAsia/Hong_KongrWrX) rNrSrrrCr]r3assert_almost_equalr)rrfr`rarrrtest_conversion_floatsz+TestDateTimeConverter.test_conversion_floatcCstdddtdddg}|j|dd}tj|}t|||j|ddd}tj|d}||kspJtddddtddddg}|j|dd}tj|}t|||j|ddd}tj|d}||ksJdS)NirXrgrr") rrNrSrrCr]r3Zassert_numpy_array_equalr)rr,r`rarrr$test_conversion_outofbounds_datetimes     z:TestDateTimeConverter.test_conversion_outofbounds_datetimeztime,format_expected))rz00:00)g@z23:59:59.999999)i_z01:00)iz01:02:03)gfffffe@z 11:02:03.200cCs||}||ksJdSrM)rO)rtimeformat_expectedresultrrrtest_time_formatters z)TestDateTimeConverter.test_time_formattercCsPd}dD]B}tjd|d}|j|dd}tj|}tj|||dqdS)Ng& .>)BLS )rEfreqre) r3Z makeDateIndexrNrSrrCr]Z _mpl_reprri)rrfrtZ dateindexr`rarrrtest_dateindex_conversions z/TestDateTimeConverter.test_dateindex_conversioncsJfdd}td}|||t|||t|||tddS)Ncs@j|dd}j|dd}||ks._assert_lessrV2)rrr r )rryrbrrxrtest_resolutions  z%TestDateTimeConverter.test_resolutioncsTtdtdg}||g}j|dd}fdd|D}t||ksPJdS)Nz 2017-01-01z 2017-01-02csg|]}j|ddqSrMrR).0xrxrr "z=TestDateTimeConverter.test_convert_nested..)rrNrSr^r_all)rinnerdatarnexpectedrrxrtest_convert_nesteds z)TestDateTimeConverter.test_convert_nestedN)rHrIrJrQrUrcrjrkr'mark parametrizerorur{rrrrrrLs.  rLc@s4eZdZddZddZddZddZd d Zd S) TestPeriodConvertercCs,t|_Gddd}||_d|j_dS)Nc@s eZdZdS)z.TestPeriodConverter.setup_method..AxisN)rHrIrJrrrrAxis*srD)rZPeriodConverterpcaxisrt)rrPrrrrrQ's z TestPeriodConverter.setup_methodcCs4|jdd|j}|jdd|j}||ks0JdS)NrVrrSrrTrrrrU0sz0TestPeriodConverter.test_convert_accepts_unicodecCs^|jdgd|jd}tdj}||ks.J|jdd|j}||ksLJ|jtdddgd|jd}||ksxJ|jtdddd|j}||ksJ|jtdgd|jd}||ksJ|jtdd|j}||ksJ|jdd|j}||ksJ|jdd|j}||ks(J|jtjddgdd d|j}|d|ksZJdS) NrVrrWrXrYrZr[zdatetime64[ns])Zdtype) rrSrrZordinalrrr^r_rr`rarrrrc5s4        z#TestPeriodConverter.test_conversioncCs.|jddgd|j}ddg}||ks*JdS)NrrXrrrrrtest_integer_passthroughYsz,TestPeriodConverter.test_integer_passthroughcsFddgjgdj}fddtdD}||ksBJdS)NrVz2012-1-2csg|]}jdjqSrMr)r|r0rrrrr~brz;TestPeriodConverter.test_convert_nested..rg)rrSrr)rTrrrr_sz'TestPeriodConverter.test_convert_nestedN)rHrIrJrQrUrcrrrrrrr&s  $rc@sDeZdZdZejdgdddZejdddgd d Zd S) TestTimeDeltaConverterzTest timedelta converterzx, decimal, format_expected))rz00:00:00)lxqAsrXz 01:06:12.3)l62"rgz8 days 06:07:13.43)lKVuz 09:00:23.4320cCs&tj}|j|d|d}||ks"JdS)N)posZ n_decimals)rTimeSeries_TimedeltaFormatterZformat_timedelta_ticks)rr}decimalrmtdcrnrrrtest_format_timedelta_ticksis z2TestTimeDeltaConverter.test_format_timedelta_ticks view_interval)rXrg)rgrXcs8Gfddd}t}||d||dddS)NcseZdZfddZdS)zNTestTimeDeltaConverter.test_call_w_different_view_intervals..mock_axiscsSrMrrxrrrget_view_interval{sz`TestTimeDeltaConverter.test_call_w_different_view_intervals..mock_axis.get_view_intervalN)rHrIrJrrrrr mock_axiszsrrrr)rrsetattr)rrZ monkeypatchrrrrr$test_call_w_different_view_intervalswsz;TestTimeDeltaConverter.test_call_w_different_view_intervalsN) rHrIrJ__doc__r'rrrrrrrrrfs r).rrrrZnumpyr^r'Zpandas._config.config_configconfigr9Zpandas.util._test_decoratorsutilZ_test_decoratorsrKZpandasrrrrrZpandas._testingZ_testingr3Zpandas.plottingr r Zpandas.tseries.offsetsr r r rZpandas.plotting._matplotlibr ImportErrorr(rCrZslowZ pytestmarkrrrrLrrrrrrs0    [@