
    Mh0                         S SK Jr  S SKrS SKrS SKrS SKrS SKJr  S SK	J
r
  S SKrS SKJrJrJrJr  S SKJr   " S S5      rg)    )deepcopyN)using_pyarrow_string_dtype)option_context)	DataFrameSeries
date_rangetimedelta_rangec            
          \ rS rSrS rS rS rS rS rS r	\
R                  R                  \" 5       SS	9S
 5       rS rS rS rS rS rS rS rS r\
R                  R/                  S\" 5       \" S/S9\" S/S9\" S/ 05      /5      S 5       rS rS rS rS rS rS r\
R                  R/                  S/ SQ5      S  5       r S! r!S" r"S#r#g$)%TestDataFrameMisc   c                    US   nUR                   S:X  d   eUR                  S5      nUR                   S:X  d   eUR                  S S 2S4   nUR                   S:X  d   eUR                  S S  nUR                   S:X  d   eg )NAB)namepoploc)selffloat_framess2s       M/var/www/html/env/lib/python3.13/site-packages/pandas/tests/frame/test_api.pytest_getitem_pop_assign_name.TestDataFrameMisc.test_getitem_pop_assign_name   sy    vv}}OOC vv}}OOAsF#vv}}UU1Xww#~~    c                 t   UnUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      S:X  d   eUR                  S5      UR                  L d   eUR                  S5      UR                  L d   e[
        R                  " [        SS9   UR                  S5        S S S 5        [
        R                  " [        S	S9   UR                  S
5        S S S 5        [
        R                  " [        SS9   UR                  S 5        S S S 5        [
        R                  " [        SS9   UR                  S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       Nd= f! , (       d  f       g = f)Nr      indexrowscolumnszNo axis namedmatch   zNo axis.*foofoozNo axis.*None)_get_axis_number_get_axis_name	_get_axisr   r   pytestraises
ValueError)r   r   fs      r   test_get_axisTestDataFrameMisc.test_get_axis#   s   !!!$)))!!!$)))!!'*a///!!&)Q...!!),111"g---"i///(G333'7222	*i777{{1~((({{1~***]]:_=q! > ]]:^<U# = ]]:_=T" > ]]:_=t$ >= >= =< >= >=s0   G65H(HH)6
H
H
H&)
H7c                     [         R                  " [        SS9   UR                  U;     S S S 5        g ! , (       d  f       g = f)Nzunhashable type: 'Index'r    )r'   r(   	TypeErrorr   )r   r   s     r   test_column_contains_raises-TestDataFrameMisc.test_column_contains_raises@   s,    ]]9,FG;. HGGs	   3
Ac                 X   [        [        S5      [        S5      /[        S5      S9n[        S5       H  nU[        U5      ;   a  M   e   [        UR	                  S5      [
        5      (       d   e[        [        S5      [        S5      /[        R                  R                  [        [        SS5      5      5      S9n[        S5       H  nU[        U5      ;   a  M   e   [        S5       H  nU[        U5      ;  a  M   e   [        UR	                  S5      [         5      (       d   eg )NabcdefghABCDr   r   EFGH)
r   listdir
isinstance__getitem__r   pd
MultiIndexfrom_tupleszip)r   dfkeys      r   test_tab_completion%TestDataFrameMisc.test_tab_completionD   s    Vd6l3T&\J<C#b'>!>  "..-v6666 &\4<(MM--d3vv3F.GH
 <C#b'>!>  <Cc"g%%%  "..-y9999r   c                    [        S5       Vs/ s H  nS[        U5      -   PM     nn[        S5      [        S5      /n[        X2S9nS[        U5      ;   d   eS[        U5      ;  d   e[	        SS5         [        X2S9nS[        U5      ;   d   eS	[        U5      ;  d   e S S S 5        [	        SS 5         [        X2S9nS
[        U5      ;   d   e S S S 5        g s  snf ! , (       d  f       ND= f! , (       d  f       g = f)Ni  ar5   a99a100zdisplay.max_dir_itemsi,  a299a300a419)rangestrr   r8   r   )r   ir   valuesr?   s        r   test_display_max_dir_items,TestDataFrameMisc.test_display_max_dir_itemsW   s    ).s4A3Q<4*eCj)v/ BSW$$$3S963BSW$$$R((( :
 3T:63BSW$$$ ;: 5 :9
 ;:s   C5,C 6C1 
C.1
C?c                 ,   [        5       n[        S/5      nSn[        R                  " [        US9   [	        U5        S S S 5        [        R                  " [        US9   [	        U5        S S S 5        g ! , (       d  f       N<= f! , (       d  f       g = f)Nr   zunhashable type: 'DataFrame'r    )r   r'   r(   r.   hash)r   empty_framer?   msgs       r   test_not_hashable#TestDataFrameMisc.test_not_hashablej   sa    ks^,]]9C0H 1]]9C0 10 1000s   A4B4
B
Bzsurrogates not allowed)reasonc                 n    Sn[        U/ 05      nU[        U5      ;  d   eUR                  S   U:X  d   eg )Nu   r   )r   r8   r   )r   colnamer?   s      r   +test_column_name_contains_unicode_surrogate=TestDataFrameMisc.test_column_name_contains_unicode_surrogatet   s?     }%c"g%%%zz!}'''r   c                 4   [        [        R                  R                  S5      R	                  S5      5      n[        [        R                  R                  S5      R	                  S5      5      nSUR
                  l        UR
                  R                  b   eg )Nr"   )r      r#   )r   nprandomdefault_rngstandard_normalr   r   )r   df1df2s      r   test_new_empty_index&TestDataFrameMisc.test_new_empty_index}   si    		--a0@@HI		--a0@@HI		yy~~%%%r   c                    UR                  S5      nX!R                  L d   eUR                  S5      nX1R                  L d   eSn[        R                  " [
        US9   UR                  S5        S S S 5        g ! , (       d  f       g = f)Nr   r   zAxis must be 0 or 1 \(got 2\)r    r"   )_get_agg_axisr   r   r'   r(   r)   )r   r   colsidxrS   s        r   test_get_agg_axis#TestDataFrameMisc.test_get_agg_axis   ss    ((+*****''*'''''.]]:S1%%a( 211s   A99
Bc                     [        5       nUR                  (       d   eUR                  (       a   eUR                  (       a   e[        / SQ/ SQS.[        R                  " S5      S9nUS	 UR                  (       a   eg )N)g      ?g       @g      @rD   bcr   r   r\   r   r   )r   emptyr]   arange)r   r   float_string_framerR   r?   s        r   
test_emptyTestDataFrameMisc.test_empty   sg    k    $$$$%++++ _?C299UV<XsG88|8r   c                     [        U5      [        UR                  5      :X  d   eUSS/   R                  nUR                  SS/S9R                  n[        R
                  " X#5        g )Nr   r   r5   )lenr   rM   reindextmassert_almost_equal)r   r   arrexpecteds       r   test_lenTestDataFrameMisc.test_len   sa    ;3{'8'8#9999 3*%,,&&Sz&:AA
s-r   c                     UnUR                  SS9nUR                  SS9n[        R                  " XC5        UR                  SS9nUR                  SS9n[        R                  " XC5        g )Nr   )axisr   r   r   )sumry   assert_series_equal)r   r   r*   r|   results        r   test_axis_aliases#TestDataFrameMisc.test_axis_aliases   sc     55a5=G$
v055a5=I&
v0r   c                     [         R                  " [        R                  5      (       d   e[         R                  " [        R                  5      (       d   eg N)pydocgetdocr   r   r   )r   s    r   test_class_axis!TestDataFrameMisc.test_class_axis   s8     ||IOO,,,,||I--....r   c                 r    UR                   nUR                  5        H  u  p4UR                  U:X  a  M   e   g r   )_seriesitemsr   )r   rs   serieskvs        r   test_series_put_names'TestDataFrameMisc.test_series_put_names   s/    #++LLNDA66Q;; #r   c                    [        / SQ5      nUR                  (       a   e[        S/S/S9nUR                  (       a   e[        SS/SS/S9R                  5       nUR                  (       d   eUR                  R                  (       d   eg )N)r   r"   r\   r   )r   r   rD   rm   rn   d)r   rq   dropnaTr   r?   s     r   test_empty_nonzero$TestDataFrameMisc.test_empty_nonzero   sl    y!88|aS1#.88|c3Z#s<CCExxxttzzzr   r?   r   rp   r5   c                 d    UR                   (       d   eUR                  R                   (       d   eg r   )rq   r   r   s     r   test_empty_like!TestDataFrameMisc.test_empty_like   s     xxxttzzzr   c                     [        [        SSS9[        SSS9S.5      nUR                  nUR                  R                  5       n[        [        R                  " S5      S0SS9n[        R                  " X45        g )	N20130101
   )periodsz1 dayro   objectcount)r   )r   r   r	   r   dtypesvalue_countsr   r]   dtypery   r   )r   r?   tr   r|   s        r   test_with_datetimelikes)TestDataFrameMisc.test_with_datetimelikes   sj    
B7$Wb9
 DD&&(288H-r2A
v0r   c                 h    [        U5      nSUR                  S'   UR                  U5      (       a   eg )Nr   )r   r   )r   r   equals)r   r   cps      r   test_deepcopyTestDataFrameMisc.test_deepcopy   s3    k"v%%b)))))r   c                    [        / SQ/ SQ/ SQS.5      nS nS nU" UR                  5       U5        S nU" UR                  S5      U5        S	 nU" UR                  5       U5        S
 nU" UR                  5       U5        S nU" UR                  5       U5        S nU" UR                  5       U5        S nU" UR                  5       U5        S nU" UR                  5       U5        UR                  5       S   nS nU" UR                  S5      S   U5        S nU" UR                  5       U5        S nU" UR                  5       U5        S nU" UR                  5       U5        g )N)r#   barbazqux)r   r   r   r   )r   r"   r\      rl   c                     U" U 5      nUb   eg r    )baser*   r   s      r   _check_f<TestDataFrameMisc.test_inplace_return_self.<locals>._check_f   s    tWF>!>r   c                 "    U R                  SSS9$ )NrD   Tinplace)	set_indexxs    r   <lambda><TestDataFrameMisc.test_inplace_return_self.<locals>.<lambda>   s    akk#tk4r   c                      U R                  SS9$ NTr   reset_indexr   s    r   r   r      s    ammDm1r   rD   c                      U R                  SS9$ r   )drop_duplicatesr   s    r   r   r      s    a'''5r   c                 "    U R                  SSS9$ )Nrm   Tr   )sort_valuesr   s    r   r   r      s    ammCm6r   c                      U R                  SS9$ r   )
sort_indexr   s    r   r   r      s    all4l0r   c                 "    U R                  SSS9$ Nr   Tr   fillnar   s    r   r   r         ahhq$h/r   c                 $    U R                  SSSS9$ Nr   r   Tr   replacer   s    r   r   r         aii1di3r   c                 &    U R                  SS0SS9$ Nr   r#   Tr   renamer   s    r   r   r         ahh5z4h8r   rn   c                 "    U R                  SSS9$ )NT)r   dropr   r   s    r   r   r     s    ammDtm<r   c                 "    U R                  SSS9$ r   r   r   s    r   r   r     r   r   c                 $    U R                  SSSS9$ r   r   r   s    r   r   r     r   r   c                 &    U R                  SS0SS9$ r   r   r   s    r   r   r     r   r   )r   copyr   )r   datar   r*   r   s        r   test_inplace_return_self*TestDataFrameMisc.test_inplace_return_self   sN    .\U
	" 5a  2$a( 6a  7a  1a  0a  4a  9a  IIK =$S)1- 01 41 91r   c                 d   [         R                  " SSS9  SSKJn  U[        L a  SnOSnUR                  U5        [        R                  " S SS	9   U" S
5         [        UR                  R                  SS5      5        S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)NIPythonz6.0.0)
minversionr   )provisionalcompleterz/from pandas import DataFrame; obj = DataFrame()z5from pandas import Series; obj = Series(dtype=object)F)raise_on_extra_warningsignorezobj.r   )r'   importorskipIPython.core.completerr   r   run_cellry   assert_produces_warningr7   	Completercompletions)r   ipframe_or_seriesr   codes        r   test_tab_complete_warning+TestDataFrameMisc.test_tab_complete_warning!  s    I':?i'DDJD
D ''eL%h/R\\--fa89 0 ML// MLs$   	B!&B?B!
B	B!!
B/c                     [        SSS/05      nUR                  0 :X  d   eSUR                  S'   UR                  [        S9nUR                  SS0:X  d   eg )Nr   r"   r\   r   versionr5   r   attrsr   rK   r   r?   r   s      r   
test_attrsTestDataFrameMisc.test_attrs2  sX    aV}%xx2~~3'||	1~---r   c                    [        SSS/05      nUR                  0 :X  d   eSS1UR                  S'   UR                  [        S9nUR                  UR                  :X  d   eUR                  S   UR                  S   Ld   eg )Nr   r"   r\   spamhamtagsr5   r   r   s      r   test_attrs_deepcopy%TestDataFrameMisc.test_attrs_deepcopy:  s|    aV}%xx2~~"E?3'||rxx'''||F#288F+;;;;r   allows_duplicate_labels)TFNc                    [        SSS/05      nSnU[        L a  US   nSnUR                  US9nUc  UR                  R                  SL d   eOUR                  R                  UL d   eXWLd   eUR                  R                  SL d   eU[        L a3  [
        R                  " UR                  UR                  5      (       d   eO8[
        R                  " US   R                  US   R                  5      (       d   e[        R                  " U5         SUR                  U'   S S S 5        U(       a  UR                  U   S:X  d   eOCUR                  U   S:X  d   e[        R                  " U5         SUR                  U'   S S S 5        UR                  SUS9nS	UR                  U'   UR                  U   S:X  d   eg ! , (       d  f       N= f! , (       d  f       NT= f)
Nr   r   r"   )r   r   r   )r  T)r   r  r   )r   r   	set_flagsflagsr  r]   may_share_memoryrM   ry   assert_cow_warningiloc)r   r  r   using_copy_on_writewarn_copy_on_writeobjr@   r   s           r   test_set_flags TestDataFrameMisc.test_set_flagsC  s    q!f&f$c(CC7NO"*<<774???<<77;RRRR     yy00D888 f$&&szz6==AAAA&&s3xs8J8JKKKK""#56 FKK 788C=A%%%88C=A%%%&&'9:#$C  ; /F  
 Cxx}!!! 76 ;:s   G6G
G
G"c                     [        5       nSn[        R                  " [        US9   UR	                  [
        R                  " S5      R                  SSS5      5        S S S 5        g ! , (       d  f       g = f)Nz<'DataFrame' object has no attribute '_constructor_expanddim'r       r\   )r   r'   r(   AttributeError_constructor_expanddimr]   rr   reshaper   r?   rS   s      r   test_constructor_expanddim,TestDataFrameMisc.test_constructor_expanddimv  sO     [L]]>5%%biim&;&;Aq!&DE 655s   7A&&
A4c                     [         R                  " S5        [        5       nSn[        R                  " [
        USS9   [        R                  " U5        S S S 5        g ! , (       d  f       g = f)Njinja2zDataFrame._data is deprecatedF)r!   check_stacklevel)r'   r   r   ry   r   DeprecationWarninginspect
getmembersr  s      r   test_inspect_getmembers)TestDataFrameMisc.test_inspect_getmembers  sQ    H%[-''cE
 r"
 
 
s   A
A+r   N)$__name__
__module____qualname____firstlineno__r   r+   r/   rA   rN   rT   r'   markxfailr   rY   rc   ri   rt   r}   r   r   r   r   parametrizer   r   r   r   r   r   r   r   r  r  r  __static_attributes__r   r   r   r   r      s   %:/:&%& [[13<TU( V(&	)
.
1/
 [[KQC qc"q"g		
1*
>@:".< [[68KL0" M0"dF#r   r   )r   r   r  r   numpyr]   r'   pandas._configr   pandas._config.configr   pandasr;   r   r   r   r	   pandas._testing_testingry   r   r   r   r   <module>r+     s8         5 0   s# s#r   