
    Mh\                      d    S SK rS SKrS SKJr  S SKrS SKJrJr  S SK	J
r  S SKJr   " S S5      rg)    N)
is_integer)IndexSeries)	Timestampc                      \ rS rSrS rS rS rS rS r\	R                  R                  S\" S5      \" S	5      \" S
5      /\" SSS9\" S	SS9\" S
SS9/\R                  " S5      \R                  " S5      \R                  " S5      /\" S5      \" S	5      \" S
5      \R                  /\" SSS9\" S	SS9\" S
SS9\R                  /\R                  " S5      \R                  " S5      \R                  " S5      \R                  //5      S 5       rS rS r\	R                  R                  S/ SQS4/ SQS4/5      S 5       rS rS rS r\	R                  R                  S\\S/5      S 5       rS rS  rS!rg")#TestSeriesQuantile   c                    UR                  S5      nU[        R                  " UR                  5       S5      :X  d   eUR                  S5      nU[        R                  " UR                  5       S5      :X  d   e[	        U[
        S9R                  S5      nU[        R                  " UR                  5       S5      :X  d   eUR                  R                  5       nUR                  S5      nU[        S5      :X  d   eUR                  5       nUR                  S5      nU[        R                  " S	5      :X  d   e[	        [        R                  " S
5      /5      R                  5       nU[        R                  " S5      :X  d   eSnSSSS/SS/4 H6  n[        R                   " ["        US9   UR                  U5        S S S 5        M8     [	        [        R$                  R'                  S5      R)                  S5      5      n/ SQn	[        R                   " ["        US9   UR                  U	5        S S S 5        g ! , (       d  f       M  = f! , (       d  f       g = f)N皙?
   ?Z   dtype皙?2000-01-10 19:12:00      ?z24:00:00NaTr   z2percentiles should all be in the interval \[0, 1\]         ?)matchd   )g      r   g      ?)quantilenp
percentiledropnar   objectindex	to_seriesr   diffpdto_timedeltatimedelta64sum	Timedeltapytestraises
ValueErrorrandomdefault_rngstandard_normal)
selfdatetime_seriesqdtstdsresultmsginvalidspercentile_arrays
             [/var/www/html/env/lib/python3.13/site-packages/pandas/tests/series/methods/test_quantile.pytest_quantile TestSeriesQuantile.test_quantile   s   $$S)BMM/"8"8":B????$$S)BMM/"8"8":B???? ?&1::3?BMM/"8"8":B???? ##--/LLI34444 hhjLLBOOJ//// ./0446a(((DARy3(3Gz5((1 65 4 299((+;;C@A,]]:S1JJ'( 21 65
 21s   ,H9I9
I	
Ic           	         UR                   R                  U5      Ul         SS/nUR                  U5      n[        [        R
                  " UR                  5       S5      [        R
                  " UR                  5       S5      /UUR                  S9n[        R                  " XE5        UR                   R                  5       nSUl        UR                  S5      n[        [        S5      [        S5      /S	S	/SS
U S3S9n[        R                  " XE5        UR                  / 5      n[        / UR                  [        / [        S9SS9n[        R                  " XE5        g )Nr   r   r   r   r   namexxx)r   r   r   r   zM8[])r   r<   r   r   float64)r<   r   r   )r   as_unitr   r   r   r   r   r<   tmassert_series_equalr    r   r   float)r-   r.   unitqsr2   expectedr0   s          r7   test_quantile_multi&TestSeriesQuantile.test_quantile_multi3   s;    / 5 5 = =d C3Z ))"-o446;o446;  %%
 	v0##--/j),-y9N/OP*vQ-	
 	v0 ))"-_))r1Gy
 	v0    c                     UR                  SSS9nU[        R                  " UR                  5       S5      :X  d   eUR                  S5      nU[        R                  " UR                  5       S5      :X  d   eX#:X  d   eg )Nr   linearinterpolationr   )r   r   r   r   )r-   r.   r/   q1s       r7   test_quantile_interpolation.TestSeriesQuantile.test_quantile_interpolationR   sx     $$S$ABMM/"8"8":B????%%c*R]]?#9#9#;R@@@@ wwrI   c                    [        / SQ5      R                  SSS9nU[        R                  " [        R                  " / SQ5      S5      :X  d   e[        U5      (       d   e[        / SQ5      R                  SSS9nU[        R                  " [        R                  " / SQ5      S5      :X  d   e[        U5      (       d   eg )N)         r   lowerrL   2   higher)r   r   r   r   arrayr   )r-   r/   s     r7   !test_quantile_interpolation_dtype4TestSeriesQuantile.test_quantile_interpolation_dtype^   s     9&&s'&BBMM"((9"5r::::!}}}9&&s(&CBMM"((9"5r::::!}}}rI   c           	      z   [        SSSS[        R                  /5      nUR                  S5      nSnX#:X  d   e[        / [        S9nU[        [        R                  [        R                  /5      /nU H  nUR                  S5      n[        R
                  " U5      (       d   eUR                  S/5      n[        R                  " U[        [        R                  /S/S95        UR                  S	S
/5      n[        R                  " U[        [        R                  [        R                  /S	S
/S95        M     g )NrR   r   rS   rT   r   g      @r   r   r   g333333?)r   r   nanr   r   isnanrA   rB   )r-   serr2   rF   s1casesress          r7   test_quantile_nan$TestSeriesQuantile.test_quantile_nanj   s    aAq"&&)*c"!!! Bf%VRVVRVV,-.C,,s#C88C== =,,u%C""3xu(EF,,Sz*C""3/?Sz(RS rI   casez
2011-01-01z
2011-01-02z
2011-01-03z
US/Eastern)tzz1 daysz2 daysz3 daysc                     [        USS9nUR                  S5      nX1S   :X  d   eUR                  S/5      n[        US   /S/SS9n[        R                  " X45        g )NXXX)r<   r   rR   r;   )r   r   rA   rB   )r-   re   r_   rb   exps        r7   test_quantile_box$TestSeriesQuantile.test_quantile_box   sa    H T&ll31g~~llC5!d1gYse%8
s(rI   c                     [         R                  " [        / SS9R                  S5      5      (       d   e[         R                  " [        / SS9R                  S5      5      (       d   eg )NzM8[ns]r   r   zm8[ns])r"   isnar   r   )r-   s    r7   !test_datetime_timedelta_quantiles4TestSeriesQuantile.test_datetime_timedelta_quantiles   sP    wwvb1::3?@@@@wwvb1::3?@@@@rI   c                 n   [        [        R                  [        R                  /5      R                  S5      nU[        R                  L d   e[        [        R                  [        R                  /5      R                  S/5      n[        R
                  " U[        [        R                  /S/S95        g )Nr   r\   )r   r"   r   r   rA   rB   )r-   rb   s     r7   test_quantile_nat$TestSeriesQuantile.test_quantile_nat   sw    bffbff%&//4bff}}bffbff%&//6
sFBFF8C5$ABrI   zvalues, dtype)r   r   r   rR   r   rS   zSparse[int])g        Ng      ?g       @Sparse[float]c                     [        XS9nUR                  S/5      n[        [        R                  " U5      5      R                  S/5      R	                  S5      n[
        R                  " XE5        g )Nr   r   rs   )r   r   r   asarrayastyperA   rB   )r-   valuesr   r_   r2   rF   s         r7   test_quantile_sparse'TestSeriesQuantile.test_quantile_sparse   sV    
 V)se$"**S/*33SE:AA/R
v0rI   c                     [        / SS9nUR                  S5      n[        R                  " U5      (       d   eUR                  S/5      n[        [        R                  /S/S9n[
        R                  " X#5        g )Nr?   r   r   r\   r   r   r   r^   r]   rA   rB   r-   r_   rb   ri   s       r7   test_quantile_empty_float64.TestSeriesQuantile.test_quantile_empty_float64   s`    Ry)ll3xx}}}llC5!bffXcU+
s(rI   c                     [        / SS9nUR                  S5      n[        R                  " U5      (       d   eUR                  S/5      n[        [        R                  /S/S9n[
        R                  " X#5        g )Nint64r   r   r\   r{   r|   s       r7   test_quantile_empty_int64,TestSeriesQuantile.test_quantile_empty_int64   s`    Rw'll3xx}}}llC5!bffXcU+
s(rI   c                     [        / SS9nUR                  S5      nU[        R                  L d   eUR                  S/5      n[        [        R                  /S/UR                  S9n[
        R                  " X#5        g )Nzdatetime64[ns]r   r   )r   r   )r   r   r"   r   r   rA   rB   r|   s       r7   test_quantile_empty_dt64+TestSeriesQuantile.test_quantile_empty_dt64   se    R/0ll3bff}}llC5!bffXcU#))<
s(rI   r   Int64c                 ,   [        / SQUS9R                  [        R                  " SSS5      5      n[        [        R                  " SSS5      [        R                  " SSS5      S9nUS	:X  a  UR	                  S
5      n[
        R                  " X#5        g )N)rR   r   rS   r   r   rR   r   rS   r   r\   r   Float64)r   r   r   arangerv   rA   rB   )r-   r   r2   rF   s       r7   test_quantile_dtypes'TestSeriesQuantile.test_quantile_dtypes   sp    	/881a9NO"))Aq#.bii1d6KLGy1H
v0rI   c                 ^   [        [        R                  [        R                  /US9n[        R                  " S 5         UR                  SS/5      nS S S 5        [        [        R                  [        R                  /USS/S9n[        R                  " WU5        g ! , (       d  f       NQ= f)Nr   r   r   r   r   )r   r"   NArA   assert_produces_warningr   rB   r-   any_int_ea_dtyper_   r2   rF   s        r7   test_quantile_all_na'TestSeriesQuantile.test_quantile_all_na   sx    beeRUU^+;<''-\\3*-F .255"%%.0@c
S
vx0 .-s   B
B,c                     [        [        R                  [        R                  S/US9nUR                  SS/5      n[        SS/USS/S9n[        R
                  " X45        g )NrR   r   r   r   r   )r   r"   r   r   rA   rB   r   s        r7   test_quantile_dtype_size+TestSeriesQuantile.test_quantile_dtype_size   sU    beeRUUA&.>?sCj)1a&(8c
K
v0rI    N)__name__
__module____qualname____firstlineno__r8   rG   rO   rY   rc   r'   markparametrizer   r"   r&   r   rj   rn   rq   rx   r}   r   r   intrC   r   r   r   __static_attributes__r   rI   r7   r   r      s   !)F1>

T* [[ ,',',' ,<8,<8,<8
 \\(#R\\(%;R\\(=ST ,',','	 ,<8,<8,<8	 X&X&X&	3	
"F)G"F)A
C [[
m	,/Do.VW1	1	)	)	) [[WsE7&;<1 =111rI   r   )numpyr   r'   pandas.core.dtypes.commonr   pandasr"   r   r   pandas._testing_testingrA   pandas.core.indexes.datetimesr   r   r   rI   r7   <module>r      s*      0   3h1 h1rI   