
    4i                     L    S r SSKrSSKrSSKJrJrJr  SSKJ	r
   " S S5      rg)z[
Tests for DataFrame cumulative operations

See also
--------
tests.series.test_cumulative
    N)	DataFrameSeries	Timestampc                       \ rS rSrS rS rS rS r\R                  R                  S/ SQ5      \R                  R                  SS	S
/5      S 5       5       rSrg)TestDataFrameCumulativeOps   c                 n   [        S[        R                  " S5      0[        R                  " S5      S9nUR                  5         UR	                  5         UR                  5         [        [        R                  " S5      R                  SS5      [        S5      [        S5      S9nUR                  5         g )NAr   )index      )r   columns)r   nparangecummaxcummincumsumreshaperange)selfdfdms      d/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/pandas/tests/frame/test_cumulative.pytest_cumulative_ops_smoke4TestDataFrameCumulativeOps.test_cumulative_ops_smoke   ss    RYYr]+299R=A
		
		
		ryy},,Q2%(ERSHU
		    c                    [         R                  UR                  SS2S4'   [         R                  UR                  SS2S4'   [         R                  UR                  SS 2S4'   UR                  S5      R	                  [
        5      nUR                  S5        UR                  S5        UR                  S5      R	                  [         R                  5      nUR                  S5        UR                  S5        g )Nr   
   r            )r   nanilocfillnaastypeintcumprodint32)r   datetime_framer   s      r   test_cumprod_smoke-TestDataFrameCumulativeOps.test_cumprod_smoke#   s    ')vvAbD!G$(*BrE1H%&(ffBCF# ""1%,,S1


1


1 ""1%,,RXX6


1


1r   c                 :   [         R                  UR                  SS2S4'   [         R                  UR                  SS2S4'   [         R                  UR                  SS 2S4'   [        X5      " 5       nUR	                  [        [
        U5      5      n[        R                  " X45        [        X5      " SS9nUR	                  [        [
        U5      SS9n[        R                  " X45        [         R                  " U5      [         R                  " U5      :X  d   eg )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_applyATestDataFrameCumulativeOps.test_cumulative_ops_match_series_apply2   s     (*vvAbD!G$(*BrE1H%&(ffBCF# CE!''8Q(RS
f/ CK!''F56Q ( 
 	f/ xx288N#;;;;r   c                    [        / SQ/ SQ/ SQS.5      nUR                  5       n[        [        / SQ[        R                  S9[        / SQ[        R
                  S9US   R                  5       S.5      n[        R                  " X#5        g )Nr    r!      )r    r!   g      @TFF)r
   BC)r    r;      )dtyper>   )r   r   r   r   int64float64r1   r2   )r   r   r5   r6   s       r   test_cumsum_preserve_dtypes6TestDataFrameCumulativeOps.test_cumsum_preserve_dtypesH   sh    Y[?STUIRXX6IRZZ8W^^%
 	f/r   method)r   r'   r   r   r.   r   r    c                    [        / SQ/ SQ/ SQ/ SQ[        SSS5      [        SSS5      [        SSS5      /S	.5      nUR                  S
S/SS9n[        X15      " USS9n[        XA5      " U5      n[        R
                  " XV5        g )Nr:   r<   )abc)g      ?g      @g      @i  r    i  i  )r&   boolstringfloatdatetimerK   rM   r-   T)r.   numeric_only)r   r   dropr/   r1   r2   )r   rE   r.   r   df_numeric_onlyr5   r6   s          r   test_numeric_only_flag1TestDataFrameCumulativeOps.test_numeric_only_flagW   s      ,)(dAq)dAq)dAq)

 ''8Z"8q'A$$TB?3D9
f/r    N)__name__
__module____qualname____firstlineno__r   r*   r7   rC   pytestmarkparametrizerQ   __static_attributes__rS   r   r   r   r      sZ    	<,0 [[X'PQ[[VaV,0 - R0r   r   )__doc__numpyr   rX   pandasr   r   r   pandas._testing_testingr1   r   rS   r   r   <module>ra      s-      
 W0 W0r   