U
    ÎZjö  ã                   @   sD   d Z ddlZddlmZmZ ddlmZ G dd„ deƒZ	dd„ Z
dS )z)
Tests involving custom Index subclasses
é    N)Ú	DataFrameÚIndexc                   @   s   e Zd Zddd„ZdS )ÚCustomIndexNc                 C   sL   t dd„ |D ƒƒrtdƒ‚|d kr2t|dƒr2|j}tj|dd�}|  ||¡S )Nc                 s   s   | ]}t |tƒV  qd S )N)Ú
isinstanceÚstr)Ú.0Úval© r	   úe/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/pandas/tests/indexes/test_subclass.pyÚ	<genexpr>   s     z&CustomIndex.__new__.<locals>.<genexpr>zCustomIndex cannot hold stringsÚnameÚO©Zdtype)ÚanyÚ	TypeErrorÚhasattrr   ÚnpÚarrayZ_simple_new)ÚclsÚdatar   r	   r	   r
   Ú__new__   s    zCustomIndex.__new__)N)Ú__name__Ú
__module__Ú__qualname__r   r	   r	   r	   r
   r      s   r   c                  C   sx   t dddgƒ} |  dd¡}tddddgtd�}t ||¡ ttj 	dd¡| tddgdd�d�}| 
¡ }t |j|¡ d S )	Né   é   é   r   Ústringr   )r   )ÚcolumnsÚindex)r   Úinsertr   ÚobjectÚtmZassert_index_equalr   r   ÚrandomZrandnZreset_indexr   )ÚidxÚresultÚexpectedZdfr	   r	   r
   Ú"test_insert_fallback_to_base_index   s      ÿr'   )Ú__doc__Únumpyr   Zpandasr   r   Zpandas._testingZ_testingr"   r   r'   r	   r	   r	   r
   Ú<module>   s
   