2 from unittest import TestCase
5 from xml.etree import ElementTree
6 except ImportError as e:
7 from elementtree import ElementTree
9 from htsworkflow.util.ethelp import indent, flatten
11 class testETHelper(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 from unittest import TestSuite, defaultTestLoader
30 suite.addTests(defaultTestLoader.loadTestsFromTestCase(testETHelper))
34 if __name__ == "__main__":
35 from unittest import main
36 main(defaultTest="suite")