U
|
¬ý°d< ã @ s> d dl Zd dlZd dlmZmZ d dlmZ G dd dZ dS )é N)Ú CategoricalÚIndexc @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestCategoricalSortc C sl t dddddgdd}t dddddg¡}tj|jdd |d
|
d |d d d
}tj|jd
|
d |d
|
d d S ) Né é é é é T©Úorderedr ©Ú ascendingF©Z check_dtypeéÿÿÿÿ)r ÚnpÚarrayÚtmÚassert_numpy_array_equalÚargsort)ÚselfÚcÚexpected© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_sorting.pyÚ test_argsort s
|
ÿ
|
ÿz TestCategoricalSort.test_argsortc C s¶ t dddddgdd}t dddddg¡}tjt |¡|d d
|
tjtj|d d |d d
|
d }tjt|d tj|dd W 5 Q R X d}tjt|d tj|dd W 5 Q R X d S )Nr r r r r Tr
|
r Fr Z mergesort)Úkindz%the 'axis' parameter is not supported)Úmatch)Zaxisz&the 'order' parameter is not supportedÚC)Úorder) r r r r r r ÚpytestZraisesÚ
|
ValueError)r r r Úmsgr r r Útest_numpy_argsort s ÿz&TestCategoricalSort.test_numpy_argsortc C sØ t ddddgdd}| ¡ t ddddgdd}| ¡ }tjddddgtd}t | ¡ |¡ t |j |j ¡ t ddddgddddgdd }| ¡ }tjddddgtd}t | ¡ |¡ t |j |j ¡ |jdd
|
}tjddddgtd}t | ¡ |¡ t |j |j ¡ |
|
¡ }|j }|jdd |j |ks6t tjddddgtd}t | ¡ |¡ t |j |j ¡ t dddddgdd}|jdd
|
}tjdddddgtd}t ddddg}t | ¡ |¡ t |j |¡ d S ) NÚaÚbFr
|
r ÚdT©Zdtype)Ú
|
categoriesr r )Zinplace)r Ú sort_valuesr r Úobjectr r Ú __array__Úassert_index_equalr' ÚcopyÚ_codesÚAssertionErrorr )r ÚcatÚresÚexpZcat1Z
|
orig_codesÚexp_valÚexp_categoriesr r r Útest_sort_values+ sB
|
|
ÿ z$TestCategoricalSort.test_sort_valuesc C sT t ddtjdtjgdd}tddg}t dddtjtjg¡}| ¡ }t | ¡ |¡ t |j
|
|¡ t tjtjdddg¡}|jddd}t | ¡ |¡ t |j
|
|¡ t tjtjdddg¡}|jd dd}t | ¡ |¡ t |j
|
|¡ t dddtjtjg¡}|jdd
|
d}t | ¡ |¡ t |j
|
|¡ t dddtjtjg¡}|jd d
|
d}t | ¡ |¡ t |j
|
|¡ t d d d dtjgdd}|jd d
|
d}tjdd d d tjgt d}td d d dg}t | ¡ |¡ t |j
|
|¡ t d d d dtjgdd}|jd dd}tjtjdd d d gt d}td d d dg}t | ¡ |¡ t |j
|
|¡ d S )Nr r Tr
|
g @g @Úfirst)r Z na_positionFÚlastr# r r$ r% r& ) r r Únanr r r( r r r* r+ r' r) )r r/ r3 r1 r0 r2 r r r Ú test_sort_values_na_positionV sD z0TestCategoricalSort.test_sort_values_na_positionN)Ú__name__Ú
|
__module__Ú __qualname__r r" r4 r8 r r r r r s +r )
|
Únumpyr r Zpandasr r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
