U
|
ý°d ã @ sD d Z ddlZddlmZmZ ddlmZ G dd deZ dd Z
|
dS )z)
|
Tests involving custom Index subclasses
|
é N)Ú DataFrameÚIndexc @ s e Zd ZdddZdS )Ú CustomIndexNc C sL t dd |D rtd|d kr2t|dr2|j}tj|dd}| ||¡S )Nc s s | ]}t |tV qd S )N)Ú
|
isinstanceÚstr)Ú.0Úval© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/test_subclass.pyÚ <genexpr> s z&CustomIndex.__new__.<locals>.<genexpr>z CustomIndex cannot hold stringsÚnameÚO©Zdtype)ÚanyÚ TypeErrorÚhasattrr ÚnpÚarrayZ _simple_new)ÚclsÚdatar r r r
|
Ú__new__ s zCustomIndex.__new__)N)Ú__name__Ú
|
__module__Ú __qualname__r r r r r
|
r s r c C sx t dddg} | dd¡}tddddgtd}t ||¡ ttj dd¡| tddgddd}|
|
¡ }t |j |¡ d S ) Né é é r Ústringr )r )ÚcolumnsÚindex) r Úinsertr ÚobjectÚtmZassert_index_equalr r ÚrandomZrandnZ reset_indexr )ÚidxÚresultÚexpectedZdfr r r
|
Ú"test_insert_fallback_to_base_index s ÿr' ) Ú__doc__Únumpyr Zpandasr r Zpandas._testingZ_testingr" r r' r r r r
|
Ú<module> s
|
|
