def test_setAttribute(self):
foo = Element('foo')
# Using standard DOM methods
foo.setAttribute('alpha', '1')
self.assertEqual('1', foo.getAttribute('alpha'))
# Using modified __setitem__ and __getitem__
foo['bravo'] = '2'
self.assertEqual('2', foo['bravo'])
def test_toprettyxml(self):
svg = Element('svg')
self.assertEqual('<svg/>', svg.toprettyxml())
svg.setAttribute('width', '200')
self.assertEqual('<svg width="200"/>', svg.toprettyxml())
p = Element('p')
p.append('some text')
self.assertEqual('<p>some text</p>', p.toprettyxml())
div = Element('div')
p = Element('p')
p.append('one')
div.append(p)
self.assertEqual('<div>\n\t<p>one</p>\n</div>', div.toprettyxml())
p = Element('p')
p.append('two')
div.append(p)
self.assertEqual('<div>\n\t<p>one</p>\n\t<p>two</p>\n</div>',
div.toprettyxml())
