Lc@sddkZddkZddkZeiiedZdZeZ ei ddjodZe Z nyei Z y ddklZlZlZWnej o eZnXyXeiido'e oddkZn ddkZn ddkZeZZZWnej o eZnXWn1ej o%ddklZeZZZnXyeWn#ej odd klZnXd eifd YZd efd YZ de fdYZ!de!fdYZ"de"fdYZ#de!fdYZ$de fdYZ%de fdYZ&e'djoeii(doei)dne"gZ*xZe*D]RZ+ei,i de+i'ddfei-i.e+Z/ei0dd i1e/q}WndS(!iNitnextit__next__(tpybeltrdkittcdktwin(R(tSett myTestCasecBs5eZdZddZddZdddZRS(s,Additional methods not present in Jython 2.2cCs|o|i|ndS(s(Fail the test if the expression is true.N(tfailureException(tselftexprtmsg((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt assertFalse,scCs|p|i|ndS(s,Fail the test unless the expression is true.N(R(R R R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt assertTrue/sicCsEt|||djo'|i|pd|||fndS(sKFail if the two objects are unequal as determined by their difference rounded to the given number of decimal places (default 7) and comparing to zero. Note that decimal places (from zero) are usually not the same as significant digits (measured from the most signficant digit). is%r != %r within %r placesN(troundR(R tfirsttsecondtplacesR ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pytassertAlmostEqual2sN(t__name__t __module__t__doc__tNoneR R R(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyR)s  t TestToolkitcBs^eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'RS(&cCsu|iidd|iiddg|_t|iidtiitd|_ |i di d|_ dS(NtsmitCCCCtCCCNtsdfshead.sdfii( ttoolkitt readstringtmolstlisttreadfiletostpathtjointheretheadtatomstatom(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pytsetUp@s-cCs|ii|ii}}|it|iiid|it|iiid|it|iid|it|iid|it|iiddS(s/Test attributes like informats, descs and so oniN( Rt informatst outformatstassertNotEqualtlentkeystdescst forcefieldstfps(R R)R*((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestattributesFs ""cCs|idiiS(Ni(Rtcalcfpt nosuchname(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt FPaccesstestOscCsMg}|iD]}||iq~}|i|d|dB|idS(s0Test the calculation of the Tanimoto coefficientiiN(RR2t assertEqualttanimotoresult(R t_[1]txR0((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestFPTanimotoSs*cCsu|it|idid|it|it|idi}|id}|it ||i dS(s0Test the string representation and corner cases.it Nosuchnames, N( t assertRaisest ValueErrorRR2tAttributeErrorR4tstrtsplitR5R,tNfpbits(R trtt((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestFPstringreprXs c Csg}|iD]}||iiq~}|it|d|ig}|D]}|t|qU~}t|d|d@tt|d|dB}|i||idS(s(Test whether the bits are set correctly.iiN( RR2tbitsR5R,tNbitstsettfloatR6(R R7R8RDt_[2]ttanimoto((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testFPbits`s -'4cCs|idiS(Ni(RR3(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt RSaccesstestiscCs<|it|iidd|it|iidddS(s(Test that invalid formats raise an errortnoeltjkjkRs &*)(%)($)N(R;R<RRtIOError(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRSformaterrormscCsp|ii|id}|i|i|idi|ii|id}|i|i|ididS(s/Test that the toolkit can eat its own dog-food.iN(RtMoleculeR%R5t _exchangeR(R tnewmol((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestselfconversionrs   cCs|ididi}|idi|ididi}|i|||id}|i|i|ididdS(s4Test that local optimisation affects the coordinatesiigN(ggg(R%R&tcoordstlocaloptR+R(R t oldcoordst newcoordstmol((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testLocalOpt{s  cCs5|id}|i|i|ididdS(s1Test that 3D coordinate generation does somethingiigN(ggg(Rtmake3DR+R&RT(R RX((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testMake3Ds  c Cs|ididtdd|ii|ididt|ididtdtg}|ididd!D]}||iqw~}|i|d d g|ididtdtdd |iid S( sCreate a 2D depictionitshowtfilenames%s.pngtupdateigt usecoordss%s_b.pngN(ggg(ggg( RtdrawtFalseRRtTrueR&RTR+(R R7R8RT((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestDraws5cCsT|i|ididd|it|idid|it|idS(s!Get the values of the properties.ig(\M@iiN( RRtmolwtR5R,R&R;R=RK(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRSgetpropss cCsy|idid}d}|id|id}}|it|t||i|d|didS(sConvert to moliRXsE OpenBabel04220815032D 4 3 0 0 0 0 0 0 0 0999 V2000 0.0000 0.0000 0.0000 C 0 0 0 0 0 0.0000 0.0000 0.0000 C 0 0 0 0 0 0.0000 0.0000 0.0000 C 0 0 0 0 0 0.0000 0.0000 0.0000 C 0 0 0 0 0 1 2 1 0 0 0 2 3 1 0 0 0 3 4 1 0 0 0 M END s iN(RtwriteR?R5R,trstrip(R tas_molttesttdatatresult((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRSconversiontoMOLs  cCs'|it|ididdS(s,Test the string representation of a moleculeiRN(R5R>Rtstrip(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRSstringreprscCs|it|iddS(s4Is the right number of molecules read from the file?iN(R5R,R(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testRFreadscCs%t|iiddt}dS(NRsnosuchfile.sdf(tgetattrRR t nextmethod(R RX((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt RFreaderrorscCs|it|idS(s;Test that reading from a non-existent file raises an error.N(R;RNRr(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRFmissingfilescCs%t|iiddt}dS(NRLshead.sdf(RpRR Rq(R RX((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt RFformaterrorscCs|it|idS(s(Test that invalid formats raise an errorN(R;R<Rt(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRFformaterrorscCs|idi}dS(Ni(Rtunitcell(R Rv((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pytRFunitcellerrorscCs|it|idS(s0Test that accessing the unitcell raises an errorN(R;R=Rw(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRFunitcellerrorscCsg}|iD]#}||ididdq~}g}x:|D]2}t|}|i|idi|qGWddg}|i||dS(sConvert to smilesRs itRRN(RRfR?RtsorttappendR#R5(R R7RXtas_smitansRRBRi((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRFconversions:   cCs|id}|iddd}tdd}|ididdi}|i|i|||it |iddt i d|it |idddS( s Test the molecule.write() methodiRstestoutput.txtRRAs RLN( RRftopent readlinesR?RmtcloseR5R;RNR!tremoveR<(R RXRitinputt filecontents((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRFsingletofiles #  cCs |it|iidd|iidd}x|iD]}|i|q;W|i|it|i||it|iiddtdd}t g}|i D]$}|i djo ||qq~}|it i d|i|ddS(sTest the Outputfile classRLstestoutput.txtRRAs$$$$iN(R;R<Rt OutputfileR%RfRRNRR,RRgR!RR5(R t outputfileRXRR7R8t numdollar((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRFoutputfiles  *  cCs|ididdS(Nit BadDescName(Rtcalcdesc(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt RFdesctestscCs|iidjo|idin|idi}|it|dj|i||idd|i t |i dS(sTest the descriptorss cinfony.cdkiigQ:@iN( RRRtaddhRR R,RttpsanameR;R<R(R tdesc((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testRFdescs cCs|ididS(NiRL(R%Rj(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt MDaccesstest scCs|idi}|it|id|d<|i|ddhdd6dd6}|i||i|dd|id|ijd S( sChange the value of a fieldit testvalueRLttheretheyityot1N( R%RjR;tKeyErrorRR5R^R tvalues(R Rjt newvalues((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testMDaccess s  cCs|idi}|i|id|it|t|ix'|D]}|i||ijtqLWt|}|i |ddjo|ddj|dS(sCheck out the keysitNoelt{is'}N( R%RjR thas_keyR5R,tdatakeysRbtreprR (R RjtkeyRA((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestMDglobalaccesss cCsg|idi}|i|id|d=|i|id|i|it|ddS(sDelete some keysitNSCN(R%RjR RR tclearR5R,(R Rj((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testMDdelete"s  cCsEg}|idD] }||q~}|it||idS(s*Test the ability to iterate over the atomsiN(R%R5R,tNatoms(R R7R'R&((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestAiteration+s(cCs |iiS(N(R'R3(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pytAtomaccesstest0scCs4|it|i|i|iiddddS(s!Get the values of some propertiesigTR'iN(R;R=RRR'RT(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestAattributes4scCs#d}|it|i|dS(s*Test the string representation of the AtomsAtom: 8 (-0.07 5.24 0.03)N(R5R>R'(R Ri((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestAstringrepr9scCs|iidS(Ns[#NOEL][#NOEL](RtSmarts(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pytinvalidSMARTStest>scCs|iidd}|iid}|i|}|it|d|iiii|iii |i t |i |iiii|iii dS(s+Searching for ethyl groups in triethylamineRs CCN(CC)CCs[#6][#6]iN(RRRtfindallR5R,tobt obErrorLogtSetOutputLeveltobErrorR;RNRt obWarning(R RXtsmartsR}((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testSMARTSBscCs|it|idid|idi|it|idid|idi|it|ididdS(sAdding and removing hydrogensiiiN(R5R,RR&Rtremoveh(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestAddhLs   ((RRR(R1R4R9RCRJRKRORSRYR[RcReRlRnRoRrRsRtRuRwRxR~RRRRRRRRRRRRRRR(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyR>sL                         t TestPybelcBs~eZeZddZdZdZdZdZdddgZ d Z d Z d Z d Z d ZdZdZRS(ig@itTPSAii RtCommentsOpenBabel Symmetry ClassescCsMg}|iD]}||idq~}|i|d|dBddS(sChecking the results from FP3tFP3iigN(RR2R5(R R7R8R0((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testFP_FP3]s-cCsVt|iidtiitdt}|i}|i |i dddS(sTesting unit cell accesstcifs hashizume.cifg9W@iN( RpRR R!R"R#R$RqRvRtGetAlpha(R RXtcell((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testunitcellbs0 cCs\|idi}|id|jt|i|ddd|d<|i|dddS(s!Mess about with the comment fieldiRsCORINA 2.61 0041 25.10.2001s New commentN(R%RjR5Rb(R Rj((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testMDcommenths  cCs!|ididtdtdS(NiR\R_(RR`Rb(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt importtestpscCs0|idid}d}|i||dS(sConvert to mol2itmol2s@MOLECULE ***** 4 3 0 0 0 SMALL GASTEIGER @ATOM 1 C 0.0000 0.0000 0.0000 C.3 1 LIG1 0.0000 2 C 0.0000 0.0000 0.0000 C.3 1 LIG1 0.0000 3 C 0.0000 0.0000 0.0000 C.3 1 LIG1 0.0000 4 C 0.0000 0.0000 0.0000 C.3 1 LIG1 0.0000 @BOND 1 1 2 1 2 2 3 1 3 3 4 1 N(RRfR5(R tas_mol2Ri((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestRSconversiontoMOL2sscCsq|i|ididd|i|ididd|it|idid|it|i dS(s!Get the values of the properties.igDl M@igV-M@iN( RRt exactmassRdR5R,R&R;R=RK(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRes cCs?tt|iii|idi}|i|ddS(sCheck out the OB iteratorsiiN(R,RRRt OBMolAtomIterRtOBMolR5(R tnumatoms((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyt testIteratorss+(RRRRR6RRRER@RRRRRRReR(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRTs       t TestOBPybelcBseZeZRS((RRtobpybelR(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRstTestOBPybelNoDrawcBseZdZRS(cCsdS(sNo drawing doneN((R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRcs(RRRc(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRstTestPybelWithDrawcBseZdZRS(cCs|ii}d|i_|ididtdtdd|ii|it |i ||i_|ii }d|i_ |it |i dS(sTesting the draw dependenciesiR\R_R]s%s_b.pngN( RttkRRR`RaRbRR;t ImportErrorRtoasa(R RB((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyttestDrawdependenciess        (RRR(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRst TestRDKitcBs9eZeZddZdZdZdZdZdgZ RS(ig@i Ri i@R( RRRRR6RRRER@R(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRs tTestCDKcBs_eZeZdZdZdZdZdZdddgZ dZ dZ d Z d Z RS( g?ittpsaiRtRemarktTitlecCsdS(sNo SMARTS testing doneN((R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRscCsdS(sNo local opt testing doneN((R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRYscCsdS(s No 3D coordinate generation doneN((R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyR[scCse|idi|i|ididd|it|idid|it|i dS(s!Get the values of the properties.ig(\M@iiN( RRRRdR5R,R&R;R=RK(R ((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRes (RRRRR6RRRER@RRRYR[Re(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyRs   t__main__stestoutput.txts TESTING %s %s s== i t verbosityi(2R!tsystunittestR"R?t__file__R$RqRbtispy2t version_infoRaRfRitcinfonyRRRRRtplatformt startswitht pybel_py2xRt pybel_py3xR=RFt NameErrortsetsRtTestCaseRRRRRRRRRtisfileRt testcasesttestcasetstdouttdefaultTestLoadertloadTestsFromTestCaset myunittesttTextTestRunnertrun(((sS/Volumes/Home/ghutchis/Devel/openbabel/test/../scripts/python/examples/testpybel.pyts`        @   !