ó jˆú\c@@sLdZddlmZddlZddlZddlZddlmZddlm Z ddlm Z ddl m Z de fd „ƒYZd e fd „ƒYZd eje fd „ƒYZdeje fd„ƒYZdeje fd„ƒYZd„ZedkrHddlZejejƒejƒe jddƒndS(s0Test module for enumerated types under PyTables.i(tabsolute_importN(tcommon(tunittest(tPyTablesTestCase(trangetCreateColTestCasecB@s‰eZdZddd„Zd„Zd„Zd„Zd„Zd„Zd„Z d „Z d „Z d „Z d „Z d „Zd„ZRS(s-Test creating enumerated column descriptions.tuint32cC@s¯tj||d|d|ƒ}tj|ƒ}|j|jdƒ|j|jjj|jjƒ|jt|j ƒt||ƒƒ|j|jj |ƒ|j|j |ƒdS(s2Create and check an enumerated column description.tbasetshapetenumN( ttablestEnumColtEnumt assertEqualttypetdtypeRtnametinttdfltRR (tselfR RRRtenumcoltsameEnum((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt _createCols#cC@s,tjdddgƒ}|j|dƒdS(s4Describing an enumerated column from an enumeration.tredtgreentblueN(R R R(Rtcolors((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest00a_validFromEnum*scC@s/idd6dd6dd6}|j|dƒdS(s2Describing an enumerated column from a dictionary.iRiRiRN(R(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest00b_validFromDict0scC@s#dddg}|j|dƒdS(s,Describing an enumerated column from a list.RRRN(R(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest00c_validFromList6scC@s#d}|jt|j|dƒdS(s7Describing an enumerated column from an invalid object.i{RN(t assertRaisest TypeErrorR(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest00d_invalidFromType<scC@s8idd6dd6dd6}|jt|j|dƒdS( s?Describing an enumerated column with an invalid default object.iRiRiRtblackN(RtKeyErrorR(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest01_invalidDfltBscC@s2idd6dd6dd6}|j|ddƒdS( s4Describing an enumerated column with a broader type.iRiRiRtint64N(R(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest02a_validDtypeBroaderHscC@sCgtdƒD]}d|^q }|jt|j|ddƒdS(s7Describing an enumerated column with a too narrow type.i,se%dte0tuint8N(RRRR(RtiR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest02b_invalidDtypeTooNarrowNs#cC@s)dddg}|j|dddƒdS(s<Describing an enumerated column with multidimensional shape.RRRRiN(i(R(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest03a_validShapeMDTscC@sˆtjdddgƒ}tj|ddddd ƒ}d}gtjdƒD]}|d j|ƒ^qL}|jt|ƒ|kƒd S(s5Checking the string representation of an enumeration.RRRRRRs[EnumCol(enum=Enum({%s}), dflt='red', base=UInt32Atom(shape=(), dflt=0), shape=(), pos=None)s 'blue': 2s 'green': 1s'red': 0s, N((s 'blue': 2s 'green': 1s'red': 0(R R R t itertoolst permutationstjoint assertTruetrepr(RRRttemplatetitemst permitations((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest04a_validReprEnumZs  &cC@s6idd6}|jt|j|ddtjƒƒdS(s<Describing an enumerated column of floats (not implemented).gð?RRN(RtNotImplementedErrorRR t FloatAtom(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest99a_nonIntEnumks cC@s/dddg}|jt|j|ddƒdS(sXDescribing an enumerated column encoded as floats. (not implemented). RRRtfloat64N(RR4R(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest99b_nonIntDtypersc C@s<idd6}|jt|j|ddtjddƒƒdS( sADescribing an enumerated column of non-scalars (not implemented).iiiRRRN(iii(RR4RR tIntAtom(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest99b_nonScalarEnum}s ((t__name__t __module__t__doc__RRRRR R#R%R)R*R3R6R8R:(((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRs           tCreateAtomTestCasecB@sweZdZdd d„Zd„Zd„Zd„Zd„Zd„Zd„Z d „Z d „Z d „Z d „Z RS(sTest creating enumerated atoms.RcC@s‰tj||d|d|ƒ}tj|ƒ}|j|jdƒ|j|jjj|jjƒ|j|j|ƒ|j|j |ƒdS(s$Create and check an enumerated atom.RRR N( R tEnumAtomR R RRRRRR (RR RRRtenumatomR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt _createAtomˆs cC@s,tjdddgƒ}|j|dƒdS(s2Describing an enumerated atom from an enumeration.RRRN(R R RA(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR’scC@s/idd6dd6dd6}|j|dƒdS(s0Describing an enumerated atom from a dictionary.iRiRiRN(RA(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR˜scC@s#dddg}|j|dƒdS(s*Describing an enumerated atom from a list.RRRN(RA(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRžscC@s#d}|jt|j|dƒdS(s5Describing an enumerated atom from an invalid object.i{RN(RRRA(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR ¤scC@s5idd6dd6dd6}|j|dddƒd S( s2Describing an enumerated atom with a broader type.iRiRiRRR$N(RA(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR%ªscC@sCgtdƒD]}d|^q }|jt|j|ddƒdS(s5Describing an enumerated atom with a too narrow type.i,se%dRR'N(RRRRA(RR(R((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR)°s#cC@s)dddg}|j|dddƒdS(s:Describing an enumerated atom with multidimensional shape.RRRRiN(i(RA(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR*¶scC@s6idd6}|jt|j|ddtjƒƒdS(s:Describing an enumerated atom of floats (not implemented).gð?RRN(RR4RAR R5(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR6¼s cC@s/dddg}|jt|j|ddƒdS(sWDescribing an enumerated atom encoded as a float. (not implemented). RRRR7N(RR4RA(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR8Ãsc C@s<idd6}|jt|j|ddtjddƒƒdS( s?Describing an enumerated atom of non-scalars (not implemented).iiiRRRN(iii(RR4RAR R9(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR:Îs ((R;R<R=RARRRR R%R)R*R6R8R:(((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR>…s         tEnumTableTestCasecB@sŸeZdZejidd6dd6dd6dd6ƒZdZejZd Z d Z dd „Z d „Z d „Z d„Zd„Zd„Zd„Zd„ZRS(s$Test tables with enumerated columns.iRiRiRiR!iÒtuint16c@s&dtjf‡‡fd†ƒY}|S(NtTestDescriptionc @sGeZejddƒZejˆjˆjdˆjdˆddƒZ RS(tposiRRi( R;R<R tIntColtridR R t defaultNametenumTypetrcolor((RR(s5lib/python2.7/site-packages/tables/tests/test_enum.pyRDàs(R t IsDescription(RRRD((RRs5lib/python2.7/site-packages/tables/tests/test_enum.pyt _descriptionßs"cC@s^|jjdd|jƒd|jƒƒ|jƒ|j|jjjjdƒ|j dƒdS(s3Reopening a file with tables using enumerated data.t/ttestttitleRJs5Enumerated type was not restored correctly from disk.N( th5filet create_tableRLt_getMethodNamet_reopenR trootRNtget_enumR (R((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest00a_reopenès   cC@sa|jjdd|jdƒd|jƒƒ|jƒ|j|jjjjdƒ|j dƒdS( sMReopening a file with tables using enumerated multi-dimensional data.RMRNiRORJs5Enumerated type was not restored correctly from disk.N(i( RPRQRLRRRSR RTRNRUR (R((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest00b_reopenMDôs  " cC@s |jjdd|jƒd|jƒƒ}d|jfd|jfg}|j}|dd|d<|dd|d <|jƒ|dd|d<|jt t j |d |ddƒ|j ƒd |_ |jƒ}tjd |d |fƒ|j|d |d ƒdS(s3Appending enumerated values using ``row.append()``.RMRNROi iiRGiRJtpythons&* appended value: %s * read value: %s iÿÿÿÿsWritten and read values differ.N(RPRQRLRRt valueInEnumtvalueOutOfEnumtrowtappendRt ValueErrortoperatortsetitemtflushtflavortreadRt verbosePrintR (RttbltappendedR[Rb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest01_rowAppends& !      cC@s¢|jjdd|jƒd|jƒƒ}d|jfd|jfg}|j|ƒ|jƒd|_|j ƒ}t j d||fƒ|j ||dƒd S( s5Appending enumerated values using ``table.append()``.RMRNROi iRXs&* appended value: %s * read value: %s sWritten and read values differ.N( RPRQRLRRRYRZR\R`RaRbRRcR (RRdReRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt test02_appends !     cC@s½|jjdd|jƒd|jƒƒ}d|jfd|jfg}|j|ƒd|jfd|jfg}||(d|_|jƒ}t j d||fƒ|j ||dƒd S( s9Changing enumerated values using ``table.__setitem__()``.RMRNROi iRXs%* written value: %s * read value: %s sWritten and read values differ.N( RPRQRLRRRYR\RZRaRbRRcR (RRdRetwrittenRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest03_setitem3s !     cC@sb|jjdd|jdƒd|jƒƒ}d|j|jffd|j|jffg}|j}|dd|d<|jtt j |d |dd ƒ|j |ƒ|j ƒd |_ |jƒ}x“tt|ƒƒD]}|j||d||dd ƒ|j||d d||d dd ƒ|j||d d ||d d d ƒqÛWd S(s,Appending multi-dimensional enumerated data.RMRNiROi iiRGRJiRXsWritten and read values differ.N(i(RPRQRLRRRYRZR[RR]R^R_R\R`RaRbRtlenR (RRdReR[RbR(((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest04_multidimJs& $     $$cC@sí|jjdd|jƒd|jƒƒ}d|jfd|jfd|jfg}|j|ƒ|jƒg|jdi|jd6ƒD]}|d |d f^qˆ}t j d |j|j j ||d fƒ|j ||d d ƒdS(sSearching enumerated data.RMRNROi iis rcolor == vtvRGRJs\* ``valueInEnum``: %s * ``rcolor`` column: ``%s`` * ``searched``: %s * Should look like: %s iÿÿÿÿs"Search returned incorrect results.N(RPRQRLRRRYRZR\R`twhereRRctcolsRJR (RRdReR[tsearched((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt test05_wherees !    :!((R;R<R=R R R RHRRYRZRIRLRVRWRfRgRiRkRp(((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRBÖs+     tEnumEArrayTestCasecB@s«eZdZejidd6dd6dd6dd6ƒZejZd Zd Z dd „Z d „Z d „Z d„Z d„Zd„Zd„Zd„Zd„Zd„ZRS(s,Test extendable arrays of enumerated values.iRiRiRiR!iÒRCcC@s"tj|jdd|jd|ƒS(NRRR(R R?R RI(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt_atom‡scC@ss|jjdd|jƒdd d|jƒƒd|jjj_|jƒ|j|jjjj ƒ|j dƒdS( s>Reopening a file with extendable arrays using enumerated data.RMRNRiRORXs5Enumerated type was not restored correctly from disk.N(i( RPt create_earrayRrRRRTRNRaRSR RUR (R((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRV‹s   cC@ss|jjdd|jƒdd d|jƒƒd|jjj_|jƒ|j|jjjj ƒ|j dƒd S( sXReopening a file with extendable arrays using enumerated multi-dimensional data.RMRNRiiRORXs5Enumerated type was not restored correctly from disk.N(ii( RPRsRrRRRTRNRaRSR RUR (R((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRW™s   cC@s®d}tj|j|d|jdd ƒ}|jjdd|dd d|jƒƒ|jƒ|j|jj j j ƒ|jdƒ|j|jj j j j |j|d ƒdS( NRRRRMRNiROs5Enumerated type was not restored correctly from disk.sJThe default value of enumerated type was not restored correctly from disk.((i(R R?R RIRPRsRRRSR RTRNRUtatomR(RRRt((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt!test_enum_default_persistence_red¨s  cC@s®d}tj|j|d|jdd ƒ}|jjdd|dd d|jƒƒ|jƒ|j|jj j j ƒ|jdƒ|j|jj j j j |j|d ƒdS( NRRRRMRNiROs5Enumerated type was not restored correctly from disk.sJThe default value of enumerated type was not restored correctly from disk.((i(R R?R RIRPRsRRRSR RTRNRURtR(RRRt((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt#test_enum_default_persistence_greenºs  cC@s®d}tj|j|d|jdd ƒ}|jjdd|dd d|jƒƒ|jƒ|j|jj j j ƒ|jdƒ|j|jj j j j |j|d ƒdS( NRRRRMRNiROs5Enumerated type was not restored correctly from disk.sJThe default value of enumerated type was not restored correctly from disk.((i(R R?R RIRPRsRRRSR RTRNRURtR(RRRt((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt"test_enum_default_persistence_blueÌs  cC@s®d}tj|j|d|jdd ƒ}|jjdd|dd d|jƒƒ|jƒ|j|jj j j ƒ|jdƒ|j|jj j j j |j|d ƒdS( NR!RRRMRNiROs5Enumerated type was not restored correctly from disk.sJThe default value of enumerated type was not restored correctly from disk.((i(R R?R RIRPRsRRRSR RTRNRURtR(RRRt((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt#test_enum_default_persistence_blackÞs  cC@s…|jjdd|jƒdd d|jƒƒ}d|_|j|jg}|j|ƒ|jƒ|j ƒ}|j ||dƒdS( s/Appending scalar elements of enumerated values.RMRNRiRORXsWritten and read values differ.N(i( RPRsRrRRRaRYRZR\R`RbR (RtearrReRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt test01_appendðs     cC@s—|jjdd|jƒdd d|jƒƒ}d|_|j|jg|j|jgg}|j|ƒ|jƒ|j ƒ}|j ||dƒd S( s:Appending multi-dimensional elements of enumerated values.RMRNRiiRORXsWritten and read values differ.N(ii( RPRsRrRRRaRYRZR\R`RbR (RRyReRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyttest02_appendMDÿs     cC@s”|jjdd|jƒdd d|jƒƒ}d|_|j|jf}|j|ƒ|j|jg}||(|jƒ}|j ||dƒdS( s:Changing enumerated values using ``earray.__setitem__()``.RMRNRiRORXsWritten and read values differ.N(i( RPRsRrRRRaRYR\RZRbR (RRyReRhRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRis    ((R;R<R=R R R RRYRZRIRrRVRWRuRvRwRxRzR{Ri(((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRqs+          tEnumVLArrayTestCasecB@s‡eZdZejidd6dd6dd6dd6ƒZejZd Zd Z dd „Z d „Z d „Z d„Z d„Zd„ZRS(s1Test variable-length arrays of enumerated values.iRiRiRiR!iÒRCcC@s"tj|jdd|jd|ƒS(NRRR(R R?R RI(RR((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRr)scC@sm|jjdd|jƒd|jƒƒd|jjj_|jƒ|j|jjjj ƒ|j dƒdS(sKReopening a file with variable-length arrays using enumerated data.RMRNRORXs5Enumerated type was not restored correctly from disk.N( RPtcreate_vlarrayRrRRRTRNRaRSR RUR (R((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRV-s   cC@sp|jjdd|jdƒd|jƒƒd|jjj_|jƒ|j|jjjj ƒ|j dƒdS( s]Reopening a file with variable-length arrays using enumerated multi-dimensional data.RMRNiRORXs5Enumerated type was not restored correctly from disk.N(i( RPR}RrRRRTRNRaRSR RUR (R((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRW<s   cC@s·|jjdd|jƒd|jƒƒ}d|_|jg|j|jgg}|j|dƒ|j|dƒ|jƒ|j ƒ}t j d||fƒ|j ||dƒd S( s/Appending scalar elements of enumerated values.RMRNRORXiis&* appended value: %s * read value: %s sWritten and read values differ.N( RPR}RrRRRaRYRZR\R`RbRRcR (RtvlarrReRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRzKs     cC@sÕ|jjdd|jd ƒd|jƒƒ}d|_|j|jgg|j|jg|j|jggg}|j|dƒ|j|dƒ|jƒ|j ƒ}t j d||fƒ|j ||d ƒd S( s:Appending multi-dimensional elements of enumerated values.RMRNiRORXiis&* appended value: %s * read value: %s sWritten and read values differ.N(i( RPR}RrRRRaRYRZR\R`RbRRcR (RR~ReRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR{as    cC@s¬|jjdd|jƒd|jƒƒ}d|_|j|jf}|j|ƒ|j|jg}||d<|jƒ}t j d||fƒ|j ||ddƒdS( s;Changing enumerated values using ``vlarray.__setitem__()``.RMRNRORXis%* written value: %s * read value: %s sWritten and read values differ.N( RPR}RrRRRaRYR\RZRbRRcR (RR~ReRhRb((s5lib/python2.7/site-packages/tables/tests/test_enum.pyRixs     ((R;R<R=R R R RRYRZRIRrRVRWRzR{Ri(((s5lib/python2.7/site-packages/tables/tests/test_enum.pyR|!s+      cC@sÍddl}ddlm}tjƒ}d}x˜t|ƒD]Š}|j|j|ƒƒ|jtjt ƒƒ|jtjt ƒƒ|jtjt ƒƒ|jtjt ƒƒ|jtjt ƒƒq;W|S(sCReturn a test suite consisting of all the test cases in the module.iN(R i(tdoctestt tables.miscR Rt TestSuiteRtaddTestt DocTestSuitet makeSuiteRR>RBRqR|(RR ttheSuitetniterR(((s5lib/python2.7/site-packages/tables/tests/test_enum.pytsuites  t__main__t defaultTestR‡(R=t __future__RR^R+R t tables.testsRttables.tests.commonRRtTestCaset six.movesRRR>t TempFileMixinRBRqR|R‡R;tsyst parse_argvtargvtprint_versionstmain(((s5lib/python2.7/site-packages/tables/tests/test_enum.pyt s&   kQ©¢l