U
|
¬ý°dr3 ã @ sâ d dl Z d dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z m Z mZmZ d dlmZ d dlm mZ G dd dZG dd dZG dd dZej d d
|
d g¡d d Z ej d d
|
d g¡dd Z ej dd Z dd Z dS )é N) ÚNAÚ CategoricalÚCategoricalIndexÚIndexÚIntervalÚ IntervalIndexÚNaTÚ PeriodIndexÚSeriesÚ TimedeltaÚ Timestampc
|
@ s e Zd Zdd Zdd Zej deddgeddgddgdg¡d d
|
Z ej deddgddd gdeddgddd gdeddgdgdeddgdgdg¡d d Z
|
ej deddgeddgddgddeddgddd gddg¡dd Z dd Z dd Z dS )Ú!TestCategoricalIndexingWithFactorc C s |d dkst |d dks t |dddg }t |jtjdddgtjd¡ |t |¡dk }t |jtjdddgtjd¡ d S )Nr ÚaéÿÿÿÿÚcé é ©Údtype)ÚAssertionErrorÚtmÚassert_numpy_array_equalÚ_codesÚnpÚarrayÚint8Zasarray)ÚselfÚfactorZsubf© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_indexing.pyÚ test_getitem s z.TestCategoricalIndexingWithFactor.test_getitemc C s | ¡ }d|d< |d dks td|d< |d dks8t| ¡ }tjt|dd}d|d< d|d< d||< tddddddddgdd }t ||¡ d S )
|
NÚbr r r Úboolr Tr ©Úordered)Úcopyr r ÚzerosÚlenr r Úassert_categorical_equal)r r r ÚindexerÚexpectedr r r Ú test_setitem" s z.TestCategoricalIndexingWithFactor.test_setitemÚotherr! r ©Ú
|
categoriesc C sR t ddgddgd}t ddg¡}|| ||< t ddgddgd}t ||¡ d S )Nr r! r- TF)r r r r r( )r r, ÚtargetÚmaskr* r r r Ú test_setitem_same_but_unordered4 s
|
zATestCategoricalIndexingWithFactor.test_setitem_same_but_unorderedr c C sP t ddgddgd}t ddg¡}d}tjt|d || ||< W 5 Q R X d S )Nr r! r- TFúCCannot set a Categorical with another, without identical categories©Úmatch©r r r ÚpytestÚraisesÚ TypeError©r r, r/ r0 Úmsgr r r Ú'test_setitem_different_unordered_raises@ s
|
zITestCategoricalIndexingWithFactor.test_setitem_different_unordered_raisesT©r. r$ c C sR t ddgddgdd}t ddg¡}d}tjt|d || ||< W 5 Q R X d S )Nr r! Tr< Fr2 r3 r5 r9 r r r Ú test_setitem_same_ordered_raisesQ s
|
|
zBTestCategoricalIndexingWithFactor.test_setitem_same_ordered_raisesc C s. t dddg}|d |d< |d dks*td S )N)r r )r r r r )r r ©r Úcatr r r Útest_setitem_tuplea s z4TestCategoricalIndexingWithFactor.test_setitem_tuplec C s t j d¡ tt jjdddd t j¡ dg¡}t dg¡ t j ¡}d||< |j
|
t dg¡ t j ¡ }t |t jdgdd ¡ d S )
|
Nr r é éðI ©Úsizeiüÿÿé r r ) r ÚrandomÚseedr ÚrandintÚastyper Zadd_categoriesr Úint64Úcodesr r )r r? r) Úresultr r r Útest_setitem_listlikei s ÿþz7TestCategoricalIndexingWithFactor.test_setitem_listlikeN)Ú__name__Ú
|
__module__Ú __qualname__r r+ r6 ÚmarkÚ parametrizer r1 r; r= r@ rM r r r r r s4
|
þ
|
üþ
|
|
ýþ
|
r c @ s, e Zd Zdd Zdd Zdd Zej dde j
|
e e e j
|
d d
|
d d d dg ¡dd Zej dddddgddgg¡dd Zej dddddgddddgddddgddddgg¡ej dddgddgddgddgg¡ej deeg¡ej d dd d g¡d d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* ZdS )+ÚTestCategoricalIndexingc C s^ t dddddddg}|d }|dks*t|dd
}t ddgddddgd}t ||¡ d S )Nr r! r Údé rA r- )r r r r( )r r? Zslicedr* r r r Útest_getitem_slicez s z*TestCategoricalIndexing.test_getitem_slicec C sn t j d¡ tt jjdddd t j¡}|jt dg¡ t j ¡ }|t dg¡ t j ¡ j}t
|
||¡ d S )Nr r rA rB rC rE ) r rF rG r rH rI r rK r rJ r r )r r rL r* r r r Útest_getitem_listlike s
|
z-TestCategoricalIndexing.test_getitem_listlikec
|
C sf t ddddddgdd}t|}t| tjddddddgtjd }t dddgdd}t |j|¡ t |j
|
|¡ t ddddddgdd}t|d
|
d }t| tjddddddgtjd }t dddgdd}t |j|¡ t |j
|
|¡ t d d dddddgdd}t|d
|
d } tjdddddddgtjd }t dddddd d gdd}t | j|¡ t | j
|
|¡ d S )Nz2014-01z2014-02z2014-03ÚM©Úfreqr r r r Tr# z2013-12z2013-11z2013-10z2013-09z2013-08z2013-07z2013-05é rA é rU ) r r Ústrr r r r r r Úassert_index_equalr. )
|
r Zidx1Zcat1Zexp_arrZexp_idxZidx2Zcat2Zexp_idx2Zidx3Zcat3r r r Útest_periodindex s\ ÿ ÿ ù ö ù ö z(TestCategoricalIndexing.test_periodindexÚnull_valNr ZnatZNATÚnanÚNaNZNANc C sD t dd|gdd}t dddgdd}|d tks4tt ||¡ d S ) Nz
|
2022-04-06z
|
2022-04-07ÚDrY r z period[D]r r )r r r r r^ )r r` rL r* r r r Ú test_periodindex_on_null_types¿ s z6TestCategoricalIndexing.test_periodindex_on_null_typesÚnew_categoriesr r rU r\ c C s< t ddddg}d}tjt|d | |¡ W 5 Q R X d S )Nr r! r zKnew categories need to have the same number of items as the old categories!r3 )r r6 r7 Ú
|
ValueErrorZrename_categories)r re r? r: r r r Ú/test_categories_assignments_wrong_length_raisesÊ s
|
ÿzGTestCategoricalIndexing.test_categories_assignments_wrong_length_raisesÚ
|
idx_valuesÚ
|
key_valuesrA Ú key_classr ÚcategoryÚkeyc C s ||t ddd}|dkr |j}t||d}| |¡\}}| |¡\} }
|
t || ¡ t ||
|
¡ | ¡ |¡} | ¡ |¡} t | | ¡ d S )Nr rA r- rl r )Úranger r Zget_indexer_non_uniquer r ÚuniqueZ get_indexer) r rh ri rj r rl Úidxr* Zexp_missrL Zres_missZ
|
exp_uniqueZ
|
res_uniquer r r Útest_get_indexer_non_uniqueÕ s z3TestCategoricalIndexing.test_get_indexer_non_uniquec C s t tddg}| ddg¡}t tdd gddgd}t ||¡ t tddg}| ddg¡}t td d gddgd}t ||¡ d S )Nr r! TFr- )r
|
r Úwherer Úassert_series_equal©r ÚserrL r* r r r Útest_where_unobserved_nanï s z1TestCategoricalIndexing.test_where_unobserved_nanc C s\ t tdddgddddgd}|jdddgdd}t tdddg|jjd}t ||¡ d S ) Nr r! r rT r- TF)r, )r
|
r rq r? r. r rr rs r r r Ú test_where_unobserved_categoriesû s z8TestCategoricalIndexing.test_where_unobserved_categoriesc C sr t tdddgddddgd}tdddgddddgd}| dddg|¡}t tdddg|jd}t ||¡ d S ) Nr r! r rT r- TFr )r
|
r rq r r rr )r rt r, rL r* r r r Ú test_where_other_categorical s
|
z4TestCategoricalIndexing.test_where_other_categoricalc C sF t tdddg}d}tjt|d | dddgd¡ W 5 Q R X d S ) Nr r! r z3Cannot setitem on a Categorical with a new categoryr3 TFrT ©r
|
r r6 r7 r8 rq )r rt r: r r r Ú test_where_new_category_raises s z6TestCategoricalIndexing.test_where_new_category_raisesc C sl t tdddgddddgdd}tdddgddddgdd}tjtdd | dd dg|¡ W 5 Q R X d S )
|
Nr r! r rT Tr< z without identical categoriesr3 Frx )r rt r, r r r Ú!test_where_ordered_differs_rasies s ÿ
|
ÿz9TestCategoricalIndexing.test_where_ordered_differs_rasies)rN rO rP rV rW r_ r6 rQ rR r ra r r Úmathrd rg r r rp ru rv rw ry rz r r r r rS y s. 3 þ
|
|
|
*ÿ$ rS c @ sf e Zd Zdd Zejjdedddfdedd d
|
fd edd
|
fe dd
|
fge
|
d d d Z dd Z dS )Ú TestContainsc C s¨ t tdtdd}d|ks td|ks,ttj|ks:ttjtdd dg|ksXtW 5 Q R X d |ksntd|ksztt tdtjg tdd}tj|ks¤td S )
|
NZaabbcaZcabr- r! Úzzunhashable type: 'list'r3 r r )r Úlistr r ra r6 r7 r8 r> r r r Ú test_contains s zTestContains.test_containszitem, expectedr r T)ç ø?Tg à?r F)r F)Úidsc C s* t t td¡}||k}||ks&td S )NrU )r r Z from_breaksrm r )r Úitemr* r? rL r r r Útest_contains_interval+ s z#TestContains.test_contains_intervalc C sh t dddg}d|ksttjtdd dg|k W 5 Q R X tjtdd ddg|k W 5 Q R X d S )Nr r rU r zunhashable typer3 r! )r r r6 r7 r8 r> r r r Útest_contains_list= s z TestContains.test_contains_listN) rN rO rP r r6 rQ rR r r r r] r r r r r r r| s
|
|
úö
|
r| ÚindexTFc C sZ t td}tdddg}| r&t|}t |¡s4t|| }|| d¡ }t ||¡ d S )NrU TFÚobject)
|
r
|
rm r r ÚcomZis_bool_indexerr rI r rr ©r
rt ro rL r* r r r Útest_mask_with_booleanJ s r c C sL t td}tddd g}| r&t|}|| }|| d¡ }t ||¡ d S )NrU TF)r
|
rm r r Zfillnar rr r r r r Ú*test_mask_with_boolean_na_treated_as_falseW s r c c s6 ddd}| ¡ }| td|¡ dV W 5 Q R X dS )z¡
|
Monkeypatch Categorical.__array__ to ensure no implicit conversion.
|
|
Raises
|
------
|
ValueError
|
When Categorical.__array__ is called.
|
Nc S s t dd S )NzI cannot be converted.)rf )r r r r r r r s z(non_coercible_categorical.<locals>.arrayZ __array__)N)ÚcontextÚsetattrr )Z monkeypatchr Úmr r r Únon_coercible_categoricale s
|
|
r c C s0 t dddg} t| }|jd }|dks,td S )Nr r! r r )r r
|
Úatr )Zarrrt rL r r r Útest_series_atz s
|
r )!r{ Únumpyr r6 Zpandasr r r r r r r r r
|
r r Zpandas._testingZ_testingr Zpandas.core.commonÚcoreÚcommonr r rS r| rQ rR r r Zfixturer r r r r r Ú<module> s 4 b !1
|
|
|
|
