5 from xml.etree import ElementTree
7 from elementtree import ElementTree
9 from htsworkflow.util.ethelp import indent, flatten
11 class testETHelper(unittest.TestCase):
13 self.foo = '<foo><bar>asdf</bar><br/></foo>'
14 self.foo_tree = ElementTree.fromstring(self.foo)
16 def test_indent(self):
17 flat_foo = ElementTree.tostring(self.foo_tree)
18 self.failUnlessEqual(len(flat_foo.split('\n')), 1)
21 pretty_foo = ElementTree.tostring(self.foo_tree)
22 self.failUnlessEqual(len(pretty_foo.split('\n')), 5)
24 def test_flatten(self):
25 self.failUnless(flatten(self.foo_tree), 'asdf')
28 return unittest.makeSuite(testETHelper, 'test')
30 if __name__ == "__main__":
31 unittest.main(defaultTest='suite')