U
|
¬ý°dÝ2 ã @ sv d dl Z d dlZd dlZd dlZd dlmZ d dlmZm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlmZ G dd dZdS )é N)ÚPYPY)Ú CategoricalÚCategoricalDtypeÚIndexÚNaTÚSeriesÚ
|
date_range)Ú is_scalarc @ sø e Zd Zej dddg¡dd Zdd Zej ded e j
|
fd
|
d d ge j
|
fej e e d d dddeejjdddg¡ej dddg¡dd Zej dddde jgedfd
|
d d e jgd d d
|
gfg¡ej dd d g¡ej d ddg¡d d Zej d ddg¡ej dd d g¡d!d" Zej d#ddg¡d$d% Zej d#ddg¡d&d' Zej d(d)d*d+g¡ej d#ddg¡d,d- Zej d.d/d0g¡d1d2 Zej d3d
|
d
|
d d4d5d5d5gd5d4d d d
|
gd5gfd
|
d
|
d
|
d4d5d5d5gd5d4d d d
|
gd5d
|
gfd
|
d d d4d5gd5d4d d d
|
gd5d4d d d
|
gfe je je jd4d5gd5d4d d d
|
gd5d4gfe je je jd4d5d4gd5d4d d d
|
gd4gfe je jd4d5d4gd5d4d d d
|
gd4gfg¡d6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Z d@dA Z dBdC Z dDdE Z ej dFd
|
dGd
|
d d gdHg¡dIdJ Z dKdL Z!dMS )NÚTestCategoricalAnalyticsÚ aggregationÚminÚmaxc C s t ddddgdd}d| }t||}tjt|d | W 5 Q R X |d krVtjntj}tjt|d | |¡ W 5 Q R X d S )
|
NÚaÚbÚcÚdF©Úorderedú)Categorical is not ordered for operation ©Úmatchr ) r ÚgetattrÚpytestÚraisesÚ TypeErrorÚnpÚminimumÚmaximumÚreduce)Úselfr ÚcatÚmsgÚagg_funcZufunc© r# úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_analytics.pyÚ test_min_max_not_ordered_raises s
|
|
z8TestCategoricalAnalytics.test_min_max_not_ordered_raisesc C sæ t ddddgdd}||}| ¡ }| ¡ }|dks8t|dksDttj |¡dksXttj |¡dksltt ddddgddddgdd}||}| ¡ }| ¡ }|dks®t|dksºttj |¡dksÎttj |¡dksâtd S )Nr r r r Tr ©Ú
|
categoriesr )r r r ÚAssertionErrorr r r r )r Zindex_or_series_or_arrayr ÚobjZ_minZ_maxr# r# r$ Útest_min_max_ordered$ s(
|
|
ÿ z-TestCategoricalAnalytics.test_min_max_orderedzcategories,expectedÚABCé é é z
|
2020-01-01)ZperiodsÚcategory©Údtypez1https://github.com/pandas-dev/pandas/issues/29962)Úreason)Zmarksc C s. t g |dd}t||}| }||ks*td S ©NTr& )r r r( )r r' Úexpectedr r r" Úresultr# r# r$ Útest_min_max_ordered_empty; s
|
z3TestCategoricalAnalytics.test_min_max_ordered_emptyzvalues, categoriesr r r ÚcbaÚskipnaTFÚfunctionc C s^ t ||dd}t|||d}|dkr6|tjksZtn$|dkrF|d n|d }||ksZtd S )NTr& ©r8 Fr r r- )r r r Únanr( )r Úvaluesr' r9 r8 r r5 r4 r# r# r$ Útest_min_max_with_nanR s z.TestCategoricalAnalytics.test_min_max_with_nanc C s8 t tjgddgdd}t|||d}|tjks4td S )Nr, r- Tr& r: )r r r; r r( )r r9 r8 r r5 r# r# r$ Útest_min_max_only_nanc s z.TestCategoricalAnalytics.test_min_max_only_nanÚmethodc C sR t tjddtjgdddddgdd}tjtdd t||dd
|
W 5 Q R X d S ) Nr, r- é é r. Tr& z .* got an unexpected keywordr )Z numeric_only)r r r; r r r r )r r? r r# r# r$ Ú test_numeric_only_min_max_raisesk s ÿz9TestCategoricalAnalytics.test_numeric_only_min_max_raisesc C sV t ddddgdd}d| d}tt|}tjtt |¡d || W 5 Q R X d S ) Nr r r Fr r zF
|
you can use .as_ordered() to change the Categorical to an ordered oner )r r r r r r ÚreÚescape)r r? r r! r# r# r$ Útest_numpy_min_max_raisest s
|
ÿ
|
z2TestCategoricalAnalytics.test_numpy_min_max_raisesÚkwargÚaxisÚoutZkeepdimsc C sl t ddddgdd}d| d| }|dkr0d }|d
|
i}tt|}tjt|d ||f| W 5 Q R X d S ) Nr r r Tr zthe 'z=' parameter is not supported in the pandas implementation of rG z8`axis` must be fewer than the number of dimensions \(1\)é* r )r r r r r Ú
|
ValueError)r r? rF r r! Úkwargsr# r# r$ Ú,test_numpy_min_max_unsupported_kwargs_raises s ÿ
|
zETestCategoricalAnalytics.test_numpy_min_max_unsupported_kwargs_raiseszmethod, expected)r r )r r c C s: t ddddgdd}tt|}||d d}||ks6td S )Nr r r Tr )rG )r r r r( )r r? r4 r r5 r# r# r$ Ú#test_numpy_min_max_axis_equals_none s
|
z<TestCategoricalAnalytics.test_numpy_min_max_axis_equals_nonezvalues,categories,exp_moderA r@ c C s: t ||dd}t| ¡ j}t ||dd}t ||¡ d S r3 )r r ÚmodeZ_valuesÚtmÚassert_categorical_equal)r r< r' Zexp_moder ÚresÚexpr# r# r$ Ú test_mode s z"TestCategoricalAnalytics.test_modec C s t dddddgddddg|d}t|}| d¡}|dks>tt|sJt| d¡}|dks`tt|slt| dg¡}| dg¡}tjdgtjd}t ||¡ t ||¡ |jddgd d
|
}|jddgd d
|
}tjdd gtjd}t ||¡ t ||¡ t
|
j t d d | d ¡ W 5 Q R X t
|
j t d d | d ¡ W 5 Q R X d}t
|
j t |d | dd g¡ W 5 Q R X t
|
j t |d | dd g¡ W 5 Q R X d S )NZcheeseZmilkZappleZbreadr& r- r. r0 Úright)Zsider@ Zcucumberr zMCannot setitem on a Categorical with a new category, set the categories first) r r Z searchsortedr( r r ÚarrayZintprO Zassert_numpy_array_equalr r r )r r r ZserZres_catZres_serrR r! r# r# r$ Útest_searchsorted¦ s@
|
ý
|
|
ÿz*TestCategoricalAnalytics.test_searchsortedc C sî t dddg|d}tdddg|d}| ¡ }t ||¡ tddddg|d}| ¡ }t |tddg|d¡ tdddddg|d}| ¡ }tdddg|d}t ||¡ tdtjdtjdg|d}| ¡ }tdtjdg|d}t ||¡ d S )Nr r r r r0 )r r ÚuniquerO rP r r; )r r r1 r rQ Zexp_catr# r# r$ Ú test_uniqueØ s z$TestCategoricalAnalytics.test_uniquec C sÚ t dddg|d}tdddddg|d}tdddg|d}t | ¡ |¡ t t| ¡ t|¡ t t| ¡ |¡ tddddg|d}tddg|d}t | ¡ |¡ t t| ¡ t|¡ t t| ¡ |¡ d S )Nr. r- r, r r0 )r r rO rP rW Úassert_index_equalr r )r r r1 r rR r# r# r$ Útest_unique_index_seriesð s z1TestCategoricalAnalytics.test_unique_index_seriesc C s t dddddg}| d¡}t tjddddg}t ||¡ t |d d
|dd
¡ | d¡}t dddtjtjgddddgd}t ||¡ t |d d
|d d
¡ t || d
|
¡¡ d S ) Nr r r r r, éÿÿÿÿéþÿÿÿ)r' r- r )r Úshiftr r; rO rP )r r Zsp1Zxp1Zsn2Zxp2r# r# r$ Ú
|
test_shift s
|
|
|
ÿ z#TestCategoricalAnalytics.test_shiftc C s$ t dddg}d}|j|ks td S )Nr, r- r. é )r Únbytesr( )r r rR r# r# r$ Ú test_nbytes s z$TestCategoricalAnalytics.test_nbytesc C s¤ t dddg}d|j k r(| ¡ ks.n td|j k rL|jddksRn tt dddg}|jdd|jksvtts |jddt |¡ }t|d k s td S )
|
Nr, r- r. r T)ÚdeepZfooÚbaréd )r r` Z memory_usager( r ÚsysÚ getsizeofÚabs)r r Zdiffr# r# r$ Útest_memory_usage s $z*TestCategoricalAnalytics.test_memory_usagec C sÀ t tdtddd}| dd ¡}t tdtddd}t ||¡ t tdtd d
|
d}| d d ¡}t tdtd d
|
d}t ||¡ | d d ¡}t |ttjdgd tj d¡ d S )NZABABCZCBATr& c S s | ¡ S ©N©Úlower©Úxr# r# r$ Ú<lambda>/ ó z3TestCategoricalAnalytics.test_map.<locals>.<lambda>Zababcr7 r+ Fc S s | ¡ S ri rj rl r# r# r$ rn 4 ro Úabcc S s dS )Nr, r# rl r# r# r$ rn 8 ro r, r@ r0 )
|
r ÚlistÚmaprO rP rY r r rU Úint64)r r r5 rR r# r# r$ Útest_map- s z!TestCategoricalAnalytics.test_mapÚvalueÚTrueg @c C sL t dddddg}dt|j }tjt|d |j|d W 5 Q R X d S )NÚAÚBÚCz9For argument "inplace" expected type bool, received type r )Zinplace)r ÚtypeÚ__name__r r rJ Z sort_values)r ru r r! r# r# r$ Ú test_validate_inplace_raises< s
|
ÿz5TestCategoricalAnalytics.test_validate_inplace_raisesc C sb t ddg}tddg}|d d
j|dd}|jjtjks@t|jddgd d
|
}t
|
||¡ d S ) Nrw rx g g à?r Zlinear)Ú interpolationr[ T)Z
|
allow_fill) r r Z _quantileÚ_codesr1 r Zint8r( ZtakerO Z assert_extension_array_equal)r r Úidxr5 r4 r# r# r$ Útest_quantile_emptyG s z,TestCategoricalAnalytics.test_quantile_emptyN)"r{ Ú
|
__module__Ú __qualname__r ÚmarkZ parametrizer% r* rq r ÚNaNÚparamr r r Zxfailr6 r; r= r> rB rE rL rM rS rV rX rZ r^ ra rh rt r| r r# r# r# r$ r
|
st
|
ÿýýþ ,þ
|
|
|
|
"$&&&"úþ
|
2
|
|
r
|
)rC re Únumpyr r Z pandas.compatr Zpandasr r r r r r Zpandas._testingZ_testingrO Zpandas.api.typesr r
|
r# r# r# r$ Ú<module> s
|
