a
    oÝEb©>  ã                   @   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#g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                   @   sd  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edd„ ƒZej dd	dg¡ej ddejded�g¡ej ddejddgdd�g¡d d!„ ƒƒƒZed"d#„ ƒZejjd$d%�d&d'„ ƒZej d(ejdddgd�jdd)�ejdd*d+gd�jdd)�gi fejdddgd�jdd)�ejdddgd�jdd)�gd,difejdddgd�jdd)�ejdddgd�jdd)�gd-dif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)�gi f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)�gd,dif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)�gd-difej
dddgiddgd�jdd)�ejddgddgd.d/�jdd)�gd-difg¡d0d1„ ƒZej d2e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dd3œde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�ddd3œded�ej
dddgiddgd�ej
d.ddgidd+gd�ddd3œdfg¡d4d5„ ƒZed6d7„ ƒZej dddg¡ed8d9„ ƒƒZ d:S );Ú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   új/home/ja/django-apps/lartica_env/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   r   r   r   r   Útest_to_frameB   s    zTestPreserves.test_to_frameÚsubÚframeFT©Zmarks©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_binopsG   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_getitemX   s    z#TestPreserves.test_preserve_getitemzUnclear behavior.r   c                 C   sL   t  ddgi¡jdd�}|d jjdu s,J ‚d|j_|d jjdu sHJ ‚d S )Nr   r   Fr   T)r   r*   r   r
   r	   r,   r   r   r   Ú"test_ndframe_getitem_caching_issuea   s    z0TestPreserves.test_ndframe_getitem_caching_issueúobjs, kwargsr   r#   ÚdZignore_indexÚaxisr)   ©r   r   c                 C   s&   t j|fi |¤Ž}|jjdu s"J ‚d S )NF)r   Úconcatr
   r	   )r   ÚobjsÚkwargsr   r   r   r   Útest_concatj   s    VzTestPreserves.test_concatzleft, right, kwargs, expected©Z
left_indexZright_indexc                 C   s(   t j||fi |¤Ž}|jj|u s$J ‚d S ©N)r   Úmerger
   r	   )r   ÚleftÚrightrC   Ú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__ÚpytestÚmarkÚparametrizer   r   ÚnpÚarrayr*   r   r8   Ú
itemgetterr9   ÚstrÚupperr   r    Únot_implementedr%   r.   r2   Úparamr:   r;   Úxfailr<   r   rD   rK   rN   rR   r   r   r   r   r      s  üþ	



úþ
$ÿ
ÿ

ÿÿü÷ÿÿü÷ÿÿü÷ÿÿü÷ÿÿü÷ÿÿü÷ÿÿü÷ºþU
ÿÿ÷ÿùüèþ"


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je dddgf¡ded�e ddgdf¡dfe ddg¡dfeje dddgf¡ded�e e ddgdf¡d¡g¡dd„ ƒZej dejdddgdd�jdd�ejdddgdd�j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   Úcolumnsrb   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	   rV   Ú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   ©rb   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 )Nrc   rd   FT)rV   rf   r   rg   rh   r
   r	   )r   r   ri   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   rc   rd   )r   r   r   rV   rf   rg   rh   rA   ©r   r   r   ri   r   r   r   Útest_series_raises)  s
    zTestRaises.test_series_raiseszgetter, targetr   Nr+   r5   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   rc   rd   )r   r*   r   ÚgetattrrV   rf   rg   rh   )r   ÚgetterÚtargetr-   ri   r   r   r   Útest_getitem_raises0  s     ÿzTestRaises.test_getitem_raisesr=   r   r@   Fr   r   r?   c                 C   sL   d}t jtjj|d��" tj|fi |¤Ž W d   ƒ n1 s>0    Y  d S )Nrc   rd   )rV   rf   r   rg   rh   rA   )r   rB   rC   ri   r   r   r   Útest_concat_raisesP  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   rc   rd   TrE   )r   r*   r   rV   rf   rg   rh   rG   rm   r   r   r   Útest_merge_raisese  s    ÿzTestRaises.test_merge_raises)rS   rT   rU   rV   rW   rX   r   r   r0   r*   rj   rl   rn   r8   r[   r_   r^   rr   r   rs   rt   r   r   r   r   ra     sd   üþ	
ýþ
ÿÿóþ
ÿÿü÷ÿþ
ra   Úidxr   r   gš™™™™™ñ?rO   ÚDr   Z1D)r   r   c                 C   s
   t | ƒjS rF   )ÚtyperS   ©Úxr   r   r   Ú<lambda>}  ó    rz   )Ú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 )
Nrc   rd   r   r   Fr   r   r   rk   )rV   rf   r   rg   rh   r   r   r*   )ru   ri   r   r   r   Útest_raises_basicp  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   Úlabelr6   r   )r   ÚIndexÚ_format_duplicate_messager*   ÚtmÚassert_frame_equal©ru   r   rJ   r   r   r   Ú$test_format_duplicate_labels_message‹  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_multi”  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 specifyrd   r   r&   r'   T)Zallow_duplicates)r   r*   r   rV   rf   Ú
ValueErrorÚinsert)r-   ri   r   r   r   Útest_dataframe_insert_raisesž  s    rŒ   zmethod, frame_onlyZ	set_indexr   T)ZinplaceZset_axisr)   FZreset_indexr   c                 C   s   | S rF   r   rx   r   r   r   rz   «  r{   c                 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‰   rd   )r   r*   r   r
   r	   rV   rf   rŠ   )ÚmethodZ
frame_onlyr-   r   ri   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__r8   ÚnumpyrY   rV   Zpandasr   Zpandas._testingZ_testingrƒ   rW   r`   r^   r   ra   rX   r�   ZPeriodIndexZPeriodZDatetimeIndexÚ	TimestampZTimedeltaIndexZ	TimedeltaZCategoricalIndexZIntervalIndexZIntervalr‡   Úfrom_tuplesr}   r†   rˆ   rŒ   r9   rŽ   r�   r   r   r   r   Ú<module>   sJ    yh÷ó
	
üþ	
