ó šxŠ\c @sâdZddlZddlZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZd„Zejde e e eeeeegƒd „ƒZejd „ƒZejd „ƒZejd „ƒZejd „ƒZejd„ƒZejd„ƒZejd„ƒZdejfd„ƒYZ dej!fd„ƒYZ"dej#fd„ƒYZ$dej%fd„ƒYZ&dej'fd„ƒYZ(dej)fd„ƒYZ*dej+fd„ƒYZ,dej-fd „ƒYZ.d!ej/fd"„ƒYZ0d#ej1fd$„ƒYZ2d%ej3fd&„ƒYZ4d'ej5fd(„ƒYZ6d)ej7fd*„ƒYZ8d+ej9fd,„ƒYZ:d-ej;fd.„ƒYZ<d/ej=fd0„ƒYZ>dS(1sj 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(tis_extension_array_dtype(t integer_array(t Int8Dtypet Int16Dtypet Int32Dtypet Int64Dtypet UInt8Dtypet UInt16Dtypet UInt32Dtypet UInt64Dtype(tbasecCsDttddƒƒtjgttddƒƒtjgddgS(Nii i ibicid(tlisttrangetnptnan(((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyt make_datastparamscCs |jƒS(N(tparam(trequest((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytdtype"scCsttƒd|ƒS(NR(RR(R((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytdata(scCsttjdgd|ƒS(NiR(RR R(R((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyt data_missing-scCstdddgd|ƒS(NiiiR(R(R((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytdata_for_sorting2scCstdtjdgd|ƒS(NiiR(RR R(R((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytdata_missing_for_sorting7scCsd„S(NcSstj|ƒotj|ƒS(N(R tisnan(txty((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyt?s((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytna_cmp<scCstjS(N(R R(((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytna_valueBsc CsCd}d}d}tj}t||||||||gd|ƒS(NiiiR(R RR(Rtbtatctna((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytdata_for_groupingGs  t TestDtypecBs&eZejjddƒd„ƒZRS(treasonsusing multiple dtypescCsdS(N((tself((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyttest_is_dtype_unboxes_dtypeRs(t__name__t __module__tpytesttmarktskipR&(((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyR#PstTestArithmeticOpscBsJeZdd„Zed„Zdd„Zejj ddƒd„ƒZ RS(cCs&tt|ƒj|||ddƒdS(Ntexc(tsuperR,t check_opnametNone(R%tstop_nametotherR-((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyR/ZscCs®|dkr‡|jjr4|dkr4tjdƒnt|dƒrƒt|jƒ rƒtjj j |jƒrƒ|j |jj ƒ}n|||ƒ}|j ||ƒ}|dkrÀtjdƒn„|d krü|j tƒ}|dkrD|j tƒ}qDnH|jd ƒr2|j |jƒ}|j |jƒ}n|j |jƒ}|d krtt|tjƒrt|jd ƒ}n|j||ƒn#tj|ƒ|||ƒWdQXdS( Nt__rsub__s*unsigned subtraction gives negative valuesRt__rdiv__s skipping reverse div in python 2t __rtruediv__t __truediv__t__div__t__rt__rpow__i(R6R7R8(R0Rtis_unsigned_integerR)R+thasattrRtpdtapittypestis_integer_dtypetastypet numpy_dtypetcombinetfloatt startswitht isinstancetSeriestfillnatassert_series_equaltraises(R%R1topR3R2R-tresulttexpected((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyt _check_op_s0    cCs#tt|ƒj|||dƒdS(N(R.R,t_check_divmod_opR0(R%R1RKR3R-((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyROŠsR$sintNA does not error on opscCsdS(N((R%Rtall_arithmetic_operators((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyt test_errorsN( R'R(R0R/tNotImplementedErrorRNROR)R*R+RQ(((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyR,Xs  + tTestComparisonOpscBseZdd„Zd„ZRS(cCs&tt|ƒj|||ddƒdS(NR-(R.RSR/R0(R%R1R2R3R-((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyR/•scCs|j|||ƒdS(N(R/(R%R1RR2R3((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyt_compare_other™sN(R'R(R0R/RT(((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRS“s t TestInterfacecBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRUstTestConstructorscBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRV¡st TestReshapingcBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRW¥st TestGetitemcBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRX­st TestSetitemcBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRY±st TestMissingcBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRZµst TestMethodscBs,eZejjdeegƒd„ƒZRS(tdropnacCs¡|d }|r-tj||jƒƒ}n|}tj|ƒjd|ƒjƒ}tj|ƒjd|ƒjƒ}|jj|j ƒ|_|j ||ƒdS(Ni R\( R tarraytisnaR=RGt value_countst sort_indextindexRARRI(R%tall_dataR\R3RLRM((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyttest_value_counts»s !(R'R(R)R*t parametrizetTruetFalseRc(((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyR[¹st TestCastingcBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRgËst TestGroupbycBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRhÏstTestNumericReducecBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRiÓstTestBooleanReducecBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRj×st TestPrintingcBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRkÛst TestParsingcBseZRS((R'R((((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pyRlßs(?t__doc__tnumpyR R)tpandas.core.dtypes.commonRtpandasR=tpandas.core.arraysRtpandas.core.arrays.integerRRRRRRRR tpandas.tests.extensionR RtfixtureRRRRRRRR"tBaseDtypeTestsR#tBaseArithmeticOpsTestsR,tBaseComparisonOpsTestsRStBaseInterfaceTestsRUtBaseConstructorsTestsRVtBaseReshapingTestsRWtBaseGetitemTestsRXtBaseSetitemTestsRYtBaseMissingTestsRZtBaseMethodsTestsR[tBaseCastingTestsRgtBaseGroupbyTestsRhtBaseNumericReduceTestsRitBaseBooleanReduceTestsRjtBasePrintingTestsRktBaseParsingTestsRl(((sBlib/python2.7/site-packages/pandas/tests/extension/test_integer.pytsB   :  ;