ó šxŠ\c@s6dZddlZddlZddlZddlZddlmZddlm Z ddl m Z d„Z ej d„ƒZej d„ƒZej d „ƒZej d „ƒZej d „ƒZej d „ƒZej d „ƒZde jfd„ƒYZde jfd„ƒYZde jfd„ƒYZde jfd„ƒYZde jfd„ƒYZde jfd„ƒYZ de j!fd„ƒYZ"de j#fd„ƒYZ$de j%fd„ƒYZ&d e j'fd!„ƒYZ(d"e j)fd#„ƒYZ*d$e j+fd%„ƒYZ,d&e j-fd'„ƒYZ.dS((sj This file contains a minimal set of tests for compliance with the extension array interface test suite, and should contain no other tests. The test suite for the full functionality of the array is located in `pandas/tests/arrays/`. The tests in this file are inherited from the BaseExtensionTests, and only minimal tweaks should be applied to get the tests passing (by overwriting a parent method). Additional tests should either be added to one of the BaseExtensionTests classes (if they are relevant for the extension interface for all dtypes), or be added to the array-specific tests in `pandas/tests/arrays/`. iÿÿÿÿN(t Categorical(tCategoricalDtype(tbasecCsJxCtrEtjjttjƒddƒ}|d|dkrPqqW|S(Ntsizeidii(tTruetnptrandomtchoicetlisttstringt ascii_letters(tvalues((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyt make_datas  !cCstƒS(N(R(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pytdtype&scCs ttƒƒS(s‰Length-100 array for this type. * data[0] and data[1] should both be non missing * data[0] and data[1] should not gbe equal (RR (((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pytdata+scCsttjdgƒS(sLength 2 array with [NA, Valid]tA(RRtnan(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyt data_missing5scCs(tdddgddddgdtƒS(NRtBtCt categoriestordered(RR(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pytdata_for_sorting;s!cCs%tdddgdddgdtƒS(NRRRR(RtNoneR(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pytdata_missing_for_sortingAscCstjS(N(RR(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pytna_valueGsc Cs"tddddddddgƒS(Ntatbtc(RR(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pytdata_for_groupingLst TestDtypecBseZRS((t__name__t __module__(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRQst TestInterfacecBs&eZejjddƒd„ƒZRS(treasonsMemory usage doesn't matchcCstt|ƒj|ƒdS(N(tsuperR!ttest_memory_usage(tselfR((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR$Vs(RR tpytesttmarktskipR$(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR!UstTestConstructorscBseZRS((RR (((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR)\st TestReshapingcBseZRS((RR (((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR*`st TestGetitemcBsïeZejjddƒZejjddƒd„ƒZed„ƒZed„ƒZed„ƒZ ed„ƒZ ed„ƒZ ejjdd ƒd „ƒZ ed „ƒZ ejjdd ƒd „ƒZejjddƒd„ƒZRS(R"s GH-20664.sBackwards compatibilitycCstt|ƒj|ƒdS(N(R#R+ttest_getitem_scalar(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR,gscCs tt|ƒj|||ƒdS(N(R#R+t test_take(R%RRtna_cmp((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR-nscCstƒj|ƒdS(N(R#ttest_take_negative(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR/sscCstƒj||ƒdS(N(R#t&test_take_pandas_style_negative_raises(R%RR((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR0wscCstƒj|ƒdS(N(R#ttest_take_non_na_fill_value(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR1{scCstƒj||ƒS(N(R#ttest_take_out_of_bounds_raises(R%Rt allow_fill((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR2ss GH-20747. Unobserved categories.cCstƒj|ƒdS(N(R#ttest_take_series(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR4ƒscCstƒj|ƒdS(N(R#ttest_reindex_non_na_fill_value(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR5‡ssCategorical.take buggycCstƒj|||ƒdS(N(R#ttest_take_empty(R%RRR.((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR6‹ss*test not written correctly for categoricalcCstƒj||ƒdS(N(R#t test_reindex(R%RR((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR7s(RR R&R'R(t skip_takeR,R-R/R0R1R2R4R5R6R7(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR+dst TestSetitemcBseZRS((RR (((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR9”st TestMissingcBsDeZejjddƒd„ƒZejjddƒd„ƒZRS(R"sNot implementedcCstƒj|ƒdS(N(R#ttest_fillna_limit_pad(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR;šscCstƒj|ƒdS(N(R#ttest_fillna_limit_backfill(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR<žs(RR R&R'R(R;R<(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR:˜st TestReducecBseZRS((RR (((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR=£st TestMethodscBsVeZejjddƒd„ƒZd„Zejjddƒd„ƒZd„ZRS(R"sUnobserved categories includedcCstƒj||ƒS(N(R#ttest_value_counts(R%tall_datatdropna((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR?¨sc Csø|dƒ\}}tj|ƒ}tj|ƒ}|j|d„ƒ}tjgtt|ƒt|ƒƒD]\}}||^qgƒ} |j|| ƒ|jd} |j| d„ƒ}tjgt|ƒD]}|| ^q˃} |j|| ƒdS(NicSs||S(N((tx1tx2((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyt²sicSs||S(N((RBRC((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRD¸s(tpdtSeriestcombinetzipRtassert_series_equaltiloc( R%t data_repeatedt orig_data1t orig_data2ts1ts2tresultRRtexpectedtval((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyttest_combine_add¬s 8 ,sNot ApplicablecCstƒj|ƒdS(N(R#ttest_fillna_length_mismatch(R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRT¼scCs"|jstjddƒ‚ndS(NR"s#searchsorted requires ordered data.(RR&R((R%R((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyttest_searchsortedÀs ( RR R&R'R(R?RSRTRU(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyR>§s t TestCastingcBseZRS((RR (((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRVÅstTestArithmeticOpscBs&eZd„Zd„Zed„ZRS(cCs?|}|dkr.tt|ƒj||ƒn tjdƒdS(Nt__rmod__s/rmod never called when string is first argument(R#RWttest_arith_series_with_scalarR&R((R%Rtall_arithmetic_operatorstop_name((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRYËs   cCs7tj|ƒ}tjtddƒ ||WdQXdS(Ntmatchscannot perform(RERFR&traisest TypeError(R%Rtser((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyt$test_add_series_with_extension_arrayÔscCs"tt|ƒj|||dtƒS(Ntexc(R#RWt_check_divmod_opR^(R%tstoptotherRa((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRbÙs(RR RYR`tNotImplementedErrorRb(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRWÉs tTestComparisonOpscBseZd„ZRS(cCsÌ|j|ƒ}|dkrZ|||ƒ}|j|d„ƒ}||kjƒsÈt‚nn|dkr¥|||ƒ}|j|d„ƒ}||kjƒsÈt‚n#tjtƒ|||ƒWdQXdS(Nt__eq__cSs ||kS(N((txty((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRDåst__ne__cSs ||kS(N((RiRj((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRDês(tget_op_from_nameRGtalltAssertionErrorR&R]R^(R%RcRR[ReRdRPRQ((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyt_compare_otherás  (RR Ro(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRgßst TestParsingcBseZRS((RR (((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyRpòs(/t__doc__R tnumpyRR&tpandasRERtpandas.api.typesRtpandas.tests.extensionRR tfixtureR RRRRRRtBaseDtypeTestsRtBaseInterfaceTestsR!tBaseConstructorsTestsR)tBaseReshapingTestsR*tBaseGetitemTestsR+tBaseSetitemTestsR9tBaseMissingTestsR:tBaseNoReduceTestsR=tBaseMethodsTestsR>tBaseCastingTestsRVtBaseArithmeticOpsTestsRWtBaseComparisonOpsTestsRgtBaseParsingTestsRp(((sFlib/python2.7/site-packages/pandas/tests/extension/test_categorical.pyts8      0