o
    ôT·j�  ã                   @   s6   d dl Z d dlmZmZ d dlmZ G dd„ dƒZdS )é    N)ÚCategoricalIndexÚIndexc                   @   sR   e Zd Zej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S )Ú
TestAppendc                 C   s   t dƒ}tt dƒ|dd�S )NÚcabÚaabbcaF)Ú
categoriesÚordered)Úlistr   )Úselfr   © r   úo/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/pandas/tests/indexes/categorical/test_append.pyÚci   s   zTestAppend.cic                 C   sv   |d d…   |dd … ¡}tj||dd� |d d… |dd… |dd … g}|d   |dd … ¡}tj||dd� d S )Né   T©Úexacté   r   ©ÚappendÚtmÚassert_index_equal)r
   r   ÚresultÚfoosr   r   r   Útest_append   s
   "zTestAppend.test_appendc                 C   s   |  g ¡}tj||dd� d S )NTr   r   )r
   r   r   r   r   r   Útest_append_empty   s   
zTestAppend.test_append_emptyc                 C   s’   d}t jt|d�� | |j tdƒ¡¡ W d   ƒ n1 sw   Y  t jt|d�� | |j tdƒ¡¡ W d   ƒ d S 1 sBw   Y  d S )Nzall inputs must be Index)ÚmatchÚabcdÚabc)ÚpytestÚraisesÚ	TypeErrorr   ÚvaluesÚset_categoriesr	   Úreorder_categories)r
   r   Úmsgr   r   r   Ú!test_append_mismatched_categories   s   ÿ"ÿz,TestAppend.test_append_mismatched_categoriesc                 C   s8   |  tddgƒ¡}ttdƒ|jd�}tj||dd� d S )NÚcÚaÚaabbcaca)r   Tr   )r   r   r   r	   r   r   r   ©r
   r   r   Úexpectedr   r   r   Útest_append_category_objects&   s   z'TestAppend.test_append_category_objectsc                 C   s2   |  tddgƒ¡}tg d¢ƒ}tj||dd� d S )Nr&   Úd)r&   r&   Úbr,   r%   r&   r&   r+   Tr   )r   r   r   r   r(   r   r   r   Útest_append_non_categories,   ó   z%TestAppend.test_append_non_categoriesc                 C   s2   t ddgƒ |¡}t tdƒƒ}tj||dd� d S )Nr%   r&   ÚcaaabbcaTr   )r   r   r	   r   r   r(   r   r   r   Útest_append_object2   r.   zTestAppend.test_append_objectc                 C   s>   t ddgƒ}tddgƒ}| |¡}t g d¢ƒ}t ||¡ d S )Nr&   r,   r+   Úe)r&   r,   r+   r1   )r   r   r   r   r   )r
   ÚfstÚsndr   r)   r   r   r   Útest_append_to_another8   s
   
z!TestAppend.test_append_to_anotherN)Ú__name__Ú
__module__Ú__qualname__r   Úfixturer   r   r   r$   r*   r-   r0   r4   r   r   r   r   r   
   s    
	r   )r   Úpandasr   r   Úpandas._testingÚ_testingr   r   r   r   r   r   Ú<module>   s    