o 5c<@sdddlZddlZddlZddlZddlmZmZmZm Z ddl m Z GdddZ GdddZdS)N) Categorical DataFrameSeries date_rangec@seZdZddZddZdS)TestCategoricalOpsWithFactorcCs tgddd}t||dS)N)abrrrcr r Tordered)rtmassert_categorical_equal)selffactorrd/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/arrays/categorical/test_operators.py test_categories_none_comparisonssz=TestCategoricalOpsWithFactor.test_categories_none_comparisonscCs||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|}|tj|}||k}t|t|k}t|||dk}tjt|t d}t||t gdgddd }t gd gddd }t gddd }t gd |j dd } ||k} t gd } t| | ||k} t gd } t| | || k} t gd } t| | d}t jt|d ||kWdn 1s%wYt gd gdd}t jt|d ||kWdn 1sKwY|d}||kr^Jt jt|d ||kWdn 1suwYtgd }d}t jt|d ||kWdn 1swYt jt|d ||kWdn 1swYt jt|d ||kWdn 1swYt jt|d ||kWdn 1swYt gd }t jt|d ||kWdn 1swYt jt|d ||kWdn 1s,wYt tdtddd }t gd } |dk} t| | |t dk} t| | dS)Nrr rddtype)rrr )r rrT categoriesr )rrrr TFFFFT>Categoricals can only be compared if 'categories' are the samematch)r rrrrFMCannot compare a Categorical for op __gt__ with type abccba)npasarrayr r lenrandom permutationassert_numpy_array_equalzerosboolrrarraypytestraises TypeError set_orderedanyrlist)rrresultexpectednothercat_rev cat_rev_basecatcat_baseres_revexp_revresexpmsg cat_rev_base2 cat_unorderdsrrrrtest_comparisonss                             z-TestCategoricalOpsWithFactor.test_comparisonsN)__name__ __module__ __qualname__rr@rrrrrs rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ e j dedededfgdgdgdfgddZe j dddddgddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Ze j d*gd+d,d-Zd.d/Zd0S)1TestCategoricalOpscCsngd}t|}t|}||jk}tgdg}t|||ddd|jk}tgdg}t||dS)Nrrr)TTTT)FTTF)rrTr assert_frame_equal)rdatar6dfr0r1rrrtest_compare_frames  z%TestCategoricalOps.test_compare_framecCsZ|}tgd}t|}d}tjt|d|||WddS1s&wYdS)NrEz5Unable to coerce to Series, length must be 1: given 4r)rrr*r+ ValueError)r comparison_opopr6rKr<rrrtest_compare_frame_raisess  "z,TestCategoricalOps.test_compare_frame_raisescCsTttddddd}t||dktgdt|d|ktgddS)Nz 2014-01-01)periodsTr rFTT)rrr r&r!r))rdt_catrrr$test_datetime_categorical_comparisons"z7TestCategoricalOps.test_datetime_categorical_comparisoncCsPtgddd}t||dktgdt|d|ktgddS)NrFrQTr rrS)rr r&r!r))rr6rrr&test_reflected_comparison_with_scalarss"z9TestCategoricalOps.test_reflected_comparison_with_scalarscCs$tgddd}d}tjt|d |dkWdn1s wYtjt|d |dkWdn1s;wYtjt|d d|kWdn1sVwYtjt|d d|kWdn1sqwYt|dktgdt|dktgddS) NrVTr z1Invalid comparison between dtype=category and intr)FFF)TTT)rr*r+r,r r&r!r))rr6r<rrr$test_comparison_with_unknown_scalarss     z7TestCategoricalOps.test_comparison_with_unknown_scalarscCs|ttjgdtd}|dk}tjgdtd}t|||dk}tjgdtd}t|||dk}t||dS)N)foorrWrQr\rr[)TFFFr\)FTFT)rr!r)objectr(r r&)rr6r0r1rrrtest_comparison_with_tuples  z-TestCategoricalOps.test_comparison_with_tuplecCstgdgddd}d}ttdttt|||}Wdn1s,wYt|||}t ||dS)NrWrFrQNrVTrrFignore rwarningscatch_warnings simplefilterRuntimeWarninggetattrr!r)r r&)rcompare_operators_no_eq_ner6scalarr1actualrrr9test_comparison_of_ordered_categorical_with_nan_to_scalars  zLTestCategoricalOps.test_comparison_of_ordered_categorical_with_nan_to_scalarcCstgdgddd}tgdgddd}ttdttt||d}Wdn1s5wYt|||}t ||dS)Nr_rVTr)rFrFrFrFr`rFra)rrgr6r3r1rirrr;test_comparison_of_ordered_categorical_with_nan_to_listlikes  zNTestCategoricalOps.test_comparison_of_ordered_categorical_with_nan_to_listlikezdata,reverse,baserr bbbrV)rQrFrW)rFrFrFcCs4tt||dd}tt||dd}tt|dd}tt||jjdd}t|}t|} ||k} tgd} t| | ||k} tgd} t| | ||k} tgd} t| | |d}||k} tgd} |j|k}t| | t | j|||k} tgd} |j|k}t| | t | j|d}t j t |d ||kWdn1swYd }t j t |d ||kWdn1swYt j t |d ||kWdn1swYt j t |d || kWdn 1swYt j t |d || kWdn 1s"wYt j t |d ||kWdn 1s>wYt j t |d ||kWdn 1sZwYt j t |d | |kWdn 1svwYt j t |d | |kWddS1swYdS) NTrr rrrWrrr) rrr6rr!r)r assert_series_equalvaluesr&r*r+r,)rrJreversebaser4r5r6r7r?rr8r9r:r;rhexp2exp_rev2r<rrrr@st                      $z#TestCategoricalOps.test_comparisonsctorcOst|i|SN)rargskwargsrrr.szTestCategoricalOps.cOstt|i|Srt)rrrurrrrx/scCs|ddgddgdd}|ddgddgdd}||ksJ|ddgddgdd}|ddgddgdd}||ks^sz.iT)by ascendingi)F)rightlabels value_group)__add__z\+)__sub__-)__mul__z\*) __truediv__/$Series cannot perform the operation |unsupported operandr)kurtskewvarstdmeansummedianzdoes not support reduction '') numeric_only)rr!r$randintranger sort_valuespdcutrr*r+r,rf)rrKr cat_labelsrOstr_repr<r?rrrtest_numeric_like_ops[s*    z(TestCategoricalOps.test_numeric_like_opscCsNttgd}tjtddt|WddS1s wYdS)NrWrFrQrYz does not support reduction 'sum'r)rrr*r+r,r!r)rr?rrrtest_numeric_like_ops_series{s "z/TestCategoricalOps.test_numeric_like_ops_seriesz op, str_reprcCs^ttgd}d|d}tjt|dt||dWddS1s(wYdS)NrrrrrF)rrr*r+r,rf)rrOrr?r<rrr"test_numeric_like_ops_series_ariths  "z5TestCategoricalOps.test_numeric_like_ops_series_arithcCsRttgd}d}tjt|dt|WddS1s"wYdS)Nrz:Object with dtype category cannot perform the numpy op logr)rrr*r+r,r!log)rr?r<rrr$test_numeric_like_ops_series_invalids  "z7TestCategoricalOps.test_numeric_like_ops_series_invalidN)rArBrCrLrPrUrXrZr^rjrkr*mark parametrizer/r@r|r~rrrrrrrrrrrDs@ * A   rD)rbnumpyr!r*pandasrrrrrpandas._testing_testingr rrDrrrrs r