U
    b                     @   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 )NZcabZaabbcaF)
categoriesZordered)listr   )selfr    r   P/tmp/pip-unpacked-wheel-ck39h295/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
   resultZ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   sh   d}t jt|d ||jtd W 5 Q R X t jt|d ||jtd W 5 Q R X d S )Nzall inputs must be Index)matchZabcdabc)pytestZraises	TypeErrorr   valuesZset_categoriesr   Z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aZ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   s>   | tddg}tddddddddg}tj||dd d S )Nr   dbr   Tr   )r   r   r   r   r   r   r   r	   test_append_non_categories,   s    z%TestAppend.test_append_non_categoriesc                 C   s2   t ddg|}t td}tj||dd d S )Nr   r   ZcaaabbcaTr   )r   r   r   r   r   r   r   r   r	   test_append_object2   s    zTestAppend.test_append_objectc                 C   sB   t ddg}tddg}||}t ddddg}t|| d S )Nr   r#   r"   e)r   r   r   r   r   )r   ZfstZsndr   r    r   r   r	   test_append_to_another8   s
    
z!TestAppend.test_append_to_anotherN)__name__
__module____qualname__r   Zfixturer
   r   r   r   r!   r$   r%   r'   r   r   r   r	   r   
   s   
	r   )r   Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r	   <module>   s   