#!/usr/bin/python
# Copyright (c) 2005 by Mark T. Holder, Florida State University. (see end of file)
from PIPRes.util.crude_xml import *
import unittest, sys, cStringIO

class CrudeXMLTest(unittest.TestCase):
    def testTreeAndSplits(self):
        treeString = '(1:0.2,2:3.0,(3:0.1,4:0.001):0.001)'
        inXML = '''<pipres-crude>
    <tree newick="%s" />
    <split index="2"/>
    <split decimal="6"/>
    <split newick="(1,3,4,(2,5,6))"/>
    <split dotstar=".*..**"/>
</pipres-crude>
''' % treeString
        expecteAutoXML = '''<pipres-crude>
    <tree newick="%s"/>
    <split decimal="4"/>
    <split decimal="6"/>
    <split decimal="50"/>
    <split decimal="50"/>
</pipres-crude>
''' % treeString
        xmlFileObj = cStringIO.StringIO(inXML)
        objs = xmlToPipres(xmlFileObj)
        tree = numberedLeafTree(treeString)
        expected = [
            ['tree', tree],
            ['split', 4L],
            ['split', 6L],
            ['split', 50L],
            ['split', 50L],
            ]
        self.assertEquals(expected, objs)
        autoXML = toPipresXML(expected)
        self.assertEquals(autoXML, expecteAutoXML)
    
if __name__ == '__main__':
    unittest.main()