U
|
®ý°dn% ã @ sv d dl Z d dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z d dl mZ d dlmZ d dlmZ G dd dZdS )é N)Ú CategoricalÚ DataFrameÚIndexÚSeriesÚ TimestampÚ
|
date_rangeÚ period_rangeÚtimedelta_range)ÚCategoricalAccessor)Ú
|
Propertiesc @ sÊ e Zd Zej ddd dd dd dd dd dd d d d
|
d g¡d d Zd d Zdd Zdd Z dd Z
|
ej de ddde dddde dddd e d d g¡d d Zd!d" Zd#d$ Zd%d& Zd'S )(ÚTestCatAccessorÚmethodc C s | j dddg¡S ©Né é é )ÚcatÚset_categories©Úx© r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/accessors/test_cat_accessor.pyÚ<lambda> ó zTestCatAccessor.<lambda>c C s | j jdddgddS )Nr r r T©Úordered)r Zreorder_categoriesr r r r r r c C s | j dddg¡S r )r Úrename_categoriesr r r r r r c C s
|
| j ¡ S ©N)r Úremove_unused_categoriesr r r r r r c C s | j dg¡S )Nr )r Zremove_categoriesr r r r r r c C s | j dg¡S )Né )r Zadd_categoriesr r r r r r c C s
|
| j ¡ S r )r Ú
|
as_orderedr r r r r r c C s
|
| j ¡ S r )r Ú as_unorderedr r r r r r c C s6 t dddgdd d¡}d}||j}||ks2td S )Nr r r ÚA)ÚnameÚcategory)r Úastyper# ÚAssertionError)Úselfr ÚserÚexpectedÚresultr r r Ú!test_getname_categorical_accessor s
|
z1TestCatAccessor.test_getname_categorical_accessorc C s¦ t tddtjdg}t |jjtddg¡ |jj r>t
|
dtddtjdgddgd}|j ddg¡}t |j |¡ d|d d
< |j ¡ }t |jjtdg¡ d S )NÚaÚbF©Ú
|
categories)r r ÚnpÚnanÚtmÚassert_index_equalr r/ r r r& r Zassert_categorical_equalÚvaluesr )r' r( ÚexpÚresr r r Útest_cat_accessor* s
|
z!TestCatAccessor.test_cat_accessorc C sl t jtkstt td d¡}t|jts0tt dg}tjt dd |j W 5 Q R X t
|
|drhtd S )NÚaabbcder$ r zonly use .cat accessor©Úmatchr ) r r r
|
r& Úlistr% Ú
|
isinstanceÚpytestÚraisesÚAttributeErrorÚhasattr)r' r( Úinvalidr r r Útest_cat_accessor_api8 s
|
z%TestCatAccessor.test_cat_accessor_apic C s8 t td d¡}tjtdd d|j_W 5 Q R X d S )Nr8 r$ z You cannot add any new attributer9 r, )r r; r% r= r> r? r Zxlabel)r' r r r r Ú#test_cat_accessor_no_new_attributesD s z3TestCatAccessor.test_cat_accessor_no_new_attributesc C s( d}t jt|d tdddgj W 5 Q R X t jt|d tdddg ¡ W 5 Q R X t jt|d tdddgj W 5 Q R X t jt|d tt d ¡j W 5 Q R X t jt|d ttd
|
gj W 5 Q R X ttddddgd d }t dddg}t
|
|jj |¡ |j dddg¡}t dddg}t
|
|jj |¡ td ddd gdd}t
|
|jj|¡ |jjspt|j ¡ }|jjrt|j ¡ }|jjs tttddddgd d }t dddg}tjddddgtjd}|j dddg¡}t
|
|jj |¡ t
|
|j ¡ |¡ t
|
| ¡ |¡ ttddddgdddgd}t ddg}tjddddgtjd}|j ¡ }t
|
|jj |¡ t
|
|j ¡ |¡ t
|
| ¡ |¡ d}t jt|d | ddddg¡ W 5 Q R X ttddddgd d }|j dd ¡}ttddddgdddgd d}t
|
||¡ d S )Nz3Can only use \.cat accessor with a 'category' dtyper9 r r r r, r- Úcg @Z20130101Tr r Zint8©Údtyper. z1'Series' object has no attribute 'set_categories'r c S s | ¡ S r )Úupperr r r r r r z>TestCatAccessor.test_categorical_delegations.<locals>.<lambda>r" ÚBÚC©r/ r )r= r> r? r r r0 Zaranger r r r2 r3 r/ r Zassert_series_equalÚcodesr r& r! r ÚarrayZobject_r Zassert_numpy_array_equalr4 Z __array__r )r' Úmsgr( Zexp_categoriesZ exp_codesZ
|
exp_valuesr* r) r r r Ú test_categorical_delegationsJ s^
|
|
|
ÿz,TestCatAccessor.test_categorical_delegationsÚidxz1/1/2015é )ÚperiodsZMET)rQ ÚtzÚD)ÚfreqrQ z1 daysz10 daysc
|
sª t |}| d¡}t|jjt|jts.tddi fddi fddi fddi fddi fd d
|
i fg}|j d krd d i f}|
|
|¡ n |j j dkr¢ddi f}|
|
|¡ dd |D dd dg fddt |jD }dd |D }|D ] }|d t |jkrè|
|
|¡ qè|D ]f\} }
|
} t ¡ @ | dkr8t dt¡ t|j| |
|
| } t|j| |
|
| } W 5 Q R X t | | ¡ qD ]*}t|j|} t|j|} t | | ¡ qzd S )Nr$ Ústrftime)z%Y-%m-%dÚround)rS ÚfloorÚceilZasfreqZas_unitÚszM8[ns]Z tz_localize)ÚUTCÚMZ
|
tz_convert)ÚESTc S s g | ] }|d qS )r r )Ú.0Úfr r r Ú
|
<listcomp>´ s zHTestCatAccessor.test_dt_accessor_api_for_categorical.<locals>.<listcomp>Ú
|
componentsc s2 g | ]*}| d ¡s|ks|ks| ks|qS )Ú_)Ú
|
startswith©r] Úfname©Z _ignore_namesZ_special_func_namesZ
|
attr_namesr r r_ ¸ s
|
úc S s g | ]}|d i fqS )r r rc r r r r_ Ã s r Z to_periodÚignore)r r% ÚtypeZ_valuesZ_datetimelike_opsr< Údtr r& rF ÚappendÚkindÚdirÚwarningsÚcatch_warningsÚ simplefilterÚ UserWarningÚgetattrr2 Z assert_equal)r' rO r( r Zspecial_func_defsÚtupZ
|
func_namesZ func_defsZf_defÚfuncÚargsÚkwargsr6 r5 Úattrr re r Ú$test_dt_accessor_api_for_categorical sJ
|
ú
|
|
|
|
|
þ
|
|
z4TestCatAccessor.test_dt_accessor_api_for_categoricalc C sJ t dddg d¡}d}tjt|d |j W 5 Q R X t|drFtd S )Nr r r r$ z+Can only use .dt accessor with datetimeliker9 Ústr)r r% r= r> r? rh r@ r& )r' rA rM r r r Ú,test_dt_accessor_api_for_categorical_invalidÙ s
|
z<TestCatAccessor.test_dt_accessor_api_for_categorical_invalidc C sî t dddgdddgddd}|d j ddg¡|d< |d j d
|
d g¡|d < t|d d
|
d d gkshtt|d dddgkstt|d d
|
d gd d |d < t|d ddgd d |d< t|d d
|
d d gksÐtt|d dddgksêtd S )Nr r )ÚSurvivedÚSexr$ rE ry ZNoZYesrz ZfemaleZmaleFrJ )r r r r; r& r )r' Údfr r r Útest_set_categories_setitemá s ÿ
|
z+TestCatAccessor.test_set_categories_setitemc C sH t dddgdddgddd}|d jjj}t tj¡}||ksDtd S )
|
Nr r r TF)Zint_catÚbool_catr$ rE r} )r r r/ rF r0 Zbool_r& )r' r{ Úvaluer) r r r Ú'test_categorical_of_booleans_is_booleanö s ÿ z7TestCatAccessor.test_categorical_of_booleans_is_booleanN)Ú__name__Ú
|
__module__Ú __qualname__r= ÚmarkZ parametrizer+ r7 rB rC rN r r r rv rx r| r r r r r r s: øþ
|
D
|
üþ
|
Br )rl Únumpyr0 r= Zpandasr r r r r r r r Zpandas._testingZ_testingr2 Z pandas.core.arrays.categoricalr
|
Z pandas.core.indexes.accessorsr r r r r r Ú<module> s (
|
|
