U
|
ý°d. ã @ sF d dl Zd dlZd dlmZmZmZmZ d dlm Z
|
G dd dZ dS )é N)Ú CategoricalÚCategoricalDtypeÚCategoricalIndexÚIndexc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestCategoricalIndexConstructorsc C s^ d}t jt|d tdtddd W 5 Q R X t jt|d ttddd W 5 Q R X d S )Nz-must be called with a collection of some kind©Úmatché ÚabcdF)ÚdataÚ
|
categoriesÚordered©r r )ÚpytestÚraisesÚ TypeErrorr Úlist)ÚselfÚmsg© r úid:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/categorical/test_constructors.pyÚ"test_construction_disallows_scalar s
|
zCTestCategoricalIndexConstructors.test_construction_disallows_scalarc
|
C s t tdtddd}|j}t|}tj||dd |jr>tt|j}tj||dd |jrbtt g |d}t |jt|¡ t |j
|
t j g dd ¡ |jr¢tt td|d}t |jt|¡ t |j
|
t j d
|
d
|
d d d d
|
gdd ¡ t td}t |}t |jttd ¡ t |j
|
t j d
|
d
|
d d d d
|
gdd ¡ |jrBtt ||d}t |jt|¡ t |j
|
t j d
|
d
|
d d d d
|
gdd ¡ |jrtt |tdd}t |}t |jt|¡ t |j
|
t j d
|
d
|
d d d d
|
gdd ¡ |jrêtt |tdd}t |jttd¡ t |j
|
t j d
|
d
|
d d dd
|
gdd ¡ |jr@tt |tddd}t |jttd¡ t |j
|
t j d
|
d
|
d d dd
|
gdd ¡ |jstt |tddd}t |tdddd}tj||dd tt |¡}t|tsìtt|t rütd S )NÚaabbcar
|
Fr T©Úexact©r Zint8©Údtyper r é ÚabcÚabéÿÿÿÿÚcategory)r r r )r r r r ÚtmÚassert_index_equalr ÚAssertionErrorÚvaluesZassert_numpy_array_equalÚcodesÚnpÚarrayr Ú
|
isinstance)r Úcir ÚresultÚcÚexpectedr r r Útest_construction s
|
|
|
|
ÿ ÿ ÿ ÿ ÿ ÿ ÿz2TestCategoricalIndexConstructors.test_constructionc C sà t tdtddd}tt |¡dd}tj||dd tt |¡ ¡ dd}tj||dd t tdtd dd}tt |¡dd |j ¡}tj||dd tt
|
d
|
}t d d d g|dd}t ||dd}tj||dd d S )Nr r Fr r" r Tr Zcabé r r r ) r r r r( r) r# r$ ÚtolistZreorder_categoriesr Úrange)r r+ r, Úidxr. r r r Ú test_construction_with_dtype^ s z=TestCategoricalIndexConstructors.test_construction_with_dtypec C s2 t g ddgd}t|j ¡ }|ddgks.td S )NTFr )r Úsortedr r1 r% )r Úcatr r r r Ú,test_construction_empty_with_bool_categoriest s zMTestCategoricalIndexConstructors.test_construction_empty_with_bool_categoriesc C s d ¡ d ¡ d }}}t||d}t||d}t|||d}tj||dd t||d}tj||dd d}tjt|d t|||d W 5 Q R X tjt|d t|||d
|
W 5 Q R X d S ) Nza a b bzc b aTr r r z?Cannot specify `categories` or `ordered` together with `dtype`.r )r r )r r ) Úsplitr r r# r$ r r r Ú
|
ValueError)r r Zcatsr r r, r. r r r r Ú(test_construction_with_categorical_dtypez s zITestCategoricalIndexConstructors.test_construction_with_categorical_dtypeN)Ú__name__Ú
|
__module__Ú __qualname__r r/ r4 r7 r: r r r r r s
|
Ir ) Únumpyr( r Zpandasr r r r Zpandas._testingZ_testingr# r r r r r Ú<module> s
|
