def test_pretty_print(self):
e = E('root').add(E('child'))
s = b"""<root>
<child/>
</root>
"""
self.assertEqual(to_s(e, pretty_print=True), s)
def test_add_elem_text(self):
e = E('root').add('text', E('child'))
s = b'<root>text<child/></root>'
self.assertEqual(to_s(e), s)
e = E('root').add('text1', 'text2', E('child'))
s = b'<root>text1text2<child/></root>'
self.assertEqual(to_s(e), s)
def test_add_text(self):
e = E('root').add('text')
s = b'<root>text</root>'
self.assertEqual(to_s(e), s)
e = E('root').add('text1', 'text2')
s = b'<root>text1text2</root>'
self.assertEqual(to_s(e), s)
def test_iterable(self):
e = E('root').add_for(range(3), lambda i: E('foo%s' % i))
s = b'<root><foo0/><foo1/><foo2/></root>'
self.assertEqual(to_s(e), s)
e = E('root').add_for([], lambda i: E('foo%s' % i))
s = b'<root/>'
self.assertEqual(to_s(e), s)
def test_true_condition(self):
e = E('root').add_if(True, E('SE'))
s = b'<root><SE/></root>'
self.assertEqual(to_s(e), s)
e = E('root').add_if(True, 'foo')
s = b'<root>foo</root>'
self.assertEqual(to_s(e), s)
def test_false_condition(self):
e = E('root').add_if(False, E('SE'))
s = b'<root/>'
self.assertEqual(to_s(e), s)
e = E('root').add_if(False, 'foo')
s = b'<root/>'
self.assertEqual(to_s(e), s)
def test_add_elem_text_tail(self):
e = E('root').add('text', E('child'), 'tail')
s = b'<root>text<child/>tail</root>'
self.assertEqual(to_s(e), s)
e = E('root').add('text1', 'text2', E('child'), 'tail1', 'tail2')
s = b'<root>text1text2<child/>tail1tail2</root>'
self.assertEqual(to_s(e), s)
def test_complex(self):
e = E('root', atr=100).add(
'text1',
E('child', atr="atr").add(
E('superchild', atr=None).add('sctext1'), 'tail1', 'tail2'),
'tail',
E('child', atr="").add('text'))
s = b'<root atr="100">text1<child atr="atr"><superchild atr="">sctext1</superchild>tail1tail2</child>tail<child atr="">text</child></root>'
self.assertEqual(to_s(e), s)
def test_add_element(self):
e = E('root').add()
s = b'<root/>'
self.assertEqual(to_s(e), s)
e = E('root').add(E('child1'))
s = b'<root><child1/></root>'
self.assertEqual(to_s(e), s)
e = E('root').add(E('child1'), E('child2'))
s = b'<root><child1/><child2/></root>'
self.assertEqual(to_s(e), s)
e = E('root').add(E('child1'), E('child2'), E('child3'))
s = b'<root><child1/><child2/><child3/></root>'
self.assertEqual(to_s(e), s)
def test_element(self):
e = E('root')
s = b'<root/>'
self.assertEqual(to_s(e), s)
e = E('root', foo='bar')
s = b'<root foo="bar"/>'
self.assertEqual(to_s(e), s)
e = E('root', bar=0)
s = b'<root bar="0"/>'
self.assertEqual(to_s(e), s)
e = E('root', baz=None)
s = b'<root baz=""/>'
self.assertEqual(to_s(e), s)
e = E('root', baz='Zażółć gęślą jaźń')
s = b'<root baz="Zażółć gęślą jaźń"/>'
self.assertEqual(to_s(e), s)
def test_add_elem_tail(self):
e = E('root').add(E('child'), 'tail')
s = b'<root><child/>tail</root>'
self.assertEqual(to_s(e), s)
e = E('root').add(E('child'), 'tail1', 'tail2')
s = b'<root><child/>tail1tail2</root>'
self.assertEqual(to_s(e), s)
e = E('root').add(E('child1'), 'tail1', 'tail2', E('child2'))
s = b'<root><child1/>tail1tail2<child2/></root>'
self.assertEqual(to_s(e), s)
def test_add_none(self):
e = E('root').add(None)
s = b'<root/>'
self.assertEqual(to_s(e), s)
def test_add_with_elems(self):
e = E('root')
se = SE(e, 'child').add(E('child2'))
s = b'<root><child><child2/></child></root>'
self.assertEqual(to_s(e), s)
def test_add_with_text(self):
e = E('root')
se = SE(e, 'child').add('text')
s = b'<root><child>text</child></root>'
self.assertEqual(to_s(e), s)
def test_add(self):
e = E('root')
se = SE(e, 'child')
s = b'<root><child/></root>'
self.assertEqual(to_s(e), s)
def test_add_empty_string(self):
e = E('root').add('t1', '', 't3')
s = b'<root>t1t3</root>'
self.assertEqual(to_s(e), s)
def test_add_number(self):
e = E('root').add(17)
s = b'<root>17</root>'
self.assertEqual(to_s(e), s)