Ë
    ž…jy  ã                   ó`   — d dl Zd dlZd dlZd dlmZ d dlmZ d dl	m
Z
 d dlmZ  G d„ d«      Zy)é    N)Ú	Timedelta)Únanops)ÚTimedeltaArrayc            
       ób  — e Zd Zej                  j                  dg d¢«      d„ «       Zd„ Zd„ Zd„ Z	d„ Z
d„ Zej                  j                  d	 ed
«      j                  d«       ej                  d«       ej                  dd¬«       ej                  dd¬«      g«      d„ «       Zd„ Zd„ Zd„ Zy)ÚTestReductionsÚname)ÚstdÚminÚmaxÚmedianÚmeanc                 óâ   — t        j                  g «      }|j                  } t        ||«      |¬«      }|t         j                  u sJ ‚ t        ||«      |¬«      }|t         j                  u sJ ‚y )N©Úskipna)ÚpdÚTimedeltaIndexÚarrayÚgetattrÚNaT)Úselfr   r   ÚtdiÚarrÚresults         úo/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/arrays/timedeltas/test_reductions.pyÚtest_reductions_emptyz$TestReductions.test_reductions_empty   se   € ä×Ñ Ó#ˆØ�i‰iˆà#”˜˜dÓ#¨6Ô2ˆØœŸ™ÑÐÐà#”˜˜dÓ#¨6Ô2ˆØœŸ™ÑÐÑó    c                 ó  — t        j                  g «      }|j                  }|j                  |¬«      }t	        |t
        «      sJ ‚|t        d«      k(  sJ ‚|j                  |¬«      }t	        |t
        «      sJ ‚|t        d«      k(  sJ ‚y )Nr   r   )r   r   r   ÚsumÚ
isinstancer   )r   r   r   r   r   s        r   Útest_sum_emptyzTestReductions.test_sum_empty   s   € Ü×Ñ Ó#ˆØ�i‰iˆà—‘ �Ó'ˆÜ˜&¤)Ô,Ð,Ð,Øœ 1›Ò%Ð%Ð%à—‘ �Ó'ˆÜ˜&¤)Ô,Ð,Ð,Øœ 1›Ò%Ð%Ñ%r   c                 ób  — d|› d�}t        j                  g d¢|¬«      }|j                  «       }t        d«      }||k(  sJ ‚|j	                  «       }t        d«      }||k(  sJ ‚|j                  d¬«      }|t
        j                  u sJ ‚|j	                  d¬«      }|t
        j                  u sJ ‚y )	Nzm8[ú]©Ú3hr$   r   Ú2hÚ5hÚ4h©Údtyper%   r&   Fr   )r   Ú_from_sequencer
   r   r   r   r   )r   Úunitr)   r   r   Úexpecteds         r   Útest_min_maxzTestReductions.test_min_max#   s®   € Ø�d�V˜1�ˆÜ×+Ñ+Ú1¸ô
ˆð —‘“ˆÜ˜T“?ˆØ˜Ò!Ð!Ð!à—‘“ˆÜ˜T“?ˆØ˜Ò!Ð!Ð!à—‘ �Ó&ˆØœŸ™ÑÐÐà—‘ �Ó&ˆØœŸ™ÑÐÑr   c                 óê  — t        j                  g d¢«      }|j                  }|j                  d¬«      }t	        d¬«      }t        |t        «      sJ ‚||k(  sJ ‚|j                  d¬«      }t        |t        «      sJ ‚||k(  sJ ‚|j                  d¬«      }|t         j                  u sJ ‚|j                  d¬«      }|t         j                  u sJ ‚|j                  d¬«      }|t         j                  u sJ ‚|j                  d¬«      }|t         j                  u sJ ‚|j                  d	¬«      }t        |t        «      sJ ‚||k(  sJ ‚|j                  d	¬«      }t        |t        «      sJ ‚||k(  sJ ‚y )
Nr#   Tr   é   ©ÚhoursFé	   )Ú	min_counté   )r   r   r   r   r   r   r   ©r   r   r   r   r,   s        r   Útest_sumzTestReductions.test_sum7   sf  € Ü×ÑÒ EÓFˆØ�i‰iˆà—‘ �Ó%ˆÜ 2Ô&ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!à—‘ �Ó%ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!à—‘ �Ó&ˆØœŸ™ÑÐÐà—‘ �Ó&ˆØœŸ™ÑÐÐà—‘ 1�Ó%ˆØœŸ™ÑÐÐà—‘ 1�Ó%ˆØœŸ™ÑÐÐà—‘ 1�Ó%ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!à—‘ 1�Ó%ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ñ!r   c                 ó  — t        j                  g d¢«      }|j                  }t        j                  |«      }t        d¬«      }t        |t
        «      sJ ‚||k(  sJ ‚t        j                  |«      }t        |t
        «      sJ ‚||k(  sJ ‚y )N)r$   r$   r%   r&   r'   r/   r0   )r   r   r   Únpr   r   r   r5   s        r   Ú
test_npsumzTestReductions.test_npsumX   s|   € ä×ÑÒ >Ó?ˆØ�i‰iˆä—‘˜“ˆÜ 2Ô&ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!ä—‘˜“ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ñ!r   c                 ó   — t        j                  d«      j                  t         j                  «      j	                  d«      j                  d«      j                  dd«      }d|d<   t        j                  |«      }|j                  d¬	«      }|t        j                  u sJ ‚|j                  d
d¬«      }t        j                  t        d¬«      t        j                  gd¬«      j                  }t        j                  ||«       |j                  dd¬«      }t        j                  t        d¬«      t        d¬«      t        d¬«      t        j                  gd¬«      j                  }t        j                  ||«       y )Né   zm8[s]zm8[ns]é   é   ÚNat)éÿÿÿÿr?   Fr   r   )Úaxisr   é   )Úsecondsr(   r4   é   r2   )r8   ÚarangeÚastypeÚint64ÚviewÚreshaper   r*   r   r   r   r   r   Ú_valuesÚtmÚassert_timedelta_array_equal)r   r   Útdar   r,   s        r   Útest_sum_2d_skipna_falsez'TestReductions.test_sum_2d_skipna_falsef   s4  € Ü�i‰i˜‹l×!Ñ!¤"§(¡(Ó+×0Ñ0°Ó9×@Ñ@ÀÓJ×RÑRÐSTÐVWÓXˆØˆˆF‰ä×+Ñ+¨CÓ0ˆà—‘ �Ó&ˆØœŸ™ÑÐÐà—‘˜a¨�Ó.ˆÜ×$Ñ$Ü˜rÔ"¤B§F¡FÐ+°8ô
ç
‰'ð 	ô 	×'Ñ'¨°Ô9à—‘˜a¨�Ó.ˆÜ×$Ñ$ä !Ô$Ü !Ô$Ü !Ô$Ü—‘ð	ð ô
÷ ‰'ð 	ô 	×'Ñ'¨°Õ9r   Úaddr   Úusz
2021-01-01ÚUTC)Útzz
Asia/Tokyoc                 óT  — t        j                  g d¢«      |z   }|j                  }|j                  d¬«      }t	        d¬«      j                  d«      }t        |t        «      sJ ‚||k(  sJ ‚|j                  d¬«      }t        |t        «      sJ ‚||k(  sJ ‚t        |dd «      €Mt        j                  t        j                  |«      d¬«      }t        |t        j                  «      sJ ‚||k(  sJ ‚|j                  d¬«      }|t         j                  u sJ ‚|j                  d¬«      }|t         j                  u sJ ‚t        |dd «      €^t        j                  t        j                  |«      d¬«      }t        |t        j                  «      sJ ‚t        j                  |«      sJ ‚y y )	N)Ú0hr'   r   r'   rS   r%   Tr   r=   r0   rO   rQ   F)r   r   r   r	   r   Úas_unitr   r   r   Únanstdr8   ÚasarrayÚtimedelta64r   Úisnat)r   rN   r   r   r   r,   s         r   Útest_stdzTestReductions.test_std‚   ss  € ô ×ÑÒ EÓFÈÑLˆØ�i‰iˆà—‘ �Ó%ˆÜ 1Ô%×-Ñ-¨dÓ3ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!à—‘ �Ó%ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!ä�3˜˜dÓ#Ð+Ü—]‘]¤2§:¡:¨c£?¸4Ô@ˆFÜ˜f¤b§n¡nÔ5Ð5Ð5Ø˜XÒ%Ð%Ð%à—‘ �Ó&ˆØœŸ™ÑÐÐà—‘ �Ó&ˆØœŸ™ÑÐÐä�3˜˜dÓ#Ð+Ü—]‘]¤2§:¡:¨c£?¸5ÔAˆFÜ˜f¤b§n¡nÔ5Ð5Ð5Ü—8‘8˜FÔ#Ð#Ñ#ð ,r   c                 ó¦  — t        j                  g d¢«      }|j                  }|j                  d¬«      }t	        d¬«      }t        |t        «      sJ ‚||k(  sJ ‚|j                  d¬«      }t        |t        «      sJ ‚||k(  sJ ‚|j                  d¬«      }|t         j                  u sJ ‚|j                  d¬«      }|t         j                  u sJ ‚y )N©rS   r$   r   Ú5h06mrS   r%   Tr   r=   r0   F)r   r   r   r   r   r   r   r5   s        r   Útest_medianzTestReductions.test_median¨   sÆ   € Ü×ÑÒ HÓIˆØ�i‰iˆà—‘ 4�Ó(ˆÜ 1Ô%ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!à—‘ 4�Ó(ˆÜ˜&¤)Ô,Ð,Ð,Ø˜Ò!Ð!Ð!à—‘ 5�Ó)ˆØœŸ™ÑÐÐà—‘ 5�Ó)ˆØœŸ™ÑÐÑr   c                 ó¦  — t        j                  g d¢«      }|j                  }t        |j	                  «       j
                  j                  «       «      }|j                  «       }||k(  sJ ‚|j                  d¬«      }|t         j                  u sJ ‚|j	                  «       j                  d¬«      }||k(  sJ ‚|j                  d¬«      }||k(  sJ ‚y )Nr[   Fr   r   ©r@   )r   r   Ú_datar   ÚdropnaÚ_ndarrayr   r   )r   r   r   r,   r   s        r   Ú	test_meanzTestReductions.test_mean»   s½   € Ü×ÑÒ HÓIˆØ�i‰iˆô ˜SŸZ™Z›\×2Ñ2×7Ñ7Ó9Ó:ˆà—‘“ˆØ˜Ò!Ð!Ð!Ø—‘ �Ó'ˆØœŸ™ÑÐÐà—‘“×"Ñ"¨%Ð"Ó0ˆØ˜Ò!Ð!Ð!à—‘˜q�Ó!ˆØ˜Ò!Ð!Ñ!r   c                 ó°  — t        j                  dd¬«      }|j                  j                  dd«      }|j	                  d¬«      }|d   }t        j                  ||«       |j	                  d¬«      }|d d …df   t        d	¬
«      j                  d«      z   }t        j                  ||«       |j	                  d ¬«      }|j	                  «       }||k(  sJ ‚y )Nz14 daysé   )Úperiodsé   r=   r   r_   r4   rA   r0   rO   )	r   Útimedelta_ranger`   rH   r   rJ   rK   r   rT   )r   r   rL   r   r,   s        r   Útest_mean_2dzTestReductions.test_mean_2dÍ   s¾   € Ü× Ñ  °AÔ6ˆØ�i‰i×Ñ  1Ó%ˆà—‘˜q�Ó!ˆØ�q‘6ˆÜ
×'Ñ'¨°Ô9à—‘˜q�Ó!ˆØ’q˜!�t‘9œy¨rÔ2×:Ñ:¸4Ó@Ñ@ˆÜ
×'Ñ'¨°Ô9à—‘˜t�Ó$ˆØ—8‘8“:ˆØ˜Ò!Ð!Ñ!r   N)Ú__name__Ú
__module__Ú__qualname__ÚpytestÚmarkÚparametrizer   r    r-   r6   r9   rM   r   rT   r   Ú	TimestamprY   r]   rc   ri   © r   r   r   r      s»   „ Ø‡[�[×Ñ˜VÒ%LÓMñ ó Nð ò
&ò ò("òB"ò:ð8 ‡[�[×ÑØá�a‹L× Ñ  Ó&ØˆB�L‰L˜Ó&ØˆB�L‰L˜¨%Ô0ØˆB�L‰L˜¨,Ô7ð		
óñ$óð$ò: ò&"ó$"r   r   )Únumpyr8   rm   Úpandasr   r   Úpandas._testingÚ_testingrJ   Úpandas.corer   Úpandas.core.arraysr   r   rq   r   r   ú<module>rx      s&   ðÛ Û ã Ý Ý Ý Ý -÷P"ò P"r   