
    A>i                     D    S SK rS SKrS SKJrJr  S SKJr   " S S5      r	g)    N)	DataFrameSeriesc                      \ rS rSrS rS rS rS r\R                  R                  SSS/5      S	 5       r\R                  R                  SSS/5      \R                  R                  S
/ SQ\R                  " / SQ5      /5      \R                  R                  SS/ SQ/ SQ/ SQ/4S/ SQ/ SQ/ SQ/4/5      S 5       5       5       r\R                  R                  S/ SQ5      S 5       rS rS rS rS rSrg)TestDataFrameClip   c                 "   UR                  5       R                  5       nUR                  5       nUR                  X"S9nUR                  U:g  R	                  5       (       a   eUR                  UR                  :H  R                  5       (       d   eg )Nupperlower)mediancopyclipvaluesanyall)selffloat_framer   originaldoubles        c/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pandas/tests/frame/methods/test_clip.py	test_clipTestDataFrameClip.test_clip   s~    ##%,,.##%!!!=MMV+002222 ""hoo5::<<<<    c                     UR                  5       R                  5       nUR                  5       nUR                  X"SS9nXCL d   eUR                  U:g  R	                  5       (       a   eg )NT)r
   r   inplace)r   r   r   r   r   )r   r   r   
frame_copyresults        r   test_inplace_clip#TestDataFrameClip.test_inplace_clip   sh    ##%,,. %%'
vTJ###%%/4466666r   c                 "   [        [        R                  R                  S5      R	                  S5      5      nS H  u  p#UR                  X#5      n[        X#5      [        X25      p2UR                  U:*  nUR                  U:  nU) U) -  nUR                  U   U:H  R                  5       (       d   eUR                  U   U:H  R                  5       (       d   eUR                  U   UR                  U   :H  R                  5       (       a  M   e   g )N     r!   ))   )r%   r$   )
r   nprandomdefault_rngstandard_normalr   minmaxr   r   )r   dflbub
clipped_dflb_maskub_maskmasks           r   test_dataframe_clip%TestDataFrameClip.test_dataframe_clip   s    ryy,,Q/??	JK(FBJ[#b+ii2oGii2oG8wh&D%%g."499;;;;%%g."499;;;;%%d+ryy>CCEEEE )r   c                 r   [        / SQS[        R                  S/S.5      nUR                  SS5      n[        / SQS[        R                  S/S.5      n[        R
                  " X#5        [        / S	Q/ S
Q// SQS9nUR                  nUR                  SS9R                  n[        R                  " X#5        g )Nr%   r!            ?      @)ABr%   r!   )r%   r!   r!          @)r%   r!   g333333@)r7      gffffff@)foobarbazcolumnsr7   r
   )r   r&   nanr   tmassert_frame_equaldtypesassert_series_equalr   r,   r   expecteds       r   test_clip_mixed_numeric)TestDataFrameClip.test_clip_mixed_numeric.   s     Yc2663-?@AA9C3EFG
f/[1;PQ99q!((
v0r   r   TFc                    [        [        R                  R                  S5      R	                  S5      5      n[        [        R                  R                  S5      R	                  S5      5      nUS-   nUR                  5       nUR                  X4SUS9nU(       a  XbL d   e[        S5       H  nUR                  S S 2U4   U:*  nUR                  S S 2U4   U:  n	U) U	) -  n
UR                  X4   n[        R                  " XU   SS9  UR                  U:X  d   eUR                  X4   n[        R                  " XU	   SS9  UR                  U:X  d   e[        R                  " UR                  X4   UR                  X4   5        M     g )	Nr!   r"   r#   r%   r   )axisr   F)check_names)r   r&   r'   r(   r)   r   r   r   rangeiloclocrE   rH   name)r   r   r,   r-   r.   r   r/   ir0   r1   r2   r   s               r   test_clip_against_series*TestDataFrameClip.test_clip_against_series;   s^    ryy,,Q/??	JKBII))!,<<TBC!V779WWR!WW=
###qAmmAqD)R/GmmAqD)R/G8wh&D^^GJ/F""6g;EJ;;!###^^GJ/F""6g;EJ;;!###"":>>$'#:BFF47OL r   r   )r!   r7   r=   zaxis,resr   )r<   r<   r9         @      @      @)      @r[   r[   r%   )r<   r9   rX   )rY   rZ   r[   c                    [         R                  " / SQ/ SQ/ SQ/5      n[        U/ SQ/ SQS9nUR                  U/ SQX1S9n[        XFR                  UR
                  S9nU(       a  XvL d   e[        R                  " XxS	S
9  g )N)r8   r<   r9   rW   )r[   g       @g      "@)onetwothree)abc)rB   index)         )r   r
   rN   r   T)check_exact)r&   arrayr   r   rB   rc   rE   rF   )	r   r   r   rN   resarrr   r   rJ   s	            r   test_clip_against_list_like-TestDataFrameClip.test_clip_against_list_likeX   sv     hh/JK0
 U)$XS*:*:(..Q%%%
fDAr   rN   )r   r%   Nc                    [        [        R                  R                  S5      R	                  S5      5      n[        [        R                  R                  S5      R	                  S5      5      nUS-   nUR                  X4US9nX#:*  nX$:  nU) U) -  n[        R                  " XV   X6   5        [        R                  " XW   XG   5        [        R                  " XX   X(   5        g )Nr!   r"   r%   )rN   )r   r&   r'   r(   r)   r   rE   rF   )	r   rN   r,   r-   r.   r/   r0   r1   r2   s	            r   test_clip_against_frame)TestDataFrameClip.test_clip_against_framep   s    ryy,,Q/??	JKryy,,Q/??	JK!VWWR$W/
((x7("
j12;?
j12;?
j.9r   c                    [        [        R                  R                  S5      R	                  S5      / SQS9n[        [        R                  R                  S5      R	                  S5      / SQS9n[        UR
                  S-
  / SQS9nUR                  SUS	9nUR                  SX!R                     S	9nUR                  US
S	9nUR                  X1R                     S
S	9nUR                  X2S	9nUR                  X1R                     X!R                     S	9n	[        R                  " XE5        [        R                  " Xg5        [        R                  " X5        g )Nr!   )r#   r=   )r:   r;   CDrA   )rr   r:   r;   rq   r%   )r;   rr   rq   r:   r   )r   r
   r7   )
r   r&   r'   r(   r)   r   r   rB   rE   rF   )
r   df1df2df3result_upperexpected_upperresult_lowerexpected_lowerresult_lower_upperexpected_lower_uppers
             r   #test_clip_against_unordered_columns5TestDataFrameClip.test_clip_against_unordered_columns   s"   II!!!$44Y?(
 II!!!$44Y?(
 

Q0DExxasx3[[1ABxxcx3KK(8B XXCX;"xxc++.>c++FVxW
l;
l;
0Gr   c                 *   [         R                  " UR                  [        R                  5      U5        [         R                  " UR                  [        R                  [        R                  S9U5        [        / SQ/ SQ/ SQS.5      nUR                  SS[        R                  /SS	9n[        [        / S
QSS9/ SQ/ SQS.5      n[         R                  " X45        UR                  SS[        R                  /SS	9n[        / SQ/ SQ/ SQS.5      n[         R                  " X45        / SQ/ SQS.n[        U5      n[        SS[        R                  SS/5      nUR                  USS	9n[        / SQ/ SQS.SS9n[         R                  " X45        g)z&Should process np.nan argument as Noner	   r6   )r=   rd   re   )rf      	   )col_0col_1col_2r=   rd   r   )r   rN   )r=   rd   r7   float)dtyper%   )r=   r=   r=   )rd   rd   re   )r   r   r$   rd   )ire   r   )r   r   r!   re   r7   )r   r   r   re   rd   )r!   r   re   r   r7   N)rE   rF   r   r&   rD   r   r   )r   r   r,   r   rJ   datats          r   test_clip_with_na_args(TestDataFrameClip.test_clip_with_na_args   sO    	k..rvv6D
k..RVV266.JKX YST1bff~A6	9""
 	f/1bff~A6)iH
 	f/ +5GHt_Ar2661a()qq)&1AB'
 	f/r   c                     [        S/ SQ05      nUR                  SS9n[        S/ SQ05      n[        R                  " X#5        g )Nr`   r6         ?)r   )r   r<   r9   r   r   rE   rF   rI   s       r   #test_clip_int_data_with_float_bound5TestDataFrameClip.test_clip_int_data_with_float_bound   s>    Y'(s#c?34
f/r   c                     [        SS/5      n[        SS/5      nUR                  S/5      n[        R                  " X25        [        SS/5      nUR                  S/S9n[        R                  " X25        g )Nr%   rd   r7   rC   r   )r   r,   rJ   r   s       r   test_clip_with_list_bound+TestDataFrameClip.test_clip_with_list_bound   si    1vaV$!
f/aV$s#
f/r    N)__name__
__module____qualname____firstlineno__r   r   r3   rK   pytestmarkparametrizerU   r&   asarrayrk   rn   r|   r   r   r   __static_attributes__r   r   r   r   r      s   =7F1 [[Yu6M 7M8 [[Yu6[[Wy"**Y2G&HI[[/?CD/?CD	
B J 7B [[V\2: 3:H*!0F0	0r   r   )
numpyr&   r   pandasr   r   pandas._testing_testingrE   r   r   r   r   <module>r      s!      }0 }0r   