j\c@@sddlmZddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z mZmZmZmZddlmZddlmZddlmZdd lmZd ejfd YZd ejefd YZdejefdYZdejefdYZdejefdYZ dejefdYZ!dZ"e#dkrej$ej%ej&ej'ddndS(i(tprint_function(tabsolute_importN(tGrouptLeaftTabletArray(t StringColtIntColtInt16ColtFloatColt Float32Col(tcommon(tunittest(tPyTablesTestCase(trangetRecordcB@s;eZeddZeZeZeZ e Z RS(titemsizei( t__name__t __module__Rtvar1Rtvar2Rtvar3R tvar4R tvar5(((s5lib/python2.7/site-packages/tables/tests/test_tree.pyRs    t TreeTestCasecB@seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z ejejd d kddZRS(twsThis is the table titlei icC@s.tt|j|j|jjdS(N(tsuperRtsetUpt populateFileth5filetclose(tself((s5lib/python2.7/site-packages/tables/tests/test_tree.pyR$s c C@s|jj}d}xotdD]a}|jj|dt|td|jddd|j}|j }xnt|jD]]}d|j||d <||d <|||d R?RR@RARBRRCt assertTruet isinstanceRR'tgroup0RRR(RR tarrR&((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest01_getNodeClasss  c C@sUtjr1tdddtd|jjntj|jd|_|j t |jj ddddd d d g}g}g}x\|D]T}y|jj |}WnqX|j |x|D]}|j |jqWqW|j|dd d d dd ddgtjr#tdng}xO|D]G}y|jj |}Wnq0Xx|D]}|j |jq]Wq0W|j|dd ddd dddgtjrtdng}x|D]}y|jj |d}WnFt k r-tjrOtj\}}} tdt|qOqXx|D]}|j |jq5WqW|j|d dddddgtjrtdng}x|D]}y|jj |d}WnFt k rtjr!tj\}}} tdt|q!qXx|D]}|j |jqWqW|j|d dgtjrQtdndS(s%Checking the File.list_nodes() methods s-=isRunning %s.test02_listNodes...R:R;t NoSuchClasss/group0s/group0/table1s/group0/group1/group2s/var1s/table0s/var4s/group0/group1s /group0/var1s /group0/var4s list_nodes(pathname) test passeds/group0/group1/table2s/group0/group1/var1s/group0/group1/var4s#list_nodes(groupobject) test passedRs( Great!, the next TypeError was catched!s7list_nodes(groupobject, classname = 'Leaf') test passedRs8list_nodes(groupobject, classname = 'Table') test passedN(R R=R>R?RR@RARBRt assertRaisest TypeErrort list_nodestextendR-RDRERGRH( RRIRJtobjectsRKt objectlistRLRQRRRS((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest02_listNodess                                   c C@stjr1tdddtd|jjntj|jd|_|j t |jj ddddd d d g}g}g}xo|D]g}y)g|jj |D] }|^q}WnqX|j |x|D]}|j|jqWqW|j|dd d d dd ddgtjr6tdng}xb|D]Z}y)g|jj |D] }|^q_}WnqCXx|D]}|j|jqWqCW|j|dd ddd dddgtjrtdng}x|D]}y,g|jj |dD] }|^q }WnFt k rftjrtj\}} } tdt| qqXx|D]}|j|jqnWqW|j|d dddddgtjrtdng}x|D]}y,g|jj |dD] }|^q}WnFt k rKtjrmtj\}} } tdt| qmqXx|D]}|j|jqSWqW|j|d dgtjrtdndS(s%Checking the File.iter_nodes() methods s-=isRunning %s.test02b_iterNodes...R:R;R]s/group0s/group0/table1s/group0/group1/group2s/var1s/table0s/var4s/group0/group1s /group0/var1s /group0/var4s iter_nodes(pathname) test passeds/group0/group1/table2s/group0/group1/var1s/group0/group1/var4s#iter_nodes(groupobject) test passedRs( Great!, the next TypeError was catched!s7iter_nodes(groupobject, classname = 'Leaf') test passedRs8iter_nodes(groupobject, classname = 'Table') test passedN(R R=R>R?RR@RARBRR^R_R`t iter_nodesRaR-RDRERGRH( RRIRJRbRKtoRcRLRQRRRS((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest02b_iterNodess     )        )      ,         ,       cC@sJtjr1tdddtd|jjntj|jd|_g}g}g}x|jj D]v}|j |j x-|jj |dD]}|j |j qWx-|jj |dD]}|j |j qWqkW|j |dd d d g|j |d d dg|j |ddddddgtjrRtdng}g}g}x|jj d D]v}|j |j x-|jj |dD]}|j |j qWx-|jj |dD]}|j |j qWqwW|j |d d g|j |dg|j |ddgtjrFtdndS(s&Checking the File.walk_groups() methods s-=is!Running %s.test03_TraverseTree...R:RRR;s/group0s/group0/group1s/group0/group1/group2s/table0s/group0/table1s/group0/group1/table2s/var1s/var4s /group0/var1s /group0/var4s/group0/group1/var1s/group0/group1/var4swalk_groups() test passeds!walk_groups(pathname) test passedN(R R=R>R?RR@RARBRt walk_groupsR-RDR`RE(Rtgroupsttables_tarraysR&R R[((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest03_TraverseTree|sT             c C@stjr1tdddtd|jjntj|jd|_|j t t |jj ddg}g}g}g}xZ|jj dd D]C}|j |jx*|jdd D]}|j |jqWqWx-|jjjd D]}|j |jqWx-|jj dd D]}|j |jq&W|j|dd d dg|j|dddg|j|dddg|j|ddddddgtjrtdng}g}g}x|jj d dd D]p}|j |jx'|jd D]}|j |jqWx-|jj |d D]}|j |jqGWqW|j|d dg|j|dg|j|ddgtjrtdndS(sChecking File.walk_nodess s-=isRunning %s.test04_walkNodes...R:R;R]R<RRRs/group0s/group0/group1s/group0/group1/group2s/table0s/group0/table1s/group0/group1/table2s/var1s/var4s /group0/var1s /group0/var4s/group0/group1/var1s/group0/group1/var4s.File.__iter__() and Group.__iter__ test passeds+walk_nodes(pathname, classname) test passedN(R R=R>R?RR@RARBRR^R_tnextt walk_nodesR-RDt _f_iter_nodesR't _f_walknodesRE( RRittables1ttables2RkR&R R[Rj((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest04_walkNodessd                   iis2Special method `__dir__()` introduced in Python-3.cC@sHtjr1tdddtd|jjntj|jd|_t |jj }|j d||j d||j d||j d||j d ||j d ||j d ||j d ||j d ||j d||j tt|t|t |jj j}|j d||j d ||j d ||j d||j tt|t|t |jj jj}|j d||j d||j d ||j d ||j d ||j d||j tt|t|tjrDtdndS(sChecking Group.__dir__s s-=isRunning %s.test05_dir...R:t _v_childrent_v_attrst_g_get_child_group_classt_f_closeRZttable0RRVttable2RWR9sGroup.__dir__ test passedN(R R=R>R?RR@RARBRtdirR'tassertInt assertNotIntassertSequenceEqualtsortedtsetRZRW(Rtroot_dirtroot_group0_dirtroot_group0_group1_dir((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt test05_dirsF     (RRt open_modeR!R#t appendrowsRRRUR\RdRgRlRsR tskipIfRGt version_infoR(((s5lib/python2.7/site-packages/tables/tests/test_tree.pyRs  # Q  b ` 7 CtDeepTreeTestCasecB@sbeZdZdZdZdZdZdZdZe j e j ddZ RS( s3Checks for deep hierarchy levels in PyTables trees.cC@sBtt|jtjr(d|_n d|_tjrMtd|jn|jj }tjrutdddnxt |jD]}tjrtd|ddn|jj |dd d gd ||jj |d d d gd ||jj |d t ||jj |d t |}qW|jjdS(Nii@sMaximum depth tested :sDepth writing progress: tendt s%3d,tarrayis depth: %dtarray2tgroup2_R&(RRRR theavytmaxdepthR=R>RR'RR0R1R)R(RR&tdepth((s5lib/python2.7/site-packages/tables/tests/test_tree.pyR3s&         #c C@stj|dd}|j}tjr=tdddnxt|jD]}tjrstd|ddn|j|j ddg|j d|k|j d t ||k|j |d t |}qMWtjrtnWdQXdS( NtmodeR:s Depth reading progress: RRs%3d,iRRR&( R@RAR'R R=R>RRRERRXR)RC(RtfilenameRR&R((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt _check_treeXs     cC@s|j|jdS(s&Creation of a large depth object tree.N(RRB(R((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest00_deepTreepsc C@stj|jdd|_tjd}zttj|ddL}tjr[tdn|jj |jj |j dt |jj WdQX|j |Wdtjj|rtj|nXdS(s"Copy of a large depth object tree.RR:s.h5Rs Copying deep tree...t recursiveN(R@RARBRttempfiletmktempR R=R>t copy_nodeR'tTrueRRtostpathtexiststremove(Rth5fname2th5file2((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest01a_copyDeepTreeus  c C@stj|jdddd|_tjd}zztj|ddddL}tjrgtdn|jj |jj |j dt |jj Wd QX|j |Wd tjj|rtj|nXd S( s8Copy of a large depth object tree with small node cache.RR:tnode_cache_slotsi s.h5Rs Copying deep tree...RN(R@RARBRRRR R=R>RR'RRRRRRR(RRR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest01b_copyDeepTrees    c C@stj|jdddd|_tjd}zztj|ddddL}tjrgtdn|jj |jj |j dt |jj Wd QX|j |Wd tjj|rtj|nXd S( s5Copy of a large depth object tree with no node cache.RR:Ris.h5Rs Copying deep tree...RN(R@RARBRRRR R=R>RR'RRRRRRR(RRR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest01c_copyDeepTrees    sonly in heavy modec C@stj|jdddd|_tjd}zztj|ddddL}tjrgtdn|jj |jj |j dt |jj Wd QX|j |Wd tjj|rtj|nXd S( s9Copy of a large depth object tree with static node cache.RR:Ris.h5Rs Copying deep tree...RN(R@RARBRRRR R=R>RR'RRRRRRR(RRR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest01d_copyDeepTrees    (RRt__doc__RRRRRRR t skipUnlessR RR(((s5lib/python2.7/site-packages/tables/tests/test_tree.pyR0s %     tWideTreeTestCasecB@s eZdZdZdZRS(s2Checks for maximum number of children for a Group.cC@stjrd}nd}tjrVtdddtd|jjtd|nddg}tjr~td d d nxat|D]S}tjrtd |d d n|jj|jj d t ||d|qWtjrtnt j }ddg}|j tjrQtd|t j |ftdd d nxot|D]a}tjrtd |d d nt |jj d t |}|j}|j||q^WtjrtndS(sChecking creation of large number of leafs (1024) per group. Variable 'maxchildren' controls this check. PyTables support up to 4096 children per group, but this would take too much memory (up to 64 MB) for testing purposes (may be we can add a test for big platforms). A 1024 children run takes up to 30 MB. A 512 children test takes around 25 MB. iis s-=isRunning %s.test00_wideTree...s#Maximum number of children tested :isChildren writing progress: RRs%3d,Rs child: %ds/ Time spent opening a file with %d arrays: %s ss Children reading progress: N(R RR=R>R?RRRR0R'R)ttimet_reopentgetattrtreadRE(Rt maxchildrentatchildtt1tarray_tb((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt test00_LeafssB               cC@stjrd}nd}tjrVtdddtd|jjtd|ntjrrtdd d nx^t|D]P}tjrtd |d d n|jj|jj d t |d |qWtjrtnt j }|j tjr6td|t j |ftdd d nxjt|D]\}tjritd |d d nt |jj d t |}|j|jd |qCWtjrtndS(sChecking creation of large number of groups (1024) per group. Variable 'maxchildren' controls this check. PyTables support up to 4096 children per group, but this would take too much memory (up to 64 MB) for testing purposes (may be we can add a test for big platforms). A 1024 children run takes up to 30 MB. A 512 children test takes around 25 MB. iis s-=isRunning %s.test00_wideTree...s#Maximum number of children tested :sChildren writing progress: RRs%3d,R&s child: %ds/ Time spent opening a file with %d groups: %s ss Children reading progress: N(R RR=R>R?RRRR1R'R)RRRREt_v_title(RRRRR&((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest01_wideTrees<            (RRRRR(((s5lib/python2.7/site-packages/tables/tests/test_tree.pyRs :tHiddenTreeTestCasecB@seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZRS(s0Check for hidden groups, leaves and hierarchies.cC@sYtt|jg|_g|_|j}|jdddg|jdd}|j|ddg|j|ddg|j|d}|j|ddg|jjddd d d d g|jdd dg|jdd}|j|ddg|j|d dg|j|d dg|jjdddddgdS(NR;Ritgta1ta2s/as/gs/g/a1s/g/a2s/g/gs/g/g/at_p_at_p_gs/_p_as/_p_gs/_p_g/as /_p_g/_p_as/g/_p_a( RRRtvisiblethiddenRR0R1Ra(Rth5fRtg_gthg((s5lib/python2.7/site-packages/tables/tests/test_tree.pyRAs$   " cC@s|jj}tjddtx+|jD] }|j||kd|q)Wx+|jD] }|j||kd|qWWtjddtdS(s*Absence of hidden nodes in `File.objects`.tignoretcategorys2Missing visible node ``%s`` from ``File.objects``.s-Found hidden node ``%s`` in ``File.objects``.tdefaultN(RRbtwarningstfilterwarningstDeprecationWarningRRXR(RRbtvpaththpath((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt_test00_objects^s   c@s`fd}tjddt|dd |dd|ddtjd dtd S( s6Object dictionaries conformance with ``walk_nodes()``.c@sj}t||}g|jd|D]}|j^q+}g|D] }|^qG}|j|jj||d|jt|t|d|dS(NR;s8nodes in ``%s`` do not match those from ``walk_nodes()``s6length of ``%s`` differs from that of ``walk_nodes()``(RRRnRDtsortREtlen(tdictNameR<tfile_RbRKt walkPathsRt dictPaths(R(s5lib/python2.7/site-packages/tables/tests/test_tree.pyt dictCheckus %  RRRbRiRtleavesRRN(RRRR*(RR((Rs5lib/python2.7/site-packages/tables/tests/test_tree.pyt_test00b_objectsrs    cC@sO|j}x|jD]}|j|qWx|jD]}|j|q4WdS(s(Node availability via `File.get_node()`.N(RRRCR(RRRR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest01_getNodes  cC@sM|j}x=|jjdD])}|j}|j||kd|qWdS(s-Hidden group absence in `File.walk_groups()`.R;s"Walked across hidden group ``%s``.N(RRRhRDRX(RRR&tpathname((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest02_walkGroupss   cC@sM|j}x=|jjdD])}|j}|j||kd|qWdS(s+Hidden node absence in `File.walk_nodes()`.R;s!Walked across hidden node ``%s``.N(RRRnRDRX(RRRKR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest03_walkNodess   cC@sM|j}x=|jjdD])}|j}|j||kd|qWdS(s9Listing visible nodes under a visible group (list_nodes).s/gsListed hidden node ``%s``.N(RRR`RDRX(RRRKR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest04_listNodesVisibles   cC@sM|j}x=|jjdD])}|j}|j||kd|qWdS(s9Listing visible nodes under a visible group (iter_nodes).s/gsListed hidden node ``%s``.N(RRReRDRX(RRRKR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest04b_listNodesVisibles   cC@s|j}d}t}xR|jjdD]>}|j}||krLt}n|j||kd|q(W|j|d|dS(s8Listing visible nodes under a hidden group (list_nodes).s/_p_g/as/_p_gsListed hidden node ``%s``.s"Hidden node ``%s`` was not listed.N(RtFalseRR`RDRRX(RRt node_to_findt found_nodeRKR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest05_listNodesHiddens     cC@s|j}d}t}xR|jjdD]>}|j}||krLt}n|j||kd|q(W|j|d|dS(s8Listing visible nodes under a hidden group (iter_nodes).s/_p_g/as/_p_gsListed hidden node ``%s``.s"Hidden node ``%s`` was not listed.N(RRRReRDRRX(RRRRRKR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest05b_iterNodesHiddens     cC@s0|jjtj|j|_|jdS(s#Reopening a file with hidden nodes.N(RRR@RARBttest00_objects(R((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt_test06_reopens cC@su|jj}|j|d|jjddd|j|d|jjddd|j|ddS(s0Moving a node between hidden and visible groups.s/_p_g/as/gRs/g/as/_p_gN(Rtis_visible_nodet assertFalset move_nodeRX(RR((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt test07_moves  cC@sI|jd|jk|jjjjdt|jd|jkdS(s.Removing a visible group with hidden children.s/g/_p_aRN(RXRR'Rt _f_removeRR(R((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt test08_removes(RRRRRRRRRRRRRRRR(((s5lib/python2.7/site-packages/tables/tests/test_tree.pyR>s       tCreateParentsTestCasecB@sDeZdZejddZdZdZdZdZ RS(sTest the ``createparents`` flag. These are mainly for the user interface. More thorough tests on the workings of the flag can be found in the ``test_do_undo.py`` module. t complevelicC@sLtt|j|jjdddg|jjddd|jdS(NR;RiR&R"(RRRRR0R1R"(R((s5lib/python2.7/site-packages/tables/tests/test_tree.pyR scC@s|j|jj}}|jt|j|jddgdt|jt|jd|jdt|jt|jd|jdt|jt|j d|dtdS(s-Using the right type of parent node argument.R[it createparentss/arrays/groupN( RR'R^R_R0R&RRRt copy_children(RRR'((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest00_parentTypesc C@s|jtj|jjdddt|jd|jk|jtj|jjdddtdt|jd|jkdS(s4Placing a node inside a nonexistent child of itself.s/groups/group/foo/barRs /group/fooRN( R^R@t NodeErrorRRRRRRB(R((s5lib/python2.7/site-packages/tables/tests/test_tree.pyt test01_insides   cC@sf|jjdddt|jd|jkx0|jjdD]}|j|j|jqBWdS(s1Propagating the filters of created parent groups.s/group/foo/bartbazRs/group/foo/bar/bazs/groupN(RR1RRXRhRER"t _v_filters(RR&((s5lib/python2.7/site-packages/tables/tests/test_tree.pyttest02_filters's( RRRR@tFiltersR"RRRR(((s5lib/python2.7/site-packages/tables/tests/test_tree.pyRs    cC@stj}d}xt|D]t}|jtjt|jtjt|jtjt|jtjt|jtjt qW|S(Ni( R t TestSuiteRtaddTestt makeSuiteRRRRR(ttheSuitetniterR5((s5lib/python2.7/site-packages/tables/tests/test_tree.pytsuite0s t__main__t defaultTestR((t __future__RRRRGRRRR@RRRRRRRR R t tables.testsR ttables.tests.commonR R tTestCaset six.movesRt IsDescriptionRt TempFileMixinRRRRRRRt parse_argvtargvtprint_versionstmain(((s5lib/python2.7/site-packages/tables/tests/test_tree.pyts4      "(u1