| import unittest |
| |
| from html5lib.filters.whitespace import Filter |
| from html5lib.constants import spaceCharacters |
| spaceCharacters = u"".join(spaceCharacters) |
| |
| class TestCase(unittest.TestCase): |
| def runTest(self, input, expected): |
| output = list(Filter(input)) |
| errorMsg = "\n".join(["\n\nInput:", str(input), |
| "\nExpected:", str(expected), |
| "\nReceived:", str(output)]) |
| self.assertEquals(output, expected, errorMsg) |
| |
| def runTestUnmodifiedOutput(self, input): |
| self.runTest(input, input) |
| |
| def testPhrasingElements(self): |
| self.runTestUnmodifiedOutput( |
| [{"type": u"Characters", "data": u"This is a " }, |
| {"type": u"StartTag", "name": u"span", "data": [] }, |
| {"type": u"Characters", "data": u"phrase" }, |
| {"type": u"EndTag", "name": u"span", "data": []}, |
| {"type": u"SpaceCharacters", "data": u" " }, |
| {"type": u"Characters", "data": u"with" }, |
| {"type": u"SpaceCharacters", "data": u" " }, |
| {"type": u"StartTag", "name": u"em", "data": [] }, |
| {"type": u"Characters", "data": u"emphasised text" }, |
| {"type": u"EndTag", "name": u"em", "data": []}, |
| {"type": u"Characters", "data": u" and an " }, |
| {"type": u"StartTag", "name": u"img", "data": [[u"alt", u"image"]] }, |
| {"type": u"Characters", "data": u"." }]) |
| |
| def testLeadingWhitespace(self): |
| self.runTest( |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"SpaceCharacters", "data": spaceCharacters}, |
| {"type": u"Characters", "data": u"foo"}, |
| {"type": u"EndTag", "name": u"p", "data": []}], |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"SpaceCharacters", "data": u" "}, |
| {"type": u"Characters", "data": u"foo"}, |
| {"type": u"EndTag", "name": u"p", "data": []}]) |
| |
| def testLeadingWhitespaceAsCharacters(self): |
| self.runTest( |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": spaceCharacters + u"foo"}, |
| {"type": u"EndTag", "name": u"p", "data": []}], |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": u" foo"}, |
| {"type": u"EndTag", "name": u"p", "data": []}]) |
| |
| def testTrailingWhitespace(self): |
| self.runTest( |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": u"foo"}, |
| {"type": u"SpaceCharacters", "data": spaceCharacters}, |
| {"type": u"EndTag", "name": u"p", "data": []}], |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": u"foo"}, |
| {"type": u"SpaceCharacters", "data": u" "}, |
| {"type": u"EndTag", "name": u"p", "data": []}]) |
| |
| def testTrailingWhitespaceAsCharacters(self): |
| self.runTest( |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": u"foo" + spaceCharacters}, |
| {"type": u"EndTag", "name": u"p", "data": []}], |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": u"foo "}, |
| {"type": u"EndTag", "name": u"p", "data": []}]) |
| |
| def testWhitespace(self): |
| self.runTest( |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": u"foo" + spaceCharacters + "bar"}, |
| {"type": u"EndTag", "name": u"p", "data": []}], |
| [{"type": u"StartTag", "name": u"p", "data": []}, |
| {"type": u"Characters", "data": u"foo bar"}, |
| {"type": u"EndTag", "name": u"p", "data": []}]) |
| |
| def testLeadingWhitespaceInPre(self): |
| self.runTestUnmodifiedOutput( |
| [{"type": u"StartTag", "name": u"pre", "data": []}, |
| {"type": u"SpaceCharacters", "data": spaceCharacters}, |
| {"type": u"Characters", "data": u"foo"}, |
| {"type": u"EndTag", "name": u"pre", "data": []}]) |
| |
| def testLeadingWhitespaceAsCharactersInPre(self): |
| self.runTestUnmodifiedOutput( |
| [{"type": u"StartTag", "name": u"pre", "data": []}, |
| {"type": u"Characters", "data": spaceCharacters + u"foo"}, |
| {"type": u"EndTag", "name": u"pre", "data": []}]) |
| |
| def testTrailingWhitespaceInPre(self): |
| self.runTestUnmodifiedOutput( |
| [{"type": u"StartTag", "name": u"pre", "data": []}, |
| {"type": u"Characters", "data": u"foo"}, |
| {"type": u"SpaceCharacters", "data": spaceCharacters}, |
| {"type": u"EndTag", "name": u"pre", "data": []}]) |
| |
| def testTrailingWhitespaceAsCharactersInPre(self): |
| self.runTestUnmodifiedOutput( |
| [{"type": u"StartTag", "name": u"pre", "data": []}, |
| {"type": u"Characters", "data": u"foo" + spaceCharacters}, |
| {"type": u"EndTag", "name": u"pre", "data": []}]) |
| |
| def testWhitespaceInPre(self): |
| self.runTestUnmodifiedOutput( |
| [{"type": u"StartTag", "name": u"pre", "data": []}, |
| {"type": u"Characters", "data": u"foo" + spaceCharacters + "bar"}, |
| {"type": u"EndTag", "name": u"pre", "data": []}]) |
| |
| def buildTestSuite(): |
| return unittest.defaultTestLoader.loadTestsFromName(__name__) |
| |
| def main(): |
| buildTestSuite() |
| unittest.main() |
| |
| if __name__ == "__main__": |
| main() |
