Ported some more tests, fixed an encoding problem, and added rudimentary doctype handling.

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/test_tree.py b/tests/test_tree.py
index eac4e72..367489e 100644
--- a/tests/test_tree.py
+++ b/tests/test_tree.py
@@ -815,3 +815,19 @@ class TestPersistence(SoupTest):
         dumped = pickle.dumps(soup, pickle.HIGHEST_PROTOCOL)
         loaded = pickle.loads(dumped)
         self.assertEqual(loaded.decode(), soup.decode())
+
+
+class TestEncoding(SoupTest):
+    """Test the ability to encode objects into strings."""
+
+    def test_unicode_string_can_be_encoded(self):
+        html = u"<b>\N{SNOWMAN}</b>"
+        soup = self.soup(html)
+        self.assertEquals(soup.b.string.encode("utf-8"),
+                          u"\N{SNOWMAN}".encode("utf-8"))
+
+    def test_tag_containing_unicode_string_can_be_encoded(self):
+        html = u"<b>\N{SNOWMAN}</b>"
+        soup = self.soup(html)
+        self.assertEquals(
+            soup.b.encode("utf-8"), html.encode("utf-8"))