ꢩeVdZddlZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z ddl mZmZmZmZmZdZGd d eeZGd d eeZd'dZGddeeZGddeeZGddeeZGddeeZGddeZGddeeZGddeeZGddeZGdd eZ Gd!d"eeZ!d#Z"e#d$kr1ddl$Z$ee$j%eej&d%&dSdS)(z"Unit test for the filenode module.N)Path) open_filefileNoSuchNodeError)filenode)unittest TempFileMixin parse_argvprint_versionsPyTablesTestCasec>ddlm}|d|z S)Nr) resourcesztables.nodes.tests) importlibrfiles)namers @lib/python3.11/site-packages/tables/nodes/tests/test_filenode.py test_filers,###### ??/ 0 04 77c*eZdZdZdZdZdZdZdS)NewFileTestCasez None This method sets the following instance attributes: * 'h5fname', the name of the temporary HDF5 file * 'h5file', the writable, temporary HDF5 file with a '/test' node * 'fnode', the closed file node in '/test' rrrNsupersetUprrrr&closer% __class__s rr?zClosedFileTestCase.setUp?sK  &t{#FKKK  rcVd|_tdS)zItearDown() -> None Closes 'h5file'; removes 'h5fname'. N)r&r>tearDownrAs rrDzClosedFileTestCase.tearDownMs(  rc |jdS#t$r|dYdSwxYw)zClosing a closed file.z'Could not close an already closed file.N)r&r@ ValueErrorr$r,s r test00_ClosezClosedFileTestCase.test00_CloseYs] A J        A A A II? @ @ @ @ @ @ AsAAcP|t|jjdS)zFlushing a closed file.N)r*rFr&flushr,s r test01_FlushzClosedFileTestCase.test01_Flushas% *dj&677777rcR|tt|jdS)z'Getting the next line of a closed file.N)r*rFnextr&r,s r test02_NextzClosedFileTestCase.test02_Nextfs$ *dDJ77777rcP|t|jjdS)zReading a closed file.N)r*rFr&readr,s r test03_ReadzClosedFileTestCase.test03_Readk$ *djo66666rcP|t|jjdS)z"Reading a line from a closed file.N)r*rFr&readliner,s rtest04_Readlinez"ClosedFileTestCase.test04_Readlinep% *dj&9:::::rcP|t|jjdS)z!Reading lines from a closed file.N)r*rFr& readlinesr,s rtest05_Readlinesz#ClosedFileTestCase.test05_Readlinesus% *dj&:;;;;;rcR|t|jjddS)zSeeking a closed file.rN)r*rFr&seekr,s r test06_SeekzClosedFileTestCase.test06_Seekzs& *djoq99999rcP|t|jjdS)z.Getting the pointer position in a closed file.N)r*rFr&tellr,s r test07_TellzClosedFileTestCase.test07_TellrQrcP|t|jjdS)zTruncating a closed file.N)r*rFr&truncater,s rtest08_Truncatez"ClosedFileTestCase.test08_TruncaterUrcR|t|jjddS)zWriting a closed file.sfooNr*rFr&writer,s r test09_WritezClosedFileTestCase.test09_Write' *dj&6?????rcT|t|jjdgdS)zWriting lines to a closed file.sfoo N)r*rFr& writelinesr,s rtest10_Writelinesz$ClosedFileTestCase.test10_Writeliness) *dj&;hZHHHHHr)r5r6r7r8r?rDrGrJrMrPrTrXr[r^rareri __classcell__rBs@rr;r;<s 99     AAA888 888 777 ;;; <<< ::: 777 ;;; @@@ IIIIIIIrr;c||}t|dkr?||||}t|dk=dSdS)zcopyFileToFile(srcfile, dstfile[, blocksize]) -> None Copies a readable opened file 'srcfile' to a writable opened file 'destfile' in blocks of 'blocksize' bytes (4 KiB by default). rN)rOlenrd)srcfiledstfile blocksizedatas rcopyFileToFilerss` << " "D d))a-- d||I&& d))a------rcBeZdZdZdZfdZfdZdZdZdZ xZ S)WriteFileTestCasez6Tests writing, seeking and truncating a new file node.test_filenode.datcttj|jdd|_t |j|_dSasetUp() -> None This method sets the following instance attributes: * 'h5fname', the name of the temporary HDF5 file * 'h5file', the writable, temporary HDF5 file with a '/test' node * 'fnode', the writable file node in '/test' rrrN)r>r?rrrr&r datafnamerAs rr?zWriteFileTestCase.setUpsF  &t{#FKKK "4>22rc|jd|_tdSzUtearDown() -> None Closes 'fnode' and 'h5file'; removes 'h5fname'. Nr&r@r>rDrAs rrDzWriteFileTestCase.tearDown<   rct|jd} t||j|dS#|wxYw)zWriting a whole file node.rbN)openryrsr&r@)r%datafiles rtest00_WriteFilez"WriteFileTestCase.test00_WriteFilesX--  8TZ 0 0 0 NN     HNN    s AAc |jd|jd|jd|jd|jd}||dd|jd|jd |jd|jd }||d d |jd d |jd }||d d|jdd |j}|jdd |jd |j}|||d z d zddS)zSeeking and writing file node.s0123s4567rs34z3Gap caused by forward seek was not properly filled.rtestz)Data was overwritten instead of appended.zWritten data was not appended.z*Pointer was not correctly moved on append.N)r&rdrZrOr!assertNotEqualr])r%rr oldendoff newendoffs rtest01_SeekFilez!WriteFileTestCase.test01_SeekFiles !!!  !!! zq!!  " A C C C  !!! zq!!  'F H H H Azq!! w(HIII 1JOO%%  A !!!JOO%%   y1}q( 8 : : : : :rcz|jd|jd|t|jj|jd|j|jd|j}||dd|jd|jd|j}||dd d S) zTruncating a file node.rrrrstestz)File was not grown to the current offset.rstestz'File was not grown to an absolute size.N)r&rdrZr*IOErrorr`rOr!)r%rrs rtest02_TruncateFilez%WriteFileTestCase.test02_TruncateFiles% !!!  '4:#6777   z    +J L L L A z    /#L N N N N Nr) r5r6r7r8ryr?rDrrrrjrks@rrurus@@#I 3 3 3 3 3     :::BNNNNNNNrruc4eZdZdZfdZdZdZdZxZS)OpenFileTestCasez None This method sets the following instance attributes: * 'h5fname', the name of the temporary HDF5 file * 'h5file', the writable, temporary HDF5 file with a '/test' node rrrN)r>r?rrrr@r%r&rBs rr?zOpenFileTestCase.setUps@  !$+SvFFF rcv|jd}tj|}||j|d||jddt|jz||dd| dS)z*Opening an existing file node for reading.r+filenode.open_node() opened the wrong node.r(File was opened with an invalid mode %s.r7Pointer is not positioned at the beginning of the file.N rr r open_noder!r"modereprr]r@r%r"r&s rtest00_OpenFileReadz$OpenFileTestCase.test00_OpenFileReads{##G,,"4((  JK M M M  J 6ej9I9I I K K K  JJLL! E G G G  rcx|jd}tj|d}||j|d||jddt|jz||dd| dS)z8Opening an existing file node for reading and appending.ra+rrrrNrrs rtest01_OpenFileReadAppendz*OpenFileTestCase.test01_OpenFileReadAppends{##G,,"4..  JK M M M  J 6ej9I9I I K K K  JJLL! E G G G  rc|ttj|jdddS)z3Opening an existing file node with an invalid mode.rwN)r*rrrrr r,s rtest02_OpenFileInvalidModez+OpenFileTestCase.test02_OpenFileInvalidMode/sG  X')=)=g)F)F M M M M Mr) r5r6r7r8r?rrrrjrks@rrrsrFF         MMMMMMMrrcHeZdZdZdZfdZfdZdZdZdZ dZ xZ S) ReadFileTestCasez)Tests reading from an existing file node.ztest_filenode.xbmct|j|_t|jd|_t t j|jdd}t|j|| |j dt j |j d|_dS)aRsetUp() -> None This method sets the following instance attributes: * 'datafile', the opened data file * 'h5fname', the name of the temporary HDF5 file * 'h5file', the writable, temporary HDF5 file with a '/test' node * 'fnode', the readable file node in '/test', with data in it rrrrrrN)rryrrr>r?rrrrsr@rZrr r&rs rr?zReadFileTestCase.setUpFs#4>22T^T22   !$+SvFFFt}e,,,  1' (<( None Closes 'fnode', 'h5file' and 'datafile'; removes 'h5fname'. N)r&r@rr>rDrAs rrDzReadFileTestCase.tearDown]sW     rc4ddl}||j}||j}|||ddS)z(Reading and comparing a whole file node.rNz?Data read from file node differs from that in the file on disk.)hashlibmd5rrOdigestr&r!)r%r dfiledigest fnodedigests rtest00_CompareFilez#ReadFileTestCase.test00_CompareFilels kk$-"4"4"6"677>>@@ kk$*//"3"344;;==    M O O O O OrcR|t|jjddS)zWriting on a read-only file.zno wayN)r*rr&rdr,s r test01_WritezReadFileTestCase.test01_Writews' '4:#3X>>>>>rc ddlm}||jdS#t$rYdSt $r|dYdSwxYw)z.Using a file node with Python Imaging Library.r)Imagez7PIL was not able to create an image from the file node.N)PILrrr& ImportErrorOSErrorr$)r%rs rtest02_UseAsImageFilez&ReadFileTestCase.test02_UseAsImageFile|s K ! ! ! ! ! ! JJtz " " " " "    DD K K K III K K K K K K Ks $ AAAc`||jddS)N) assertIsNotr&filenor,s r test_filenozReadFileTestCase.test_filenos, **,,d33333r) r5r6r7r8ryr?rDrrrrrjrks@rrrAs33#IGGGGG.      O O O??? K K K4444444rrcPeZdZdZfdZfdZdZdZdZdZ dZ d Z xZ S) ReadlineTestCasezBase class for text line-reading test cases. It provides a set of tests independent of the line separator string. Sub-classes must provide the 'line_separator' attribute. c(t|j}tj|jdd}||d|dfdzz}|d}|||d|z|d| tj |j d |_ d S) aThis method sets the following instance attributes: * ``h5fname``: the name of the temporary HDF5 file. * ``h5file``: the writable, temporary HDF5 file with a ``/test`` node. * ``fnode``: the readable file node in ``/test``, with text in it. rrrzshort line%sshort line%s%sasciirlong line long line long line long line long line long line long line long line long line long line long line long line long line long line long line long line long line long line long line long line unterminatedrN) r>r?line_separatorrrrrddecodeencoder@rr r&)r%linesepr&rrrBs rr?zReadlineTestCase.setUps  %!$+SvFFF G+w0G0G/IA/MN{{7## D &0111 O$$$ ' (<(eZdZdZfdZfdZdZdZdZxZ S) AttrsTestCasez/Tests setting and getting file node attributes.cttj|jdd|_dSrx)r>r?rrrr&rAs rr?zAttrsTestCase.setUp}s5  &t{#FKKK rc|jd|_tdSr{r|rAs rrDzAttrsTestCase.tearDownr}rct|jjdd}||tjdt|jjdd}||tjvddS)z,Mangling the type attributes on a file node. NODE_TYPENz6File node does not have a valid 'NODE_TYPE' attribute.NODE_TYPE_VERSIONz>File node does not have a valid 'NODE_TYPE_VERSION' attribute.)getattrr&attrsr!rNodeTyperNodeTypeVersions)r%nodeTypenodeTypeVersions rtest00_MangleTypeAttrsz$AttrsTestCase.test00_MangleTypeAttrss4:+[$??  h' D F F F"$*"24GNN  x8 8 L N N N N Nrc|t|jjddddd|jj_|t|jjddddd|jj_|t|jjdddd|jj`|t|jjdddddS) z(Setting a user attribute on a file node.userAttrNz4Inexistent attribute has a value that is not 'None'.foobar%User attribute was not correctly set.bazquux)User attribute was not correctly changed.User attribute was not deleted.)r!rr&rrr,s rtest02_SetGetDelUserAttrz&AttrsTestCase.test02_SetGetDelUserAttrs  DJ$j$ 7 7 B D D D%- !  DJ$j$ 7 7 3 5 5 5%. !  DJ$j$ 7 7 7 9 9 9 J  %  DJ$j$ 7 7 - / / / / /rc|j|tt|jddS)z+Accessing attributes on a closed file node.rN)r&r@r*AttributeErrorrr,s rtest03_AttrsOnClosedFilez&AttrsTestCase.test03_AttrsOnClosedFiles:  .'4:wGGGGGr) r5r6r7r8r?rDrrrrjrks@rrrzs99 L L L L L     & N N NF///2HHHHHHHrrc8eZdZdZfdZfdZdZdZxZS)ClosedH5FileTestCasez6Tests accessing a file node in a closed PyTables file.cttj|jdd|_|jdS)asetUp() -> None This method sets the following instance attributes: * 'h5fname', the name of the temporary HDF5 file * 'h5file', the closed HDF5 file with a '/test' node * 'fnode', the writable file node in '/test' rrrNr=rAs rr?zClosedH5FileTestCase.setUpsK  &t{#FKKK  rcZtjdt |jn#t $rYnwxYwtjdtn #tjdtwxYwd|_t dS)zHtearDown() -> None Closes 'fnode'; removes 'h5fname'. ignore)categorydefaultN)warningsfilterwarnings UserWarningr&r@rFr>rDrAs rrDzClosedH5FileTestCase.tearDowns ;???? E J          D   #I D D D D DH #I D D D D D  s&8A$ AA$AA$$BcR|t|jjddS)z1Writing to a file node in a closed PyTables file.rrNrcr,s r test00_Writez!ClosedH5FileTestCase.test00_WriterfrcT|tt|jddS)zJAccessing the attributes of a file node in a closed PyTables file.rN)r*rFrr&r,s r test01_Attrsz!ClosedH5FileTestCase.test01_Attrs s& *gtz7CCCCCr) r5r6r7r8r?rDr rrjrks@rrrs@@     *@@@ DDDDDDDrrc>eZdZdZfdZfdZdZdZdZxZ S)OldVersionTestCasezBase class for old version compatibility test cases. It provides some basic tests for file operations and attribute handling. Sub-classes must provide the 'oldversion' attribute and the 'oldh5fname' attribute. cttjd|_t |j|_t|j}||j| t|jdd|_ tj |j j jd|_dS)aThis method sets the following instance attributes: * ``h5fname``: the name of the temporary HDF5 file. * ``h5file``: the writable, temporary HDF5 file with a ``/test`` node. * ``fnode``: the readable file node in ``/test``. .h5suffixzr+z,Test for file node old version compatibility)titlerN)r>r?tempfilemktemph5fnamer oldh5fnamer copy_filer@rrrrootrr&)r%oldh5frBs rr?zOldVersionTestCase.setUps  e444 #DO444?++&&&  L$@BBB ' (8(=tDD rc|jd|_|jd|_t|jt dS)z5Closes ``fnode`` and ``h5file``; removes ``h5fname``.N)r&r@rrrunlinkr>rDrAs rrDzOldVersionTestCase.tearDown1ss     T\!!### rc |j}||d|j}||d|j}||d|jz|j}||d|jd|j}||ddS)z!Reading an old version file node.z This is only z a test file zfor FileNode version %d rN)r&rSr! oldversionrZr%rs r test00_ReadzOldVersionTestCase.test00_Read;s z""$$ /000z""$$ ///z""$$ :T_LMMMz""$$ r""" z""$$ /00000rc|jd|jdd|j}||ddS)z!Writing an old version file node.zfoobar irN)r&rdrZrSr!r"s rrzOldVersionTestCase.test01_WritePs_ $$$ Az""$$ z*****rcd|jj_|t |jjddddd|jj_|t |jjdddd|jj`|t |jjdddddS)z1Accessing attributes in an old version file node.rrNrrrr)r&rrr!rr,s rtest02_Attributesz$OldVersionTestCase.test02_AttributesZs%- !  DJ$j$ 7 7 3 5 5 5%. !  DJ$j$ 7 7 7 9 9 9 J  %  DJ$j$ 7 7 - / / / / /r) r5r6r7r8r?rDr#rr&rjrks@rrrsEEEEE,111*+++///////rrceZdZdZdZdZdS)Version1TestCasez.Basic test for version 1 format compatibility.rztest_filenode_v1.h5N)r5r6r7r8r!rr9rrr(r(ms88J&JJJrr(cPeZdZdZfdZfdZdZdZdZdZ dZ d Z xZ S) DirectReadWriteTestCaservcdtt|j|_t j|_t jd|_t|j |_ t j |_ dS)aThis method sets the following instance attributes: * ``h5fname``: the name of the temporary HDF5 file. * ``h5file``, the writable, temporary HDF5 file with a '/test' node * ``datafname``: the name of the data file to be stored in the temporary HDF5 file. * ``data``: the contents of the file ``datafname`` * ``testfname``: the name of a temporary file to be written to. rrN) r>r?rryrr testfname testh5fnamer read_bytesrrmkdtemptestdirrAs rr?zDirectReadWriteTestCase.setUpxs|  "4>22!**#?%888((3355 ')) rctj|jtjr&t |jtj|jtjr&t |jtj|j t dSr{) osaccessr,R_OKrrr-shutilrmtreer0r>rDrAs rrDz DirectReadWriteTestCase.tearDowns 9T^RW - - *  ' ' ) ) ) 9T%rw / / , ! " " ) ) + + + dl### rct|j}t|j}tj||ddS)N/test1)rr-ryrsave_to_filenode)r%r-rys rtest01_WriteToPathlibPathz1DirectReadWriteTestCase.test01_WriteToPathlibPaths<4+,, (( !+y(CCCCCrc.tj|j|jd|t tj|j|jdtj|j|jddtj|j|jddtj|j|jd|t|j |j tj 5tjd|t tj|j|jddddn #1swxYwYtj|j|jddd|t|j |j t|jt|jdS) Nr8T overwritertest2rr)rr=)rr9r-ryr*rread_from_filenoder,r!rr.rrrcatch_warnings simplefilterrr,s rtest01_WriteToFilenamez.DirectReadWriteTestCase.test01_WriteToFilenames8!$"2DNHMMM '8#N.( 4 4 4 !$"2DNH,0 2 2 2 2 !$"2DNC'. 0 0 0 0 #D$4dnhOOO dn--88::DIFFF  $ & & J J  !( + + +   gx'B". J J J J J J J J J J J J J J J J J J #D$4dnc)0D B B B B dn--88::DIFFF T^##%%% T %%'''''s AEE #E ctj|j|jd|t tj|j|jdtj|j|jddtj|j|jd|t|j |j |t tj|j|jd|t|jtjd||jjddS)Nr8Tr<r)rr9rryr*rr@r,r!rr.rr isinstancerFilerr,s rtest02_WriteToHDF5Filez.DirectReadWriteTestCase.test02_WriteToHDF5Files(!$+t~xHHH '8# .( 4 4 4 DK;;TBBB )3/////rc tj|j|jdt |jj}tj|j|jd|dd| tj t |j|z tj d| t |j|z |jdS)Nr._r?T)rr9r-ryrrr@r0replacer!r2r3r4r.rrr%rys rtest03_AutomaticNameGuessingz4DirectReadWriteTestCase.test03_AutomaticNameGuessings!$"2DNCHHH((- #D$4dlC)2):):3)D)D F F F F 4 #5#5 #A27KK    $t|,,y8DDFF $ $ $ $ $rctj|j|jdt |jj}tj|j|jd||tj t |j|z tj d|t |j|z |j dS)Nrr?T)rr9r-ryrrr@r0r!r2r3r4r.rrrLs r1test04_AutomaticNameGuessingWithFilenameAttributezIDirectReadWriteTestCase.test04_AutomaticNameGuessingWithFilenameAttributes!$"2DNCHHH((- #D$4dlC)2 4 4 4 4 4 #5#5 #A27KK    $t|,,y8DDFF $ $ $ $ $rctj|j|jd|t tj|j|jdddS)NrTHISNODEDOESNOTEXISTr?)rr9r-ryr*rr@r0r,s r$test05_ReadFromNonexistingNodeRaisesz test suite Returns a test suite consisting of all the test cases in the module. ) r TestSuiteaddTest makeSuiterr;rurrrrrr*)theSuites rsuiterXs7!##H X'88999 X'(:;;<<< X'(9::;;; X'(899::: X'(899::: X'(<==>>> X' 66777 X'(<==>>> X'(?@@AAA Or__main__rX) defaultTest)rl)'r8r2r5rrpathlibrr rrrnodesr tests.commonr r r r r TestCaserrr;rsrurrrrrrrr(r*rXr5sysargvmainr9rrrbs(( 0000000000"""""""""""""" 888 $G$G$G$G$GmX$G$G$GNUIUIUIUIUIUIUIUIp ' ' ' '[N[N[N[N[N x[N[N[N|3M3M3M3M3M}h3M3M3MBJ4J4J4J4J4}hJ4J4J4Zg8g8g8g8g8}hg8g8g8T+DcHcHcHcHcHM8cHcHcHL/D/D/D/D/D=(/D/D/DdX/X/X/X/X/X/X/X/v''''')'''}7}7}7}7}7mX}7}7}7@0 zJJJJsxNHMg&&&&&& r