htsworkflow/util/test/test_ethelp.py

   1 import os
   2 from unittest import TestCase
   3
   4 try:
   5   from xml.etree import ElementTree
   6 except ImportError, e:
   7   from elementtree import ElementTree
   8
   9 from htsworkflow.util.ethelp import indent, flatten
  10
  11 class testETHelper(TestCase):
  12     def setUp(self):
  13         self.foo = '<foo><bar>asdf</bar><br/></foo>'
  14         self.foo_tree = ElementTree.fromstring(self.foo)
  15
  16     def test_indent(self):
  17         flat_foo = ElementTree.tostring(self.foo_tree)
  18         self.failUnlessEqual(len(flat_foo.split('\n')), 1)
  19
  20         indent(self.foo_tree)
  21         pretty_foo = ElementTree.tostring(self.foo_tree)
  22         self.failUnlessEqual(len(pretty_foo.split('\n')), 5)
  23
  24     def test_flatten(self):
  25         self.failUnless(flatten(self.foo_tree), 'asdf')
  26
  27 def suite():
  28     from unittest import TestSuite, defaultTestLoader
  29     suite = TestSuite()
  30     suite.addTests(defaultTestLoader.loadTestsFromTestCase(testETHelper))
  31     return suite
  32
  33
  34 if __name__ == "__main__":
  35     from unittest import main
  36     main(defaultTest="suite")