Ë
    ž…jÎ  ã                   óL   — d Z ddlZddlZddlmZmZmZ ddlm	Z
  G d„ d«      Zy)z[
Tests for DataFrame cumulative operations

See also
--------
tests.series.test_cumulative
é    N)Ú	DataFrameÚSeriesÚ	Timestampc                   ó®   — e Zd Zd„ Zd„ Zd„ Zd„ Zej                  j                  dg d¢«      ej                  j                  ddd	g«      d
„ «       «       Z
y)ÚTestDataFrameCumulativeOpsc                 ót  — t        dt        j                  d«      it        j                  d«      ¬«      }|j                  «        |j	                  «        |j                  «        t        t        j                  d«      j                  dd«      t        d«      t        d«      ¬«      }|j                  «        y )NÚAé   )Úindexé   é   )r   Úcolumns)r   ÚnpÚarangeÚcummaxÚcumminÚcumsumÚreshapeÚrange)ÚselfÚdfÚdms      úc/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/frame/test_cumulative.pyÚtest_cumulative_ops_smokez4TestDataFrameCumulativeOps.test_cumulative_ops_smoke   ss   € ä˜œRŸY™Y r›]Ð+´2·9±9¸R³=ÔAˆØ
�	‰	ŒØ
�	‰	ŒØ
�	‰	Œä”r—y‘y “}×,Ñ,¨Q°Ó2¼%À»(ÌEÐRSËHÔUˆà
�	‰	�ó    c                 óö  — t         j                  |j                  dd…df<   t         j                  |j                  dd…df<   t         j                  |j                  dd …df<   |j                  d«      j	                  t
        «      }|j                  d«       |j                  d«       |j                  d«      j	                  t         j                  «      }|j                  d«       |j                  d«       y )Nr   é
   r   é   é   é   )r   ÚnanÚilocÚfillnaÚastypeÚintÚcumprodÚint32)r   Údatetime_framer   s      r   Útest_cumprod_smokez-TestDataFrameCumulativeOps.test_cumprod_smoke#   s¾   € Ü')§v¡vˆ×Ñ˜A˜b˜D !˜GÑ$Ü(*¯©ˆ×Ñ˜B˜r˜E 1˜HÑ%Ü&(§f¡fˆ×Ñ˜B™C ˜FÑ#ð ×"Ñ" 1Ó%×,Ñ,¬SÓ1ˆØ
�
‰
�1ŒØ
�
‰
�1Œð ×"Ñ" 1Ó%×,Ñ,¬R¯X©XÓ6ˆØ
�
‰
�1ŒØ
�
‰
�1�r   c                 óD  — t         j                  |j                  dd…df<   t         j                  |j                  dd…df<   t         j                  |j                  dd …df<    t        ||«      «       }|j	                  t        t
        |«      «      }t        j                  ||«        t        ||«      d¬«      }|j	                  t        t
        |«      d¬«      }t        j                  ||«       t        j                  |«      t        j                  |«      k(  sJ ‚y )Nr   r   r   r   r   r    ©Úaxis)	r   r!   r"   ÚgetattrÚapplyr   ÚtmÚassert_frame_equalÚshape)r   r(   Úall_numeric_accumulationsÚresultÚexpecteds        r   Ú&test_cumulative_ops_match_series_applyzATestDataFrameCumulativeOps.test_cumulative_ops_match_series_apply2   s   € ô (*§v¡vˆ×Ñ˜A˜b˜D !˜GÑ$Ü(*¯©ˆ×Ñ˜B˜r˜E 1˜HÑ%Ü&(§f¡fˆ×Ñ˜B™C ˜FÑ#ð D”˜Ð)BÓCÓEˆØ!×'Ñ'¬´Ð8QÓ(RÓSˆÜ
×Ñ˜f hÔ/ð D”˜Ð)BÓCÈÔKˆØ!×'Ñ'Ü”FÐ5Ó6¸Qð (ó 
ˆô 	×Ñ˜f hÔ/ô �x‰x˜Ó¤2§8¡8¨NÓ#;Ò;Ð;Ñ;r   c                 ó&  — t        g d¢g d¢g d¢dœ«      }|j                  «       }t        t        g d¢t        j                  ¬«      t        g d¢t        j
                  ¬«      |d   j                  «       dœ«      }t        j                  ||«       y )N©r   r    é   )r   r    g      @©TFF)r	   ÚBÚC)r   r8   é   )Údtyper;   )r   r   r   r   Úint64Úfloat64r/   r0   )r   r   r3   r4   s       r   Útest_cumsum_preserve_dtypesz6TestDataFrameCumulativeOps.test_cumsum_preserve_dtypesH   sj   € äšYª[Ò?SÑTÓUˆà—‘“ˆääšI¬R¯X©XÔ6ÜšI¬R¯Z©ZÔ8Ø˜‘W—^‘^Ó%ñó
ˆô 	×Ñ˜f hÕ/r   Úmethod)r   r&   r   r   r,   r   r   c                 ó   — t        g d¢g d¢g d¢g d¢t        ddd«      t        ddd«      t        ddd«      gd	œ«      }|j                  d
dgd¬«      } t        ||«      |d¬«      } t        ||«      |«      }t	        j
                  ||«       y )Nr7   r9   )ÚaÚbÚc)g      ð?g      @g      @iâ  r   iã  iä  )r%   ÚboolÚstringÚfloatÚdatetimerG   rI   r+   T)r,   Únumeric_only)r   r   Údropr-   r/   r0   )r   rA   r,   r   Údf_numeric_onlyr3   r4   s          r   Útest_numeric_only_flagz1TestDataFrameCumulativeOps.test_numeric_only_flagW   s™   € ô â Ú,Ú)Ú(ä˜d A qÓ)Ü˜d A qÓ)Ü˜d A qÓ)ðñ
ó
ˆð Ÿ'™' 8¨ZÐ"8¸q˜'ÓAˆà$”˜˜VÓ$¨$¸TÔBˆØ3”7˜?¨FÓ3°DÓ9ˆÜ
×Ñ˜f hÕ/r   N)Ú__name__Ú
__module__Ú__qualname__r   r)   r5   r@   ÚpytestÚmarkÚparametrizerM   © r   r   r   r      sZ   „ ò	òò<ò,0ð ‡[�[×Ñ˜XÒ'PÓQØ‡[�[×Ñ˜V a¨ VÓ,ñ0ó -ó Rñ0r   r   )Ú__doc__Únumpyr   rQ   Úpandasr   r   r   Úpandas._testingÚ_testingr/   r   rT   r   r   ú<module>rZ      s-   ðñó Û ÷ñ õ
 ÷W0ò W0r   