Imported Upstream version 2.3.5

[platform/upstream/python-lxml.git] / src / lxml / tests / test_etree.py

diff --git a/src/lxml/tests/test_etree.py b/src/lxml/tests/test_etree.py
index 0b608ad..50141d4 100644 (file)
--- a/src/lxml/tests/test_etree.py
+++ b/src/lxml/tests/test_etree.py
@@ -432,6 +432,19 @@ class ETreeOnlyTestCase(HelperTestCase):
         self.assertEquals(_bytes('<test>TESTA<b>B<c xmlns="urn:c"></c>CT</b>BTAT<x>X<a>ABT<c xmlns="urn:x"></c>CT</a>AT</x>XT</test>'),
                           self._writeElement(root))
 
+    def test_strip_tags_and_remove(self):
+        # previously crashed
+        HTML = self.etree.HTML
+        root = HTML(_bytes('<div><h1>title</h1> <b>foo</b> <p>boo</p></div>'))[0][0]
+        self.assertEquals(_bytes('<div><h1>title</h1> <b>foo</b> <p>boo</p></div>'),
+                          self.etree.tostring(root))
+        self.etree.strip_tags(root, 'b')
+        self.assertEquals(_bytes('<div><h1>title</h1> foo <p>boo</p></div>'),
+                          self.etree.tostring(root))
+        root.remove(root[0])
+        self.assertEquals(_bytes('<div><p>boo</p></div>'),
+                          self.etree.tostring(root))
+
     def test_pi(self):
         # lxml.etree separates target and text
         Element = self.etree.Element