B ™i‚\/Yã@sðdZddlmZddlZddlZddlZddlmZddlm Z ddlm Z ddl m Z Gdd „d e ƒZGd d „d e ƒZGd d „d eje ƒZGdd„deje ƒZGdd„deje ƒZdd„ZedkrìddlZe ej¡e ¡e jdddS)z0Test module for enumerated types under PyTables.é)Úabsolute_importN)Úcommon)Úunittest)ÚPyTablesTestCase)Úrangec@szeZdZdZddd„Zdd„Zdd „Zd d „Zd d „Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„ZdS) ÚCreateColTestCasez-Test creating enumerated column descriptions.Úuint32©cCs|tj||||d}t |¡}| |jd¡| |jjj|jj¡| t|j ƒt||ƒ¡| |jj |¡| |j |¡dS)z2Create and check an enumerated column description.)ÚbaseÚshapeÚenumN) ÚtablesÚEnumColÚEnumÚ assertEqualÚtypeÚdtyper ÚnameÚintÚdfltr r )Úselfr rr r ÚenumcolÚsameEnumr r ú5lib/python3.7/site-packages/tables/tests/test_enum.pyÚ _createCols zCreateColTestCase._createColcCs t dddg¡}| |d¡dS)z4Describing an enumerated column from an enumeration.ÚredÚgreenÚblueN)r rr)rÚcolorsr r rÚtest00a_validFromEnum*sz'CreateColTestCase.test00a_validFromEnumcCsddddœ}| |d¡dS)z2Describing an enumerated column from a dictionary.ééé)rrrrN)r)rrr r rÚtest00b_validFromDict0s z'CreateColTestCase.test00b_validFromDictcCsdddg}| |d¡dS)z,Describing an enumerated column from a list.rrrN)r)rrr r rÚtest00c_validFromList6s z'CreateColTestCase.test00c_validFromListcCsd}| t|j|d¡dS)z7Describing an enumerated column from an invalid object.é{rN)Ú assertRaisesÚ TypeErrorr)rrr r rÚtest00d_invalidFromType<sz)CreateColTestCase.test00d_invalidFromTypecCs"ddddœ}| t|j|d¡dS)z?Describing an enumerated column with an invalid default object.r r!r")rrrÚblackN)r&ÚKeyErrorr)rrr r rÚtest01_invalidDfltBs z$CreateColTestCase.test01_invalidDfltcCsddddœ}| |dd¡dS)z4Describing an enumerated column with a broader type.r r!r")rrrrÚint64N)r)rrr r rÚtest02a_validDtypeBroaderHs z+CreateColTestCase.test02a_validDtypeBroadercCs*dd„tdƒDƒ}| t|j|dd¡dS)z7Describing an enumerated column with a too narrow type.cSsg|] }d|‘qS)ze%dr )Ú.0Úir r rú QszCCreateColTestCase.test02b_invalidDtypeTooNarrow..i,Ze0Úuint8N)rr&r'r)rrr r rÚtest02b_invalidDtypeTooNarrowNsz/CreateColTestCase.test02b_invalidDtypeTooNarrowcCsdddg}|j|ddddS)z.)z 'blue': 2z 'green': 1z'red': 0N)r rrÚ itertoolsÚ permutationsZ assertTrueÚrepr)rrrZ permitationsr )r6rÚtest04a_validReprEnumZs  z'CreateColTestCase.test04a_validReprEnumcCs&ddi}|jt|j|dt ¡ddS)z.i,rr1N)rr&r'rI)rrr r rr2°sz0CreateAtomTestCase.test02b_invalidDtypeTooNarrowcCsdddg}|j|ddddS)z:Describing an enumerated atom with multidimensional shape.rrr)r!)r N)rI)rrr r rr3¶s z'CreateAtomTestCase.test03a_validShapeMDcCs&ddi}|jt|j|dt ¡ddS)z:Describing an enumerated atom of floats (not implemented).rgð?)r N)r&r;rIr r<)rrr r rr=¼sz%CreateAtomTestCase.test99a_nonIntEnumcCs"dddg}| t|j|dd¡dS)zWDescribing an enumerated atom encoded as a float. (not implemented). rrrr>N)r&r;rI)rrr r rr?Ãs z&CreateAtomTestCase.test99b_nonIntDtypecCs*ddi}|jt|j|dtjddddS)z?Describing an enumerated atom of non-scalars (not implemented).r)r"r!r@r@)r )r N)r&r;rIr rA)rrr r rrBÎsz(CreateAtomTestCase.test99b_nonScalarEnumN)rr )rCrDrErFrIrr#r$r(r-r2r3r=r?rBr r r rrG…s  rGc@sxeZdZdZe dddddœ¡ZdZejZ dZ d Z dd d „Z d d„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„ZdS)ÚEnumTableTestCasez$Test tables with enumerated columns.r r!r"r)rrrr)r)iÒÚuint16r csG‡‡fdd„dtjƒ}|S)Ncs2eZdZejddZej”j”j”j ”ddZ dS)z7EnumTableTestCase._description..TestDescriptionr)Úposr")r r rLN) rCrDrEr ZIntColÚridrr Ú defaultNameÚenumTypeÚrcolorr )rr r rÚTestDescriptionàs rQ)r Z IsDescription)rr rQr )rr rÚ _descriptionßszEnumTableTestCase._descriptioncCsD|jjdd| ¡| ¡d| ¡| |jjj d¡|j d¡dS)z3Reopening a file with tables using enumerated data.ú/Útest)ÚtitlerPz5Enumerated type was not restored correctly from disk.N) Úh5fileÚ create_tablerRÚ_getMethodNameÚ_reopenrÚrootrTÚget_enumr )rr r rÚtest00a_reopenès z EnumTableTestCase.test00a_reopencCsF|jjdd| d¡| ¡d| ¡| |jjj d¡|j d¡dS)zMReopening a file with tables using enumerated multi-dimensional data.rSrT)r!)rUrPz5Enumerated type was not restored correctly from disk.N) rVrWrRrXrYrrZrTr[r )rr r rÚtest00b_reopenMDôs z"EnumTableTestCase.test00b_reopenMDcCsÔ|jjdd| ¡| ¡d}d|jfd|jfg}|j}|dd|d<|dd|d <| ¡|dd|d<| t t j |d |dd¡|  ¡d |_ | ¡}t d |d d …|f¡| |d d …|d¡d S)z3Appending enumerated values using ``row.append()``.rSrT)rUé érrMr"rPÚpythonz&* appended value: %s * read value: %s NéÿÿÿÿzWritten and read values differ.)rVrWrRrXÚ valueInEnumÚvalueOutOfEnumÚrowÚappendr&Ú ValueErrorÚoperatorÚsetitemÚflushÚflavorÚreadrÚ verbosePrintr)rÚtblÚappendedrdrkr r rÚtest01_rowAppends& z"EnumTableTestCase.test01_rowAppendcCst|jjdd| ¡| ¡d}d|jfd|jfg}| |¡| ¡d|_|  ¡}t   d||f¡|  ||d¡d S) z5Appending enumerated values using ``table.append()``.rSrT)rUr^r_r`z&* appended value: %s * read value: %s zWritten and read values differ.N) rVrWrRrXrbrcrerirjrkrrlr)rrmrnrkr r rÚ test02_appends   zEnumTableTestCase.test02_appendcCsŒ|jjdd| ¡| ¡d}d|jfd|jfg}| |¡d|jfd|jfg}||dd…<d|_| ¡}t   d||f¡|  ||d ¡dS) z9Changing enumerated values using ``table.__setitem__()``.rSrT)rUr^r_Nr`z%* written value: %s * read value: %s zWritten and read values differ.) rVrWrRrXrbrercrjrkrrlr)rrmrnÚwrittenrkr r rÚtest03_setitem3s     z EnumTableTestCase.test03_setitemcCs|jjdd| d¡| ¡d}d|j|jffd|j|jffg}|j}|dd|d<| tt j |d |dd ¡|  |¡|  ¡d |_ | ¡}x~tt|ƒƒD]n}| ||d||dd ¡| ||d d||d dd ¡| ||d d ||d d d ¡qžWd S)z,Appending multi-dimensional enumerated data.rSrT)r!)rUr^r_rrMrPr"r`zWritten and read values differ.N)rVrWrRrXrbrcrdr&rfrgrhrerirjrkrÚlenr)rrmrnrdrkr/r r rÚtest04_multidimJs&   z!EnumTableTestCase.test04_multidimc Cs¤|jjdd| ¡| ¡d}d|jfd|jfd|jfg}| |¡| ¡dd„| d d |ji¡Dƒ}t   d |j|j j ||d d …f¡|  ||d d …d¡d S)zSearching enumerated data.rSrT)rUr^r_écSsg|]}|d|df‘qS)rMrPr )r.rdr r rr0ssz2EnumTableTestCase.test05_where..z rcolor == vÚvz\* ``valueInEnum``: %s * ``rcolor`` column: ``%s`` * ``searched``: %s * Should look like: %s Nraz"Search returned incorrect results.)rVrWrRrXrbrcreriÚwhererrlZcolsrPr)rrmrnZsearchedr r rÚ test05_wherees  zEnumTableTestCase.test05_whereN)r )rCrDrErFr rr rNrrbrcrOrRr\r]rorprrrtrxr r r rrJÖs   rJc@s„eZdZdZe dddddœ¡ZejZdZ dZ dd d „Z d d „Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„ZdS) ÚEnumEArrayTestCasez,Test extendable arrays of enumerated values.r r!r"r)rrrr)iÒrKr cCstj|jd|j|dS)Nr)r r )r rHr rO)rr r r rÚ_atom‡szEnumEArrayTestCase._atomcCsP|jjdd| ¡d| ¡dd|jjj_| ¡| |jjj  ¡|j d¡dS)z>Reopening a file with extendable arrays using enumerated data.rSrT)r)r rUr`z5Enumerated type was not restored correctly from disk.N) rVÚ create_earrayrzrXrZrTrjrYrr[r )rr r rr\‹s   z!EnumEArrayTestCase.test00a_reopencCsP|jjdd| ¡d| ¡dd|jjj_| ¡| |jjj  ¡|j d¡dS)zXReopening a file with extendable arrays using enumerated multi-dimensional data.rSrT)rr!)r rUr`z5Enumerated type was not restored correctly from disk.N) rVr{rzrXrZrTrjrYrr[r )rr r rr]™s   z#EnumEArrayTestCase.test00b_reopenMDcCsxd}tj|j||jdd}|jjdd|d| ¡d| ¡| |jj j   ¡|jd¡| |jj j j j |j|d ¡dS) Nrr )r r rSrT)r)r rUz5Enumerated type was not restored correctly from disk.zJThe default value of enumerated type was not restored correctly from disk.)r rHr rOrVr{rXrYrrZrTr[Úatomr)rrr|r r rÚ!test_enum_default_persistence_red¨s z4EnumEArrayTestCase.test_enum_default_persistence_redcCsxd}tj|j||jdd}|jjdd|d| ¡d| ¡| |jj j   ¡|jd¡| |jj j j j |j|d ¡dS) Nrr )r r rSrT)r)r rUz5Enumerated type was not restored correctly from disk.zJThe default value of enumerated type was not restored correctly from disk.)r rHr rOrVr{rXrYrrZrTr[r|r)rrr|r r rÚ#test_enum_default_persistence_greenºs z6EnumEArrayTestCase.test_enum_default_persistence_greencCsxd}tj|j||jdd}|jjdd|d| ¡d| ¡| |jj j   ¡|jd¡| |jj j j j |j|d ¡dS) Nrr )r r rSrT)r)r rUz5Enumerated type was not restored correctly from disk.zJThe default value of enumerated type was not restored correctly from disk.)r rHr rOrVr{rXrYrrZrTr[r|r)rrr|r r rÚ"test_enum_default_persistence_blueÌs z5EnumEArrayTestCase.test_enum_default_persistence_bluecCsxd}tj|j||jdd}|jjdd|d| ¡d| ¡| |jj j   ¡|jd¡| |jj j j j |j|d ¡dS) Nr)r )r r rSrT)r)r rUz5Enumerated type was not restored correctly from disk.zJThe default value of enumerated type was not restored correctly from disk.)r rHr rOrVr{rXrYrrZrTr[r|r)rrr|r r rÚ#test_enum_default_persistence_blackÞs z6EnumEArrayTestCase.test_enum_default_persistence_blackcCs\|jjdd| ¡d| ¡d}d|_|j|jg}| |¡| ¡|  ¡}|  ||d¡dS)z/Appending scalar elements of enumerated values.rSrT)r)r rUr`zWritten and read values differ.N) rVr{rzrXrjrbrcrerirkr)rÚearrrnrkr r rÚ test01_appendðs    z EnumEArrayTestCase.test01_appendcCsh|jjdd| ¡d| ¡d}d|_|j|jg|j|jgg}| |¡| ¡|  ¡}|  ||d¡dS)z:Appending multi-dimensional elements of enumerated values.rSrT)rr!)r rUr`zWritten and read values differ.N) rVr{rzrXrjrbrcrerirkr)rrrnrkr r rÚtest02_appendMDÿs    z"EnumEArrayTestCase.test02_appendMDcCsl|jjdd| ¡d| ¡d}d|_|j|jf}| |¡|j|jg}||dd…<| ¡}|  ||d¡dS)z:Changing enumerated values using ``earray.__setitem__()``.rSrT)r)r rUr`NzWritten and read values differ.) rVr{rzrXrjrbrercrkr)rrrnrqrkr r rrrs      z!EnumEArrayTestCase.test03_setitemN)r )rCrDrErFr rr rrbrcrOrzr\r]r}r~rr€r‚rƒrrr r r rrys ryc@sdeZdZdZe dddddœ¡ZejZdZ dZ dd d „Z d d „Z dd„Z dd„Zdd„Zdd„ZdS)ÚEnumVLArrayTestCasez1Test variable-length arrays of enumerated values.r r!r"r)rrrr)iÒrKr cCstj|jd|j|dS)Nr)r r )r rHr rO)rr r r rrz)szEnumVLArrayTestCase._atomcCsN|jjdd| ¡| ¡dd|jjj_| ¡| |jjj  ¡|j d¡dS)zKReopening a file with variable-length arrays using enumerated data.rSrT)rUr`z5Enumerated type was not restored correctly from disk.N) rVÚcreate_vlarrayrzrXrZrTrjrYrr[r )rr r rr\-s   z"EnumVLArrayTestCase.test00a_reopencCsP|jjdd| d¡| ¡dd|jjj_| ¡| |jjj  ¡|j d¡dS)z]Reopening a file with variable-length arrays using enumerated multi-dimensional data.rSrT)r!)rUr`z5Enumerated type was not restored correctly from disk.N) rVr…rzrXrZrTrjrYrr[r )rr r rr]<s   z$EnumVLArrayTestCase.test00b_reopenMDcCs†|jjdd| ¡| ¡d}d|_|jg|j|jgg}| |d¡| |d¡| ¡|  ¡}t   d||f¡|  ||d¡d S) z/Appending scalar elements of enumerated values.rSrT)rUr`rr"z&* appended value: %s * read value: %s zWritten and read values differ.N) rVr…rzrXrjrbrcrerirkrrlr)rÚvlarrrnrkr r rr‚Ks   z!EnumVLArrayTestCase.test01_appendcCsš|jjdd| d¡| ¡d}d|_|j|jgg|j|jg|j|jggg}| |d¡| |d¡| ¡|  ¡}t   d||f¡|  ||d ¡d S) z:Appending multi-dimensional elements of enumerated values.rSrT)r!)rUr`rr"z&* appended value: %s * read value: %s zWritten and read values differ.N) rVr…rzrXrjrbrcrerirkrrlr)rr†rnrkr r rrƒas     z#EnumVLArrayTestCase.test02_appendMDcCs||jjdd| ¡| ¡d}d|_|j|jf}| |¡|j|jg}||d<| ¡}t   d||f¡|  ||dd¡dS) z;Changing enumerated values using ``vlarray.__setitem__()``.rSrT)rUr`rz%* written value: %s * read value: %s zWritten and read values differ.N) rVr…rzrXrjrbrercrkrrlr)rr†rnrqrkr r rrrxs      z"EnumVLArrayTestCase.test03_setitemN)r )rCrDrErFr rr rrbrcrOrzr\r]r‚rƒrrr r r rr„!s r„cCs–ddl}ddlm}t ¡}d}xpt|ƒD]d}| | |¡¡| t t ¡¡| t t ¡¡| t t ¡¡| t t ¡¡| t t ¡¡q*W|S)zCReturn a test suite consisting of all the test cases in the module.rN)r r")ÚdoctestZ tables.miscr rZ TestSuiterZaddTestZ DocTestSuiteZ makeSuiterrGrJryr„)r‡r ZtheSuiteZniterr/r r rÚsuites rˆÚ__main__)Z defaultTest)rFZ __future__rrgr7r Z tables.testsrZtables.tests.commonrrZTestCaseZ six.movesrrrGZ TempFileMixinrJryr„rˆrCÚsysZ parse_argvÚargvZprint_versionsÚmainr r r rÚ s*     kQ*#l