o
    ôT·jÊg  ã                   @   sœ   d Z ddlmZmZ ddl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d„ ƒZG dd„ dƒZG d	d
„ d
ƒZG dd„ dƒZdS )z/
See also: test_reindex.py:TestReindexSetIndex
é    ©ÚdatetimeÚ	timedeltaN)
ÚCategoricalÚCategoricalIndexÚ	DataFrameÚDatetimeIndexÚIndexÚ
MultiIndexÚSeriesÚ
date_rangeÚperiod_rangeÚto_datetimec               	   C   sP   t g d¢g d¢g d¢tj d¡ d¡tj d¡ d¡tj d¡ d¡dœƒ} | S )aM  
    Fixture for DataFrame of columns that can be used for indexing

    Columns are ['A', 'B', 'C', 'D', 'E', ('tuple', 'as', 'label')];
    'A' & 'B' contain duplicates (but are jointly unique), the rest are unique.

         A      B  C         D         E  (tuple, as, label)
    0  foo    one  a  0.608477 -0.012500           -1.664297
    1  foo    two  b -0.633460  0.249614           -0.364411
    2  foo  three  c  0.615256  2.154968           -0.834666
    3  bar    one  d  0.234246  1.085675            0.718445
    4  bar    two  e  0.533841 -0.005702           -3.533912
    )Úfoor   r   Úbarr   )ÚoneÚtwoÚthreer   r   )ÚaÚbÚcÚdÚeé   é   )ÚAÚBÚCÚDÚE©ÚtupleÚasÚlabel)r   ÚnpÚrandomÚdefault_rngÚstandard_normal©Údf© r*   úl/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/pandas/tests/frame/methods/test_set_index.pyÚframe_of_index_cols   s   úÿ
r,   c                   @   sh  e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zej dddddgdg¡ej dddg¡ej dddg¡dd„ ƒƒƒZej dddddgdg¡ej dddg¡dd„ ƒƒZej dddddgdg¡ej dddg¡d d!„ ƒƒZd"d#„ Zej d$eeejed%d&„ d'd&„ g¡ej d(g d)¢¡ej dddg¡d*d+„ ƒƒƒZej d$eeejed,d&„ g¡ej d(g d-¢¡ej dddg¡d.d/„ ƒƒƒZej d0eeejeed1d&„ d2d&„ g¡ej d3eeejeed4d&„ d5d&„ g¡ej d(g d6¢¡ej dddg¡d7d8„ ƒƒƒƒZej d9ddg¡ej dddg¡d:d;„ ƒƒZd<d=„ Zd>d?„ Zd@dA„ ZdBdC„ Z dDS )EÚTestSetIndexc                 C   sF   g d¢g d¢dœ}t |ƒ}g d¢}||d< t |d ¡}| |¡ d S )N)r   g      @é   )é   r   é   )Út1Út2))r   é   )r   r   )r3   r   Útuples)r   r
   Úfrom_tuplesÚ	set_index)Úselfr   r)   r4   Úindexr*   r*   r+   Útest_set_index_multiindex9   s   z&TestSetIndex.test_set_index_multiindexc                 C   s~   t dddœdddœddd	d
œdddd
œgg d¢d�}| ddg¡}|ddg }tj|d |d gddgd�|_t ||¡ d S )Nr3   r   )r   Úpr   é
   )r   Úmr.   é   é   )r   r<   r:   r/   é   é   )r   r<   r:   Úx©Úcolumnsr   rA   r<   r:   ©Únames)r   r6   r
   Úfrom_arraysr8   ÚtmÚassert_frame_equal)r7   r)   ÚresultÚexpectedr*   r*   r+   Útest_set_index_empty_columnD   s   

üù
 z(TestSetIndex.test_set_index_empty_columnc                 C   sT   t tdd�tdd�g dœƒ}| ddg¡}|j ¡ j}|ddg j}t ||¡ d S )Nzdatetime64[ns])ÚdtypeÚint64©r   r   r   r   r   )r   r   r6   r8   Úto_frameÚdtypesrG   Úassert_series_equal)r7   Údf1Údf2rI   rJ   r*   r*   r+   Útest_set_index_empty_dataframeV   s   ÿz+TestSetIndex.test_set_index_empty_dataframec                 C   s„   t  g d¢¡}ttj d¡ d¡|d�}| |jd ¡}|j	d d …dd …f }|j	d d …df j
|_|jd g|j_t ||¡ d S )N))r   r3   )r   r   )r   r3   r   )r.   r.   rB   r   r3   )r
   r5   r   r$   r%   r&   r'   r6   rC   ÚilocÚvaluesr8   rE   rG   rH   )r7   rC   r)   rI   rJ   r*   r*   r+   Ú test_set_index_multiindexcolumnsa   s   ÿz-TestSetIndex.test_set_index_multiindexcolumnsc                 C   sv   t dgdd� d¡}td|iƒ}| |¡jd jdksJ ‚t t|jƒƒd jdks+J ‚| |j¡jd jdks9J ‚d S )Nz2014-01-01 10:10:10ÚUTC©ÚtzzEurope/Romer   r   r=   )r   Ú
tz_convertr   r6   r8   Úhourr   r   )r7   Úidxr)   r*   r*   r+   Útest_set_index_timezonen   s
    z$TestSetIndex.test_set_index_timezonec                 C   sH   t dd„ tdƒD ƒtj d¡ d¡dœƒ}| d¡}t|jt	ƒs"J ‚d S )Nc                 S   s    g | ]}t d ddƒt|ƒ ‘qS )iÐ  r3   r   ©Ú.0Úir*   r*   r+   Ú
<listcomp>z   s     zBTestSetIndex.test_set_index_cast_datetimeindex.<locals>.<listcomp>iè  r   ©r   r   r   )
r   Úranger$   r%   r&   r'   r6   Ú
isinstancer8   r   )r7   r)   Úidfr*   r*   r+   Ú!test_set_index_cast_datetimeindexw   s   þÿ
z.TestSetIndex.test_set_index_cast_datetimeindexc                 C   sÀ   t ddddd�}tg d¢g d¢dœ|d	� ¡ }| d
¡}tg d¢g d¢dœt|d
d�d	�}|j d ¡|_t ||¡ | d
dg¡}t	j
|g d¢gd
dgd�}tdg d¢i|d�}t ||¡ d S )Nz2006-10-29 00:00:00r.   Úhú
US/Pacific)ÚperiodsÚfreqrZ   ©r   r3   r   )r.   r/   r   )r   r   )Údatar8   r8   ©Únamer   rD   r   ©r8   )r   r   Úreset_indexr6   r	   r8   Ú
_with_freqrG   rH   r
   rF   )r7   Údir)   ÚresÚexpÚ	exp_indexr*   r*   r+   Útest_set_index_dst‚   s   

þzTestSetIndex.test_set_index_dstc                 C   s€   |}t t t|ƒ¡d d d… ƒ}| |¡}t |j|¡ tj	t
dd�� | |d d d… ¡ W d   ƒ d S 1 s9w   Y  d S )NéÿÿÿÿzLength mismatch©Úmatchr   )r	   r$   ÚarangeÚlenr6   rG   Úassert_index_equalr8   ÚpytestÚraisesÚ
ValueError)r7   Úfloat_string_framer)   r]   r*   r*   r+   Útest_set_index•   s   
"ÿzTestSetIndex.test_set_indexc                 C   s.  t t d¡ttdƒƒtdd„ tdƒD ƒƒd�}d|j_| |j¡jj	dgks(J ‚t
j|dd	g jjdd	gd
�}t
j|g d¢ jjg d¢d
�}| dd	g¡}| |j¡jj	dd	gks[J ‚t| |j¡jt
ƒsgJ ‚t | |j¡j|¡ |j ddg¡}t| |j|g¡jt
ƒsˆJ ‚t | |j|g¡j|¡ d S )N)r;   r/   ÚABCDc                 S   s   g | ]}d |› �‘qS )zi-r*   r_   r*   r*   r+   rb   ¢   s    z5TestSetIndex.test_set_index_names.<locals>.<listcomp>r;   )rC   r8   ro   r   r   rD   )r   r   r   r   )r   r   r   r   r   r   )r   r$   Úonesr	   Úlistrd   r8   ro   r6   rE   r
   rF   ÚTrV   re   rG   r}   Úrename)r7   r)   ÚmiÚmi2Úidx2r*   r*   r+   Útest_set_index_namesž   s$   
ýÿz!TestSetIndex.test_set_index_namesÚkeysr   r   r   r    ÚinplaceTFÚdropc           	         s    |‰ t |tƒrtj‡ fdd„|D ƒ|d�}ntˆ | |d�}|r'ˆ j|dd�nˆ }||_|rAˆ  ¡ }|j||dd�}|d u s@J ‚nˆ j||d	�}t	 
||¡ d S )
Nc                    ó   g | ]}ˆ | ‘qS r*   r*   ©r`   rA   r(   r*   r+   rb   È   ó    z<TestSetIndex.test_set_index_drop_inplace.<locals>.<listcomp>rD   rn   r3   ©ÚaxisT)rŽ   r�   )rŽ   )re   r…   r
   rF   r	   rŽ   r8   Úcopyr6   rG   rH   )	r7   r,   rŽ   r�   rŒ   r]   rJ   rI   Úreturn_valuer*   r(   r+   Útest_set_index_drop_inplaceÁ   s   
z(TestSetIndex.test_set_index_drop_inplacec                    s‚   |‰ t |tƒr	|n|g}tjˆ jg‡ fdd„|D ƒ d g| d�}|r*ˆ j|dd�nˆ  ¡ }||_ˆ j||dd�}t 	||¡ d S )Nc                    r�   r*   r*   r�   r(   r*   r+   rb   ß   r‘   z6TestSetIndex.test_set_index_append.<locals>.<listcomp>rD   r3   r’   T©rŽ   Úappend)
re   r…   r
   rF   r8   rŽ   r”   r6   rG   rH   )r7   r,   rŽ   rŒ   r]   rJ   rI   r*   r(   r+   Útest_set_index_appendØ   s    ÿz"TestSetIndex.test_set_index_appendc                 C   s\   |j dg|dd�}t|tƒr|n|g}|j dg| |dd�}|j ||dd�}t ||¡ d S )Nr   Tr—   )r6   re   r…   rG   rH   )r7   r,   rŽ   rŒ   r)   rJ   rI   r*   r*   r+   Ú#test_set_index_append_to_multiindexé   s
   z0TestSetIndex.test_set_index_append_to_multiindexc                 C   sd   t g d¢g d¢dœƒ}t dddgitddgd	d
�ƒ}|j|j dd„ ¡ }| d	¡}t ||¡ d S )Nrl   rN   )ÚvalÚkeyr›   r3   r   r   r   rœ   rn   c                 S   s   | dkS )Nr3   r*   )Úindxr*   r*   r+   Ú<lambda>û   s    z<TestSetIndex.test_set_index_after_mutation.<locals>.<lambda>)r   r	   Úlocr8   Úmapr6   rG   rH   )r7   r)   rJ   rS   rI   r*   r*   r+   Útest_set_index_after_mutationö   s
   
z*TestSetIndex.test_set_index_after_mutationÚboxc                 C   s
   t | ƒgS ©N)r…   ©rA   r*   r*   r+   rž   	  s   
 zTestSetIndex.<lambda>c                 C   ó   t  | g¡S r£   ©r
   rF   r¤   r*   r*   r+   rž   
  ó    zappend, index_name)©TN©Tr   ©TÚtest©FNc                 C   sÖ   |}||j _||d ƒ}|tkr4d}tjt|d�� |j|||d� W d   ƒ d S 1 s-w   Y  d S t|dd ƒ}	|	d u rEt|dd ƒgn|	}
|j|||d�}|jdgd|d�}|r_|g|
 n|
|j _t	 
||¡ d S )Nr   z%['one', 'two', 'three', 'one', 'two']ry   r—   rE   ro   F)r8   ro   r…   r~   r   ÚKeyErrorr6   ÚgetattrrE   rG   rH   )r7   r,   rŽ   r˜   Ú
index_namer¢   r)   rœ   ÚmsgÚname_miro   rI   rJ   r*   r*   r+   Ú test_set_index_pass_single_array  s   "ÿz-TestSetIndex.test_set_index_pass_single_arrayc                 C   r¥   r£   r¦   r¤   r*   r*   r+   rž   .  r§   )r¨   ©Tr   r©   rª   r¬   c                 C   sš   |}||j _d||d ƒg}d|tjtttfv rd ndg}|j|||d�}	|jddgd|d�}
|r8|
jddd�n|
}
|rA|g| n||
j _	t
 |	|
¡ d S )Nr   r   r—   Fr3   r’   )r8   ro   r$   Úarrayr…   r!   Úiterr6   rŽ   rE   rG   rH   )r7   r,   rŽ   r˜   r¯   r¢   r)   rŒ   rE   rI   rJ   r*   r*   r+   Útest_set_index_pass_arrays-  s   z'TestSetIndex.test_set_index_pass_arraysÚbox2c                 C   r¥   r£   r¦   r¤   r*   r*   r+   rž   T  r§   c                 C   ó   | j S r£   rn   r¤   r*   r*   r+   rž   U  ó    Úbox1c                 C   r¥   r£   r¦   r¤   r*   r*   r+   rž   `  r§   c                 C   r¸   r£   rn   r¤   r*   r*   r+   rž   a  r¹   )r¨   r³   rª   r¬   c                 C   sÄ   |}||j _||d ƒ||d ƒg}|j|||d�}	||d ƒ||d ƒg}t|d tƒrB|d dkrBt|d tƒrB|d dkrBdn|}
|j|d g|
|d�}|j|d g|dd�}t |	|¡ d S )Nr   r—   r   r3   FT)r8   ro   r6   re   ÚstrrG   rH   )r7   r,   rŽ   r˜   r¯   rº   r·   r)   rŒ   rI   Ú
first_droprJ   r*   r*   r+   Ú$test_set_index_pass_arrays_duplicateL  s"   þÿûøz1TestSetIndex.test_set_index_pass_arrays_duplicater˜   c                 C   sV   |}t j|d |d gddgd�}|j|||d�}|jddgd|d�}t ||¡ d S )Nr   r   rD   r—   F)r
   rF   r6   rG   rH   )r7   r,   rŽ   r˜   r)   rŒ   rI   rJ   r*   r*   r+   Útest_set_index_pass_multiindexˆ  s
   z+TestSetIndex.test_set_index_pass_multiindexc                 C   sž   t tdƒd dd�}ttj d¡ d¡|jdœƒ}| d¡}t	 
|j|¡ ttj d¡ d¡|dœƒ}| d¡}t	 
|j|¡ | ¡  d¡}t	 
|j|¡ d S )NÚabr   r   rn   r   r;   rc   )r   r…   r   r$   r%   r&   r'   rV   r6   rG   r}   r8   rq   )r7   Úcir)   rf   r*   r*   r+   Ú(test_construction_with_categorical_index•  s   ÿ

z5TestSetIndex.test_construction_with_categorical_indexc              	   C   sŠ   t g d¢g d¢ttdƒtdƒdd�ttdƒtdƒdd�dœƒ}d	d
dd	gdd
gd	d
gfD ]}| |¡ ¡ }|j|jd�}t ||¡ q,d S )N)r3   r   r3   r3   r   )r;   é   é   é   é"   ÚabaabÚbacF)Ú
categoriesÚorderedT)r   r   ÚC1ÚC2rÊ   rË   r   rB   )	r   r   r…   r6   rq   ÚreindexrC   rG   rH   )r7   r)   ÚcolsrI   r*   r*   r+   Ú)test_set_index_preserve_categorical_dtype¨  s   üÿýz6TestSetIndex.test_set_index_preserve_categorical_dtypec                 C   s  t g d¢g d¢tdƒdœƒ}t| d¡dd�|_|j d¡|_tg d	¢dd
�}| d¡ d¡}|jddd�}t	 
|jjd |¡ t	 
|jjd tddgdd
�¡ |jjddgksZJ ‚| dd¡}t	 
|jjd tddgdd
�¡ t	 
|jjd |¡ |jjddgks„J ‚t tj d¡ d¡ƒ}tg d¢dd�}tg d¢dd�}tdddd�}| d ¡}| |¡}|j|dd�}|j|dd�}tg d	¢dd�}tddgdd�}t	 
|jjd |¡ t	 
|jjd |¡ t	 
|jjd |¡ t	 
|j d¡|¡ t	 
|j d¡|¡ t	 
|j d¡|¡ d S )N)r   r   r   r   r   r   )ú2011-07-19 07:00:00ú2011-07-19 08:00:00ú2011-07-19 09:00:00rÏ   rÐ   rÑ   r0   )r#   r   Úvaluer   T)Úutcri   )rÏ   rÐ   rÑ   rn   rX   r#   ©r˜   r   r3   r   r   r   z
US/EasternrY   )ú2012-04-01 09:00rÕ   rÕ   ú2012-04-02 09:00rÖ   rÖ   z2011-01-01 09:00z
Asia/Tokyo)rj   rZ   rÕ   rÖ   )r   rd   r   Úpopr8   r[   r   Útz_localizer6   rG   r}   Úlevelsr	   rE   Ú	swaplevelr$   r%   r&   r   rr   Úget_level_values)r7   r)   rJ   Úidx1rŠ   Úidx3Ú	expected1Ú	expected2r*   r*   r+   Útest_set_index_datetime·  s`   öÿþ  ÷÷

þÿz$TestSetIndex.test_set_index_datetimec                 C   s  t tj d¡ d¡ƒ}tdddd�}| |¡}tdddd�}| |¡ |¡}td	dd
d�}| |¡}|j|dd�}|j|dd�}tdddd�}tdddd�}t |j	j
d |¡ t |j	j
d |¡ t |j	j
d |¡ t |j	 d¡|¡ t |j	 d¡|¡ t |j	 d¡|¡ d S )Nr   r0   z
2011-01-01r.   ÚM©rj   rk   z2013-01-01 09:00rh   Ú2005ÚYTrÔ   r   r3   )r   r$   r%   r&   r   r˜   r6   rG   r}   r8   rÙ   rÛ   )r7   r)   rÜ   rŠ   rÝ   rÞ   rß   r*   r*   r+   Útest_set_index_period	  s"   

z"TestSetIndex.test_set_index_periodN)!Ú__name__Ú
__module__Ú__qualname__r9   rK   rT   rW   r^   rg   rw   r‚   r‹   r~   ÚmarkÚparametrizer–   r™   rš   r¡   r   r	   r$   r´   r…   r²   r¶   rµ   r½   r¾   rÁ   rÎ   rà   rå   r*   r*   r*   r+   r-   8   sž    		#úþÿÿþùþùþÿ Rr-   c                   @   sî   e Zd Zdd„ Zej dddg¡ej dddg¡dd„ ƒƒZej dddg¡ej dddg¡ejjd	egd
gd�dd„ ƒƒƒZ	ejjd	e
eejedd„ gg d¢d�ejjdddgddgd�ej dddg¡ej dddg¡dd„ ƒƒƒƒZdS )ÚTestSetIndexInvalidc                 C   sŽ   |}t jtdd�� |jddd� W d   ƒ n1 sw   Y  t jtdd�� |j|d |d gdd� W d   ƒ d S 1 s@w   Y  d S )NzIndex has duplicate keysry   r   T)Úverify_integrity)r~   r   r€   r6   )r7   r,   r)   r*   r*   r+   Útest_set_index_verify_integrity#  s   ÿ"ÿz3TestSetIndexInvalid.test_set_index_verify_integrityr˜   TFrŽ   c                 C   s2  |}t jtdd�� |jg d¢||d� W d   ƒ n1 sw   Y  t jtdd�� |j|d |d dg||d� W d   ƒ n1 sDw   Y  d}t jt|d�� |jt|d ƒ||d� W d   ƒ n1 siw   Y  t jt|d�� |jd|d t|d ƒg||d� W d   ƒ d S 1 s’w   Y  d S )	Nz['foo', 'bar', 'baz']ry   )r   r   Úbazr—   ÚXr   r   z%[('foo', 'foo', 'foo', 'bar', 'bar')])r~   r   r­   r6   r!   )r7   r,   rŽ   r˜   r)   r°   r*   r*   r+   Útest_set_index_raise_keys,  s   þ ÿÿ$"ÿz-TestSetIndexInvalid.test_set_index_raise_keysr¢   Úset)Úidsc                 C   s¤   |}d}t jt|d�� |j||d ƒ||d� W d   ƒ n1 s"w   Y  t jt|d�� |jd|d ||d ƒg||d� W d   ƒ d S 1 sKw   Y  d S )Nú,The parameter "keys" may be a column key, .*ry   r   r—   )r~   r   Ú	TypeErrorr6   )r7   r,   r¢   rŽ   r˜   r)   r°   r*   r*   r+   Útest_set_index_raise_on_typeB  s   ÿ$"ÿz0TestSetIndexInvalid.test_set_index_raise_on_typec                 C   r¥   r£   r¦   r¤   r*   r*   r+   rž   T  r§   zTestSetIndexInvalid.<lambda>)r   r	   znp.arrayrµ   r
   Úlengthr/   r0   Ú	too_shortÚtoo_longc           	      C   s²   |}t j d¡ dd|f¡}d}tjt|d�� |j||ƒ||d� W d   ƒ n1 s,w   Y  tjt|d�� |jd|j||ƒg||d� W d   ƒ d S 1 sRw   Y  d S )Nr   r   r;   z<Length mismatch: Expected 5 rows, received array of length.*ry   r—   r   )	r$   r%   r&   Úintegersr~   r   r€   r6   r   )	r7   r,   r¢   rö   rŽ   r˜   r)   rV   r°   r*   r*   r+   Útest_set_index_raise_on_lenR  s   ÿ"ÿz/TestSetIndexInvalid.test_set_index_raise_on_lenN)ræ   rç   rè   rí   r~   ré   rê   rð   rñ   rõ   r   r	   r$   r´   rµ   rú   r*   r*   r*   r+   rë   "  s$    	ýrë   c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚTestSetIndexCustomLabelTypec           	      C   s  G dd„ dƒ}|ddƒ}|ddƒ}t |ddg|d	d
giƒ}t |ddgitd	d
g|d�d�}| |¡}t ||¡ | |g¡}t ||¡ |ddƒ}d}tjt|d�� | |¡ W d   ƒ n1 sbw   Y  tjt|d�� | |g¡ W d   ƒ d S 1 s€w   Y  d S )Nc                   @   s(   e Zd Zddd„Zdefdd„ZeZdS )zKTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.ThingÚreturnNc                 S   ó   || _ || _d S r£   ©ro   Úcolor©r7   ro   rÿ   r*   r*   r+   Ú__init__r  ó   
zTTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thing.__init__c                 S   ó   dt | jƒ› d�S ©Nz<Thing ú>©Úreprro   ©r7   r*   r*   r+   Ú__str__v  ó   zSTestSetIndexCustomLabelType.test_set_index_custom_label_type.<locals>.Thing.__str__©rü   N)ræ   rç   rè   r  r»   r	  Ú__repr__r*   r*   r*   r+   ÚThingq  s    
r  ÚOneÚredÚTwoÚbluer   r3   r   r.   rn   rp   ÚThreeÚpinkz<Thing 'Three'>ry   )r   r	   r6   rG   rH   r~   r   r­   ©	r7   r  Úthing1Úthing2r)   rJ   rI   Úthing3r°   r*   r*   r+   Ú test_set_index_custom_label_typen  s"   

 

þ"þz<TestSetIndexCustomLabelType.test_set_index_custom_label_typec           	      C   s  G dd„ dt ƒ}|ddgƒ}|ddgƒ}t|ddg|d	d
giƒ}t|ddgitd	d
g|d�d�}| |¡}t ||¡ | |g¡}t ||¡ |ddgƒ}d}tjt|d�� | |¡ W d   ƒ n1 sfw   Y  tjt|d�� | |g¡ W d   ƒ d S 1 s„w   Y  d S )Nc                   @   s   e Zd Zdefdd„ZdS )zXTestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterable.<locals>.Thingrü   c                 S   s$   t | ƒ}d tt|ƒ¡}d|› d�S )Nz, zfrozenset({z}))ÚsortedÚjoinr    r  )r7   ÚtmpÚjoined_reprsr*   r*   r+   r  �  s   zaTestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterable.<locals>.Thing.__repr__N)ræ   rç   rè   r»   r  r*   r*   r*   r+   r  ›  s    r  r  r  r  r  r   r3   r   r.   rn   rp   r  r  z frozenset\(\{'Three', 'pink'\}\)ry   )	Ú	frozensetr   r	   r6   rG   rH   r~   r   r­   r  r*   r*   r+   Ú-test_set_index_custom_label_hashable_iterable”  s"    
þ"þzITestSetIndexCustomLabelType.test_set_index_custom_label_hashable_iterablec                 C   s¼   G dd„ dt ƒ}|ddƒ}|ddƒ}tddgd	d
gg||gd�}d}tjt|d�� | |¡ W d   ƒ n1 s9w   Y  tjt|d�� | |g¡ W d   ƒ d S 1 sWw   Y  d S )Nc                   @   s$   e Zd Zddd„Zdefdd„ZdS )zRTestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thingrü   Nc                 S   rý   r£   rþ   r   r*   r*   r+   r  À  r  z[TestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thing.__init__c                 S   r  r  r  r  r*   r*   r+   r	  Ä  r
  zZTestSetIndexCustomLabelType.test_set_index_custom_label_type_raises.<locals>.Thing.__str__r  )ræ   rç   rè   r  r»   r	  r*   r*   r*   r+   r  ¿  s    
r  r  r  r  r  r   r   r3   r.   rB   ró   ry   )rñ   r   r~   r   rô   r6   )r7   r  r  r  r)   r°   r*   r*   r+   Ú'test_set_index_custom_label_type_raises»  s   

þ"þzCTestSetIndexCustomLabelType.test_set_index_custom_label_type_raisesc                 C   sf   t tj d¡ d¡ƒ}tdddd�}tdddd�}| |¡}t |j|¡ | |¡}t |j|¡ d S )Nr   r0   z
2011/01/01rá   râ   Ú2013rä   )	r   r$   r%   r&   r   r6   rG   r}   r8   )r7   r)   rÜ   rŠ   r*   r*   r+   Útest_set_index_periodindexÕ  s   

z6TestSetIndexCustomLabelType.test_set_index_periodindexN)ræ   rç   rè   r  r  r  r!  r*   r*   r*   r+   rû   m  s
    &'rû   )Ú__doc__r   r   Únumpyr$   r~   Úpandasr   r   r   r   r	   r
   r   r   r   r   Úpandas._testingÚ_testingrG   Úfixturer,   r-   rë   rû   r*   r*   r*   r+   Ú<module>   s    0
   mK