a
    é´«jª8  ã                   @   s|  d Z ddlZddlZddlZddlZddlmZ	 ej
jdd�ZG dd„ dƒZG dd„ dƒZej
jd	e d
d
g¡e ddg¡e ddg¡e e dd¡gd ¡e e d¡gd ¡e e d¡gd ¡e ddg¡e e dd
¡gd ¡ej ddg¡g	dd„ d�dd„ ƒZdd„ Zdd„ Zdd„ Zej
 dej ddd d!�d fej d"d d!�d fej d#d$d„ d d!�d%fg¡d&d'„ ƒZ!d(d)„ Z"dS )*z1Tests dealing with the NDFrame.allows_duplicates.é    NzNot implemented.©Úreasonc                   @   s¼  e Zd Zej deje 	g ¡fejddgfej
i fej
dddgifg¡dd„ ƒZej de dg¡e d	d¡e d
ej¡e d
d¡e d¡ejg¡dd„ ƒZej dejdg d¢d�ejdddgd�g¡edd„ ƒƒZdd„ Zdd„ Zej dd	dg¡ej dddg¡ej ddejddgdd�g¡dd „ ƒƒƒZd!d"„ Zd#d$„ Zej d%ejdddgd�ejdd&d'gd�gi fejdddgd�ejdddgd�gd(difejdddgd�ejdddgd�gd)difej
dddgiddgd�ej
dddgid&d'gd�gi fej
dddgiddgd�ej
dddgiddgd�gd(difej
dddgiddgd�ej
d*ddgiddgd�gd)difej
dddgiddgd�ejddgddgd*d+�gd)difg¡d,d-„ ƒZej d.ejej
dddgiddgd�jdd/�ej
d*ddgidd'gd�jdd/�ded0�ejej
dddgiddgd�jdd/�ej
d*ddgidd'gd�ded0�ej
dddgiddgd�ej
d*ddgidd'gd�dfg¡d1d2„ ƒZed3d4„ ƒZej dddg¡ed5d6„ ƒƒZd7S )8ÚTestPreservesz	cls, dataé   é   ÚAc                 C   s<   ||ƒ}|j jdu sJ ‚||ƒjdd�}|j jdu s8J ‚d S )NTF©Úallows_duplicate_labels)Úflagsr	   Ú	set_flags)ÚselfÚclsÚdataÚresult© r   úi/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/generic/test_duplicate_labels.pyÚtest_construction_ok   s    
z"TestPreserves.test_construction_okÚfuncÚaÚaddÚrenameÚnameÚabsc                 C   s6   t jddgddgd�jdd�}||ƒjjdu s2J ‚d S ©Nr   r   r   Úb©ÚindexFr   )ÚpdÚSeriesr   r
   r	   )r   r   Úsr   r   r   Útest_preserved_series!   s    z#TestPreserves.test_preserved_seriesÚotherr   ©r   r   Úcr   r   c                 C   sP   t jddgddgd�jdd�}| |¡\}}|jjdu s<J ‚|jjdu sLJ ‚d S r   )r   r   r   Úalignr
   r	   )r   r!   r   r   r   r   r   r   Ú
test_align0   s    zTestPreserves.test_alignc                 C   sf   t jddgddgdœddgd�jd	d
�}|jdg jjd	u s@J ‚|jd d …ddgf jjd	u sbJ ‚d S )Nr   r   é   é   ©r   ÚBr   r   r   Fr   r   r)   )r   Ú	DataFramer   Úlocr
   r	   ©r   Údfr   r   r   Útest_preserved_frame;   s
     ÿz"TestPreserves.test_preserved_framec                 C   s,   t jtd�jdd�}| ¡ jjdu s(J ‚d S )N)ÚdtypeFr   )r   r   Úfloatr   Úto_framer
   r	   )r   Zserr   r   r   Útest_to_frameB   s    zTestPreserves.test_to_frameÚsubÚframeFT©r   c                 C   sx   t jddgdddgd�jdd�}|r,| ¡ }t|t jƒrD|rD| ¡ }t ||¡}|jjdu s`J ‚||ƒjjdu stJ ‚d S )	Nr   r   r   r   r   )r   r   Fr   )	r   r   r   r1   Ú
isinstanceÚoperatorÚmethodcallerr
   r	   )r   r   r!   r4   r-   r   r   r   Útest_binopsF   s    ÿzTestPreserves.test_binopsc                 C   s’   t  dddgi¡jdd�}|dg jjdu s0J ‚|d jjdu sDJ ‚|jd jjdu sZJ ‚|jdg jjdu srJ ‚|jddgf jjdu sŽJ ‚d S )Nr   r   r   Fr   r   )r   r*   r   r
   r	   r+   r,   r   r   r   Útest_preserve_getitemU   s    z#TestPreserves.test_preserve_getitemc                 C   sh   |s|s|  tjjdd�¡ t ddgi¡jdd�}|d jjdu sHJ ‚d|j_|d jjdu sdJ ‚d S )NzUnclear behavior.r   r   r   Fr   T)	ZapplymarkerÚpytestÚmarkÚxfailr   r*   r   r
   r	   )r   ÚrequestZusing_copy_on_writeZwarn_copy_on_writer-   r   r   r   Ú"test_ndframe_getitem_caching_issue]   s    z0TestPreserves.test_ndframe_getitem_caching_issueúobjs, kwargsr#   ÚdZignore_indexÚaxisr)   ©r   r   c                 C   s4   dd„ |D ƒ}t j|fi |¤Ž}|jjdu s0J ‚d S )Nc                 S   s   g | ]}|j d d�‘qS ©Fr   ©r   ©Ú.0Úxr   r   r   Ú
<listcomp>£   ó    z-TestPreserves.test_concat.<locals>.<listcomp>F)r   Úconcatr
   r	   )r   ÚobjsÚkwargsr   r   r   r   Útest_concati   s    :zTestPreserves.test_concatzleft, right, expectedr   )Zmarksc                 C   s&   t j||ddd�}|jj|u s"J ‚d S )NT©Z
left_indexZright_index)r   Úmerger
   r	   )r   ÚleftÚrightÚexpectedr   r   r   r   Ú
test_merge§   s     zTestPreserves.test_mergec                 C   sB   t  dg d¢i¡jdd�}| g d¢¡ d¡}|jjdu s>J ‚d S )Nr   )r   r   r&   Fr   )r   r   r   Úcount)r   r*   r   ÚgroupbyZaggr
   r	   )r   r-   r   r   r   r   Útest_groupbyÊ   s    zTestPreserves.test_groupbyc                 C   sz   t jdt jddd�ddd�}|r(| ¡ }| d¡ ¡ jjdu sBJ ‚| d¡ ¡ jjdu s\J ‚| 	d¡ ¡ jjdu svJ ‚d S )	Nr   Ú2000é   )Zperiodsr   F)r   r   r	   r&   )
r   r   Z
date_ranger1   ZrollingÚmeanr
   r	   ZewmZ	expanding)r   r4   r-   r   r   r   Útest_windowÕ   s    üzTestPreserves.test_windowN) Ú__name__Ú
__module__Ú__qualname__r;   r<   Úparametrizer   r   ÚnpÚarrayr*   r   r7   Ú
itemgetterr8   ÚstrÚupperr   r    Únot_implementedr%   r.   r2   r9   r:   r?   rN   Úparamr   rT   rW   r[   r   r   r   r   r      sÒ   üþ	



úþ
$ÿþû	þû	þû
þû	þû	þû
þûÒþ9
ÿÿøÿú
ýêþ


r   c                   @   s¦  e Zd Zej dejddgedœfej	dddgifej	ddgddgdœfej	dddgifg¡dd	„ ƒZ
ej d
ejddged�ej	ddgd�ej	ddgd�g¡dd„ ƒZdd„ Zej de ddg¡dfe ddg¡dfe e dddgf¡d¡e ddgdf¡dfe ddg¡dfe e dddgf¡d¡e e ddgdf¡d¡g¡dd„ ƒZej dejdddgdd�ejdddgdd�gddifg¡dd„ ƒZed d!„ ƒZdS )"Ú
TestRaisesz	cls, axesr   )r   r/   r   r   )r   Úcolumnsrh   c                 C   sn   |f i |¤Ž}|j jdu sJ ‚d}tjtjj|d��& |f i |¤Žjdd� W d   ƒ n1 s`0    Y  d S )NTúIndex has duplicates.©ÚmatchFr   )r
   r	   r;   Úraisesr   ÚerrorsÚDuplicateLabelErrorr   )r   r   Zaxesr   Úmsgr   r   r   Útest_set_flags_with_duplicatesê   s
    
z)TestRaises.test_set_flags_with_duplicatesr   r   r   ©rh   c                 C   sR   d}t jtjj|d�� d|j_W d   ƒ n1 s40    Y  |jjdu sNJ ‚d S )Nri   rj   FT)r;   rl   r   rm   rn   r
   r	   )r   r   ro   r   r   r   Ú+test_setting_allows_duplicate_labels_raisesû   s    	&z6TestRaises.test_setting_allows_duplicate_labels_raisesc                 C   sx   t jdddgd�}t jddgddgd�jdd�}d}tjt jj|d	�� t  ||g¡ W d   ƒ n1 sj0    Y  d S )
Nr   r   r   r   r   Fr   ri   rj   )r   r   r   r;   rl   rm   rn   rK   ©r   r   r   ro   r   r   r   Útest_series_raises
  s
    zTestRaises.test_series_raiseszgetter, targetr   Nr+   Zilocc                 C   s~   t jddgddgdœddgd�jd	d
�}|r8t||ƒ}n|}d}tjt jj|d�� ||ƒ W d   ƒ n1 sp0    Y  d S )Nr   r   r&   r'   r(   r   r   r   Fr   ri   rj   )r   r*   r   Úgetattrr;   rl   rm   rn   )r   ÚgetterÚtargetr-   ro   r   r   r   Útest_getitem_raises  s     ÿzTestRaises.test_getitem_raisesr@   r   rC   r   rB   c                 C   sZ   dd„ |D ƒ}d}t jtjj|d��" tj|fi |¤Ž W d   ƒ n1 sL0    Y  d S )Nc                 S   s   g | ]}|j d d�‘qS rD   rE   rF   r   r   r   rI   :  rJ   z1TestRaises.test_concat_raises.<locals>.<listcomp>ri   rj   )r;   rl   r   rm   rn   rK   )r   rL   rM   ro   r   r   r   Útest_concat_raises-  s    zTestRaises.test_concat_raisesc                 C   sˆ   t jdg d¢ig d¢d�jdd�}t jdg d¢ig d¢d�}d	}tjt jj|d
��" t j||ddd� W d   ƒ n1 sz0    Y  d S )Nr   )r   r   r   r"   r   Fr   r)   )r   r   r   ri   rj   TrO   )r   r*   r   r;   rl   rm   rn   rP   rs   r   r   r   Útest_merge_raises?  s    ÿzTestRaises.test_merge_raises)r\   r]   r^   r;   r<   r_   r   r   r0   r*   rp   rr   rt   r7   rb   rf   rx   ry   re   rz   r   r   r   r   rg   é   sT   üþ	
ýþ
÷þ
þûÿþ
rg   Úidxr   r   gš™™™™™ñ?rX   ÚDr   Z1D)r   r   c                 C   s
   t | ƒjS ©N)Útyper\   ©rH   r   r   r   Ú<lambda>W  rJ   r€   )Úidsc                 C   sî   d}t jtjj|d��& tjd| d�jdd� W d   ƒ n1 sB0    Y  t jtjj|d��. tjdddgi| d�jdd� W d   ƒ n1 s’0    Y  t jtjj|d��, tjddgg| d	�jdd� W d   ƒ n1 sà0    Y  d S )
Nri   rj   r   r   Fr   r   r   rq   )r;   rl   r   rm   rn   r   r   r*   )r{   ro   r   r   r   Útest_raises_basicJ  s    4<r‚   c                  C   sR   t  g d¢¡} |  ¡ }t jdddgddggit jddgd	d
�d�}t ||¡ d S )N©r   r   r   r   r#   Ú	positionsr   r   r   r&   r   r   Úlabelr5   r   )r   ÚIndexÚ_format_duplicate_messager*   ÚtmÚassert_frame_equal©r{   r   rS   r   r   r   Ú$test_format_duplicate_labels_messagee  s    "ÿr‹   c                  C   s^   t j dgg d¢g¡} |  ¡ }t jdddgddggit j dgdd	gg¡d
�}t ||¡ d S )Nr   rƒ   r„   r   r   r   r&   r   r   r   )r   Ú
MultiIndexZfrom_productr‡   r*   rˆ   r‰   rŠ   r   r   r   Ú*test_format_duplicate_labels_message_multin  s    þr�   c                  C   sf   t  dddgi¡jdd�} d}tjt|d��& | jddd	d
gdd� W d   ƒ n1 sX0    Y  d S )Nr   r   r   Fr   úCannot specifyrj   r   r&   r'   T)Zallow_duplicates)r   r*   r   r;   rl   Ú
ValueErrorÚinsert)r-   ro   r   r   r   Útest_dataframe_insert_raisesx  s    r‘   zmethod, frame_onlyZ	set_indexr   T)ZinplaceZreset_indexr   c                 C   s   | S r}   r   r   r   r   r   r€   „  rJ   Fc                 C   s¨   t  ddgddgdœ¡jdd�}|d }d|j_d}tjt|d	�� | |ƒ W d   ƒ n1 s`0    Y  |s¤tjt|d	�� | |ƒ W d   ƒ n1 sš0    Y  d S )
Nr   r   r   r(   Fr   r   rŽ   rj   )r   r*   r   r
   r	   r;   rl   r�   )ÚmethodZ
frame_onlyr-   r   ro   r   r   r   Útest_inplace_raises  s    	ÿ&r“   c                  C   s\   t  ddg¡jdd�} t | ¡}t | |¡ t  dg i¡jdd�} t | ¡}t | |¡ d S )Nr   r   Fr   r   )r   r   r   rˆ   Zround_trip_pickleZassert_series_equalr*   r‰   )r   r   r   r   r   Útest_pickle–  s    

r”   )#Ú__doc__r7   Únumpyr`   r;   Zpandasr   Zpandas._testingZ_testingrˆ   r<   r=   re   r   rg   r_   r†   ZPeriodIndexZPeriodZDatetimeIndexÚ	TimestampZTimedeltaIndexZ	TimedeltaZCategoricalIndexZIntervalIndexZIntervalrŒ   Úfrom_tuplesr‚   r‹   r�   r‘   r8   r“   r”   r   r   r   r   Ú<module>   sH    Za÷ó
	
ýþ
