
    d3                        d dl mZm Z  d dlZd dlZd dlZd dlZd dlmc m	Z
 d dlmZmZmZmZmZmZmZ d dlmZ d dlmZmZ d dlmZmZmZmZ 	 d dlmZ n# e $ r Y nw xY w ej!        d            ej!        d          Z"d	 Z#d
 Z$ G d d          Z% G d d          Z& G d d          Z' G d d          Z(ej)        *                    dg d          d             Z+dS )    )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                  P    d} t           j        d| g}t          j        |           d S )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)codecalls     `/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/plotting/test_converter.pytest_registry_mpl_resetsr   .   s3    	/ 	 ND$'DD!!!!!    c                  ^    t          j        d          t          j        d          k    sJ d S )Nz00:01)r   time2num r   r   test_timtetonum_accepts_unicoder#   =   s/    g&&)*<W*E*EEEEEEEr   c                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )TestRegistrationc                 \    d}t           j        d|g}t          j        |          dk    sJ d S )Nzwimport matplotlib.units; import pandas as pd; units = dict(matplotlib.units.registry); assert pd.Timestamp not in unitsr   r   )r   r   r   
check_call)selfr   r   s      r   test_dont_register_by_defaultz.TestRegistration.test_dont_register_by_defaultB   s>    / 	 d+$T**a//////r   c                 :   t          j        d          }t          t          d          t	          dd                    }|                                \  }}t                       |                    |j        |j	                   |
                                 d S )Nr      2017periodsindex)pytestimportorskipr   ranger   subplotsr   plotr0   valuesclose)r(   plts_axs        r   test_registering_no_warningz,TestRegistration.test_registering_no_warningM   s    !"566599Jvr$B$B$BCCC2 	'(((
"""		r   c                    t          j        d          }t          t          d          t	          dd                    }t          j        d           5 }|                                 d d d            n# 1 swxY w Y   	 t          |          dk    sJ 	 |	                                 d S # |	                                 w xY w)Nr   r+   r,   r-   r/   r   )
r1   r2   r   r3   r   tmassert_produces_warningr5   lenr7   )r(   r8   r9   ws       r   test_pandas_plots_registerz+TestRegistration.test_pandas_plots_registerW   s    !"566599Jvr$B$B$BCCC'-- 	FFHHH	 	 	 	 	 	 	 	 	 	 	 	 	 	 		q66Q;;;;;IIKKKKKCIIKKKKs   A77A;>A;B/ /Cc                 (   t          j        d          }t          j        dd          5  t          j        dd          5  t          |j        vsJ 	 d d d            n# 1 swxY w Y   t          |j        v sJ 	 d d d            d S # 1 swxY w Y   d S )Nmatplotlib.units'plotting.matplotlib.register_convertersTF)r1   r2   cfoption_contextr	   registry)r(   unitss     r   test_matplotlib_formattersz+TestRegistration.test_matplotlib_formattersc   s"   #$677 H$OO 	/ 	/"#LeTT 7 7 666667 7 7 7 7 7 7 7 7 7 7 7 7 7 7.....	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/s5   B ABA"	"B%A"	&BBBc                 4   t          j        d           t          j        dd          }t          j        d          }t	          t          d          t          dd                    }|                                \  }}|5  |                    |j	        |j
                   d d d            n# 1 swxY w Y   t                       |5  |                    |j	        |j
                   d d d            n# 1 swxY w Y   |                                 d S )Nr   rE   Fr+   r,   r-   r/   )r1   r2   rF   rG   r   r3   r   r4   r5   r0   r6   r   r7   )r(   ctxr8   r9   r:   r;   s         r   test_option_no_warningz'TestRegistration.test_option_no_warningo   su   /000 I5QQ!"566599Jvr$B$B$BCCC2  	' 	'GGAGQX&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	'((( 	' 	'GGAGQX&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	'		s$   !B11B58B5!C99C= C=c                    t          j        d          }t          j        d          }t          |j                  }	 |j                                         |                                }||j        t          <   ||j        t          <   t                       |j        t                   |usJ t                       |j        t                   |u sJ 	 |j                                         |
                                D ]\  }}||j        |<   d S # |j                                         |
                                D ]\  }}||j        |<   w xY w)NrD   r   )r1   r2   dictrH   clearDateConverterr   r   r   r   items)r(   rI   datesoriginaldate_converterkvs          r   test_registry_resetsz%TestRegistration.test_registry_resets   sY   #$677#$677 ''	&N  """"0022N'5EN8$#1EN4 *,,,>$'~====,...>$'>99999 N  """ (( & &1$%q!!& & N  """ (( & &1$%q!!&s   BD ?EN)	__name__
__module____qualname__r)   r<   rB   rJ   rM   rX   r"   r   r   r%   r%   A   sn        	0 	0 	0  
 
 

/ 
/ 
/  "& & & & &r   r%   c                   `   e Zd Zej        d             Zd Zd Zd Zd Z	ej
                            dg d          d             Zej
                            d	d
          d             Zej
                            d e             e             ed          g          d             Zd ZdS )TestDateTimeConverterc                 (    t          j                    S N)r   DatetimeConverterr(   s    r   dtczTestDateTimeConverter.dtc   s    *,,,r   c                     |                     dd d           }|                     dd d           }||k    s
J d            d S )Nz2000-01-01 12:22z/DatetimeConverter.convert should accept unicodeconvert)r(   rb   r1r2s       r   test_convert_accepts_unicodez2TestDateTimeConverter.test_convert_accepts_unicode   sF    [[+T488[[+T488RxxxJxxxxxr   c                     |                     dgd d           d         }t                              t          ddd                    }||k    sJ |                     dd d           }||k    sJ |                     t	          ddd          d d           }||k    sJ |                     dd d           }||k    sJ |                     t          d          d d           }||k    sJ |                     dd d           }||k    sJ |                     dd d           }||k    sJ |                     t          j        ddg          d d           }|d         |k    sJ t          d                              d          	                    d	          }|                     |d d           }||k    sJ |                     |
                                d d           }||k    sJ |                     t          |t          d          z
  |g          d d           }|d         |k    sJ |                     t          |t          d          z
  |g          
                                d d           }|d         |k    sJ d S )
N2012-1-1r        
2012-01-012012-01-01 00:00:00+0000z2012-01-02 00:00:00+0000UTCz
US/Eastern)re   rS   date2numr   r   r	   nparraytz_localize
tz_convertto_pydatetimer   r   )r(   rb   rsxptss        r   test_conversionz%TestDateTimeConverter.test_conversion   sp   [[*tT2215^^HT1a0011Rxxxx[[T400Rxxxx[[dAq))466Rxxxx[[T400Rxxxx[[:..d;;Rxxxx [[tT22Rxxxx[[3T4@@Rxxxx[[H02LMNN
 

 !u{{{{ |$$0077BB<PP[[T4((Rxxxx[[))++T488Rxxxx[[SVVR0114>>!u{{{{[[SVVR011??AA4NN!u{{{{{{r   c           
         d}|                     t          dd          d d           }t          j                            t          dd                    }t          j        |||           |                     t          dd          d d           }t          j        |||           |                     t          dd	d	d	d
d          d d           }t          j        |||           d S )Ng&.>z2012-1-1 01:02:03ro   )tzrtolz2012-1-1 09:02:03zAsia/Hong_Kongrk   rl         )re   r	   r   mdatesrp   r>   assert_almost_equalr   )r(   rb   r}   rv   rw   s        r   test_conversion_floatz+TestDateTimeConverter.test_conversion_float   s    [[#65AAA4NN&&y1D'O'O'OPP
r2D1111[[).>???t
 
 	r2D1111[[$1aA66dCC
r2D111111r   c                    t          ddd          t          ddd          g}|                    |d d           }t          j                            |          }t          j        ||           |                    |d         d d           }t          j                            |d                   }||k    sJ t          dddd          t          dddd          g}|                    |d d           }t          j                            |          }t          j        ||           |                    |d         d d           }t          j                            |d                   }||k    sJ d S )Ni  rl   r~   r   r+   )r   re   r   r   rp   r>   assert_numpy_array_equalr   )r(   rb   r6   rv   rw   s        r   $test_conversion_outofbounds_datetimez:TestDateTimeConverter.test_conversion_outofbounds_datetime   sJ   tQ""Dq!$4$45[[t,,&&v..
#B+++[[D$//&&vay11Rxxxx4Ar**HT1a,D,DE[[t,,&&v..
#B+++[[D$//&&vay11Rxxxxxxr   ztime,format_expected))r   z00:00)g@z23:59:59.999999)i_ z01:00)i  z01:02:03)gfffffe@z11:02:03.200c                 P     t          j        d           |          }||k    sJ d S r_   )r   TimeFormatter)r(   timeformat_expectedresults       r   test_time_formatterz)TestDateTimeConverter.test_time_formatter   s5     /(..t44((((((r   freq)BLSc                     d}t          j        d|          }|                    |d d           }t          j                            |                                          }t          j        |||           d S )Ng&.>
   )rV   r   r|   )r>   makeDateIndexre   r   r   rp   	_mpl_reprr   )r(   r   rb   r}   	dateindexrv   rw   s          r   test_dateindex_conversionz/TestDateTimeConverter.test_dateindex_conversion  sq    $r555	[[D$//&&y':':'<'<==
r2D111111r   offset2   c                     t          d          }||z   }|                    |d d           }|                    |d d           }||k     st          | d| d          d S )Nrj   z is not less than .)r	   re   AssertionError)r(   r   rb   ts1ts2val1val2s          r   test_resolutionz%TestDateTimeConverter.test_resolution
  ss     
##Fl{{3d++{{3d++d{{ D!C!CD!C!C!CDDD {r   c                     t          d          t          d          g}||g}                    |d d           }fd|D             }t          j        |          |k                                    sJ d S )Nz
2017-01-01z
2017-01-02c                 >    g | ]}                     |d d           S r_   rd   ).0xrb   s     r   
<listcomp>z=TestDateTimeConverter.test_convert_nested.<locals>.<listcomp>  s)    ===1CKK4..===r   )r	   re   rq   rr   all)r(   rb   innerdatar   expecteds    `    r   test_convert_nestedz)TestDateTimeConverter.test_convert_nested  s    <(()L*A*ABu~T4..=======  H,113333333r   N)rY   rZ   r[   r1   fixturerb   rh   ry   r   r   markparametrizer   r   r   r   r   r   r   r"   r   r   r]   r]      s;       ^- - ^-K K K
, , ,\2 2 2  $ [	
 	
 	
	 	) )	 	)
 [V_552 2 652 [X%%''5599'EFFE E GFE4 4 4 4 4r   r]   c                   f    e Zd Zej        d             Zej        d             Zd Zd Zd Z	d Z
dS )TestPeriodConverterc                 (    t          j                    S r_   )r   PeriodConverterra   s    r   pczTestPeriodConverter.pc  s    (***r   c                 B     G d d          } |            }d|_         |S )Nc                       e Zd ZdS )&TestPeriodConverter.axis.<locals>.AxisN)rY   rZ   r[   r"   r   r   Axisr   $  s        Dr   r   Dr   )r(   r   axiss      r   r   zTestPeriodConverter.axis"  sC    	 	 	 	 	 	 	 	 tvv	r   c                 r    |                     dd |          }|                     dd |          }||k    sJ d S )Nrj   rd   )r(   r   r   rf   rg   s        r   rh   z0TestPeriodConverter.test_convert_accepts_unicode+  s=    ZZ
D$//ZZ
D$//Rxxxxxxr   c                 .   |                     dgd |          d         }t          d          j        }||k    sJ |                     dd |          }||k    sJ |                     t          ddd          gd |          d         }||k    sJ |                     t          ddd          d |          }||k    sJ |                     t	          d          gd |          d         }||k    sJ |                     t	          d          d |          }||k    sJ |                     dd |          }||k    sJ |                     dd |          }||k    sJ |                     t          j        ddgd	
          d |          }|d         |k    sJ d S )Nrj   r   rk   rl   rm   rn   z2012-01-01 00:00:00z2012-01-02 00:00:00zdatetime64[ns])dtype)re   r   ordinalr   r	   rq   rr   r(   r   r   rv   rw   s        r   ry   z#TestPeriodConverter.test_conversion0  s   ZZdD11!4J'RxxxxZZ
D$//RxxxxZZdAq))*D$77:RxxxxZZT1a(($55RxxxxZZ:../t<<Q?RxxxxZZ	*--tT::RxxxxZZdD11RxxxxZZ2D$??RxxxxZZH&(=>&   
 
 !u{{{{{{r   c                 P    |                     ddgd |          }ddg}||k    sJ d S )Nr   rl   rd   r   s        r   test_integer_passthroughz,TestPeriodConverter.test_integer_passthroughT  s5    ZZAd++VRxxxxxxr   c                     ddg                     gd           }fdt          d          D             }||k    sJ d S )Nrj   z2012-1-2c                 >    g | ]}                     d           S r_   rd   )r   r:   r   r   r   s     r   r   z;TestPeriodConverter.test_convert_nested.<locals>.<listcomp>]  s)    ===qbjjtT**===r   r~   )re   r3   )r(   r   r   rf   rg   r   s    ``  @r   r   z'TestPeriodConverter.test_convert_nestedZ  s^    J'ZZtdD11======E!HH===Rxxxxxxr   N)rY   rZ   r[   r1   r   r   r   rh   ry   r   r   r"   r   r   r   r     s        ^+ + ^+ ^  ^  
" " "H      r   r   c                       e Zd ZdZej                            dg d          d             Zej                            dddg          d             Zd	S )
TestTimeDeltaConverterzTest timedelta converterzx, decimal, format_expected))        r   z00:00:00)l    xqAsrl   z
01:06:12.3)l    62" r~   z8 days 06:07:13.43)l    KVu   z09:00:23.4320c                 ^    t           j        }|                    |d |          }||k    sJ d S )N)pos
n_decimals)r   TimeSeries_TimedeltaFormatterformat_timedelta_ticks)r(   r   decimalr   tdcr   s         r   test_format_timedelta_ticksz2TestTimeDeltaConverter.test_format_timedelta_ticksd  s=     5++A4G+LL((((((r   view_interval)rl   r~   )r~   rl   c                      G fdd          }t          j                    }|                    |d |                        |dd           d S )Nc                       e Zd Z fdZdS )NTestTimeDeltaConverter.test_call_w_different_view_intervals.<locals>.mock_axisc                     S r_   r"   )r(   r   s    r   get_view_intervalz`TestTimeDeltaConverter.test_call_w_different_view_intervals.<locals>.mock_axis.get_view_intervalv  s	    $$r   N)rY   rZ   r[   r   )r   s   r   	mock_axisr   u  s.        % % % % % % %r   r   r   r   r   )r   r   setattr)r(   r   monkeypatchr   r   s    `   r   $test_call_w_different_view_intervalsz;TestTimeDeltaConverter.test_call_w_different_view_intervalsr  sv    	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 577C555Cr   N)	rY   rZ   r[   __doc__r1   r   r   r   r   r"   r   r   r   r   a  s        ""[%	
 	
 	
 ) ) )
 [_vv.>??  @?  r   r   	year_span)	g     &@   P      i  i   i  i	  i  c                    d}|| dz  z   }||z
  dz   }|dk     rd S |dz  }t          j        |          \  }}t          j        ||d          }t          t	          j        t          j        d |D                       d                    }t          j        d |D                       }	t          j        d	 |D                       }
||	         }||
         }|j        |z  d
k    }|j        |z  d
k    }|j	        dk    }|j	        dk    }t          j
        |          sJ t          j
        |          sJ t          j
        |          sJ t          j
        |          sJ d S )Nir   rl   -   Qc                     g | ]
}|d          S )r   r"   r   r   s     r   r   z)test_quarterly_finder.<locals>.<listcomp>  s    $:$:$:aQqT$:$:$:r   r   c                     g | ]
}|d          S )rl   r"   r   s     r   r   z)test_quarterly_finder.<locals>.<listcomp>      ,,,qt,,,r   c                     g | ]
}|d          S )r~   r"   r   s     r   r   z)test_quarterly_finder.<locals>.<listcomp>  r   r   r   )r   _get_default_annual_spacing_quarterly_finderr   r
   PeriodArrayrq   rr   yearquarterr   )r   vminvmaxspannyears
min_anndef
maj_anndefr   quartersmajorsminorsmajor_quartersminor_quarterscheck_major_yearscheck_minor_yearscheck_major_quarterscheck_minor_quarterss                    r   test_quarterly_finderr   ~  s    D)a-D$;?DbyyAXF(DVLLZ(tS99F28$:$:6$:$:$:;;#FFF H X,,V,,,--FX,,V,,,--Ff%Nf%N&+j8A=&+j8A=)1Q6)1Q66#$$$$$6#$$$$$6&'''''6&'''''''r   ),r   r   r   r   numpyrq   r1   pandas._config.config_configconfigrF   pandasr   r   r   r   r	   r
   r   pandas._testing_testingr>   pandas.plottingr   r   pandas.tseries.offsetsr   r   r   r   pandas.plotting._matplotlibr   ImportErrorr2   rS   r   r#   r%   r]   r   r   r   r   r   r"   r   r   <module>r     s              



      " " " " " " " " "                                         	5555555 	 	 	 	D	
  ' ( ( (.//" " "F F FV& V& V& V& V& V& V& V&r@4 @4 @4 @4 @4 @4 @4 @4FA A A A A A A AH       : &V&V&VWW( ( XW( ( (s   A AA