U
|
¬ý°d! ã @ sf d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z m
|
Z
|
m Z m Z d dl mZ G dd dZdS )é N)ÚCategoricalDtype)Ú CategoricalÚ DataFrameÚIndexÚSeriesÚisnac @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d ddfi dfddidfde d ddddgidfg¡dd Z ej
|
d d d g¡d d Z d!d" Zej
|
d#d ddge d d d g¡fd dejge d d d g¡fd dejge d d d g¡fd dejge d d d g¡fg¡d$d% Zej
|
d#d ddge d d d g¡fd dejge d d d g¡fd dejge d d d g¡fd dejge d d d g¡fg¡d&d' Zej
|
d(dd)d*gejdd)gdd)d*gfd ddgejd dgd ddgfg¡d+d, Zej
|
d-ejd.fd/ejd0fejd0fg¡d1d2 Zd3S )4ÚTestCategoricalMissingc C s8 t dddg¡}tddt jg}| ¡ }t ||¡ d S )NFTÚaÚb)ÚnpÚarrayr Únanr ÚtmÚassert_numpy_array_equal)ÚselfÚexpÚcatÚres© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_missing.pyÚ test_isna s z TestCategoricalMissing.test_isnac C sX t td}tj ddd¡}d|d d d
< t||dd}t| t t ||dk¡ d S )Né
|
r é éÿÿÿÿé T)Zfastpath)
|
ÚlistÚranger ÚrandomÚrandintr Úreprr r r )r Ú
|
categoriesÚlabelsr r r r Ú test_na_flags_int_categories s z3TestCategoricalMissing.test_na_flags_int_categoriesc C sÚ t ddtjdg}t |jtddg¡ t |jtj ddddgtj
|
d¡ tj|d< t |jtddg¡ t |jtj ddddgtj
|
d¡ t ddtjdg}t |jtddg¡ t |jtj ddddgtj
|
d¡ d S )Nr r
|
r é r ©Údtype) r r r r Úassert_index_equalr r r Ú_codesr Úint8)r Úcr r r Útest_nan_handling' s "
|
"z(TestCategoricalMissing.test_nan_handlingc C sD t ddtjg}| tddg¡}t |jtjdddgdd¡ d S )Nr r
|
r) r r r( r$ ) r r r Z
|
_set_dtyper r r Úcodesr )r r) Úresultr r r Útest_set_dtype_nans6 s z*TestCategoricalMissing.test_set_dtype_nansc C sB t dddg}tj|d< t dtjdgdddgd}t ||¡ d S )Nr# é é )r )r r r r Úassert_categorical_equal)r r r r r r Útest_set_item_nan; s
|
z(TestCategoricalMissing.test_set_item_nanzfillna_kwargs, msgr# Zffill)ÚvalueÚmethodz)Cannot specify both 'value' and 'method'.z(Must specify a fill 'value' or 'method'.r3 Úbadz%Invalid fill method. Expecting .* badr2 r. r/ é r ú3Cannot setitem on a Categorical with a new categoryc C sZ t dddd d g}t|dkr,d|kr,t}nt}tj||d |jf | W 5 Q R X d S )Nr# r. r/ r2 ©Úmatch)r ÚlenÚ TypeErrorÚ
|
ValueErrorÚpytestÚraisesÚfillna)r Z fillna_kwargsÚmsgr Úerrr r r Útest_fillna_raisesB s z)TestCategoricalMissing.test_fillna_raisesÚnamedTFc C sÐ |rt dd¡}ndd }ttj|dd|ddd gtd}| |dd¡}t|dd|dd|ddg}t ||¡ ttj|dd|ddd gtd}d}t j
|
t |d | |dd¡ W 5 Q R X d S )
|
NÚPointzx yc W s | S )Nr )Úargsr r r Ú<lambda>d ó zFTestCategoricalMissing.test_fillna_iterable_category.<locals>.<lambda>r r# r$ r6 r7 ) Ú collectionsÚ
|
namedtupler r r Úobjectr> r r0 r< r= r: )r rB rC r r, Úexpectedr? r r r Ú test_fillna_iterable_category^ s $ $z4TestCategoricalMissing.test_fillna_iterable_categoryc C s t dddd d g}| d¡}| |¡}t ||¡ t|d sBtt dddddg¡}| |¡}t dddddg|jd}t ||¡ t|d std S )NÚAÚBÚCr r$ ) r r> r r0 r ÚAssertionErrorr r r% )r r Úotherr, rJ r r r Útest_fillna_arrayr s
|
|
|
z(TestCategoricalMissing.test_fillna_arrayzvalues, expectedc C sx t dd¡b t|}| ¡ }t ||¡ t| ¡ }t|}t ||¡ t| ¡ }t|}t ||¡ W 5 Q R X d S ©Nzmode.use_inf_as_naT)
|
ÚpdÚoption_contextr r r r r Úassert_series_equalr Úassert_frame_equal©r ÚvaluesrJ r r, r r r Útest_use_inf_as_na s z)TestCategoricalMissing.test_use_inf_as_nac C sx t |}t dd¡Z t|}t ||¡ tt|}t|}t ||¡ tt|}t|}t ||¡ W 5 Q R X d S rR )
|
r rS rT r r r r rU r rV rW r r r Ú"test_use_inf_as_na_outside_context s z9TestCategoricalMissing.test_use_inf_as_na_outside_contextza1, a2, categoriesr
|
r) c C st t |}t||dt||dk}t|t|k}t ||¡ t||dt||dk}t|t|k}t ||¡ d S ©Nr$ )r r r rU )r Za1Za2r Zcat_typer, rJ r r r Ú%test_compare_categorical_with_missing³ s z<TestCategoricalMissing.test_compare_categorical_with_missingzna_value, dtypezdatetime64[ns])NÚfloat64r] c C s& t ||g}t |jtg |d¡ d S r[ )r r r& r r )r Zna_valuer% r, r r r Ú,test_categorical_only_missing_values_no_castÈ s zCTestCategoricalMissing.test_categorical_only_missing_values_no_castN)Ú__name__Ú
|
__module__Ú __qualname__r r" r* r- r1 r< ÚmarkZ parametrizer rA rK rQ r r r ÚinfrS ZNArY rZ r\ ZNaTr^ r r r r r sn þ
|
þùþ
|
|
üþ
|
üþ
|
þþ
|
üþ r )rG Únumpyr r< Zpandas.core.dtypes.dtypesr ZpandasrS r r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
