mirror of
https://github.com/moparisthebest/SickRage
synced 2024-10-31 15:35:01 -04:00
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
|
import os
|
||
|
import unittest
|
||
|
from support import html5lib_test_files, TestData, test_dir
|
||
|
|
||
|
from html5lib import HTMLParser, inputstream
|
||
|
|
||
|
import re, unittest
|
||
|
|
||
|
class Html5EncodingTestCase(unittest.TestCase):
|
||
|
def test_codec_name(self):
|
||
|
self.assertEquals(inputstream.codecName("utf-8"), "utf-8")
|
||
|
self.assertEquals(inputstream.codecName("utf8"), "utf-8")
|
||
|
self.assertEquals(inputstream.codecName(" utf8 "), "utf-8")
|
||
|
self.assertEquals(inputstream.codecName("ISO_8859--1"), "windows-1252")
|
||
|
|
||
|
def buildTestSuite():
|
||
|
for filename in html5lib_test_files("encoding"):
|
||
|
test_name = os.path.basename(filename).replace('.dat',''). \
|
||
|
replace('-','')
|
||
|
tests = TestData(filename, "data")
|
||
|
for idx, test in enumerate(tests):
|
||
|
def encodingTest(self, data=test['data'],
|
||
|
encoding=test['encoding']):
|
||
|
p = HTMLParser()
|
||
|
t = p.parse(data, useChardet=False)
|
||
|
|
||
|
errorMessage = ("Input:\n%s\nExpected:\n%s\nRecieved\n%s\n"%
|
||
|
(data, repr(encoding.lower()),
|
||
|
repr(p.tokenizer.stream.charEncoding)))
|
||
|
self.assertEquals(encoding.lower(),
|
||
|
p.tokenizer.stream.charEncoding[0],
|
||
|
errorMessage)
|
||
|
setattr(Html5EncodingTestCase, 'test_%s_%d' % (test_name, idx+1),
|
||
|
encodingTest)
|
||
|
|
||
|
try:
|
||
|
import chardet
|
||
|
def test_chardet(self):
|
||
|
data = open(os.path.join(test_dir, "encoding" , "chardet", "test_big5.txt")).read()
|
||
|
encoding = inputstream.HTMLInputStream(data).charEncoding
|
||
|
assert encoding[0].lower() == "big5"
|
||
|
setattr(Html5EncodingTestCase, 'test_chardet', test_chardet)
|
||
|
except ImportError:
|
||
|
print "chardet not found, skipping chardet tests"
|
||
|
|
||
|
|
||
|
return unittest.defaultTestLoader.loadTestsFromName(__name__)
|
||
|
|
||
|
def main():
|
||
|
buildTestSuite()
|
||
|
unittest.main()
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main()
|