summaryrefslogtreecommitdiff
path: root/src/lxml/tests/test_elementtree.py
diff options
context:
space:
mode:
authorscoder <none@none>2007-09-15 23:44:54 +0200
committerscoder <none@none>2007-09-15 23:44:54 +0200
commited4299b1b40e24ef83a0968505d4463ddb996b83 (patch)
tree080e69fe1a80cc6cfc7f9bce2b439823574be615 /src/lxml/tests/test_elementtree.py
parentb7b9db53ba34688b7724a8c0c2bf883dd606e500 (diff)
downloadpython-lxml-lxml-2.0alpha2.tar.gz
[svn r2884] support for selection output method on serialisationlxml-2.0alpha2
--HG-- branch : trunk
Diffstat (limited to 'src/lxml/tests/test_elementtree.py')
-rw-r--r--src/lxml/tests/test_elementtree.py72
1 files changed, 72 insertions, 0 deletions
diff --git a/src/lxml/tests/test_elementtree.py b/src/lxml/tests/test_elementtree.py
index 43c44f02..f663704b 100644
--- a/src/lxml/tests/test_elementtree.py
+++ b/src/lxml/tests/test_elementtree.py
@@ -716,6 +716,47 @@ class ETreeTestCaseBase(unittest.TestCase):
self.assertEquals(
'<doc%s>This is a test.</doc%s>' % (i, i),
canonicalize(data))
+
+ def test_write_method_html(self):
+ ElementTree = self.etree.ElementTree
+ Element = self.etree.Element
+ SubElement = self.etree.SubElement
+
+ html = Element('html')
+ body = SubElement(html, 'body')
+ p = SubElement(body, 'p')
+ p.text = "html"
+ SubElement(p, 'br').tail = "test"
+
+ tree = ElementTree(element=html)
+ f = StringIO()
+ tree.write(f, method="html")
+ data = f.getvalue()
+
+ self.assertEquals('<html><body><p>html<br>test</p></body></html>',
+ data)
+
+ def test_write_method_text(self):
+ ElementTree = self.etree.ElementTree
+ Element = self.etree.Element
+ SubElement = self.etree.SubElement
+
+ a = Element('a')
+ a.text = "A"
+ a.tail = "tail"
+ b = SubElement(a, 'b')
+ b.text = "B"
+ b.tail = "TAIL"
+ c = SubElement(a, 'c')
+ c.text = "C"
+
+ tree = ElementTree(element=a)
+ f = StringIO()
+ tree.write(f, method="text")
+ data = f.getvalue()
+
+ self.assertEquals('ABTAILCtail',
+ data)
def test_write_fail(self):
ElementTree = self.etree.ElementTree
@@ -2099,6 +2140,37 @@ class ETreeTestCaseBase(unittest.TestCase):
self.assert_(tostring(b) == '<b/>Foo' or
tostring(b) == '<b />Foo')
+ def test_tostring_method_html(self):
+ tostring = self.etree.tostring
+ Element = self.etree.Element
+ SubElement = self.etree.SubElement
+
+ html = Element('html')
+ body = SubElement(html, 'body')
+ p = SubElement(body, 'p')
+ p.text = "html"
+ SubElement(p, 'br').tail = "test"
+
+ self.assertEquals('<html><body><p>html<br>test</p></body></html>',
+ tostring(html, method="html"))
+
+ def test_tostring_method_text(self):
+ tostring = self.etree.tostring
+ Element = self.etree.Element
+ SubElement = self.etree.SubElement
+
+ a = Element('a')
+ a.text = "A"
+ a.tail = "tail"
+ b = SubElement(a, 'b')
+ b.text = "B"
+ b.tail = "TAIL"
+ c = SubElement(a, 'c')
+ c.text = "C"
+
+ self.assertEquals('ABTAILCtail',
+ tostring(a, method="text"))
+
def test_iterparse(self):
iterparse = self.etree.iterparse
f = StringIO('<a><b></b><c/></a>')