
    Mh                     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        V/var/www/html/env/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Ub   eUR                  U:g  R	                  5       (       a   eg )NT)r
   r   inplace)r   r   r   r   r   )r   r   r   
frame_copyreturn_values        r   test_inplace_clip#TestDataFrameClip.test_inplace_clip   sh    ##%,,. %%'
!V4P###%%/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,   result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  Un[        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   rJ   s               r   test_clip_against_series*TestDataFrameClip.test_clip_against_series;   sW    ryy,,Q/??	JKBII))!,<<TBC!V779WWR!WW=
J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   rY   )rZ   r[   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  Un[        R                  " XxS	S
9  g )N)r8   r<   r9   rX   )r\   g       @g      "@)onetwothree)abc)rB   index)         )r   r
   rO   r   T)check_exact)r&   arrayr   r   rB   rd   rE   rF   )	r   r   r   rO   resarrr   rJ   rK   s	            r   test_clip_against_list_like-TestDataFrameClip.test_clip_against_list_likeX   so     hh/JK0
 U)$XS*:*:(..QF
fDAr   rO   )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%   )rO   )r   r&   r'   r(   r)   r   rE   rF   )	r   rO   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   )rs   r:   r;   rr   r%   )r;   rs   rr   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Sn[         R                  " [        US9   UR                  SS	[        R                  /S
S9nSSS5        [        / SQ/ SQ/ SQS.5      n[         R                  " WU5        UR                  SS	[        R                  /SS9n[        / SQ/ SQ/ SQS.5      n[         R                  " XE5        / SQ/ SQS.n[        U5      n[        SS[        R                  SS/5      n[         R                  " [        US9   UR                  US
S9nSSS5        [        / SQ/ SQS.5      n[         R                  " XE5        g! , (       d  f       GN= f! , (       d  f       NI= f)z&Should process np.nan argument as Noner	   r6   )r=   re   rf   )rg      	   )col_0col_1col_2z<Downcasting behavior in Series and DataFrame methods 'where')matchr=   re   r   )r   rO   N)r=   re   r7   r%   )r=   r=   r=   )re   re   rf   )r   r   r$   re   )irf   r   )r   r   r!   rf   r7   )r   r   r   rf   re   )r!   r   rf   r   r7   )	rE   rF   r   r&   rD   r   assert_produces_warningFutureWarningr   )r   r   r,   msgrJ   rK   datats           r   test_clip_with_na_args(TestDataFrameClip.test_clip_with_na_args   s    	k..rvv6D
k..RVV266.JKX YSTL ''SAWWAq"&&>W:F B)iH
 	fh/1bff~A6)iH
 	f/ +5GHt_Ar2661a()''SAWW11W-F B'7BRST
f/) BA" BAs   &"GG'
G$'
G5c                     [        S/ SQ05      nUR                  SS9n[        S/ SQ05      n[        R                  " X#5        g )Nra   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%   re   r7   rC   r   )r   r,   rK   rJ   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   rL   pytestmarkparametrizerV   r&   asarrayrl   ro   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* 0D0	0r   r   )
numpyr&   r   pandasr   r   pandas._testing_testingrE   r   r   r   r   <module>r      s!      |0 |0r   