U
    sVc                     @   s"   d dl Z d dlZG dd dZdS )    Nc                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )	TestFlagsc                 C   sX   t  jddj}t  jddj}||ks0t||ks<t||ksHt|dksTtd S )NTallows_duplicate_labelsF   )pd	DataFrame	set_flagsflagsAssertionError)selfab r   ;/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/test_flags.pytest_equality   s    zTestFlags.test_equalityc                 C   sD   t  jdd}|j}d|_|jdks*td|d< |jdks@td S )NTr   Fr   )r   r   r   r	   r   r
   )r   dfr   r   r   r   test_set   s    zTestFlags.test_setc                 C   sP   t tdhjddj}|dks&tt tdhjddj}|dksLtd S )NATr   z%<Flags(allows_duplicate_labels=True)>Fz&<Flags(allows_duplicate_labels=False)>)reprr   r   r   r	   r
   )r   r   r   r   r   	test_repr   s    zTestFlags.test_reprc              	   C   s4   t  }|j}~tjtdd d|_W 5 Q R X d S )Nzobject has been deletedmatchT)r   r   r	   pytestraises
ValueErrorr   r   r   r	   r   r   r   test_obj_ref   s
    zTestFlags.test_obj_refc              	   C   s~   t  }|j}|d dkstd|d< |d dks6ttjtdd |d  W 5 Q R X tjtdd d|d< W 5 Q R X d S )Nr   TFr   r   
   )r   r   r	   r
   r   r   KeyErrorr   r   r   r   r   test_getitem%   s    zTestFlags.test_getitemN)__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r      s
   	r   )r   Zpandasr   r   r   r   r   r   <module>   s   