Ë
    ¤ehóL  ã                   óÎ   — d dl Z d dlZd dlZd dlmc mZ d dlZ	d dlm
Z
mZmZmZmZmZmZmZmZmZ d dlmZ ej.                  d„ «       Zej.                  d„ «       Z G d„ d«      Zy)é    N)
ÚCategoricalÚCategoricalDtypeÚCategoricalIndexÚ	DataFrameÚIndexÚIntervalÚSeriesÚ	TimedeltaÚ	TimestampÚoption_contextc                  ó˜   — t        dt        j                  dd¬«      it        t	        d«      t        t	        d«      «      d¬«      ¬	«      S )
NÚAé   Úint64©ÚdtypeÚaabbcaÚcabÚB©r   Úname©Úindex©r   ÚnpÚaranger   Úlistr   © ó    úd/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/indexing/test_categorical.pyÚdfr!      sC   € äà”—‘˜1 GÔ,ð	
ô Ü�‹NÔ"2´4¸³;Ó"?Àcô
ô	ð r   c                  ó˜   — t        dt        j                  dd¬«      it        t	        d«      t        t	        d«      «      d¬«      ¬	«      S )
Nr   r   r   r   r   Úcaber   r   r   r   r   r   r    Údf2r$   $   sC   € äà”—‘˜1 GÔ,ð	
ô Ü�‹NÔ"2´4¸³<Ó"@Àsô
ô	ð r   c                   ól  — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	d„ Z
d	„ Zd
„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zej(                  j+                  dd ej,                  d ej0                  d«      ¬«      g«      d„ «       Zd„ Zd„ Zej(                  j+                  dg d¢g d¢g d¢g d¢gd„ ej:                  D «       ¢d„ ej<                  D «       ¢ ej@                  g d ¢e!¬!«      ‘ e"d"d#«       e"d#d$«       e"d$d%«      g‘ e#d&d"d"«       e#d&d'd"«       e#d&d(d"«      g‘ e$d"d)«       e$d'd)«       e$d(d*«      g‘d+„ ejJ                  D «       ¢e&jN                  jQ                  g d,¢«      j@                  ‘ e&jR                  d-d(¬.«      j@                  ‘ e&jT                  d/d(¬0«      j@                  ‘«      d1„ «       Z+d2„ Z,y3)4ÚTestCategoricalIndexc                 óŽ  — t        t        d«      «      }|j                  d   }t        t        d«      d¬«      j	                  |«      }|j
                  |k(  sJ ‚t        dg d¢it        |«      ¬«      }t        j                  ||«       |j                  «       }d	|j                  d<   t        t        d
«      d¬«      j	                  |«      }|j
                  |k(  sJ ‚t        dg d¢it        |«      ¬«      }t        j                  ||«       t        j                  t        d¬«      5  |j                  d    d d d «       |j                  «       }|j                  «       }|j                  j	                  t        «      |_        d|j                  d<   d|j                  d<   t        j                  ||«       y # 1 sw Y   Œ‚xY w)Nr   ÚaÚaaar   ©r   r   )r   é   é   r   é   r   )r-   r-   é   é   é   r-   z^'d'$©ÚmatchÚdé
   )r   r   Úlocr	   Úastyper   r   r   ÚtmÚassert_frame_equalÚcopyÚpytestÚraisesÚKeyErrorr   Úobject)Úselfr!   r   ÚresultÚbidxÚexpectedÚbidx2r$   s           r    Útest_loc_scalarz$TestCategoricalIndex.test_loc_scalar1   sm  € Ü ¤ e£Ó-ˆØ—‘˜‘ˆÜ”d˜5“k¨Ô,×3Ñ3°EÓ:ˆØ�z‰z˜UÒ"Ð"Ð"ä˜c¢9Ð-´U¸4³[ÔAˆÜ
×Ñ˜f hÔ/à�W‰W‹YˆØˆ�‰ˆs‰Ü”t˜H“~¨CÔ0×7Ñ7¸Ó>ˆØ�{‰{˜eÒ#Ð#Ð#ÜàÒ*ðô ˜“,ô	
ˆô 	×Ñ˜b (Ô+ô �]‰]œ8¨8Ô4ñ 	Ø�F‰F�3ŠK÷	ð �g‰g‹iˆØ—8‘8“:ˆØ!Ÿ™×.Ñ.¬vÓ6ˆŒØˆ�‰�SÑØˆ�‰�‰Ü
×Ñ˜c 8Õ,÷	ð 	ús   Ä*F;Æ;Gc           
      óh  — d|j                   d<   |j                  «       }d|j                   d<   t        t        d«      d¬«      }t	        dg d	¢it        |«      ¬
«      }t        j                  ||«       |j                  «       }d|j                   d<   t	        ddddddt        j                  gt        j                  t        j                  t        j                  t        j                  t        j                  t        j                  dgdœt        |«      ¬
«      }t        j                  ||«       y )Nr-   r(   r4   )r3   r   Úaabbcadr   r*   r   )r-   r-   r.   r/   r0   r-   g      $@r   )r3   ÚCr.   r/   r0   )r   rF   )	r5   r9   r   r   r   r7   r8   r   Únan)r>   r!   Údf3Úbidx3Ú	expected3Údf4s         r    Ú,test_loc_setitem_with_expansion_non_categoryzATestCategoricalIndex.test_loc_setitem_with_expansion_non_categoryQ   sï   € àˆ�‰ˆs‰ð �g‰g‹iˆØˆ�‰�ÑÜ”d˜9“o¨CÔ0ˆÜàÒ0ðô ˜“,ô	
ˆ	ô 	×Ñ˜c 9Ô-ð �g‰g‹iˆØˆ�‰�ÑÜà˜"˜a  A r¬2¯6©6Ð2Ü—f‘fœbŸf™f¤b§f¡f¬b¯f©f´b·f±f¼b¿f¹fÀbÐIñô ˜“,ô
ˆ	ô 	×Ñ˜c 9Õ-r   c                 ó‚   — t        j                  t        d¬«      5  |j                  d    d d d «       y # 1 sw Y   y xY w)Nz^1$r1   r+   ©r:   r;   r<   r5   )r>   r!   s     r    Ú$test_loc_getitem_scalar_non_categoryz9TestCategoricalIndex.test_loc_getitem_scalar_non_categorym   s0   € Ü�]‰]œ8¨5Ô1ñ 	Ø�F‰F�1ŠI÷	÷ 	ñ 	úó   œ5µ>c                 óÊ  — t        t        g d¢«      «      }|d d d…   }t        j                  g d¢t        j                  ¬«      }t        j                  |j                  «       |«       t        dt        j                  d«      dz   j                  d«      i«      }t        j                  |j                  g d	¢¬
«      |d<   t        dt        dd«      gddgd¬«      }|j                  d   }t        j                   ||«       t        dt        j                  dd«      j                  d«      it        j                  dd«      j                  d«      ¬«      }t        j                  |j                  g d	¢¬
«      |d<   |j                  dd }t        j"                  ||«       t        dt        dd«      gddgd¬«      }|j$                  d   }t        j                   ||«       y )N©r+   r.   r/   r0   éÿÿÿÿ)r0   r/   r.   r+   r   Úvalueéd   r+   r   )r   é   é2   éK   rU   )ÚbinsÚDé   r   rV   r4   ©r   r   é   r-   r   é	   é   )r	   r   r   Úarrayr   r7   Úassert_numpy_array_equalÚ	__array__r   r   r6   ÚpdÚcutrT   r   ÚilocÚassert_series_equalr8   r5   )r>   ÚcatÚreverseÚexpr!   rA   r?   s          r    Útest_slicingz!TestCategoricalIndex.test_slicingq   sz  € Ü”[¢Ó.Ó/ˆØ‘d˜�d‘)ˆÜ�h‰h’|¬2¯8©8Ô4ˆÜ
×#Ñ# G×$5Ñ$5Ó$7¸Ô=ä˜¤"§)¡)¨C£.°1Ñ"4×!<Ñ!<¸WÓ!EÐFÓGˆÜ—&‘&˜Ÿ™Ò(<Ô=ˆˆ3‰ä˜2œx¨¨2›Ð/¸À°~ÈBÔOˆØ—‘˜‘ˆÜ
×Ñ˜v xÔ0äØ”b—i‘i  BÓ'×.Ñ.¨wÓ7Ð8Ü—)‘)˜B Ó#×*Ñ*¨7Ó3ô
ˆô Ÿ™˜xŸ~™~Ò4HÔIˆ�‰Ø—‘˜˜B�ˆÜ
×Ñ˜f hÔ/ä˜1œh q¨"›oÐ.°wÀ°nÈ1ÔMˆØ—‘˜‘ˆÜ
×Ñ˜v xÕ0r   c                 ón	  — t        g d¢g d¢¬«      }t        g d¢«      }g d¢}t        ||dœ|¬«      }t        dd	gg d¢¬«      }t        d
dg«      }ddg}t        ||dœ|¬«      }t        ||d¬«      }	t        ddgddgdd
¬«      }
d}|j                  dd…d d …f   }t        j                  ||«       t        |d   j                  t        «      sJ ‚|j                  dd d …f   }t        j                  ||
«       t        |d   t        «      sJ ‚|j                  d d …df   }t        j                  ||	«       t        |j                  t        «      sJ ‚|j                  d   }||k(  sJ ‚|j                  d
d…d d …f   }t        j                  ||«       t        |d   j                  t        «      sJ ‚|j                  d
d d …f   }t        j                  ||
«       t        |d   t        «      sJ ‚|j                  d d …df   }t        j                  ||	«       t        |j                  t        «      sJ ‚|j                  d   }||k(  sJ ‚|j                  d
|j                  d   f   }||k(  sJ ‚|j                  d   }||k(  sJ ‚|j                  d   }||k(  sJ ‚|j                  dg   }||d   dk(     }t        j                  ||«       ||d   dk(     }t        j                  ||«       |j                  d   }||k(  sJ ‚|j                  d   }t        j                  ||
«       t        |d   t        «      sJ ‚|j                  t!        dd«         }t        j                  ||«       t        |d   j                  t        «      sJ ‚|j                  ddg   }t        j                  ||«       t        |d   j                  t        «      sJ ‚|j                  d d …df   }t        j                  ||	«       t        |j                  t        «      sJ ‚|j                  d d …t!        dd«      f   }t        j                  ||«       t        |d   j                  t        «      sJ ‚|j                  d d …ddgf   }t        j                  ||«       t        |d   j                  t        «      sJ ‚y )N)r(   ÚcÚbrl   rl   rl   rl   ©r(   rm   rl   ©Ú
categories©ÚhÚiÚjÚkÚlÚmÚn)r+   r.   r/   r0   r,   r   é   ©ÚcatsÚvaluesr   rm   rl   rt   ru   r/   r0   r{   r\   r|   r=   )r   r   r   r.   r   )r.   r   )rt   r{   r+   )r   r   r   r	   re   r7   r8   Ú
isinstancer   r   rf   Ústrr5   ÚcolumnsÚiatÚatÚslice)r>   r{   Úidxr|   r!   Úcats2Úidx2Úvalues2Úexp_dfÚexp_colÚexp_rowÚexp_valÚres_dfÚres_rowÚres_colÚres_valÚ	exp_fancyÚ	res_fancys                     r    Útest_slicing_and_getting_opsz1TestCategoricalIndex.test_slicing_and_getting_opsŠ   s‡  € ô Ú/ºOô
ˆô Ò7Ó8ˆÚ&ˆÜ °Ñ7¸sÔCˆô ˜S #˜J²?ÔCˆÜ�c˜3�ZÓ ˆØ�a�&ˆô  E°WÑ=ÀTÔJˆô ˜ S¨vÔ6ˆô ˜#˜q˜¨&°(Ð);À8ÐRUÔVˆð ˆð —‘˜˜1˜ša˜‘ˆÜ
×Ñ˜f fÔ-Ü˜& ™.×.Ñ.Ô0@ÔAÐAÐAð —'‘'˜!šQ˜$‘-ˆÜ
×Ñ˜w¨Ô0Ü˜' &™/¬3Ô/Ð/Ð/ð —'‘'š!˜Q˜$‘-ˆÜ
×Ñ˜w¨Ô0Ü˜'Ÿ-™-Ô)9Ô:Ð:Ð:ð —'‘'˜$‘-ˆØ˜'Ò!Ð!Ð!ð —‘˜˜C˜¢˜
Ñ#ˆÜ
×Ñ˜f fÔ-Ü˜& ™.×.Ñ.Ô0@ÔAÐAÐAð —&‘&˜ša˜‘.ˆÜ
×Ñ˜w¨Ô0Ü˜' &™/¬3Ô/Ð/Ð/ð —&‘&š˜F˜Ñ#ˆÜ
×Ñ˜w¨Ô0Ü˜'Ÿ-™-Ô)9Ô:Ð:Ð:ð —&‘&˜Ñ%ˆØ˜'Ò!Ð!Ð!ð —&‘&˜˜bŸj™j¨™mÐ+Ñ,ˆØ˜'Ò!Ð!Ð!ð —&‘&˜‘,ˆØ˜'Ò!Ð!Ð!ð —%‘%˜Ñ$ˆØ˜'Ò!Ð!Ð!ð —G‘G˜Q˜C‘Lˆ	à�r˜&‘z SÑ(Ñ)ˆ	Ü
×Ñ˜i¨Ô3Ø�r˜(‘| qÑ(Ñ)ˆ	Ü
×Ñ˜i¨Ô3ð —%‘%˜Ñ$ˆØ˜'Ò!Ð!Ð!ð —'‘'˜!‘*ˆÜ
×Ñ˜w¨Ô0Ü˜' &™/¬3Ô/Ð/Ð/à—‘œ˜q !›Ñ%ˆÜ
×Ñ˜f fÔ-Ü˜& ™.×.Ñ.Ô0@ÔAÐAÐAà—‘˜!˜Q˜‘ˆÜ
×Ñ˜f fÔ-Ü˜& ™.×.Ñ.Ô0@ÔAÐAÐAà—'‘'š!˜Q˜$‘-ˆÜ
×Ñ˜w¨Ô0Ü˜'Ÿ-™-Ô)9Ô:Ð:Ð:à—‘šœE ! Q›K˜Ñ(ˆÜ
×Ñ˜f bÔ)Ü˜& ™.×.Ñ.Ô0@ÔAÐAÐAà—‘š˜Q ˜F˜Ñ#ˆÜ
×Ñ˜f bÔ)Ü˜& ™.×.Ñ.Ô0@ÔAÐAÑAr   c                 óÖ  — t        g d¢g d¢¬«      }t        g d¢«      }g d¢}t        ||dœ|¬«      }|j                  dd	…d d …f   }t        t        d
d
gg d¢¬«      ddgdœddg¬«      }t	        j
                  ||«       |j                  dd	…d d …f   j                  }t        ddgddgt        ¬«      }t	        j                  ||«       |j                  dd…df   }t        t        g d¢g d¢¬«      g d¢d¬«      }t	        j                  ||«       |j                  dd…|j                  dd f   }t        dt        g d¢g d¢¬«      ig d¢¬«      }t	        j
                  ||«       y )N)r(   rm   rm   rm   rl   rl   rl   rn   ro   rq   )r+   r.   r.   r.   r/   r0   r,   rz   r   r.   r0   rm   rt   ru   Úcategoryr   r{   r|   r   rr   )r(   rm   rm   )rr   rs   rt   r\   r   r+   )r   r   r   re   r7   r8   Údtypesr	   r=   rf   r5   r   )r>   r{   rƒ   r|   r!   r?   rA   s          r    Útest_slicing_doc_examplesz.TestCategoricalIndex.test_slicing_doc_examples  s\  € äÚ/ºOô
ˆô Ò7Ó8ˆÚ&ˆÜ °Ñ7¸sÔCˆà—‘˜˜1˜ša˜‘ˆÜä# S¨# Jº?ÔKØ˜a˜&ñð ˜�*ô
ˆô 	×Ñ˜f hÔ/à—‘˜˜1˜ša˜‘×'Ñ'ˆÜ˜: wÐ/°&¸(Ð1CÌ6ÔRˆÜ
×Ñ˜v xÔ0à—‘˜˜C˜ ˜Ñ(ˆÜÜš²OÔDÚ!Øô
ˆô
 	×Ñ˜v xÔ0à—‘˜˜C˜ §¡¨A¨a Ð0Ñ1ˆÜØ”[¢º_ÔMÐNÚ!ô
ˆô 	×Ñ˜f hÕ/r   c                 óx   — |j                   ddg   }|j                  g d¢   }t        j                  ||d¬«       y )Nrl   r(   )r0   r   r+   r,   T©Úcheck_index_type)r5   re   r7   r8   )r>   r!   r?   rA   s       r    Ú test_loc_getitem_listlike_labelsz5TestCategoricalIndex.test_loc_getitem_listlike_labels)  s4   € à—‘˜˜c˜
Ñ#ˆØ—7‘7š<Ñ(ˆÜ
×Ñ˜f hÀÖFr   c                 ó¬   — t        j                  t        t        j                  d«      ¬«      5  |j
                  g d¢    d d d «       y # 1 sw Y   y xY w)Nú['e'] not in indexr1   ©r(   rm   Úe©r:   r;   r<   ÚreÚescaper5   ©r>   r$   s     r    Ú)test_loc_getitem_listlike_unused_categoryz>TestCategoricalIndex.test_loc_getitem_listlike_unused_category/  s=   € ô �]‰]œ8¬2¯9©9Ð5IÓ+JÔKñ 	%Ø�G‰G’OÒ$÷	%÷ 	%ñ 	%úó   ¯A
Á
Ac                 ó‚   — t        j                  t        d¬«      5  |j                  d    d d d «       y # 1 sw Y   y xY w)Nz^'e'$r1   r�   rN   r¡   s     r    Ú&test_loc_getitem_label_unused_categoryz;TestCategoricalIndex.test_loc_getitem_label_unused_category5  s0   € ä�]‰]œ8¨8Ô4ñ 	Ø�G‰G�CŠL÷	÷ 	ñ 	úrP   c                 ó¬   — t        j                  t        t        j                  d«      ¬«      5  |j
                  ddg    d d d «       y # 1 sw Y   y xY w)Nz['d'] not in indexr1   r(   r3   rž   r¡   s     r    Útest_loc_getitem_non_categoryz2TestCategoricalIndex.test_loc_getitem_non_category:  s?   € ä�]‰]œ8¬2¯9©9Ð5IÓ+JÔKñ 	 Ø�G‰G�S˜#�JÒ÷	 ÷ 	 ñ 	 úr£   c                 óò   — |j                  «       }d|j                  d<   |j                  g d¢   }t        t        d«      t        d«      d¬«      }t	        dg d	¢i|¬
«      }t        j                  ||«       y )Nr-   r�   rœ   Úaaabber#   r   )rp   r   r   )r   r+   r,   r.   r/   r-   r   )r9   r5   r   r   r   r7   r8   )r>   r$   r!   r?   Ú	exp_indexrA   s         r    Ú0test_loc_setitem_expansion_label_unused_categoryzETestCategoricalIndex.test_loc_setitem_expansion_label_unused_category?  sa   € à�X‰X‹ZˆØˆ�‰ˆs‰Ø—‘šÑ(ˆÜ$¤T¨(£^ÄÀVÃÐSVÔWˆ	Ü˜cÒ#6Ð7¸yÔIˆÜ
×Ñ˜f hÕ/r   c                 óF  — t        g d¢«      }t        g d¢g d¢dœ|¬«      }|j                  ddg   }t        ddg|j                  ¬«      }t        d	d
gddgdœ|¬«      }t	        j
                  ||d¬«       |j                  g d¢   }t        g d¢|j                  ¬«      }t        g d¢g d¢dœ|¬«      }t	        j
                  ||d¬«       t        j                  t        t        j                  d«      ¬«      5  |j                  ddg    d d d «       y # 1 sw Y   y xY w)Nrn   ©r+   r.   r/   ©r0   r,   r   ©r   r   r   r(   rm   ro   r+   r.   r0   r,   Tr—   ©r(   r(   rm   )r+   r+   r.   )r0   r0   r,   ú['x'] not in indexr1   Úx)r   r   r5   rp   r7   r8   r:   r;   r<   rŸ   r    )r>   r   r!   Úresrª   ri   s         r    Útest_loc_listlike_dtypesz-TestCategoricalIndex.test_loc_listlike_dtypesH  sö   € ô !¢Ó1ˆÜšYªYÑ7¸uÔEˆð �f‰f�c˜3�ZÑ ˆÜ$ c¨3 Z¸E×<LÑ<LÔMˆ	Ü˜q !˜f¨A¨q¨6Ñ2¸)ÔDˆÜ
×Ñ˜c 3¸Õ>ð �f‰f’_Ñ%ˆä$¢_À×AQÑAQÔRˆ	ÜšiªiÑ8À	ÔJˆÜ
×Ñ˜c 3¸Õ>ä�]‰]œ8¬2¯9©9Ð5IÓ+JÔKñ 	Ø�F‰F�C˜�:Ò÷	÷ 	ñ 	ús   Ã<DÄD c                 ó  — t        g d¢«      }t        g d¢g d¢dœ|¬«      }|j                  ddg   }t        g d¢g d	¢dœt        g d
¢«      ¬«      }t        j                  ||d¬«       |j                  g d
¢   }t        g d¢g d¢dœt        g d¢«      ¬«      }t        j                  ||d¬«       t        j                  t        t        j                  d«      ¬«      5  |j                  ddg    d d d «       y # 1 sw Y   y xY w)N)r(   rm   r(   r­   r®   r¯   r   r(   rm   ©r+   r/   r.   )r0   r   r,   r°   Tr—   ©r+   r/   r+   r/   r.   )r0   r   r0   r   r,   ©r(   r(   r(   r(   rm   r±   r1   r²   )
r   r   r5   r7   r8   r:   r;   r<   rŸ   r    ©r>   r   r!   r³   ri   s        r    Ú8test_loc_listlike_dtypes_duplicated_categories_and_codeszMTestCategoricalIndex.test_loc_listlike_dtypes_duplicated_categories_and_codes_  sÛ   € ä ¢Ó1ˆÜšYªYÑ7¸uÔEˆð �f‰f�c˜3�ZÑ ˆÜÚ¢)Ñ,Ô4DÂ_Ó4Uô
ˆô 	×Ñ˜c 3¸Õ>ð �f‰f’_Ñ%ˆÜÚ!ªÑ8Ü"Ò#<Ó=ô
ˆô 	×Ñ˜c 3¸Õ>ä�]‰]œ8¬2¯9©9Ð5IÓ+JÔKñ 	Ø�F‰F�C˜�:Ò÷	÷ 	ñ 	ús   Ã C;Ã;Dc           	      óP  — t        g d¢t        d«      ¬«      }t        g d¢g d¢dœ|¬«      }|j                  dd	g   }t        g d
¢g d¢dœt        g d¢t        d«      ¬«      ¬«      }t	        j
                  ||d¬«       |j                  g d¢   }t        g d¢g d¢dœt        g d¢t        d«      ¬«      ¬«      }t	        j
                  ||d¬«       t        j                  t        t        j                  d«      ¬«      5  |j                  ddg    d d d «       y # 1 sw Y   y xY w)N©r(   rm   r(   rl   Úabcdero   rR   ©r,   r   ry   r_   r¯   r   r(   rm   r¶   )r,   ry   r   r°   Tr—   r·   )r,   ry   r,   ry   r   r¸   r±   r1   r²   )r   r   r   r5   r7   r8   r:   r;   r<   rŸ   r    r¹   s        r    Ú(test_loc_listlike_dtypes_unused_categoryz=TestCategoricalIndex.test_loc_listlike_dtypes_unused_categoryv  sì   € ä Ò!5Ä$ÀwÃ-ÔPˆÜš\²Ñ=ÀUÔKˆà�f‰f�c˜3�ZÑ ˆÜÚ¢)Ñ,Ü"¢?¼tÀG»}ÔMô
ˆô 	×Ñ˜c 3¸Õ>ð �f‰f’_Ñ%ˆÜÚ!ªÑ8Ü"Ò#<ÌÈgËÔWô
ˆô 	×Ñ˜c 3¸Õ>ä�]‰]œ8¬2¯9©9Ð5IÓ+JÔKñ 	Ø�F‰F�C˜�:Ò÷	÷ 	ñ 	ús   ÄDÄD%c                 ó‚  — t        g d¢t        d«      ¬«      }t        g d¢g d¢dœ|¬«      }t        j                  t
        d¬	«      5  |j                  d    d d d «       t        j                  t
        t        j                  d
«      ¬	«      5  |j                  ddg    d d d «       y # 1 sw Y   ŒRxY w# 1 sw Y   y xY w)Nr¼   r½   ro   rR   r¾   r¯   r   r�   r1   r›   r(   )	r   r   r   r:   r;   r<   r5   rŸ   r    )r>   r   r!   s      r    Ú9test_loc_getitem_listlike_unused_category_raises_keyerrorzNTestCategoricalIndex.test_loc_getitem_listlike_unused_category_raises_keyerror�  s–   € ä Ò!5Ä$ÀwÃ-ÔPˆÜš\²Ñ=ÀUÔKˆä�]‰]œ8¨3Ô/ñ 	à�F‰F�3ŠK÷	ô �]‰]œ8¬2¯9©9Ð5IÓ+JÔKñ 	Ø�F‰F�C˜�:Ò÷	ð 	÷		ð 	ú÷	ð 	ús   ÁB)ÂB5Â)B2Â5B>c                 ó  — t        t        j                  j                  d«      j	                  d«      t        d«      t        d«      ¬«      }|j                  «       }t        |j                  «      |_        t        |j                  «      |_	        t        |j                  dd d …f   |j                  d¬«      }t        j                  |j                  dd d …f   |«       t        |j                  d d …df   |j                  d¬«      }t        j                  |j                  d d …df   |«       t        t        d	«      g d
¢¬«      }t        |j                  ddgd d …f   |j                  |¬«      }t        j                  |j                  ddgd d …f   |«       t        t        d«      g d¢¬«      }t        |j                  d d …ddgf   |j                  |¬«      }t        j                  |j                  d d …ddgf   |«       y )Nr.   ©r/   r/   ÚABCÚXYZ©r   r   r   r\   ÚXÚAB©r   r   rF   ro   r   ©r   r   ÚXY)rÇ   ÚYÚZrÌ   )r   r   ÚrandomÚdefault_rngÚstandard_normalr   r9   r   r   r   r	   r5   r7   rf   r8   )r>   r!   ÚcdfÚexpectrª   Úexp_columnss         r    Útest_ix_categorical_indexz.TestCategoricalIndex.test_ix_categorical_index™  s€  € äÜ�I‰I×!Ñ! !Ó$×4Ñ4°VÓ<Ü�u“+Ü˜“Kô
ˆð
 �g‰g‹iˆÜ$ R§X¡XÓ.ˆŒ	Ü& r§z¡zÓ2ˆŒä˜Ÿ™˜s¢A˜v™¨c¯k©kÀÔDˆÜ
×Ñ˜sŸw™w sªA v™°Ô7ä˜Ÿ™šq #˜v™¨c¯i©i¸cÔBˆÜ
×Ñ˜sŸw™w¢q¨# v™°Ô7ä$¤T¨$£ZºOÔLˆ	Ü˜2Ÿ6™6 3¨ *ªa -Ñ0¸#¿+¹+ÈYÔWˆÜ
×Ñ˜cŸg™g s¨C j²! mÑ4°fÔ=ä&¤t¨D£zºoÔNˆÜ˜2Ÿ6™6¢! c¨3 Z -Ñ0¸¿	¹	È;ÔWˆÜ
×Ñ˜cŸg™g¢a¨#¨s¨ mÑ4°fÕ=r   Úinfer_stringFTÚpyarrow)Úmarksc           
      óž  — t        d|«      5  t        t        j                  j	                  d«      j                  d«      t        d«      t        d«      ¬«      }|j                  «       }t        |j                  «      |_	        t        |j                  «      |_
        t        t        d«      dd	g¬
«      }t        |j                  dd d …f   |j                  |¬«      }t        j                  |j                  dd d …f   |«       t        t        d«      ddg¬
«      }t        |j                  d d …df   |j                  |¬«      }t        j                  |j                  d d …df   |«       t        |j                  dd	gd d …f   |j                  t        t        d«      «      ¬«      }t        j                  |j                  dd	gd d …f   |«       t        |j                  d d …ddgf   |j                  t        t        d«      «      ¬«      }t        j                  |j                  d d …ddgf   |«       d d d «       y # 1 sw Y   y xY w)Nzfuture.infer_stringr.   rÃ   ÚABAÚXYXrÆ   ÚAAr   r   ro   rÊ   ÚXXrÇ   rÌ   ÚAABÚXXY)r   r   r   rÎ   rÏ   rÐ   r   r9   r   r   r   r5   r7   r8   )r>   rÕ   r!   rÑ   rª   rÒ   rÓ   s          r    Ú$test_ix_categorical_index_non_uniquez9TestCategoricalIndex.test_ix_categorical_index_non_unique²  sÎ  € ô
 Ð1°<Ó@ñ 	BÜÜ—	‘	×%Ñ% aÓ(×8Ñ8¸Ó@Ü˜5“kÜ˜U›ôˆBð
 —'‘'“)ˆCÜ(¨¯©Ó2ˆCŒIÜ*¨2¯:©:Ó6ˆCŒKä(¬¨d«ÀÀcÀ
ÔKˆIÜ˜rŸv™v cª1 f™~°s·{±{È)ÔTˆFÜ×!Ñ! #§'¡'¨#ªq¨&¡/°6Ô:ä*¬4°«:À3ÈÀ*ÔMˆKÜ˜rŸv™v¢a¨ f™~°S·Y±YÈÔTˆFÜ×!Ñ! #§'¡'ª!¨S¨&¡/°6Ô:äØ—‘˜˜S�z¢1�}Ñ%ØŸ™Ü&¤t¨E£{Ó3ôˆFô
 ×!Ñ! #§'¡'¨3°¨*²a¨-Ñ"8¸&ÔAäØ—‘’q˜3 ˜*�}Ñ%Ø—i‘iÜ(¬¨e«Ó5ôˆFô
 ×!Ñ! #§'¡'ª!¨c°3¨Z¨-Ñ"8¸&ÔA÷=	B÷ 	Bñ 	Bús   �H-IÉIc                 óò   — d}t        j                  t        |¬«      5  |j                  dd  d d d «       |j                  dd }|j                  g d¢   }t        j                  ||«       y # 1 sw Y   Œ@xY w)NzRcannot do slice indexing on CategoricalIndex with these indexers \[1\] of type intr1   r+   r,   rm   rl   )r.   r/   r0   )r:   r;   Ú	TypeErrorr5   re   r7   r8   )r>   r!   Úmsgr?   rA   s        r    Útest_loc_slicez#TestCategoricalIndex.test_loc_slice×  sm   € ð*ð 	ô �]‰]œ9¨CÔ0ñ 	Ø�F‰F�1�Q‰K÷	ð —‘˜˜C�ˆØ—7‘7š9Ñ%ˆÜ
×Ñ˜f hÕ/÷	ð 	ús   žA-Á-A6c                 óð   — t        ddgddgddggt        g d¢«      ¬«      }|d	   }|j                  d
   dk(  sJ ‚|j                  d
   dk(  sJ ‚|j                  d   dk(  sJ ‚|j                  d   dk(  sJ ‚y )Nr+   r.   r/   r0   r,   r   rÉ   r   r   r   )r   r+   )r   r   r5   r�   )r>   r!   Úss      r    Ú&test_loc_and_at_with_categorical_indexz;TestCategoricalIndex.test_loc_and_at_with_categorical_indexä  s‰   € äØ�ˆV�a˜�V˜a ˜VÐ$Ô,<º_Ó,Mô
ˆð ˆq‰EˆØ�u‰u�S‰z˜QŠÐˆØ�t‰t�C‰y˜AŠ~Ðˆ~à�v‰v�f‰~ Ò"Ð"Ð"Ø�u‰u�V‰} Ò!Ð!Ñ!r   Ú
idx_valuesr­   )rS   éþÿÿÿéýÿÿÿ©g      ø?g      @ç      @)g      ø¿g      Àg      Àc              #   óN   K  — | ]  }t        j                  g d ¢|¬«      –— Œ y­w©r­   r   N©r   r`   ©Ú.0r   s     r    ú	<genexpr>zTestCategoricalIndex.<genexpr>ú  s   è ø€ ÒT°5Œb�h‰h’y¨×.Ð.ÑTùó   ‚#%c              #   óN   K  — | ]  }t        j                  g d ¢|¬«      –— Œ y­w)rê   r   Nrî   )rð   Údtyps     r    rñ   zTestCategoricalIndex.<genexpr>ü  s   è ø€ ÒV¸Œb�h‰h’¨d×3Ð3ÑVùrò   )r+   rm   rë   r   r+   r0   r   r^   iã  r.   r/   r3   rZ   c              #   óN   K  — | ]  }t        j                  g d ¢|¬«      –— Œ y­wrí   )rc   r`   rï   s     r    rñ   zTestCategoricalIndex.<genexpr>  s   è ø€ ÒQ°5Œb�h‰h’y¨×.Ð.ÑQùrò   )r+   r0   r   r^   z
2019-01-01)ÚperiodsÚ1d)Ústartrö   c                 ó„  — t        ||¬«      }t        dg d¢i|¬«      }t        |d   |d   «      }|j                  |d      }t	        dgdg|d   ¬«      }t        j                  ||«       |j                  |d d	    }t        dd
g|d d	 dg¬«      }t        j                  ||«       |j                  |   }t        dd
g|d d	 dg¬«      }t        j                  ||«       |j                  «       }d|j                  |d   <   t        dg d¢i|¬«      }t        j                  ||«       |j                  «       }ddg|j                  |d d	 df<   t        dg d¢i|¬«      }t        j                  ||«       |j                  «       }ddg|j                  |df<   t        dg d¢i|¬«      }t        j                  ||«       y )N)Úorderedr   )ÚfooÚbarÚbazr   r   r+   rû   r\   r.   rü   rÆ   Úqux)rþ   rü   rý   Úqux2)rþ   rÿ   rý   )	r   r   r‚   r5   r	   r7   rf   r8   r9   )r>   rç   rú   Úcat_idxr!   Úslr?   rA   s           r    Ú+test_loc_getitem_with_non_string_categoriesz@TestCategoricalIndex.test_loc_getitem_with_non_string_categoriesñ  s»  € ô8 # :°wÔ?ˆÜ˜Ò2Ð3¸7ÔCˆÜ�:˜a‘= *¨Q¡-Ó0ˆð —‘˜
 1™Ñ&ˆÜ˜5˜'¨#¨°ZÀ±]ÔCˆÜ
×Ñ˜v xÔ0ð —‘˜
 2 A˜Ñ'ˆÜ˜e U˜^°7¸2¸A°;ÈÈÔNˆÜ
×Ñ˜f hÔ/ð —‘˜‘ˆÜ˜e U˜^°7¸2¸A°;ÈÈÔNˆÜ
×Ñ˜f hÔ/ð —‘“ˆØ$)ˆ�
‰
�:˜a‘=Ñ!Ü˜cÒ#8Ð9ÀÔIˆÜ
×Ñ˜f hÔ/ð —‘“ˆØ+0°&¨/ˆ�
‰
�:˜b˜q�> 3Ð&Ñ'Ü˜cÒ#9Ð:À'ÔJˆÜ
×Ñ˜f hÔ/ð —‘“ˆØ$ f˜oˆ�
‰
�2�s�7ÑÜ˜cÒ#9Ð:À'ÔJˆÜ
×Ñ˜f hÕ/r   c                 ó„  — t        ddt        j                  g«      }t        t	        d«      |¬«      }|t        j                     dk(  sJ ‚|j
                  t        j                     dk(  sJ ‚t        |«      }|j
                  t        j                  df   dk(  sJ ‚|j
                  t        j                     d   dk(  sJ ‚y )Nr   r   r/   r   r.   r   )r   r   rG   r	   Úranger5   r   )r>   ÚciÚserr!   s       r    Ú!test_getitem_categorical_with_nanz6TestCategoricalIndex.test_getitem_categorical_with_nan2  sž   € ä˜s C¬¯©Ð0Ó1ˆä”U˜1“X RÔ(ˆà”2—6‘6‰{˜aÒÐÐØ�w‰w”r—v‘v‰ !Ò#Ð#Ð#ä�s‹^ˆØ�v‰v”b—f‘f˜a�iÑ  AÒ%Ð%Ð%Ø�v‰v”b—f‘f‰~˜aÑ  AÒ%Ð%Ñ%r   N)-Ú__name__Ú
__module__Ú__qualname__rC   rL   rO   rj   r‘   r•   r™   r¢   r¥   r§   r«   r´   rº   r¿   rÁ   rÔ   r:   ÚmarkÚparametrizeÚparamÚtdÚ
skip_if_norß   rã   ræ   r7   ÚALL_INT_NUMPY_DTYPESÚFLOAT_NUMPY_DTYPESr   r`   r=   r   r   r
   ÚALL_INT_EA_DTYPESrc   ÚIntervalIndexÚfrom_breaksÚ
date_rangeÚtimedelta_ranger  r  r   r   r    r&   r&   0   s  „ ò-ò@.ò8ò1ò2wBòr$0òLGò%òò
 ò
0òò.ò.ò.
ò>ð2 ‡[�[×ÑØ˜  §¡¨T¸¸¿¹ÀyÓ9QÔ RÐSóñ Bóð BòD0ò"ð ‡[�[×ÑØò ÚÚÚð	
ñ U¸B×<SÑ<SÔTð	
ñ WÀ×@UÑ@UÔVð	
ð ˆB�H‰H’]¨&Ô1ð	
ñ �a˜‹^™X a¨›^©X°a¸«^Ð<ð	
ñ �t˜Q Ó"¡I¨d°A°qÓ$9¹9ÀTÈ1ÈaÓ;PÐQð	
ñ �q˜#Ó¡	¨!¨SÓ 1±9¸QÀÓ3DÐEð	
ñ" R¸B×<PÑ<PÔQð#	
ð& ×Ñ×(Ñ(ªÓ6×<Ñ<ð'	
ð( ˆB�M‰M˜,°Ô2×8Ñ8ð)	
ð* ˆB×Ñ T°1Ô5×;Ñ;ð+	
óñ4%0ó5ð4%0óN&r   r&   )rŸ   Únumpyr   r:   Úpandas.util._test_decoratorsÚutilÚ_test_decoratorsr  Úpandasrc   r   r   r   r   r   r   r	   r
   r   r   Úpandas._testingÚ_testingr7   Úfixturer!   r$   r&   r   r   r    ú<module>r     sg   ðÛ 	ã Û ç )Ð )ã ÷÷ ÷ õ ð ‡�ñó ðð ‡�ñó ð÷M&ò M&r   