gaworkflow/util/test/test_ethelp.py

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