j\c@@sdZddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZyddlZWn2ek rejd'krqddlZnXddlZddlZddlZdd lmZdd lmZe ejZe ejd d Zeaea eZ!d Z"ejddk Z$ejddk Z%ejddk Z&ejd dk Z'dZ(dZ)dZ*edZ+ddZ,dZ-e.ej/dsVdd(dYZ0de0fdYZ1de1fdYZ2ndej/fd YZ3d!e4fd"YZ5d#e4fd$YZ6d%e3fd&YZ7dS()s$Utilities for PyTables' test suites.i(tprint_function(tabsolute_importN(t LooseVersion(tresource_filenameii(tdetect_number_of_cores(tmin_blosc_bitshuffle_versiontbloscicC@sjd|kr"ta|jdnd|krDta|jdnd|krfta|jdn|S(Ntverbosetsilents--heavy(tTrueRtremovetFalsetheavy(targv((s2lib/python2.7/site-packages/tables/tests/common.pyt parse_argv:s   tzlibtlzotbzip2cC@s/|rtdn tdtdddS(Ns#Performing the complete test suite!sQPerforming only a light (yet comprehensive) subset of the test suite. If you want a more complete test, try passing the --heavy flag to this script (or set the 'heavy' parameter in case you are using tables.test() call). The whole suite will take more than 4 hours to complete on a relatively modern CPU and around 512 MB of main memory.s-=i&(tprint(R ((s2lib/python2.7/site-packages/tables/tests/common.pyt print_heavyRs  c C@stddtdtjtdtjddtdtjtjd}tjrtj}tj d |d }d |}nd }td tj|f|d&k rtd|ddfntjd}|d&k rtd|d|dfntjd}|d&k rJtd|d|dfntjd}|d&k r0|dj d}td|d|ftj }gt |jD] \}}d||df^q}tddj|dg}|dtkr|jdntddj|ny"d dlm}td|WnnXtdtjtd tjtd!tjtd"ttd#tjtd$tjtd%tjtddtjjd&S('s;Print all the versions of software that PyTables relies on.s-=i&sPyTables version: %ssHDF5 version: %sthdf5isNumPy version: %sRs[0-9.]+isusing VML/MKL %ssnot using Intel's VML/MKLsNumexpr version: %s (%s)sZlib version: %s (%s)sin Python interpreterRsLZO version: %s (%s)iRsBZIP2 version: %s (%s)RsBlosc version: %s (%s)s%s (%s)sBlosc compressors: %ss, tshufflet bitshufflesBlosc filters: %s(t __version__sCython version: %ssPython version: %ssPlatform: %ssByte-ordering: %ssDetected cores: %ssDefault encoding: %ssDefault FS encoding: %ssDefault locale: (%s, %s)N(RttablesRtwhich_lib_versiontnumpytnumexprtuse_vmltget_vml_versiontretfindalltNonetsplittblosc_get_complib_infotsortedtitemstjoinRtappendtCythontsystversiontplatformt byteorderRtgetdefaultencodingtgetfilesystemencodingtlocaletgetdefaultlocaletstdouttflush( ttinfot vml_versiont vml_availt blosc_datet blosc_cinfotktvt blosc_finfotcython_version((s2lib/python2.7/site-packages/tables/tests/common.pytprint_versions_sZ         9 cC@s td|S(Ns tables.tests(R(tfilename((s2lib/python2.7/site-packages/tables/tests/common.pyt test_filenamescC@s1ts dS|r#t|ddn t|dS(s4Print out the `string` if verbose output is enabled.Ntendt (RR(tstringtnonl((s2lib/python2.7/site-packages/tables/tests/common.pyt verbosePrints RcC@st|ds||kSt|d s8|jd krat|d sW|jd kra||kS|j|jkrtrtd|jd|jndSt|dr|jjd|jjdkrtrtd|jd|jndSt|jdkrD|d|dkrdStr=td|jd|jndSn|jdkr|jdkrfdStrtdtd t|jtd t|jndSn||k}t j |}| rtrtd n|S(s*Checks if two numerical objects are equal.tshapesShape is not equal:s!=itdtypeisdtype is not equal:slength is not equalslen(a.data) ==>slen(b.data) ==>s,Some of the elements in arrays are not equal((((( thasattrRCRRRDtstrtlentsizetdataRtall(tatbtflavortresult((s2lib/python2.7/site-packages/tables/tests/common.pytallequals@  /    cC@snt|}t|}t|dr9|j|jkpTt||pTt||s[tStj||kS(sAre both `arr1` and `arr2` equal arrays? Arguments can be regular NumPy arrays, chararray arrays or structured arrays (including structured record arrays). They are checked for type and value equality. RD(ttypeRERDt issubclassR RRJ(tarr1tarr2tt1tt2((s2lib/python2.7/site-packages/tables/tests/common.pytareArraysEquals  !t assertWarnst_BaseTestCaseContextcB@seZdZdZRS(cC@s ||_dS(N(t test_case(tselfRY((s2lib/python2.7/site-packages/tables/tests/common.pyt__init__scC@s.|jj|j|}|jj|dS(N(RYt_formatMessagetmsgtfailureException(RZt standardMsgR]((s2lib/python2.7/site-packages/tables/tests/common.pyt _raiseFailures(t__name__t __module__R[R`(((s2lib/python2.7/site-packages/tables/tests/common.pyRXs t_AssertRaisesBaseContextcB@s eZdddZdZRS(cC@stj||||_||_|dk rdy|j|_Wqmtk r`t||_qmXn d|_|dk rt j |}n||_ d|_ dS(N( RXR[texpectedRYR Ratobj_nametAttributeErrorRFRtcompiletexpected_regexR](RZRdRYt callable_objRh((s2lib/python2.7/site-packages/tables/tests/common.pyR[s       cC@sC|dkr%|jdd|_|S||||WdQXdS(s If callable_obj is None, assertRaises/Warns is being used as a context manager, so check for a 'msg' kwarg and return self. If callable_obj is not None, call it passing args and kwargs. R]N(R tpopR](RZtnameRitargstkwargs((s2lib/python2.7/site-packages/tables/tests/common.pythandle s  N(RaRbR R[Rn(((s2lib/python2.7/site-packages/tables/tests/common.pyRcs t_AssertWarnsContextcB@seZdZdZRS(cC@s|x;ttjjD]$}t|ddri|_qqWtjdt |_ |j j |_tj d|j |S(Nt__warningregistry__trecordtalways(tlistR(tmodulestvaluestgetattrR Rptwarningstcatch_warningsR twarnings_managert __enter__t simplefilterRd(RZR8((s2lib/python2.7/site-packages/tables/tests/common.pyRzscC@so|jj||||dk r&dSy|jj}Wn tk rXt|j}nXd}x|jD]}|j}t ||jsqin|dkr|}n|j dk r|j j t| rqin||_ |j |_ |j|_dSW|dk r-|jdj|j jt|n|jrU|jdj||jn|jdj|dS(Ns"{0}" does not match "{1}"s{0} not triggered by {1}s{0} not triggered(Ryt__exit__R RdRaRfRFRwtmessaget isinstanceRhtsearchtwarningR<tlinenoR`tformattpatternRe(RZtexc_typet exc_valuettbtexc_nametfirst_matchingtmtw((s2lib/python2.7/site-packages/tables/tests/common.pyR|#s:           (RaRbRzR|(((s2lib/python2.7/site-packages/tables/tests/common.pyRos tPyTablesTestCasecB@seeZdZdZdZdZeejdsKddZ ne dZ e dZ RS( cC@sTtt|jx:|jD]/}|j|jjdkrd|j|Get the name of the method currently running in the test case.Ri(RR!(RZ((s2lib/python2.7/site-packages/tables/tests/common.pyt_getMethodNameRscC@sStrO|j}|j}d||f}td|dt|fndS(s;Print a nice header for the current test method if verbose.s Running %s.%ss%s %st-N(RRRRRG(RZRkt methodNamettitle((s2lib/python2.7/site-packages/tables/tests/common.pyt_verboseHeaderVs   RWcO@s(t|||}|jd|||S(NRW(RoRn(RZtexpected_warningRiRlRmtcontext((s2lib/python2.7/site-packages/tables/tests/common.pyRWbscC@str#td|td|n|rH|j|j|jkdn|j|j|jkd|j|j|jkddS(NsGroup 1:sGroup 2:s(node1 and node2 have the same pathnames.s1node1 and node2 does not have the same pathnames.s0node1 and node2 does not have the same children.(RRt assertTruet _v_pathnamet _v_children(RZtnode1tnode2thardlink((s2lib/python2.7/site-packages/tables/tests/common.pyt_checkEqualityGroupgs  cC@str#td|td|n|rH|j|j|jkdn|j|j|jkd|jt||ddS(NsLeaf 1:sLeaf 2:s(node1 and node2 have the same pathnames.s1node1 and node2 does not have the same pathnames.s.node1 and node2 does not have the same values.(RRRRRV(RZRRR((s2lib/python2.7/site-packages/tables/tests/common.pyt_checkEqualityLeafws  N(RaRbRRRRREtunittesttTestCaseR RWR RR(((s2lib/python2.7/site-packages/tables/tests/common.pyRGs     t TestFileMixincB@s&eZdZiZdZdZRS(cC@s>tt|jtj|jd|j|j|_dS(NR( RRtsetUpRt open_fileth5fnameRt open_kwargsth5file(RZ((s2lib/python2.7/site-packages/tables/tests/common.pyRscC@s$|jjtt|jdS(sClose ``h5file``.N(RtcloseRRR(RZ((s2lib/python2.7/site-packages/tables/tests/common.pyRs N(RaRbR RRRR(((s2lib/python2.7/site-packages/tables/tests/common.pyRs t TempFileMixincB@s;eZdZiZdZdZdZddZRS(RcC@stjd|jddS(Ntprefixtsuffixs.h5(ttempfiletmktempR(RZ((s2lib/python2.7/site-packages/tables/tests/common.pyt_getTempFileNamescC@sStt|j|j|_tj|j|jd|j|j |_ dS(sSet ``h5file`` and ``h5fname`` instance attributes. * ``h5fname``: the name of the temporary HDF5 file. * ``h5file``: the writable, empty, temporary HDF5 file. RN( RRRRRRRt open_modeRRR(RZ((s2lib/python2.7/site-packages/tables/tests/common.pyRs cC@s=|jjd|_tj|jtt|jdS(s(Close ``h5file`` and remove ``h5fname``.N( RRR tosR RRRR(RZ((s2lib/python2.7/site-packages/tables/tests/common.pyRs  trcK@s,|jjtj|j|||_tS(sReopen ``h5file`` in the specified ``mode``. Returns a true or false value depending on whether the file was reopenend or not. If not, nothing is changed. (RRRRRR (RZtmodeRm((s2lib/python2.7/site-packages/tables/tests/common.pyt_reopens (RaRbRRRRRR(((s2lib/python2.7/site-packages/tables/tests/common.pyRs    t ShowMemTimecB@seZejZdZRS(cC@suxtdD]}|jdr;t|jd}q |jdrct|jd}q |jdrt|jd}q |jdrt|jd}q |jdrt|jd}q |jdr t|jd}q q Wtd tj|jtd |jtd ||ftd ||ftd ||fdS(s$Showing memory and time consumption.s/proc/self/statussVmSize:isVmRSS:sVmData:sVmStk:sVmExe:sVmLib:s WallClock time:s Memory usage: ******* %s *******sVmSize: %7s kB VmRSS: %7s kBsVmData: %7s kB VmStk: %7s kBsVmExe: %7s kB VmLib: %7s kBN(topent startswithtintR!RttimettrefR(RZtlinetvmsizetvmrsstvmdatatvmstktvmexetvmlib((s2lib/python2.7/site-packages/tables/tests/common.pyttest00s$(RaRbRRR(((s2lib/python2.7/site-packages/tables/tests/common.pyRs (ii((8t__doc__t __future__RRRRR(RR.R*RRwtdistutils.versionRt pkg_resourcesRt unittest2Rt ImportErrort version_infoRRRt tables.utilsRttables.req_versionsRt hdf5_versionRt blosc_versionR RR t show_memoryRR t zlib_availt lzo_availt bzip2_availt blosc_availRR;R=RBRORVRERRXRcRoRtobjectRRR(((s2lib/python2.7/site-packages/tables/tests/common.pyt s^              ;  5 .A*