U
|
¬ý°dê= ã @ sd d dl Z d dlZd dlZd dlZd dlmZmZmZm Z d dl
|
m Z G dd dZ G dd dZdS )é N)Ú CategoricalÚ DataFrameÚSeriesÚ
|
date_rangec @ s e Zd Zdd Zdd ZdS )Ú TestCategoricalOpsWithFactorc C s, t ddddddddgdd}t ||¡ d S )NÚaÚbÚcT©Úordered)r ÚtmÚassert_categorical_equal)ÚselfÚfactor© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_operators.pyÚ test_categories_none_comparisons s z=TestCategoricalOpsWithFactor.test_categories_none_comparisonsc C s ||dk }|t |¡dk }t ||¡ ||dk }|t |¡dk }t ||¡ ||dk }|t |¡dk }t ||¡ ||dk }|t |¡dk }t ||¡ ||dk }|t |¡dk }t ||¡ ||dk }|t |¡dk }t ||¡ t|}|t j |¡ }||k}t |¡t |¡k}t ||¡ |dk}t jt|t d}t ||¡ t
|
dddgdddgdd}t
|
dddgdddgdd}t
|
dddgdd}t
|
dddg|j dd} ||k}
|
t dd d g¡} t |
|
| ¡ ||k }
|
t d d dg¡} t |
|
| ¡ || k} t d d dg¡} t | | ¡ d
|
}t jt|d ||k W 5 Q R X t
|
dddgddddgd }t jt|d ||k W 5 Q R X | d ¡}||k ¡ r¨tt jt|d ||k W 5 Q R X tdddg}d }t jt|d ||k W 5 Q R X t jt|d ||k W 5 Q R X t jt|d ||k W 5 Q R X t jt|d ||k W 5 Q R X t dddg¡}t jt|d ||k W 5 Q R X t jt|d ||k W 5 Q R X t
|
tdtddd}t dd d g¡} |dk} t | | ¡ |t d¡k} t | | ¡ d S )Nr r r Úd©ZdtypeT©Ú
|
categoriesr r
|
Fú>Categoricals can only be compared if 'categories' are the same©Úmatch©r úMCannot compare a Categorical for op __gt__ with type <class 'numpy\.ndarray'>ÚabcÚcba)ÚnpZasarrayr r ÚlenÚrandomZ permutationÚassert_numpy_array_equalÚzerosÚboolr r ÚarrayÚpytestÚraisesÚ TypeErrorZ set_orderedÚanyÚAssertionErrorr Úlist)r r ÚresultÚexpectedÚnÚotherÚcat_revÚ cat_rev_baseÚcatÚcat_baseÚres_revÚexp_revÚresÚexpÚmsgZ cat_rev_base2Z cat_unorderdÚsr r r r Útest_comparisons s ÿ
|
ÿ z-TestCategoricalOpsWithFactor.test_comparisonsN)Ú__name__Ú
|
__module__Ú __qualname__r r9 r r r r r s r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
e j dedededfdddgdddgdddgfg¡dd Ze j ddd d d g¡d d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Ze j d*d+d,d-d.g¡d/d0 Zd1d2 Zd3S )4ÚTestCategoricalOpsc C sz ddddg}t |}t|}||jk}tddddgg}t ||¡ |d d d
|jk}tddddgg}t ||¡ d S )Nr r é TéÿÿÿÿF)r r ÚTr Zassert_frame_equal)r Údatar1 Údfr+ r, r r r Útest_compare_frame s
|
z%TestCategoricalOps.test_compare_framec C sH |}t ddddg}t|}d}tjt|d ||| W 5 Q R X d S )Nr r r> z5Unable to coerce to Series, length must be 1: given 4r )r r r% r& Ú
|
ValueError)r Z comparison_opÚopr1 rB r7 r r r Útest_compare_frame_raises s z,TestCategoricalOps.test_compare_frame_raisesc C sX t tddddd}t ||d kt dddg¡¡ t |d |k t dddg¡¡ d S )Nz
|
2014-01-01é )ZperiodsTr
|
r F)r r r r! r r$ )r Zdt_catr r r Ú$test_datetime_categorical_comparison s z7TestCategoricalOps.test_datetime_categorical_comparisonc C sV t dddgdd}t ||d kt dddg¡¡ t |d |k t dddg¡¡ d S )Né r> rG Tr
|
r F)r r r! r r$ )r r1 r r r Ú&test_reflected_comparison_with_scalars s z9TestCategoricalOps.test_reflected_comparison_with_scalarsc C sÚ t dddgdd}d}tjt|d |dk W 5 Q R X tjt|d |dk W 5 Q R X tjt|d d|k W 5 Q R X tjt|d d|k W 5 Q R X t |dkt d d d g¡¡ t |dkt dddg¡¡ d S )
|
NrI r> rG Tr
|
z1Invalid comparison between dtype=category and intr é F)r r% r& r' r r! r r$ )r r1 r7 r r r Ú$test_comparison_with_unknown_scalars¦ s z7TestCategoricalOps.test_comparison_with_unknown_scalarsc C s t tjddddgtd}|dk}tjddddgtd}t ||¡ |dk}tjddddgtd}t ||¡ |dk}t || ¡ d S )NZfoo)r rI rG r TF)r r r$ Úobjectr# r r! )r r1 r+ r, r r r Útest_comparison_with_tuple¹ s z-TestCategoricalOps.test_comparison_with_tuplec C sr t dddd gdddgdd}d}t ¡ & t dt¡ tt |¡||}W 5 Q R X t|||}t ||¡ d S ©NrI r> rG Tr Úignore©
|
r ÚwarningsÚcatch_warningsÚ simplefilterÚRuntimeWarningÚgetattrr r$ r r! )r Úcompare_operators_no_eq_ner1 Úscalarr, Úactualr r r Ú9test_comparison_of_ordered_categorical_with_nan_to_scalarÇ s
|
zLTestCategoricalOps.test_comparison_of_ordered_categorical_with_nan_to_scalarc C s t dddd gdddgdd}t ddddgdddgdd}t ¡ & t dt¡ tt |¡|d}W 5 Q R X t|||}t ||¡ d S rO rQ )r rW r1 r. r, rY r r r Ú;test_comparison_of_ordered_categorical_with_nan_to_listlike× s
|
zNTestCategoricalOps.test_comparison_of_ordered_categorical_with_nan_to_listlikezdata,reverse,baser r ÚbbbrI r> rG c C s| t t||dd}t t||dd}t t|dd}t t||jjdd}t |}t |¡} ||k}
|
t dddg} t |
|
| ¡ ||k }
|
t dddg} t |
|
| ¡ ||k} t dddg} t | | ¡ |d }||k} t dddg} |j|k}t | | ¡ t | j|¡ ||k}
|
t dddg} |j|k}t |
|
| ¡ t |
|
j|¡ d}t
|
j t |d ||k W 5 Q R X d}t
|
j t |d ||k W 5 Q R X t
|
j t |d ||k W 5 Q R X t
|
j t |d || k W 5 Q R X t
|
j t |d || k W 5 Q R X t
|
j t |d ||k W 5 Q R X t
|
j t |d ||k W 5 Q R X t
|
j t |d | |k W 5 Q R X t
|
j t |d | |k W 5 Q R X d S ) NTr r
|
FrI r r r ) r r r1 r r r$ r Zassert_series_equalÚvaluesr! r% r& r' )r rA ÚreverseÚbaser/ r0 r1 r2 r8 r r3 r4 r5 r6 rX Zexp2Zexp_rev2r7 r r r r9 æ sb ÿ
|
|
|
ÿz#TestCategoricalOps.test_comparisonsÚctorc O s
|
t | |S ©N)r ©ÚargsÚkwargsr r r Ú<lambda>. ó zTestCategoricalOps.<lambda>c O s t t| |S ra )r r rb r r r re / rf c C s |ddgddgdd}|ddgddgdd}||k ¡ s<t|ddgddgdd}|ddgddgdd}||k ¡ sxt|ddgddgdd}|ddgddgdd}||k ¡ s´t|ddgddgdd}|ddgddgdd}||k}t t |¡t ddg¡¡ d S )Nr r Fr T)Úallr) r r! r r$ )r r` Úc1Úc2r+ r r r Ú$test_unordered_different_order_equal+ s z7TestCategoricalOps.test_unordered_different_order_equalc C sR t ddgddgdd}t ddgddgdd}tjtdd ||k W 5 Q R X d S )Nr r Fr r z!Categoricals can only be comparedr ©r r% r& r' )r rh ri r r r Ú*test_unordered_different_categories_raisesE s z=TestCategoricalOps.test_unordered_different_categories_raisesc C sH t g ddgd}t g dgd}d}tjt|d ||k W 5 Q R X d S )Nr r r z?Categoricals can only be compared if 'categories' are the same.r rk )r rh ri r7 r r r Ú test_compare_different_lengthsL s
|
z1TestCategoricalOps.test_compare_different_lengthsc C s6 t dgddgd}t dgddgd}| |¡r2td S )Nr r r )r Úequalsr) )r r r r r r Ú&test_compare_unordered_different_orderT s z9TestCategoricalOps.test_compare_unordered_different_orderc
|
C sö t dtj ddd¡i}dd tdddD }t||}|jdgdd }tj|j tdd
|
dd |d |d < dD ]<\}}d| d}t
|
j t |d t ||| W 5 Q R X ql|d }dD ]:}d| d}t
|
j t |d t ||d d W 5 Q R X q¶d S )NÚvaluer i' éd c S s g | ]}| d |d qS )z - ió r )Ú.0Úir r r Ú
|
<listcomp>] s z<TestCategoricalOps.test_numeric_like_ops.<locals>.<listcomp>iô T)ZbyZ ascendingi) F)ÚrightÚlabelsZ value_group)©Ú__add__z\+©Ú__sub__ú-©Ú__mul__z\*©Ú __truediv__ú/ú$Series cannot perform the operation ú|unsupported operandr )ZkurtZskewÚvarZstdZmeanÚsumZmedianz does not support reduction 'ú')Z numeric_only)r r r ÚrandintÚranger Z sort_valuesÚpdZcutrp r% r& r' rV )r rB rv Z
|
cat_labelsrE Ústr_repr7 r8 r r r Útest_numeric_like_ops[ s&
|
|
ÿ
|
z(TestCategoricalOps.test_numeric_like_opsc C s< t tddddg}tjtdd t |¡ W 5 Q R X d S )NrI r> rG rK z does not support reduction 'sum'r )r r r% r& r' r r )r r8 r r r Ú test_numeric_like_ops_seriesx s z/TestCategoricalOps.test_numeric_like_ops_seriesz op, str_reprw ry r| r~ c C sL t tddddg}d| d}tjt|d t||d W 5 Q R X d S )NrI r> rG rK r r r )r r r% r& r' rV )r rE r r8 r7 r r r Ú"test_numeric_like_ops_series_arith~ s z5TestCategoricalOps.test_numeric_like_ops_series_arithc C s@ t tddddg}d}tjt|d t |¡ W 5 Q R X d S )NrI r> rG rK z:Object with dtype category cannot perform the numpy op logr )r r r% r& r' r Úlog)r r8 r7 r r r Ú$test_numeric_like_ops_series_invalid s z7TestCategoricalOps.test_numeric_like_ops_series_invalidN)r: r; r< rC rF rH rJ rL rN rZ r[ r% ÚmarkZ parametrizer* r9 rj rl rm ro r r r r r r r r r= sF 0þ
|
Aþþ
|
üþ
|
r= )rR Únumpyr r% Zpandasr r r r r Zpandas._testingZ_testingr r r= r r r r Ú<module> s r
|
