diff --git a/linkcheck/tests/test_strformat.py b/linkcheck/tests/test_strformat.py index 651f8e61..b8bcf960 100644 --- a/linkcheck/tests/test_strformat.py +++ b/linkcheck/tests/test_strformat.py @@ -33,20 +33,24 @@ class TestStrFormat (unittest.TestCase): """ Test quote stripping. """ - self.assertEquals(linkcheck.strformat.unquote(""), "") - self.assertEquals(linkcheck.strformat.unquote(None), None) - self.assertEquals(linkcheck.strformat.unquote("'"), "'") - self.assertEquals(linkcheck.strformat.unquote("\""), "\"") - self.assertEquals(linkcheck.strformat.unquote("\"\""), "") - self.assertEquals(linkcheck.strformat.unquote("''"), "") - self.assertEquals(linkcheck.strformat.unquote("'a'"), "a") - self.assertEquals(linkcheck.strformat.unquote("'a\"'"), "a\"") - self.assertEquals(linkcheck.strformat.unquote("'\"a'"), "\"a") - self.assertEquals(linkcheck.strformat.unquote('"a\'"'), 'a\'') - self.assertEquals(linkcheck.strformat.unquote('"\'a"'), '\'a') + u = linkcheck.strformat.unquote + self.assertEquals(u(""), "") + self.assertEquals(u(None), None) + self.assertEquals(u("'"), "'") + self.assertEquals(u("\""), "\"") + self.assertEquals(u("\"\""), "") + self.assertEquals(u("''"), "") + self.assertEquals(u("'a'"), "a") + self.assertEquals(u("'a\"'"), "a\"") + self.assertEquals(u("'\"a'"), "\"a") + self.assertEquals(u('"a\'"'), 'a\'') + self.assertEquals(u('"\'a"'), '\'a') # even mis-matching quotes should be removed... - self.assertEquals(linkcheck.strformat.unquote("'a\""), "a") - self.assertEquals(linkcheck.strformat.unquote("\"a'"), "a") + self.assertEquals(u("'a\""), "a") + self.assertEquals(u("\"a'"), "a") + # ...but not when matching is True + self.assertEquals(u("'a\"", matching=True), "'a\"") + self.assertEquals(u("\"a'", matching=True), "\"a'") def test_wrap (self): """