diff options
| author | Leonard Richardson <leonard.richardson@canonical.com> | 2012-02-27 13:52:11 -0500 |
|---|---|---|
| committer | Leonard Richardson <leonard.richardson@canonical.com> | 2012-02-27 13:52:11 -0500 |
| commit | f503a773d851ca3a0f1b690e8065a445f34acca9 (patch) | |
| tree | 170a9968ee69f4d7ce8d89c015eba55abceef826 /bs4/tests/test_tree.py | |
| parent | f224b8536ce266538bcfa492ec8d2b3b41fceae5 (diff) | |
Make prettify() return Unicode by default, so it will look nice when passed into print() under Python 3.
Diffstat (limited to 'bs4/tests/test_tree.py')
| -rw-r--r-- | bs4/tests/test_tree.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/bs4/tests/test_tree.py b/bs4/tests/test_tree.py index c75b561..4dda90e 100644 --- a/bs4/tests/test_tree.py +++ b/bs4/tests/test_tree.py @@ -1201,7 +1201,15 @@ class TestSubstitutions(SoupTest): def test_prettify_accepts_formatter(self): soup = BeautifulSoup("<html><body>foo</body></html>") pretty = soup.prettify(formatter = lambda x: x.upper()) - self.assertTrue(b"FOO" in pretty) + self.assertTrue("FOO" in pretty) + + def test_prettify_outputs_unicode_by_default(self): + soup = self.soup("<a></a>") + self.assertEqual(unicode, type(soup.prettify())) + + def test_prettify_can_encode_data(self): + soup = self.soup("<a></a>") + self.assertEqual(bytes, type(soup.prettify("utf-8"))) def test_html_entity_substitution_off_by_default(self): markup = u"<b>Sacr\N{LATIN SMALL LETTER E WITH ACUTE} bleu!</b>" |