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